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

track tool usage

Browse Source
This commit is contained in:
Braelyn Boynton
2024-03-20 18:25:41 -07:00
parent f6ee12dbc5
commit f67d0a26f1
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/tools/tool_usage.py
View File

@@ -9,7 +9,7 @@ from crewai.agents.tools_handler import ToolsHandler
from crewai.telemetry import Telemetry
from crewai.tools.tool_calling import InstructorToolCalling, ToolCalling
from crewai.utilities import I18N, Converter, ConverterError, Printer
from agentops import record, ToolEvent
from agentops import ToolEvent, ErrorEvent, record
OPENAI_BIGGER_MODELS = ["gpt-4"]
@@ -79,13 +79,16 @@ class ToolUsage:
self._printer.print(content=f"\n\n{error}\n", color="red")
self.task.increment_tools_errors()
return error
event = ToolEvent(name=calling.tool_name)
record(event)
try:
tool = self._select_tool(calling.tool_name)
record(ToolEvent(name=calling.tool_name))
except Exception as e:
error = getattr(e, "message", str(e))
self.task.increment_tools_errors()
self._printer.print(content=f"\n\n{error}\n", color="red")
record(ErrorEvent(details=error, trigger_event=event))
return error
return f"{self._use(tool_string=tool_string, tool=tool, calling=calling)}"