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

Merge branch 'main' of https://github.com/arc53/DocsGPT

Browse Source
This commit is contained in:
GH Action - Upstream Sync
2025-08-23 01:35:35 +00:00
parent 2410bd8654 44d21ab703
commit e25b988dc8
2 changed files with 21 additions and 3 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(
{ {

7
application/api/user/routes.py
View File

@@ -2026,7 +2026,12 @@ class SharedAgent(Resource):
else "" else ""
), ),
"description": shared_agent.get("description", ""), "description": shared_agent.get("description", ""),
"source": shared_agent.get("source", ""), "source": (
str(source_doc["_id"])
if isinstance(shared_agent.get("source"), DBRef)
and (source_doc := db.dereference(shared_agent.get("source")))
else ""
),
"chunks": shared_agent.get("chunks", "0"), "chunks": shared_agent.get("chunks", "0"),
"retriever": shared_agent.get("retriever", "classic"), "retriever": shared_agent.get("retriever", "classic"),
"prompt_id": shared_agent.get("prompt_id", "default"), "prompt_id": shared_agent.get("prompt_id", "default"),