mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-17 07:50:59 +00:00
feat(tenancy): implement workspace isolation
This commit is contained in:
@@ -125,6 +125,8 @@ class FakeApp:
|
||||
"""Mock SkillManager that returns no skill index addition by default."""
|
||||
skill_mgr = Mock()
|
||||
skill_mgr.skills = {}
|
||||
skill_mgr.ensure_loaded = AsyncMock()
|
||||
skill_mgr.get_skills = Mock(return_value=[])
|
||||
skill_mgr.build_skill_aware_prompt_addition = Mock(return_value='')
|
||||
skill_mgr.get_skill_index = Mock(return_value=[])
|
||||
return skill_mgr
|
||||
|
||||
@@ -14,6 +14,7 @@ import langbot_plugin.api.entities.builtin.platform.message as platform_message
|
||||
import langbot_plugin.api.entities.builtin.platform.events as platform_events
|
||||
import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
|
||||
import langbot_plugin.api.entities.builtin.provider.session as provider_session
|
||||
from langbot.pkg.api.http.context import ExecutionContext
|
||||
|
||||
|
||||
# Counter for generating unique IDs
|
||||
@@ -194,7 +195,20 @@ def _base_query(
|
||||
for key, value in overrides.items():
|
||||
base_data[key] = value
|
||||
|
||||
return pipeline_query.Query.model_construct(**base_data)
|
||||
query = pipeline_query.Query.model_construct(**base_data)
|
||||
object.__setattr__(
|
||||
query,
|
||||
'_execution_context',
|
||||
ExecutionContext(
|
||||
instance_uuid='test-instance',
|
||||
workspace_uuid='test-workspace',
|
||||
placement_generation=1,
|
||||
bot_uuid=query.bot_uuid,
|
||||
pipeline_uuid=query.pipeline_uuid,
|
||||
query_uuid=query.query_uuid,
|
||||
),
|
||||
)
|
||||
return query
|
||||
|
||||
|
||||
def text_query(
|
||||
|
||||
Reference in New Issue
Block a user