fix(client): use tracing_enabled in get_client (#1220)

Author: hassiebp

langfuse/_client/get_client.py

@@ -65,6 +65,7 @@ def get_client(*, public_key: Optional[str] = None) -> Langfuse:
                     public_key=instance.public_key,
                     secret_key=instance.secret_key,
                     host=instance.host,
+                    tracing_enabled=instance.tracing_enabled,
                 )
 
             else:
@@ -94,4 +95,5 @@ def get_client(*, public_key: Optional[str] = None) -> Langfuse:
                 public_key=public_key,
                 secret_key=instance.secret_key,
                 host=instance.host,
+                tracing_enabled=instance.tracing_enabled,
             )

langfuse/_client/resource_manager.py

@@ -114,7 +114,9 @@ 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,
+                    tracing_enabled=tracing_enabled
+                    if tracing_enabled is not None
+                    else True,
                 )
 
                 cls._instances[public_key] = instance
@@ -140,6 +142,7 @@ def _initialize_instance(
     ):
         self.public_key = public_key
         self.secret_key = secret_key
+        self.tracing_enabled = tracing_enabled
         self.host = host
         self.mask = mask
 

tests/test_core_sdk.py

@@ -381,7 +381,7 @@ def test_create_generation():
     assert generation_api.model_parameters == {
         "max_tokens": "1000",
         "temperature": "0.9",
-        "stop": ["user-1", "user-2"],
+        "stop": '["user-1","user-2"]',
     }
     assert generation_api.input == [
         {"role": "system", "content": "You are a helpful assistant."},