andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

arguments descriptions added to tool description

Browse Source 
so now the agent knows how to use the tools params
This commit is contained in:
Jakub Strnad
2024-07-05 16:30:41 +02:00
committed by GitHub
parent cf67b424c2
commit a5d2839431
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/crewai_tools/tools/base_tool.py
View File

@@ -86,13 +86,16 @@ class BaseTool(BaseModel, ABC):
) )
def _generate_description(self): def _generate_description(self):
args = [] args = []
for arg, attribute in self.args_schema.schema()["properties"].items(): args_description = []
if "type" in attribute: for arg, attribute in self.args_schema.schema()["properties"].items():
args.append(f"{arg}: '{attribute['type']}'") if "type" in attribute:
args.append(f"{arg}: '{attribute['type']}'")
if "description" in attribute:
args_description.append(f"{arg}: '{attribute['description']}'")
description = self.description.replace("\n", " ") description = self.description.replace("\n", " ")
self.description = f"{self.name}({', '.join(args)}) - {description}" self.description = f"{self.name}({', '.join(args)}) - {description} {', '.join(args_description)}"
class Tool(BaseTool): class Tool(BaseTool):