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 45d06e6b6..f7907c69a 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 @@ -1,7 +1,7 @@ +from datetime import datetime, timezone import logging import uuid import webbrowser -from datetime import datetime, timezone from rich.console import Console from rich.panel import Panel 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 28286664e..1364f718f 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 @@ -1,8 +1,8 @@ -import time from dataclasses import dataclass, field from datetime import datetime, timezone from logging import getLogger from threading import Condition, Lock +import time from typing import Any import uuid @@ -114,10 +114,10 @@ class TraceBatchManager: """Send batch initialization to backend""" if not skip_context_check and not is_tracing_enabled_in_context(): - return + return None if not self.plus_api or not self.current_batch: - return + return None try: payload = { @@ -167,14 +167,14 @@ class TraceBatchManager: f"Error initializing trace batch: {e}. Continuing without tracing." ) self.trace_batch_id = None - return + return None if response is None: logger.warning( "Trace batch initialization failed gracefully. Continuing without tracing." ) self.trace_batch_id = None - return + return None # Fall back to ephemeral on auth failure (expired/revoked token) if response.status_code in [401, 403] and not use_ephemeral: diff --git a/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py b/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py index 2feaa18d4..9d81f1d55 100644 --- a/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py +++ b/lib/crewai/src/crewai/events/listeners/tracing/trace_listener.py @@ -780,7 +780,7 @@ class TraceCollectionListener(BaseEventListener): "crew_name": getattr(source, "name", "Unknown Crew"), "crewai_version": get_crewai_version(), } - self.batch_manager.initialize_batch(user_context, execution_metadata) + self._initialize_batch(user_context, execution_metadata) self.batch_manager.begin_event_processing() try: