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

Cutting new version with tool ussage bug fix

Browse Source
This commit is contained in:
João Moura
2024-02-15 23:19:12 -03:00
parent 0ee7189b82
commit 430eb23448
6 changed files with 104 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/crewai/agents/tools_handler.py
View File

@@ -16,13 +16,11 @@ class ToolsHandler:
self.cache = cache
self.last_used_tool = {}
def on_tool_start(self, calling: ToolCalling) -> Any:
"""Run when tool starts running."""
self.last_used_tool = calling
def on_tool_end(self, calling: ToolCalling, output: str) -> Any:
def on_tool_use(self, calling: ToolCalling, output: str) -> Any:
"""Run when tool ends running."""
if self.last_used_tool.tool_name != CacheTools().name:
print(f"Tool {calling.tool_name} has been used.")
self.last_used_tool = calling
if calling.tool_name != CacheTools().name:
self.cache.add(
tool=calling.tool_name,
input=calling.arguments,