diff --git a/lib/crewai/tests/test_crew.py b/lib/crewai/tests/test_crew.py index 795ce9a55..9d47be97b 100644 --- a/lib/crewai/tests/test_crew.py +++ b/lib/crewai/tests/test_crew.py @@ -8,6 +8,7 @@ from concurrent.futures import Future from hashlib import md5 import re import sys +from typing import Any from unittest.mock import ANY, MagicMock, call, patch from crewai.agent import Agent diff --git a/lib/crewai/tests/tracing/test_tracing.py b/lib/crewai/tests/tracing/test_tracing.py index 267b30072..46bb53615 100644 --- a/lib/crewai/tests/tracing/test_tracing.py +++ b/lib/crewai/tests/tracing/test_tracing.py @@ -276,7 +276,12 @@ class TestTraceListenerSetup: expected_output="hello world", agent=agent, ) - crew = Crew(agents=[agent], tasks=[task], verbose=True) + crew = Crew( + name="Hello World Crew", + agents=[agent], + tasks=[task], + verbose=True, + ) from crewai.events.event_bus import crewai_event_bus @@ -304,7 +309,7 @@ class TestTraceListenerSetup: # Verify the first completion event has proper structure completion_event = completion_events[0] assert "crew_name" in completion_event.event_data - assert completion_event.event_data["crew_name"] == "Crew" + assert completion_event.event_data["crew_name"] == "Hello World Crew" # Verify all events have proper structure for call in add_event_mock.call_args_list: