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

adding function calling llm support

Browse Source
This commit is contained in:
João Moura
2024-02-13 02:57:12 -08:00
parent 2410d0c531
commit 55c0c186d1
11 changed files with 4378 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/agents/executor.py
View File

@@ -24,6 +24,7 @@ class CrewAgentExecutor(AgentExecutor):
task: Any = None
tools_description: str = ""
tools_names: str = ""
function_calling_llm: Any = None
request_within_rpm_limit: Any = None
tools_handler: InstanceOf[ToolsHandler] = None
max_iterations: Optional[int] = 15
@@ -194,6 +195,7 @@ class CrewAgentExecutor(AgentExecutor):
tools=self.tools,
tools_description=self.tools_description,
tools_names=self.tools_names,
function_calling_llm=self.function_calling_llm,
llm=self.llm,
task=self.task,
).use(agent_action.log)

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

@@ -22,9 +22,9 @@ class ToolsHandler:
def on_tool_end(self, calling: ToolCalling, output: str) -> Any:
"""Run when tool ends running."""
if self.last_used_tool.function_name != CacheTools().name:
if self.last_used_tool.tool_name != CacheTools().name:
self.cache.add(
tool=calling.function_name,
tool=calling.tool_name,
input=calling.arguments,
output=output,
)