feat(tenancy): implement workspace isolation

tests/unit_tests/provider/test_model_service.py

@@ -19,6 +19,8 @@ from langbot.pkg.provider.modelmgr import requester
 from langbot.pkg.provider.modelmgr.modelmgr import ModelManager
 from langbot.pkg.provider.modelmgr.token import TokenManager
 from langbot.pkg.provider.runners.localagent import LocalAgentRunner
+from langbot.pkg.api.http.context import ExecutionContext
+from langbot.pkg.workspace.entities import WorkspaceExecutionBinding
 
 
 def test_runtime_llm_model_data_preserves_uuid_after_update_payload_uuid_removed():
@@ -95,11 +97,22 @@ async def test_model_manager_initialize_skips_space_sync_after_timeout():
     ap.discover = SimpleNamespace(get_components_by_kind=Mock(return_value=[]))
     ap.instance_config = SimpleNamespace(data={'space': {'models_sync_timeout': 0.01}})
     ap.logger = Mock()
+    binding = WorkspaceExecutionBinding(
+        instance_uuid='instance-test',
+        workspace_uuid='workspace-test',
+        placement_generation=1,
+        write_fenced=False,
+        state='active',
+    )
+    ap.workspace_service = SimpleNamespace(
+        get_local_execution_binding=AsyncMock(return_value=binding),
+        get_execution_binding=AsyncMock(return_value=binding),
+    )
 
     mgr = ModelManager(ap)
     mgr.load_models_from_db = AsyncMock()
 
-    async def slow_sync():
+    async def slow_sync(_context):
         await asyncio.sleep(1)
 
     mgr.sync_new_models_from_space = AsyncMock(side_effect=slow_sync)
@@ -117,6 +130,14 @@ async def test_updated_llm_model_is_immediately_usable_by_local_agent_pipeline()
 
     model_uuid = 'qwen-model-uuid'
     provider_uuid = 'ollama-provider-uuid'
+    workspace_uuid = 'workspace-test'
+    execution_context = ExecutionContext(
+        instance_uuid='instance-test',
+        workspace_uuid=workspace_uuid,
+        placement_generation=1,
+        bot_uuid='bot-uuid',
+        pipeline_uuid='pipeline-uuid',
+    )
 
     ap = SimpleNamespace()
     ap.logger = Mock()
@@ -126,24 +147,63 @@ async def test_updated_llm_model_is_immediately_usable_by_local_agent_pipeline()
     ap.plugin_connector = SimpleNamespace(
         emit_event=AsyncMock(return_value=SimpleNamespace(event=SimpleNamespace(default_prompt=[], prompt=[])))
     )
+    binding = WorkspaceExecutionBinding(
+        instance_uuid='instance-test',
+        workspace_uuid=workspace_uuid,
+        placement_generation=1,
+        write_fenced=False,
+        state='active',
+    )
+    ap.workspace_service = SimpleNamespace(get_execution_binding=AsyncMock(return_value=binding))
 
     ap.model_mgr = ModelManager(ap)
-    runtime_provider = Mock()
-    ap.model_mgr.provider_dict = {provider_uuid: runtime_provider}
-    ap.model_mgr.llm_models = [
-        requester.RuntimeLLMModel(
-            model_entity=persistence_model.LLMModel(
-                uuid=model_uuid,
-                name='old-qwen-name',
-                provider_uuid=provider_uuid,
-                abilities=[],
-                extra_args={},
-            ),
-            provider=runtime_provider,
-        )
-    ]
+    runtime_provider = Mock(
+        execution_context=execution_context,
+        provider_entity=persistence_model.ModelProvider(
+            workspace_uuid=workspace_uuid,
+            uuid=provider_uuid,
+            name='Ollama',
+            requester='ollama',
+            base_url='http://localhost:11434',
+            api_keys=[],
+        ),
+    )
+    cache_key = ('instance-test', workspace_uuid, 1, provider_uuid)
+    ap.model_mgr.provider_dict = {cache_key: runtime_provider}
+    runtime_model = requester.RuntimeLLMModel(
+        execution_context=execution_context,
+        model_entity=persistence_model.LLMModel(
+            workspace_uuid=workspace_uuid,
+            uuid=model_uuid,
+            name='old-qwen-name',
+            provider_uuid=provider_uuid,
+            abilities=[],
+            extra_args={},
+        ),
+        provider=runtime_provider,
+    )
+    ap.model_mgr.llm_model_dict = {
+        ('instance-test', workspace_uuid, 1, model_uuid): runtime_model,
+    }
 
-    await LLMModelsService(ap).update_llm_model(
+    ap.provider_service = SimpleNamespace(
+        get_provider=AsyncMock(return_value={'uuid': provider_uuid, 'workspace_uuid': workspace_uuid})
+    )
+    model_service = LLMModelsService(ap)
+    model_service.get_llm_model = AsyncMock(
+        return_value={
+            'uuid': model_uuid,
+            'workspace_uuid': workspace_uuid,
+            'name': 'old-qwen-name',
+            'provider_uuid': provider_uuid,
+            'abilities': [],
+            'context_length': None,
+            'extra_args': {},
+            'prefered_ranking': 0,
+        }
+    )
+    await model_service.update_llm_model(
+        workspace_uuid,
         model_uuid,
         {
             'name': 'Qwen3.5-27B',
@@ -153,13 +213,17 @@ async def test_updated_llm_model_is_immediately_usable_by_local_agent_pipeline()
         },
     )
 
-    runtime_model = await ap.model_mgr.get_model_by_uuid(model_uuid)
+    runtime_model = await ap.model_mgr.get_model_by_uuid(execution_context, model_uuid)
     assert runtime_model.model_entity.uuid == model_uuid
     assert runtime_model.model_entity.name == 'Qwen3.5-27B'
 
-    session = SimpleNamespace(
+    session = provider_session.Session(
+        instance_uuid='instance-test',
+        workspace_uuid=workspace_uuid,
+        placement_generation=1,
         launcher_type=provider_session.LauncherTypes.PERSON,
         launcher_id=12345,
+        bot_uuid='bot-uuid',
     )
     conversation = SimpleNamespace(
         uuid='conversation-uuid',
@@ -194,6 +258,9 @@ async def test_updated_llm_model_is_immediately_usable_by_local_agent_pipeline()
         'output': {'misc': {'remove-think': False}},
     }
     query = pipeline_query.Query.model_construct(
+        instance_uuid='instance-test',
+        workspace_uuid=workspace_uuid,
+        placement_generation=1,
         query_id='query-id',
         launcher_type=provider_session.LauncherTypes.PERSON,
         launcher_id=12345,
@@ -215,6 +282,7 @@ async def test_updated_llm_model_is_immediately_usable_by_local_agent_pipeline()
         resp_message_chain=None,
         current_stage_name=None,
     )
+    object.__setattr__(query, '_execution_context', execution_context)
 
     result = await PreProcessor(ap).process(query, 'PreProcessor')
     processed_query = result.new_query