andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 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 108a0d36b7
commit 94cb96b288
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 self.ready = False
try: try:
telemetry_endpoint = "http://telemetry.crewai.com:4318" 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) self.provider = TracerProvider(resource=self.resource)
processor = BatchSpanProcessor( 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.provider.add_span_processor(processor)
self.ready = True self.ready = True
@@ -53,7 +55,10 @@ class Telemetry:
def set_tracer(self): def set_tracer(self):
if self.ready: 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): def crew_creation(self, crew):
"""Records the creation of a crew.""" """Records the creation of a crew."""