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

improve tool text description and args (#1512)

Browse Source 
* improve tool text descriptoin and args

* fix lint

* Drop print

* add back in docstring
This commit is contained in:
Brandon Hancock (bhancock_ai)
2024-10-25 18:42:55 -04:00
committed by GitHub
parent f29f4abdd7
commit 26afee9bed
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/crewai/agent.py
View File

@@ -1,7 +1,6 @@
import os import os
import shutil import shutil
import subprocess import subprocess
from inspect import signature
from typing import Any, List, Literal, Optional, Union from typing import Any, List, Literal, Optional, Union
from pydantic import Field, InstanceOf, PrivateAttr, model_validator from pydantic import Field, InstanceOf, PrivateAttr, model_validator
@@ -395,26 +394,26 @@ class Agent(BaseAgent):
def _render_text_description_and_args(self, tools: List[Any]) -> str: def _render_text_description_and_args(self, tools: List[Any]) -> str:
"""Render the tool name, description, and args in plain text. """Render the tool name, description, and args in plain text.
Output will be in the format of: Output will be in the format of:
.. code-block:: markdown .. code-block:: markdown
search: This tool is used for search, args: {"query": {"type": "string"}} search: This tool is used for search, args: {"query": {"type": "string"}}
calculator: This tool is used for math, \ calculator: This tool is used for math, \
args: {"expression": {"type": "string"}} args: {"expression": {"type": "string"}}
""" """
tool_strings = [] tool_strings = []
for tool in tools: for tool in tools:
args_schema = str(tool.model_fields) args_schema = {
if hasattr(tool, "func") and tool.func: name: {
sig = signature(tool.func) "description": field.description,
description = ( "type": field.annotation.__name__,
f"Tool Name: {tool.name}{sig}\nTool Description: {tool.description}" }
) for name, field in tool.args_schema.model_fields.items()
else: }
description = ( description = (
f"Tool Name: {tool.name}\nTool Description: {tool.description}" f"Tool Name: {tool.name}\nTool Description: {tool.description}"
) )
tool_strings.append(f"{description}\nTool Arguments: {args_schema}") tool_strings.append(f"{description}\nTool Arguments: {args_schema}")
return "\n".join(tool_strings) return "\n".join(tool_strings)