From 433d19d8bb88312d4984b47b2d2a7aa372a1fd78 Mon Sep 17 00:00:00 2001 From: Tiago Freire Date: Wed, 25 Mar 2026 10:10:52 -0300 Subject: [PATCH] Fix: ruff format --- .../listeners/tracing/first_time_trace_handler.py | 8 ++++++-- .../events/listeners/tracing/trace_batch_manager.py | 12 +++++++----- .../src/crewai/events/utils/console_formatter.py | 5 ++++- 3 files changed, 17 insertions(+), 8 deletions(-) diff --git a/lib/crewai/src/crewai/events/listeners/tracing/first_time_trace_handler.py b/lib/crewai/src/crewai/events/listeners/tracing/first_time_trace_handler.py index f7907c69a..436d50c27 100644 --- a/lib/crewai/src/crewai/events/listeners/tracing/first_time_trace_handler.py +++ b/lib/crewai/src/crewai/events/listeners/tracing/first_time_trace_handler.py @@ -105,7 +105,9 @@ class FirstTimeTraceHandler: ) if not self.batch_manager.trace_batch_id: - self._gracefully_fail("Backend batch creation failed, cannot send events.") + self._gracefully_fail( + "Backend batch creation failed, cannot send events." + ) self._reset_batch_state() return @@ -234,7 +236,9 @@ To enable tracing later, do any one of these: logger.debug(f"First-time trace error: {error_message}") - def _show_local_trace_message(self, events_count: int = 0, duration_ms: int = 0, batch_id: str | None = None) -> None: + def _show_local_trace_message( + self, events_count: int = 0, duration_ms: int = 0, batch_id: str | None = None + ) -> None: """Show message when traces were collected locally but couldn't be uploaded.""" if self.batch_manager is None: return diff --git a/lib/crewai/src/crewai/events/listeners/tracing/trace_batch_manager.py b/lib/crewai/src/crewai/events/listeners/tracing/trace_batch_manager.py index 1364f718f..1a25b68a9 100644 --- a/lib/crewai/src/crewai/events/listeners/tracing/trace_batch_manager.py +++ b/lib/crewai/src/crewai/events/listeners/tracing/trace_batch_manager.py @@ -212,7 +212,9 @@ class TraceBatchManager: def _mark_batch_as_failed(self, trace_batch_id: str, error_message: str) -> None: """Mark a trace batch as failed, routing to the correct endpoint.""" if self.is_current_batch_ephemeral: - self.plus_api.mark_ephemeral_trace_batch_as_failed(trace_batch_id, error_message) + self.plus_api.mark_ephemeral_trace_batch_as_failed( + trace_batch_id, error_message + ) else: self.plus_api.mark_trace_batch_as_failed(trace_batch_id, error_message) @@ -409,16 +411,16 @@ class TraceBatchManager: logger.error( f"❌ Failed to finalize trace batch: {response.status_code} - {response.text}" ) - self._mark_batch_as_failed( - self.trace_batch_id, response.text - ) + self._mark_batch_as_failed(self.trace_batch_id, response.text) except Exception as e: logger.error(f"❌ Error finalizing trace batch: {e}") try: self._mark_batch_as_failed(self.trace_batch_id, str(e)) except Exception: - logger.debug("Could not mark trace batch as failed (network unavailable)") + logger.debug( + "Could not mark trace batch as failed (network unavailable)" + ) def _cleanup_batch_data(self) -> None: """Clean up batch data after successful finalization to free memory""" diff --git a/lib/crewai/src/crewai/events/utils/console_formatter.py b/lib/crewai/src/crewai/events/utils/console_formatter.py index 744d8365e..f0f6f4caa 100644 --- a/lib/crewai/src/crewai/events/utils/console_formatter.py +++ b/lib/crewai/src/crewai/events/utils/console_formatter.py @@ -141,7 +141,10 @@ To update, run: uv sync --upgrade-package crewai""" # Don't show "disabled" message when the first-time handler will show # the trace prompt after execution completes (avoids confusing mid-flow messages) - if TraceCollectionListener._instance and TraceCollectionListener._instance.first_time_handler.is_first_time: + if ( + TraceCollectionListener._instance + and TraceCollectionListener._instance.first_time_handler.is_first_time + ): return if not is_tracing_enabled_in_context():