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 pull request #1869 from siiddhantt/refactor/tools-dict

Browse Source 
refactor: update user tools dict to use enumeration based key
This commit is contained in:
Alex
2025-07-03 12:51:33 +09:00
committed by GitHub
parent 42f48649b9 1372210004
commit 839a12bed4
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
application/agents/base.py
View File

@@ -91,8 +91,8 @@ class BaseAgent(ABC):
user_tools_collection = db["user_tools"] user_tools_collection = db["user_tools"]
user_tools = user_tools_collection.find({"user": user, "status": True}) user_tools = user_tools_collection.find({"user": user, "status": True})
user_tools = list(user_tools) user_tools = list(user_tools)
tools_by_id = {str(tool["_id"]): tool for tool in user_tools}
return tools_by_id return {str(i): tool for i, tool in enumerate(user_tools)}
def _build_tool_parameters(self, action): def _build_tool_parameters(self, action):
params = {"type": "object", "properties": {}, "required": []} params = {"type": "object", "properties": {}, "required": []}
@@ -313,7 +313,6 @@ class BaseAgent(ABC):
if hasattr(response, "message") and getattr(response.message, "content", None): if hasattr(response, "message") and getattr(response.message, "content", None):
yield {"answer": response.message.content} yield {"answer": response.message.content}
return return
processed_response_gen = self._llm_handler( processed_response_gen = self._llm_handler(
response, tools_dict, messages, log_context, self.attachments response, tools_dict, messages, log_context, self.attachments
) )