andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #71 from strnad/add_tools_params_description

Browse Source 
arguments descriptions added to tool description
This commit is contained in:
João Moura
2024-07-14 11:30:02 -07:00
committed by GitHub
parent fc52061bc9 cb1dc13a9d
commit f0d924fab1
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -87,12 +87,15 @@ class BaseTool(BaseModel, ABC):
def _generate_description(self):
args = []
args_description = []
for arg, attribute in self.args_schema.schema()["properties"].items():
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", " ")
self.description = f"{self.name}({', '.join(args)}) - {description}"
self.description = f"{self.name}({', '.join(args)}) - {description} {', '.join(args_description)}"
class Tool(BaseTool):