diff --git a/app/plugins/builtins/ai_assistant/routes.py b/app/plugins/builtins/ai_assistant/routes.py
index e708771..6a2ace4 100644
--- a/app/plugins/builtins/ai_assistant/routes.py
+++ b/app/plugins/builtins/ai_assistant/routes.py
@@ -550,6 +550,7 @@ async def chat_stream(
 ):
     tenant_id = uuid.UUID(current_user["tenant_id"])
     user_id = uuid.UUID(current_user["user_id"])
+    await set_tenant_context(db, tenant_id)
     session = await get_session_by_id(db, uuid.UUID(session_id), user_id, tenant_id)
     if not session:
         raise HTTPException(status_code=404, detail="Session not found")