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

fixing tests

Browse Source
This commit is contained in:
João Moura
2024-10-10 19:32:26 -03:00
parent 42871d9ffc
commit b887c5cf3c
2 changed files with 756 additions and 600 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1348
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

8
tests/agent_test.py
View File

@@ -73,7 +73,7 @@ def test_agent_creation():
def test_agent_default_values(): def test_agent_default_values():
agent = Agent(role="test role", goal="test goal", backstory="test backstory") agent = Agent(role="test role", goal="test goal", backstory="test backstory")
assert agent.llm.model == "gpt-4o-mini" assert agent.llm.model == "gpt-4o"
assert agent.allow_delegation is False assert agent.allow_delegation is False
@@ -206,7 +206,7 @@ def test_logging_tool_usage():
verbose=True, verbose=True,
) )
assert agent.llm.model == "gpt-4o-mini" assert agent.llm.model == "gpt-4o"
assert agent.tools_handler.last_used_tool == {} assert agent.tools_handler.last_used_tool == {}
task = Task( task = Task(
description="What is 3 times 4?", description="What is 3 times 4?",
@@ -276,7 +276,9 @@ def test_cache_hitting():
"multiplier-{'first_number': 12, 'second_number': 3}": 36, "multiplier-{'first_number': 12, 'second_number': 3}": 36,
} }
with patch.object(CacheHandler, "read") as read, patch.object(Emitter, "emit") as emit: with patch.object(CacheHandler, "read") as read, patch.object(
Emitter, "emit"
) as emit:
read.return_value = "0" read.return_value = "0"
task = Task( task = Task(
description="What is 2 times 6? Ignore correctness and just return the result of the multiplication tool, you must use the tool.", description="What is 2 times 6? Ignore correctness and just return the result of the multiplication tool, you must use the tool.",