feat(tenancy): implement workspace isolation

tests/unit_tests/provider/test_requester_base.py

@@ -15,6 +15,7 @@ from langbot.pkg.provider.modelmgr import requester
 from langbot.pkg.provider.modelmgr import token
 from langbot.pkg.entity.persistence import model as persistence_model
 from langbot.pkg.provider.modelmgr.errors import RequesterError
+from tests.unit_tests.provider.conftest import TEST_EXECUTION_CONTEXT, TEST_WORKSPACE_UUID
 
 
 # ============================================================================
@@ -134,6 +135,7 @@ async def test_requester_invoke_rerank_not_implemented():
 
     # Create fake model
     fake_provider_entity = persistence_model.ModelProvider(
+        workspace_uuid=TEST_WORKSPACE_UUID,
         uuid='provider-uuid',
         name='Provider',
         requester='test',
@@ -143,17 +145,20 @@ async def test_requester_invoke_rerank_not_implemented():
     fake_token_mgr = token.TokenManager(name='test', tokens=[])
     fake_requester = inst
     fake_provider = requester.RuntimeProvider(
+        execution_context=TEST_EXECUTION_CONTEXT,
         provider_entity=fake_provider_entity,
         token_mgr=fake_token_mgr,
         requester=fake_requester,
     )
     fake_model_entity = persistence_model.RerankModel(
+        workspace_uuid=TEST_WORKSPACE_UUID,
         uuid='model-uuid',
         name='Model',
         provider_uuid='provider-uuid',
         extra_args={},
     )
     fake_model = requester.RuntimeRerankModel(
+        execution_context=TEST_EXECUTION_CONTEXT,
         model_entity=fake_model_entity,
         provider=fake_provider,
     )
@@ -289,6 +294,7 @@ async def test_runtime_provider_invoke_llm_delegates(runtime_provider, runtime_l
         resp_message_chain=None,
         current_stage_name=None,
     )
+    object.__setattr__(query, '_execution_context', TEST_EXECUTION_CONTEXT)
 
     messages = [
         provider_message.Message(role='user', content=[provider_message.ContentElement(type='text', text='Hello')])
@@ -332,6 +338,7 @@ async def test_runtime_provider_invoke_llm_stream_yields_chunks(runtime_provider
         resp_message_chain=None,
         current_stage_name=None,
     )
+    object.__setattr__(query, '_execution_context', TEST_EXECUTION_CONTEXT)
 
     messages = [
         provider_message.Message(role='user', content=[provider_message.ContentElement(type='text', text='Hello')])
@@ -350,7 +357,11 @@ async def test_runtime_provider_invoke_embedding_returns_vectors(runtime_provide
     """Test RuntimeProvider.invoke_embedding returns embedding vectors."""
     provider = runtime_provider
 
-    result = await provider.invoke_embedding(runtime_embedding_model, ['text1', 'text2'])
+    result = await provider.invoke_embedding(
+        runtime_embedding_model,
+        ['text1', 'text2'],
+        execution_context=TEST_EXECUTION_CONTEXT,
+    )
 
     assert len(result) == 2
     assert result[0] == [0.1, 0.2, 0.3]
@@ -362,7 +373,12 @@ async def test_runtime_provider_invoke_rerank_returns_scores(runtime_provider, r
     # Need to use the correct provider for rerank model
     provider = runtime_rerank_model.provider
 
-    result = await provider.invoke_rerank(runtime_rerank_model, 'query', ['doc1', 'doc2', 'doc3'])
+    result = await provider.invoke_rerank(
+        runtime_rerank_model,
+        'query',
+        ['doc1', 'doc2', 'doc3'],
+        execution_context=TEST_EXECUTION_CONTEXT,
+    )
 
     assert len(result) == 3
     assert result[0]['index'] == 0
@@ -532,6 +548,7 @@ async def test_runtime_provider_invoke_llm_propagates_error(mock_app_for_modelmg
     await requester_inst.initialize()
 
     provider_entity = persistence_model.ModelProvider(
+        workspace_uuid=TEST_WORKSPACE_UUID,
         uuid='error-provider',
         name='Error Provider',
         requester='error-requester',
@@ -541,19 +558,25 @@ async def test_runtime_provider_invoke_llm_propagates_error(mock_app_for_modelmg
     token_mgr = token.TokenManager(name='error-provider', tokens=['error-key'])
 
     provider = requester.RuntimeProvider(
+        execution_context=TEST_EXECUTION_CONTEXT,
         provider_entity=provider_entity,
         token_mgr=token_mgr,
         requester=requester_inst,
     )
 
     model_entity = persistence_model.LLMModel(
+        workspace_uuid=TEST_WORKSPACE_UUID,
         uuid='error-model',
         name='Error Model',
         provider_uuid='error-provider',
         abilities=[],
         extra_args={},
     )
-    model = requester.RuntimeLLMModel(model_entity=model_entity, provider=provider)
+    model = requester.RuntimeLLMModel(
+        execution_context=TEST_EXECUTION_CONTEXT,
+        model_entity=model_entity,
+        provider=provider,
+    )
 
     import langbot_plugin.api.entities.builtin.provider.message as provider_message
     import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
@@ -580,6 +603,7 @@ async def test_runtime_provider_invoke_llm_propagates_error(mock_app_for_modelmg
         resp_message_chain=None,
         current_stage_name=None,
     )
+    object.__setattr__(query, '_execution_context', TEST_EXECUTION_CONTEXT)
 
     messages = [
         provider_message.Message(role='user', content=[provider_message.ContentElement(type='text', text='Hello')])