From b2772fd5b9868c0f0fcc7e9eac9bd9be10287e9b Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Wed, 16 Apr 2025 11:59:34 -0700 Subject: [PATCH] refactor: create constants.py & use in telemetry - created `constants.py` for telemetry base url and service name - updated `telemetry.py` to reflect changes --- src/crewai/telemetry/constants.py | 2 ++ src/crewai/telemetry/telemetry.py | 10 +++++++--- 2 files changed, 9 insertions(+), 3 deletions(-) create mode 100644 src/crewai/telemetry/constants.py diff --git a/src/crewai/telemetry/constants.py b/src/crewai/telemetry/constants.py new file mode 100644 index 000000000..95820a774 --- /dev/null +++ b/src/crewai/telemetry/constants.py @@ -0,0 +1,2 @@ +CREWAI_TELEMETRY_BASE_URL: str = "https://telemetry.crewai.com:4319" +CREWAI_TELEMETRY_SERVICE_NAME: str = "crewAI-telemetry" diff --git a/src/crewai/telemetry/telemetry.py b/src/crewai/telemetry/telemetry.py index 6fec368b3..276e0748d 100644 --- a/src/crewai/telemetry/telemetry.py +++ b/src/crewai/telemetry/telemetry.py @@ -9,6 +9,11 @@ from contextlib import contextmanager from importlib.metadata import version from typing import TYPE_CHECKING, Any, Optional +from crewai.telemetry.constants import ( + CREWAI_TELEMETRY_BASE_URL, + CREWAI_TELEMETRY_SERVICE_NAME, +) + @contextmanager def suppress_warnings(): @@ -52,16 +57,15 @@ class Telemetry: return try: - telemetry_endpoint = "https://telemetry.crewai.com:4319" self.resource = Resource( - attributes={SERVICE_NAME: "crewAI-telemetry"}, + attributes={SERVICE_NAME: CREWAI_TELEMETRY_SERVICE_NAME}, ) with suppress_warnings(): self.provider = TracerProvider(resource=self.resource) processor = BatchSpanProcessor( OTLPSpanExporter( - endpoint=f"{telemetry_endpoint}/v1/traces", + endpoint=f"{CREWAI_TELEMETRY_BASE_URL}/v1/traces", timeout=30, ) )