LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 00:23:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: passing sources and chunk if agent is shared

Browse Source
This commit is contained in:
Alex
2025-08-22 13:36:31 +01:00
parent e283957c8f
commit 44d21ab703
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
application/api/answer/services/stream_processor.py
View File

@@ -85,8 +85,8 @@ class StreamProcessor:
def initialize(self): def initialize(self):
"""Initialize all required components for processing""" """Initialize all required components for processing"""
self._configure_agent()
self._configure_retriever() self._configure_retriever()
self._configure_agent()
self._load_conversation_history() self._load_conversation_history()
self._process_attachments() self._process_attachments()
@@ -173,8 +173,9 @@ class StreamProcessor:
source_doc = self.db.dereference(source) source_doc = self.db.dereference(source)
data["source"] = str(source_doc["_id"]) data["source"] = str(source_doc["_id"])
data["retriever"] = source_doc.get("retriever", data.get("retriever")) data["retriever"] = source_doc.get("retriever", data.get("retriever"))
data["chunks"] = source_doc.get("chunks", data.get("chunks"))
else: else:
data["source"] = {} data["source"] = None
return data return data
def _configure_agent(self): def _configure_agent(self):
@@ -197,6 +198,12 @@ class StreamProcessor:
) )
self.initial_user_id = data_key.get("user") self.initial_user_id = data_key.get("user")
self.decoded_token = {"sub": data_key.get("user")} self.decoded_token = {"sub": data_key.get("user")}
if data_key.get("source"):
self.source = {"active_docs": data_key["source"]}
if data_key.get("retriever"):
self.retriever_config["retriever_name"] = data_key["retriever"]
if data_key.get("chunks") is not None:
self.retriever_config["chunks"] = data_key["chunks"]
elif self.agent_key: elif self.agent_key:
data_key = self._get_data_from_api_key(self.agent_key) data_key = self._get_data_from_api_key(self.agent_key)
self.agent_config.update( self.agent_config.update(
@@ -212,6 +219,12 @@ class StreamProcessor:
if self.is_shared_usage if self.is_shared_usage
else {"sub": data_key.get("user")} else {"sub": data_key.get("user")}
) )
if data_key.get("source"):
self.source = {"active_docs": data_key["source"]}
if data_key.get("retriever"):
self.retriever_config["retriever_name"] = data_key["retriever"]
if data_key.get("chunks") is not None:
self.retriever_config["chunks"] = data_key["chunks"]
else: else:
self.agent_config.update( self.agent_config.update(
{ {