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

Fix renaming issue

Browse Source
This commit is contained in:
Brandon Hancock
2024-07-01 16:13:21 -04:00
parent f36f73e035
commit 68de393534
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/agent_test.py
View File

@@ -12,7 +12,6 @@ from crewai import Agent, Crew, Task
from crewai.agents.cache import CacheHandler from crewai.agents.cache import CacheHandler
from crewai.agents.executor import CrewAgentExecutor from crewai.agents.executor import CrewAgentExecutor
from crewai.agents.parser import CrewAgentParser from crewai.agents.parser import CrewAgentParser
from crewai.tools.tool_calling import InstructorToolCalling from crewai.tools.tool_calling import InstructorToolCalling
from crewai.tools.tool_usage import ToolUsage from crewai.tools.tool_usage import ToolUsage
from crewai.utilities import RPMController from crewai.utilities import RPMController
@@ -734,7 +733,7 @@ def test_agent_llm_uses_token_calc_handler_with_llm_has_model_name():
assert len(agent1.llm.callbacks) == 1 assert len(agent1.llm.callbacks) == 1
assert agent1.llm.callbacks[0].__class__.__name__ == "TokenCalcHandler" assert agent1.llm.callbacks[0].__class__.__name__ == "TokenCalcHandler"
assert agent1.llm.callbacks[0].model == "gpt-4o" assert agent1.llm.callbacks[0].model_name == "gpt-4o"
assert ( assert (
agent1.llm.callbacks[0].token_cost_process.__class__.__name__ == "TokenProcess" agent1.llm.callbacks[0].token_cost_process.__class__.__name__ == "TokenProcess"
) )