diff --git a/langfuse/_client/get_client.py b/langfuse/_client/get_client.py index 2ce1d656c..f53886e46 100644 --- a/langfuse/_client/get_client.py +++ b/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, ) diff --git a/langfuse/_client/resource_manager.py b/langfuse/_client/resource_manager.py index 72b55b276..b0acd40d8 100644 --- a/langfuse/_client/resource_manager.py +++ b/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 diff --git a/tests/test_core_sdk.py b/tests/test_core_sdk.py index 914b2fb65..b519d5877 100644 --- a/tests/test_core_sdk.py +++ b/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."},