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

(feat:attach) fallback strategy to process docs

Browse Source
This commit is contained in:
ManishMadan2882
2025-04-03 03:26:37 +05:30
parent 19d68252cd
commit f9ad4c068a
3 changed files with 37 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
application/agents/base.py
View File

@@ -23,7 +23,7 @@ class BaseAgent(ABC):
prompt: str = "",
chat_history: Optional[List[Dict]] = None,
decoded_token: Optional[Dict] = None,
attachments: Optional[str]=None,
attachments: Optional[List[Dict]]=None,
):
self.endpoint = endpoint
self.llm_name = llm_name
@@ -44,7 +44,7 @@ class BaseAgent(ABC):
decoded_token=decoded_token,
)
self.llm_handler = get_llm_handler(llm_name)
set.attachments = attachments or []
self.attachments = attachments or []
@log_activity()
def gen(
@@ -243,8 +243,9 @@ class BaseAgent(ABC):
tools_dict: Dict,
messages: List[Dict],
log_context: Optional[LogContext] = None,
attachments: Optional[List[Dict]] = None
):
resp = self.llm_handler.handle_response(self, resp, tools_dict, messages)
resp = self.llm_handler.handle_response(self, resp, tools_dict, messages, attachments)
if log_context:
data = build_stack_data(self.llm_handler)
log_context.stacks.append({"component": "llm_handler", "data": data})