fix(resource-manager): do not register processor if tracing disabled (#1215)

Author: hassiebp

langfuse/_client/client.py

@@ -219,6 +219,7 @@ def __init__(
             media_upload_thread_count=media_upload_thread_count,
             sample_rate=sample_rate,
             mask=mask,
+            tracing_enabled=self._tracing_enabled,
         )
         self._mask = self._resources.mask
 

langfuse/_client/resource_manager.py

@@ -92,6 +92,7 @@ def __new__(
         media_upload_thread_count: Optional[int] = None,
         sample_rate: Optional[float] = None,
         mask: Optional[MaskFunction] = None,
+        tracing_enabled: Optional[bool] = None,
     ) -> "LangfuseResourceManager":
         if public_key in cls._instances:
             return cls._instances[public_key]
@@ -113,6 +114,7 @@ def __new__(
                     media_upload_thread_count=media_upload_thread_count,
                     sample_rate=sample_rate,
                     mask=mask,
+                    tracing_enabled=tracing_enabled if tracing_enabled is not None else True,
                 )
 
                 cls._instances[public_key] = instance
@@ -134,33 +136,35 @@ def _initialize_instance(
         httpx_client: Optional[httpx.Client] = None,
         sample_rate: Optional[float] = None,
         mask: Optional[MaskFunction] = None,
+        tracing_enabled: bool = True,
     ):
         self.public_key = public_key
         self.secret_key = secret_key
         self.host = host
         self.mask = mask
 
         # OTEL Tracer
-        tracer_provider = _init_tracer_provider(
-            environment=environment, release=release, sample_rate=sample_rate
-        )
+        if tracing_enabled:
+            tracer_provider = _init_tracer_provider(
+                environment=environment, release=release, sample_rate=sample_rate
+            )
 
-        langfuse_processor = LangfuseSpanProcessor(
-            public_key=self.public_key,
-            secret_key=secret_key,
-            host=host,
-            timeout=timeout,
-            flush_at=flush_at,
-            flush_interval=flush_interval,
-        )
-        tracer_provider.add_span_processor(langfuse_processor)
+            langfuse_processor = LangfuseSpanProcessor(
+                public_key=self.public_key,
+                secret_key=secret_key,
+                host=host,
+                timeout=timeout,
+                flush_at=flush_at,
+                flush_interval=flush_interval,
+            )
+            tracer_provider.add_span_processor(langfuse_processor)
 
-        tracer_provider = cast(TracerProvider, otel_trace_api.get_tracer_provider())
-        self._otel_tracer = tracer_provider.get_tracer(
-            LANGFUSE_TRACER_NAME,
-            langfuse_version,
-            attributes={"public_key": self.public_key},
-        )
+            tracer_provider = cast(TracerProvider, otel_trace_api.get_tracer_provider())
+            self._otel_tracer = tracer_provider.get_tracer(
+                LANGFUSE_TRACER_NAME,
+                langfuse_version,
+                attributes={"public_key": self.public_key},
+            )
 
         # API Clients