Prevent crew inside flow to finish trace batch prematurely on failure

2026-01-07 23:28:30 +00:00 · 2025-12-18 09:22:37 -03:00
1 changed files with 6 additions and 5 deletions
--- a/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py
+++ b/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py
@@ -221,11 +221,12 @@ class TraceCollectionListener(BaseEventListener):
        @event_bus.on(CrewKickoffFailedEvent)
        def on_crew_failed(source: Any, event: CrewKickoffFailedEvent) -> None:
            self._handle_trace_event("crew_kickoff_failed", source, event)
-            if self.first_time_handler.is_first_time:
-                self.first_time_handler.mark_events_collected()
-                self.first_time_handler.handle_execution_completion()
-            else:
-                self.batch_manager.finalize_batch()
+            if self.batch_manager.batch_owner_type == "crew":
+                if self.first_time_handler.is_first_time:
+                    self.first_time_handler.mark_events_collected()
+                    self.first_time_handler.handle_execution_completion()
+                else:
+                    self.batch_manager.finalize_batch()

        @event_bus.on(TaskStartedEvent)
        def on_task_started(source: Any, event: TaskStartedEvent) -> None: