From aa2e7c888eaffbd1207df521b43ba69e36d6a976 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Fri, 14 Feb 2025 15:57:38 -0800 Subject: [PATCH] Update test_events to validate tool usage error event handling - Modify test to assert single error event with correct attributes - Use pytest.raises() to verify error event generation - Simplify error event validation in test case --- tests/utilities/test_events.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/utilities/test_events.py b/tests/utilities/test_events.py index 011317ad9..8a1717ff5 100644 --- a/tests/utilities/test_events.py +++ b/tests/utilities/test_events.py @@ -318,14 +318,15 @@ def test_tools_emits_error_events(): ) crew = Crew(agents=[agent], tasks=[task], name="TestCrew") - crew.kickoff() + with pytest.raises(Exception): + crew.kickoff() - assert len(received_events) == 75 + assert len(received_events) == 1 assert received_events[0].agent_key == agent.key assert received_events[0].agent_role == agent.role assert received_events[0].tool_name == "error_tool" assert received_events[0].tool_args == {} - assert received_events[0].error == "Simulated tool error" + assert str(received_events[0].error) == "Simulated tool error" assert received_events[0].type == "tool_usage_error" assert isinstance(received_events[0].timestamp, datetime)