mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-07 02:02:35 +00:00
test: name crew explicitly in trace listener test and fix crew test
This commit is contained in:
@@ -8,6 +8,7 @@ from concurrent.futures import Future
|
|||||||
from hashlib import md5
|
from hashlib import md5
|
||||||
import re
|
import re
|
||||||
import sys
|
import sys
|
||||||
|
from typing import Any
|
||||||
from unittest.mock import ANY, MagicMock, call, patch
|
from unittest.mock import ANY, MagicMock, call, patch
|
||||||
|
|
||||||
from crewai.agent import Agent
|
from crewai.agent import Agent
|
||||||
|
|||||||
@@ -276,7 +276,12 @@ class TestTraceListenerSetup:
|
|||||||
expected_output="hello world",
|
expected_output="hello world",
|
||||||
agent=agent,
|
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
|
from crewai.events.event_bus import crewai_event_bus
|
||||||
|
|
||||||
@@ -304,7 +309,7 @@ class TestTraceListenerSetup:
|
|||||||
# Verify the first completion event has proper structure
|
# Verify the first completion event has proper structure
|
||||||
completion_event = completion_events[0]
|
completion_event = completion_events[0]
|
||||||
assert "crew_name" in completion_event.event_data
|
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
|
# Verify all events have proper structure
|
||||||
for call in add_event_mock.call_args_list:
|
for call in add_event_mock.call_args_list:
|
||||||
|
|||||||
Reference in New Issue
Block a user