From b623960a94e401edd056e9460204b841943ac1cb Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Tue, 18 Feb 2025 14:48:35 -0800 Subject: [PATCH] Improve type hinting for TaskCompletedEvent handler - Add explicit type annotation for TaskCompletedEvent in event_listener.py - Enhance type safety for event handling in EventListener --- src/crewai/utilities/events/event_listener.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/crewai/utilities/events/event_listener.py b/src/crewai/utilities/events/event_listener.py index 7f461429d..0dcefcd3d 100644 --- a/src/crewai/utilities/events/event_listener.py +++ b/src/crewai/utilities/events/event_listener.py @@ -140,7 +140,7 @@ class EventListener(BaseEventListener): ) @crewai_event_bus.on(TaskCompletedEvent) - def on_task_completed(source, event): + def on_task_completed(source, event: TaskCompletedEvent): if source._execution_span: self._telemetry.task_ended( source._execution_span, source, source.agent.crew