diff --git a/src/langbot/pkg/provider/runners/localagent.py b/src/langbot/pkg/provider/runners/localagent.py index f444529b..52e78b9d 100644 --- a/src/langbot/pkg/provider/runners/localagent.py +++ b/src/langbot/pkg/provider/runners/localagent.py @@ -74,7 +74,13 @@ class LocalAgentRunner(runner.RequestRunner): self.ap.logger.warning(f'Knowledge base {kb_uuid} not found, skipping') continue - result = await kb.retrieve(user_message_text) + result = await kb.retrieve( + user_message_text, + settings={ + 'sender_id': str(query.sender_id), + 'session_name': f'{query.session.launcher_type.value}_{query.session.launcher_id}', + }, + ) if result: all_results.extend(result) diff --git a/src/langbot/pkg/rag/knowledge/kbmgr.py b/src/langbot/pkg/rag/knowledge/kbmgr.py index 5831da30..6d7f76b4 100644 --- a/src/langbot/pkg/rag/knowledge/kbmgr.py +++ b/src/langbot/pkg/rag/knowledge/kbmgr.py @@ -321,13 +321,23 @@ class RuntimeKnowledgeBase(KnowledgeBaseInterface): if not plugin_id: raise ValueError(f'No RAG plugin ID configured for KB {kb.uuid}. Retrieval failed.') + # Extract session context before building retrieval_context + sender_id = settings.pop('sender_id', None) + session_name = settings.pop('session_name', None) + + filters = settings.pop('filters', {}) + if sender_id is not None: + filters['sender_id'] = sender_id + if session_name is not None: + filters['session_name'] = session_name + retrieval_context = { 'query': query, 'knowledge_base_id': kb.uuid, 'collection_id': kb.collection_id or kb.uuid, 'retrieval_settings': settings, 'creation_settings': kb.creation_settings or {}, - 'filters': settings.pop('filters', {}), + 'filters': filters, } result = await self.ap.plugin_connector.call_rag_retrieve(