tracing with phoenix

README.md

@@ -43,17 +43,16 @@ The application includes a modular tracing system that supports both:
   ```
 - View traces at http://localhost:6006
 
-You can enable either or both systems by configuring the tracing setup:
+You can configure which tracing system to use:
 
 ```python
-# Use just Langfuse
+# Use Langfuse (default)
 initialize_tracing(provider="langfuse")
 
-# Use just Phoenix
+# Use Phoenix
 initialize_tracing(provider="phoenix")
-
-# Use both (default)
-initialize_tracing(provider="both")
 ```
 
+Due to limitations in the OpenTelemetry integration, only one provider can be active at a time. Simply choose the provider that best meets your needs.
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

agent.py

@@ -14,13 +14,13 @@ load_dotenv()
 trace_provider = None
 
 
-def initialize_tracing(enabled=True, provider="both"):
+def initialize_tracing(enabled=True, provider="langfuse"):
     """
     Initialize tracing for the agent module
 
     Args:
         enabled: Whether tracing should be active
-        provider: Which provider(s) to use - "langfuse", "phoenix", or "both"
+        provider: Which provider to use - "langfuse" or "phoenix"
     """
     global trace_provider
     if trace_provider is None:
@@ -63,8 +63,9 @@ def get_agent():
 
 
 if __name__ == "__main__":
-    # Initialize tracing when run directly - enable both Langfuse and Phoenix
-    initialize_tracing(enabled=True, provider="both")
+    # Initialize tracing when run directly
+    # Choose one provider: "langfuse" (default) or "phoenix"
+    initialize_tracing(enabled=True, provider="phoenix")
 
     # Get agent with tracing already configured
     agent = get_agent()
@@ -76,6 +77,6 @@ if __name__ == "__main__":
     )
     print(f"Result: {result}")
     print(
-        "View traces in Phoenix UI (run 'python -m phoenix.server.main serve' in another terminal)"
+        "If using Phoenix: run 'python -m phoenix.server.main serve' and view at http://localhost:6006"
     )
-    print("View traces in Langfuse dashboard (if configured)")
+    print("If using Langfuse: view traces at https://cloud.langfuse.com")

tracing.py

@@ -11,15 +11,15 @@ from openinference.instrumentation.smolagents import SmolagentsInstrumentor
 def setup_tracing(
     service_name: str = "agent-service",
     enabled: bool = True,
-    provider: Literal["langfuse", "phoenix", "both"] = "langfuse",
+    provider: Literal["langfuse", "phoenix"] = "langfuse",
 ) -> Optional[TracerProvider]:
     """
-    Configure and set up OpenTelemetry tracing with Langfuse and/or Phoenix integration.
+    Configure and set up OpenTelemetry tracing with Langfuse or Phoenix integration.
 
     Args:
         service_name: Name of the service for trace identification
         enabled: Whether tracing should be active
-        provider: Which tracing provider(s) to use - "langfuse", "phoenix", or "both"
+        provider: Which tracing provider to use - "langfuse" or "phoenix"
 
     Returns:
         Configured TracerProvider instance or None if disabled
@@ -27,15 +27,12 @@ def setup_tracing(
     if not enabled:
         return None
 
-    trace_provider = None
-
-    # Setup Phoenix if requested
-    if provider in ["phoenix", "both"]:
+    # Setup Phoenix
+    if provider == "phoenix":
         try:
             from phoenix.otel import register
 
             # Create Phoenix tracer provider
-
             trace_provider = register(
                 project_name=service_name,
                 endpoint="http://127.0.0.1:6006/v1/traces",
@@ -43,11 +40,16 @@ def setup_tracing(
             )
             print(f"✅ Phoenix tracing enabled for {service_name}")
 
+            # Instrument smolagents with Phoenix provider
+            SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+            return trace_provider
+
         except ImportError:
             print("⚠️ Phoenix not installed. Run: pip install 'smolagents[telemetry]'")
+            return None
 
-    # Setup Langfuse if requested
-    if provider in ["langfuse", "both"]:
+    # Setup Langfuse (default)
+    else:
         # Configure Langfuse host
         langfuse_host = os.environ.get("LANGFUSE_HOST", "https://cloud.langfuse.com")
 
@@ -57,38 +59,30 @@ def setup_tracing(
 
         if not langfuse_public_key or not langfuse_secret_key:
             print("⚠️ Langfuse keys not found in environment")
+            return None
 
-            # If Phoenix is not set up either, return None
-            if trace_provider is None:
-                return None
-        else:
-            langfuse_auth = base64.b64encode(
-                f"{langfuse_public_key}:{langfuse_secret_key}".encode()
-            ).decode()
-
-            # Set environment variables for OTLP exporter
-            os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = (
-                f"{langfuse_host}/api/public/otel"
-            )
-            os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = (
-                f"Authorization=Basic {langfuse_auth}"
-            )
+        # Configure Langfuse
+        langfuse_auth = base64.b64encode(
+            f"{langfuse_public_key}:{langfuse_secret_key}".encode()
+        ).decode()
+
+        # Set environment variables for OTLP exporter
+        os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = f"{langfuse_host}/api/public/otel"
+        os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = (
+            f"Authorization=Basic {langfuse_auth}"
+        )
 
-            if trace_provider is None:
-                # Create new TracerProvider if Phoenix wasn't enabled
-                trace_provider = TracerProvider()
-                # Set as the global default tracer provider
-                trace.set_tracer_provider(trace_provider)
+        # Create a new TracerProvider for Langfuse
+        langfuse_provider = TracerProvider()
 
-            # Add Langfuse exporter to the tracer provider
-            trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
-            print(f"✅ Langfuse tracing enabled for {service_name}")
+        # Add Langfuse exporter
+        langfuse_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
 
-    if trace_provider:
-        # Instrument smolagents with the configured provider
-        SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+        # Set as the global default tracer provider
+        trace.set_tracer_provider(langfuse_provider)
 
-        # Create a single tracer for the entire application
-        trace.get_tracer(service_name)
+        print(f"✅ Langfuse tracing enabled for {service_name}")
 
-    return trace_provider
+        # Instrument smolagents
+        SmolagentsInstrumentor().instrument(tracer_provider=langfuse_provider)
+        return langfuse_provider