andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 15:48:23 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix(tests): update mock_kickoff_fn signatures to accept input_files

Browse Source
This commit is contained in:
Greyson LaLonde
2026-01-23 09:58:34 -05:00
parent ec116b6d42
commit eff3a7d115
1 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lib/crewai/tests/test_streaming.py
View File

@@ -207,10 +207,10 @@ class TestCrewKickoffStreaming:
original_kickoff = Crew.kickoff original_kickoff = Crew.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
crewai_event_bus.emit( crewai_event_bus.emit(
crew, crew,
@@ -274,10 +274,10 @@ class TestCrewKickoffStreaming:
original_kickoff = Crew.kickoff original_kickoff = Crew.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
crewai_event_bus.emit( crewai_event_bus.emit(
crew, crew,
@@ -329,10 +329,10 @@ class TestCrewKickoffStreamingAsync:
original_kickoff = Crew.kickoff original_kickoff = Crew.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
return mock_output return mock_output
@@ -356,7 +356,9 @@ class TestCrewKickoffStreamingAsync:
mock_output = MagicMock() mock_output = MagicMock()
mock_output.raw = "Test output" mock_output.raw = "Test output"
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(
self: Any, inputs: Any = None, input_files: Any = None, **kwargs: Any
) -> Any:
crewai_event_bus.emit( crewai_event_bus.emit(
crew, crew,
LLMStreamChunkEvent( LLMStreamChunkEvent(
@@ -439,10 +441,10 @@ class TestFlowKickoffStreaming:
original_kickoff = Flow.kickoff original_kickoff = Flow.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
crewai_event_bus.emit( crewai_event_bus.emit(
flow, flow,
@@ -484,10 +486,10 @@ class TestFlowKickoffStreaming:
original_kickoff = Flow.kickoff original_kickoff = Flow.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
return "flow result" return "flow result"
@@ -532,10 +534,10 @@ class TestFlowKickoffStreamingAsync:
original_kickoff = Flow.kickoff_async original_kickoff = Flow.kickoff_async
call_count = [0] call_count = [0]
async def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: async def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return await original_kickoff(self, inputs) return await original_kickoff(self, inputs, **kwargs)
else: else:
await asyncio.sleep(0.01) await asyncio.sleep(0.01)
crewai_event_bus.emit( crewai_event_bus.emit(
@@ -583,10 +585,10 @@ class TestFlowKickoffStreamingAsync:
original_kickoff = Flow.kickoff_async original_kickoff = Flow.kickoff_async
call_count = [0] call_count = [0]
async def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: async def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return await original_kickoff(self, inputs) return await original_kickoff(self, inputs, **kwargs)
else: else:
return "async flow result" return "async flow result"
@@ -615,10 +617,10 @@ class TestStreamingEdgeCases:
original_kickoff = Crew.kickoff original_kickoff = Crew.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
raise ValueError("Test error") raise ValueError("Test error")
@@ -673,10 +675,10 @@ class TestStreamingEdgeCases:
original_kickoff = Crew.kickoff original_kickoff = Crew.kickoff
call_count = [0] call_count = [0]
def mock_kickoff_fn(self: Any, inputs: Any = None) -> Any: def mock_kickoff_fn(self: Any, inputs: Any = None, **kwargs: Any) -> Any:
call_count[0] += 1 call_count[0] += 1
if call_count[0] == 1: if call_count[0] == 1:
return original_kickoff(self, inputs) return original_kickoff(self, inputs, **kwargs)
else: else:
crewai_event_bus.emit( crewai_event_bus.emit(
crew, crew,