andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix failing test

Browse Source
This commit is contained in:
Brandon Hancock
2025-01-14 12:56:41 -05:00
parent 79a26c1261
commit 4774420025
3 changed files with 830 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/crew_test.py
View File

@@ -3477,6 +3477,7 @@ def test_crew_guardrail_feedback_in_context():
assert task.retry_count == 1, "Task should have been retried once"
@pytest.mark.vcr(filter_headers=["authorization"])
def test_before_kickoff_callback():
from crewai.project import CrewBase, agent, before_kickoff, crew, task
@@ -3516,7 +3517,9 @@ def test_before_kickoff_callback():
def crew(self):
return Crew(agents=self.agents, tasks=self.tasks)
test_crew_instance = TestCrewClass().crew()
test_crew_instance = TestCrewClass()
crew = test_crew_instance.crew()
# Verify that the before_kickoff_callbacks are set
assert len(crew.before_kickoff_callbacks) == 1
@@ -3525,13 +3528,14 @@ def test_before_kickoff_callback():
inputs = {"initial": True}
# Call kickoff
test_crew_instance.kickoff(inputs=inputs)
crew.kickoff(inputs=inputs)
# Check that the before_kickoff function was called and modified inputs
assert test_crew_instance.inputs_modified
assert inputs.get("modified") == True
@pytest.mark.vcr(filter_headers=["authorization"])
def test_before_kickoff_without_inputs():
from crewai.project import CrewBase, agent, before_kickoff, crew, task