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: update cache tests to use input_data parameter name

Browse Source 
The CacheHandler methods use 'input_data' not 'input' as the parameter name
This commit is contained in:
Greyson LaLonde
2025-09-03 23:09:51 -04:00
parent 90ca02b9dc
commit b6e7311d2d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_crew.py
View File

@@ -904,13 +904,13 @@ def test_cache_hitting_between_agents(researcher, writer, ceo):
crew.kickoff() crew.kickoff()
assert read.call_count == 2, "read was not called exactly twice" assert read.call_count == 2, "read was not called exactly twice"
# Filter the mock calls to only include the ones with 'tool' and 'input' keywords # Filter the mock calls to only include the ones with 'tool' and 'input_data' keywords
cache_calls = [ cache_calls = [
call call
for call in read.call_args_list for call in read.call_args_list
if len(call.kwargs) == 2 if len(call.kwargs) == 2
and "tool" in call.kwargs and "tool" in call.kwargs
and "input" in call.kwargs and "input_data" in call.kwargs
] ]
# Check if we have the expected number of cache calls # Check if we have the expected number of cache calls
@@ -918,7 +918,7 @@ def test_cache_hitting_between_agents(researcher, writer, ceo):
# Check if both calls were made with the expected arguments # Check if both calls were made with the expected arguments
expected_call = call( expected_call = call(
tool="multiplier", input={"first_number": 2, "second_number": 6} tool="multiplier", input_data={"first_number": 2, "second_number": 6}
) )
assert cache_calls[0] == expected_call, f"First call mismatch: {cache_calls[0]}" assert cache_calls[0] == expected_call, f"First call mismatch: {cache_calls[0]}"
assert cache_calls[1] == expected_call, ( assert cache_calls[1] == expected_call, (
@@ -2229,7 +2229,7 @@ def test_tools_with_custom_caching():
# Verify that one of those calls was with the even number that should be cached # Verify that one of those calls was with the even number that should be cached
add_to_cache.assert_any_call( add_to_cache.assert_any_call(
tool="multiplcation_tool", tool="multiplcation_tool",
input={"first_number": 2, "second_number": 6}, input_data={"first_number": 2, "second_number": 6},
output=12, output=12,
) )