From 78348b76d14bd668d810cdd611ff2e846f1b4e78 Mon Sep 17 00:00:00 2001
From: Dinesh Yeduguru <yvdinesh@gmail.com>
Date: Fri, 22 Nov 2024 18:00:59 -0800
Subject: [PATCH] remove comments

---
 llama_stack/distribution/server/server.py     |  1 -
 .../telemetry/opentelemetry/opentelemetry.py  | 88 -------------------
 2 files changed, 89 deletions(-)

diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py
index 5e372d183..8116e2b39 100644
--- a/llama_stack/distribution/server/server.py
+++ b/llama_stack/distribution/server/server.py
@@ -211,7 +211,6 @@ def create_dynamic_typed_route(func: Any, method: str):
     return endpoint
 
 
-# Add new middleware class
 class TracingMiddleware:
     def __init__(self, app):
         self.app = app
diff --git a/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py b/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
index a7b91991d..c9830fd9d 100644
--- a/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
+++ b/llama_stack/providers/remote/telemetry/opentelemetry/opentelemetry.py
@@ -5,7 +5,6 @@
 # the root directory of this source tree.
 
 import threading
-from datetime import datetime
 
 from opentelemetry import metrics, trace
 from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
@@ -17,13 +16,11 @@ from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.semconv.resource import ResourceAttributes
 
-from llama_stack.providers.utils.telemetry.tracing import generate_short_uuid
 
 from llama_stack.apis.telemetry import *  # noqa: F403
 
 from .config import OpenTelemetryConfig
 
-# Add global storage
 _GLOBAL_STORAGE = {
     "active_spans": {},
     "counters": {},
@@ -209,88 +206,3 @@ class OpenTelemetryAdapter(Telemetry):
 
     async def get_trace(self, trace_id: str) -> Trace:
         raise NotImplementedError("Trace retrieval not implemented yet")
-
-
-# Usage example
-async def main():
-    telemetry = OpenTelemetryAdapter(OpenTelemetryConfig())
-    await telemetry.initialize()
-
-    # # Log a metric event
-    # await telemetry.log_event(
-    #     MetricEvent(
-    #         trace_id="trace123",
-    #         span_id="span456",
-    #         timestamp=datetime.now(),
-    #         metric="my_metric",
-    #         value=42,
-    #         unit="count",
-    #     )
-    # )
-
-    # Log a structured event (span start)
-    trace_id = generate_short_uuid(16)
-    span_id = generate_short_uuid(8)
-    span_id_2 = generate_short_uuid(8)
-    await telemetry.log_event(
-        StructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id,
-            timestamp=datetime.now(),
-            payload=SpanStartPayload(name="my_operation"),
-        )
-    )
-
-    # Log an unstructured event
-    await telemetry.log_event(
-        UnstructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id,
-            timestamp=datetime.now(),
-            message="This is a log message",
-            severity=LogSeverity.INFO,
-        )
-    )
-
-    await telemetry.log_event(
-        StructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id_2,
-            timestamp=datetime.now(),
-            payload=SpanStartPayload(name="my_operation_2"),
-        )
-    )
-    await telemetry.log_event(
-        UnstructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id_2,
-            timestamp=datetime.now(),
-            message="This is a log message 2",
-            severity=LogSeverity.INFO,
-        )
-    )
-
-    await telemetry.log_event(
-        StructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id_2,
-            timestamp=datetime.now(),
-            payload=SpanEndPayload(status=SpanStatus.OK),
-        )
-    )
-    await telemetry.log_event(
-        StructuredLogEvent(
-            trace_id=trace_id,
-            span_id=span_id,
-            timestamp=datetime.now(),
-            payload=SpanEndPayload(status=SpanStatus.OK),
-        )
-    )
-
-    await telemetry.shutdown()
-
-
-if __name__ == "__main__":
-    import asyncio
-
-    asyncio.run(main())