Fix telemetry endpoint (opea-project#66)

* fix telemetry endpoint Signed-off-by: Spycsh <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Spycsh <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
poussa · May 17, 2024 · 16a16eb · 16a16eb
1 parent 5d56d7f
commit 16a16eb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/comps/cores/telemetry/opea_telemetry.py b/comps/cores/telemetry/opea_telemetry.py
@@ -14,6 +14,7 @@
 
 
 import inspect
+import os
 from functools import wraps
 
 from opentelemetry import trace
@@ -23,12 +24,12 @@
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
 
+telemetry_endpoint = os.environ.get("TELEMETRY_ENDPOINT", "http://localhost:4318/v1/traces")
+
 resource = Resource.create({SERVICE_NAME: "opea"})
 traceProvider = TracerProvider(resource=resource)
 
-traceProvider.add_span_processor(
-    BatchSpanProcessor(HTTPSpanExporter(endpoint="http://10.165.57.68:4318/v1/traces"))  # change to ip
-)
+traceProvider.add_span_processor(BatchSpanProcessor(HTTPSpanExporter(endpoint=telemetry_endpoint)))
 in_memory_exporter = InMemorySpanExporter()
 traceProvider.add_span_processor(BatchSpanProcessor(in_memory_exporter))
 trace.set_tracer_provider(traceProvider)