mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 00:58:30 +00:00
Prevent crew inside flow to finish trace batch prematurely on failure
This commit is contained in:
@@ -221,11 +221,12 @@ class TraceCollectionListener(BaseEventListener):
|
|||||||
@event_bus.on(CrewKickoffFailedEvent)
|
@event_bus.on(CrewKickoffFailedEvent)
|
||||||
def on_crew_failed(source: Any, event: CrewKickoffFailedEvent) -> None:
|
def on_crew_failed(source: Any, event: CrewKickoffFailedEvent) -> None:
|
||||||
self._handle_trace_event("crew_kickoff_failed", source, event)
|
self._handle_trace_event("crew_kickoff_failed", source, event)
|
||||||
if self.first_time_handler.is_first_time:
|
if self.batch_manager.batch_owner_type == "crew":
|
||||||
self.first_time_handler.mark_events_collected()
|
if self.first_time_handler.is_first_time:
|
||||||
self.first_time_handler.handle_execution_completion()
|
self.first_time_handler.mark_events_collected()
|
||||||
else:
|
self.first_time_handler.handle_execution_completion()
|
||||||
self.batch_manager.finalize_batch()
|
else:
|
||||||
|
self.batch_manager.finalize_batch()
|
||||||
|
|
||||||
@event_bus.on(TaskStartedEvent)
|
@event_bus.on(TaskStartedEvent)
|
||||||
def on_task_started(source: Any, event: TaskStartedEvent) -> None:
|
def on_task_started(source: Any, event: TaskStartedEvent) -> None:
|
||||||
|
|||||||
Reference in New Issue
Block a user