mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 00:58:30 +00:00
Revert telemetry implementation changes while keeping CREWAI_DISABLE_TELEMETRY functionality
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -50,33 +50,31 @@ class Telemetry:
|
|||||||
|
|
||||||
if self._is_telemetry_disabled():
|
if self._is_telemetry_disabled():
|
||||||
return
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
telemetry_endpoint = "https://telemetry.crewai.com:4319"
|
telemetry_endpoint = "https://telemetry.crewai.com:4319"
|
||||||
|
self.resource = Resource(
|
||||||
otlp_exporter = OTLPSpanExporter(
|
attributes={SERVICE_NAME: "crewAI-telemetry"},
|
||||||
endpoint=f"{telemetry_endpoint}/v1/traces"
|
|
||||||
)
|
)
|
||||||
|
with suppress_warnings():
|
||||||
|
self.provider = TracerProvider(resource=self.resource)
|
||||||
|
|
||||||
span_processor = BatchSpanProcessor(otlp_exporter)
|
processor = BatchSpanProcessor(
|
||||||
|
OTLPSpanExporter(
|
||||||
tracer_provider = TracerProvider(
|
endpoint=f"{telemetry_endpoint}/v1/traces",
|
||||||
resource=Resource.create(
|
timeout=30,
|
||||||
{
|
|
||||||
"service.name": "crewai",
|
|
||||||
"service.version": version("crewai"),
|
|
||||||
}
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
tracer_provider.add_span_processor(span_processor)
|
self.provider.add_span_processor(processor)
|
||||||
|
|
||||||
trace.set_tracer_provider(tracer_provider)
|
|
||||||
|
|
||||||
self.tracer = trace.get_tracer("crewai")
|
|
||||||
self.ready = True
|
self.ready = True
|
||||||
except Exception:
|
except Exception as e:
|
||||||
pass
|
if isinstance(
|
||||||
|
e,
|
||||||
|
(SystemExit, KeyboardInterrupt, GeneratorExit, asyncio.CancelledError),
|
||||||
|
):
|
||||||
|
raise # Re-raise the exception to not interfere with system signals
|
||||||
|
self.ready = False
|
||||||
|
|
||||||
def _is_telemetry_disabled(self) -> bool:
|
def _is_telemetry_disabled(self) -> bool:
|
||||||
"""Check if telemetry should be disabled based on environment variables."""
|
"""Check if telemetry should be disabled based on environment variables."""
|
||||||
@@ -88,7 +86,9 @@ class Telemetry:
|
|||||||
def set_tracer(self):
|
def set_tracer(self):
|
||||||
if self.ready and not self.trace_set:
|
if self.ready and not self.trace_set:
|
||||||
try:
|
try:
|
||||||
self.trace_set = True
|
with suppress_warnings():
|
||||||
|
trace.set_tracer_provider(self.provider)
|
||||||
|
self.trace_set = True
|
||||||
except Exception:
|
except Exception:
|
||||||
self.ready = False
|
self.ready = False
|
||||||
self.trace_set = False
|
self.trace_set = False
|
||||||
|
|||||||
Reference in New Issue
Block a user