andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Increasing timeout for telemetry

Browse Source
This commit is contained in:
João Moura
2024-02-19 22:48:14 -03:00
parent a649eb8555
commit 72dea5310e
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/crewai/telemtry/telemetry.py
View File

@@ -41,10 +41,12 @@ class Telemetry:
self.ready = False
try:
telemetry_endpoint = "http://telemetry.crewai.com:4318"
self.resource = Resource(attributes={SERVICE_NAME: "crewAI-telemetry"})
self.resource = Resource(
attributes={SERVICE_NAME: "crewAI-telemetry"},
)
self.provider = TracerProvider(resource=self.resource)
processor = BatchSpanProcessor(
OTLPSpanExporter(endpoint=f"{telemetry_endpoint}/v1/traces")
OTLPSpanExporter(endpoint=f"{telemetry_endpoint}/v1/traces", timeout=60)
)
self.provider.add_span_processor(processor)
self.ready = True
@@ -53,7 +55,10 @@ class Telemetry:
def set_tracer(self):
if self.ready:
trace.set_tracer_provider(self.provider)
try:
trace.set_tracer_provider(self.provider)
except Exception:
self.ready = False
def crew_creation(self, crew):
"""Records the creation of a crew."""