fix tracing (#172)

Azure-Samples · Sep 5, 2024 · 61e4bbb · 61e4bbb
1 parent 1387e1f
commit 61e4bbb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/src/api/evaluate/evaluate.py b/src/api/evaluate/evaluate.py
@@ -154,7 +154,7 @@ def evaluate_row(research_context, product_context, assignment_context):
     print("value: ", os.environ["BING_SEARCH_KEY"], len(os.environ["BING_SEARCH_KEY"]))
 
 
-    tracer = init_tracing()
+    tracer = init_tracing(local_tracing=True)
 
     eval_result = evaluate_orchestrator(model_config, data_path=folder +"/eval_inputs.jsonl")
 

diff --git a/src/api/tracing.py b/src/api/tracing.py
@@ -42,20 +42,20 @@ def init_tracing(local_tracing: bool = True):
     else:
         Tracer.add("OpenTelemetry", trace_span)
 
-    azmon_logger = logging.getLogger("azure")
-    azmon_logger.setLevel(logging.INFO)
+        azmon_logger = logging.getLogger("azure")
+        azmon_logger.setLevel(logging.INFO)
 
-    # oteltrace.set_tracer_provider(TracerProvider())
+        # oteltrace.set_tracer_provider(TracerProvider())
 
-    # Configure Azure Monitor as the Exporter
-    app_insights = os.getenv("APPINSIGHTS_CONNECTIONSTRING")
+        # Configure Azure Monitor as the Exporter
+        app_insights = os.getenv("APPINSIGHTS_CONNECTIONSTRING")
 
-    # Add the Azure exporter to the tracer provider
+        # Add the Azure exporter to the tracer provider
 
-    oteltrace.set_tracer_provider(TracerProvider(sampler=ParentBasedTraceIdRatio(1.0)))
-    oteltrace.get_tracer_provider().add_span_processor(BatchSpanProcessor(AzureMonitorTraceExporter(connection_string=app_insights)))
-    # oteltrace.get_tracer_provider().add_span_processor(
-    #     SimpleSpanProcessor(trace_exporter)
-    # )
+        oteltrace.set_tracer_provider(TracerProvider(sampler=ParentBasedTraceIdRatio(1.0)))
+        oteltrace.get_tracer_provider().add_span_processor(BatchSpanProcessor(AzureMonitorTraceExporter(connection_string=app_insights)))
+        # oteltrace.get_tracer_provider().add_span_processor(
+        #     SimpleSpanProcessor(trace_exporter)
+        # )
 
-    return oteltrace.get_tracer(_tracer)
+        return oteltrace.get_tracer(_tracer)