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

feat: enhance ClassicAgent and ReActAgent with tool preparation steps

Browse Source
This commit is contained in:
Siddhant Rai
2025-03-31 17:02:36 +05:30
parent 82d377abf5
commit fd2b6c111c
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
application/agents/classic_agent.py
View File

@@ -11,11 +11,12 @@ class ClassicAgent(BaseAgent):
self, query: str, retriever: BaseRetriever, log_context: LogContext
) -> Generator[Dict, None, None]:
retrieved_data = self._retriever_search(retriever, query, log_context)
messages = self._build_messages(self.prompt, query, retrieved_data)
tools_dict = self._get_user_tools(self.user)
self._prepare_tools(tools_dict)
messages = self._build_messages(self.prompt, query, retrieved_data)
resp = self._llm_gen(messages, log_context)
if isinstance(resp, str):
@@ -46,5 +47,6 @@ class ClassicAgent(BaseAgent):
for line in completion:
if isinstance(line, str):
yield {"answer": line}
yield {"sources": retrieved_data}
yield {"tool_calls": self.tool_calls.copy()}