feat(tracing): add support for configurable tracing protocol in setup_tracing function

src/agentscope/__init__.py

@@ -73,6 +73,7 @@ def init(
     logging_level: str = "INFO",
     studio_url: str | None = None,
     tracing_url: str | None = None,
+    tracing_protocol: str = "http/protobuf",
 ) -> None:
     """Initialize the agentscope library.
 
@@ -97,6 +98,11 @@ def init(
             OpenTelemetry tracing platforms like Arize-Phoenix and Langfuse.
             If not provided and `studio_url` is provided, it will send traces
             to the AgentScope Studio's tracing endpoint.
+        tracing_protocol (`str`, optional):
+            The protocol to use for the trace exporter. Supported values are:
+            - "grpc": Use gRPC protocol
+            - "http/protobuf": Use HTTP protocol with protobuf encoding
+            Defaults to "http/protobuf".
     """
 
     if project:
@@ -148,7 +154,7 @@ def init(
     if endpoint:
         from .tracing import setup_tracing
 
-        setup_tracing(endpoint=endpoint)
+        setup_tracing(endpoint=endpoint, protocol=tracing_protocol)
         _config.trace_enabled = True
 
 

src/agentscope/tracing/_setup.py

@@ -2,22 +2,35 @@
 """The tracing interface class in agentscope."""
 
 
-def setup_tracing(endpoint: str) -> None:
+def setup_tracing(
+    endpoint: str,
+    protocol: str = "http/protobuf",
+) -> None:
     """Set up the AgentScope tracing by configuring the endpoint URL.
 
     Args:
         endpoint (`str`):
             The endpoint URL for the tracing exporter.
+        protocol (`str`, optional):
+            The protocol to use for the trace exporter. Supported values are:
+            - "grpc": Use gRPC protocol
+            - "http/protobuf": Use HTTP protocol with protobuf encoding
+            Defaults to "http/protobuf".
     """
     # Lazy import
     from opentelemetry.sdk.trace import TracerProvider
     from opentelemetry.sdk.trace.export import BatchSpanProcessor
-    from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
-        OTLPSpanExporter,
-    )
     from opentelemetry import trace
 
     tracer_provider = TracerProvider()
+    if protocol == "grpc":
+        from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+            OTLPSpanExporter,
+        )
+    else:
+        from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+            OTLPSpanExporter,
+        )
     exporter = OTLPSpanExporter(endpoint=endpoint)
     span_processor = BatchSpanProcessor(exporter)
     tracer_provider.add_span_processor(span_processor)