fix public key as constructor arg

Author: hassiebp

langfuse/_client/client.py

@@ -142,6 +142,9 @@ class Langfuse:
         ```
     """
 
+    _resources: Optional[LangfuseResourceManager] = None
+    _mask: Optional[MaskFunction] = None
+
     def __init__(
         self,
         *,
@@ -190,7 +193,6 @@ def __init__(
             langfuse_logger.warning(
                 "Authentication error: Langfuse client initialized without public_key. Client will be disabled. "
                 "Provide a public_key parameter or set LANGFUSE_PUBLIC_KEY environment variable. "
-                "See documentation: https://langfuse.com/docs/sdk/python/low-level-sdk#initialize-client"
             )
             self._otel_tracer = otel_trace_api.NoOpTracer()
             return
@@ -200,7 +202,6 @@ def __init__(
             langfuse_logger.warning(
                 "Authentication error: Langfuse client initialized without secret_key. Client will be disabled. "
                 "Provide a secret_key parameter or set LANGFUSE_SECRET_KEY environment variable. "
-                "See documentation: https://langfuse.com/docs/sdk/python/low-level-sdk#initialize-client"
             )
             self._otel_tracer = otel_trace_api.NoOpTracer()
             return
@@ -1326,7 +1327,9 @@ def create_score(
                 "timestamp": _get_timestamp(),
                 "body": new_body,
             }
-            self._resources.add_score_task(event)
+
+            if self._resources is not None:
+                self._resources.add_score_task(event)
 
         except Exception as e:
             langfuse_logger.exception(
@@ -1520,7 +1523,8 @@ def flush(self):
             # Continue with other work
             ```
         """
-        self._resources.flush()
+        if self._resources is not None:
+            self._resources.flush()
 
     def shutdown(self):
         """Shut down the Langfuse client and flush all pending data.
@@ -1544,7 +1548,8 @@ def shutdown(self):
             langfuse.shutdown()
             ```
         """
-        self._resources.shutdown()
+        if self._resources is not None:
+            self._resources.shutdown()
 
     def get_current_trace_id(self) -> Optional[str]:
         """Get the trace ID of the current active span.
@@ -1926,6 +1931,10 @@ def get_prompt(
             Exception: Propagates any exceptions raised during the fetching of a new prompt, unless there is an
             expired prompt in the cache, in which case it logs a warning and returns the expired prompt.
         """
+        if self._resources is None:
+            raise Error(
+                "SDK is not correctly initalized. Check the init logs for more details."
+            )
         if version is not None and label is not None:
             raise ValueError("Cannot specify both version and label at the same time.")
 
@@ -2051,7 +2060,8 @@ def fetch_prompts():
             else:
                 prompt = TextPromptClient(prompt_response)
 
-            self._resources.prompt_cache.set(cache_key, prompt, ttl_seconds)
+            if self._resources is not None:
+                self._resources.prompt_cache.set(cache_key, prompt, ttl_seconds)
 
             return prompt
 
@@ -2150,7 +2160,8 @@ def create_prompt(
                 )
                 server_prompt = self.api.prompts.create(request=request)
 
-                self._resources.prompt_cache.invalidate(name)
+                if self._resources is not None:
+                    self._resources.prompt_cache.invalidate(name)
 
                 return ChatPromptClient(prompt=cast(Prompt_Chat, server_prompt))
 
@@ -2169,7 +2180,8 @@ def create_prompt(
 
             server_prompt = self.api.prompts.create(request=request)
 
-            self._resources.prompt_cache.invalidate(name)
+            if self._resources is not None:
+                self._resources.prompt_cache.invalidate(name)
 
             return TextPromptClient(prompt=cast(Prompt_Text, server_prompt))
 
@@ -2200,7 +2212,9 @@ def update_prompt(
             version=version,
             new_labels=new_labels,
         )
-        self._resources.prompt_cache.invalidate(name)
+
+        if self._resources is not None:
+            self._resources.prompt_cache.invalidate(name)
 
         return updated_prompt
 

langfuse/_client/get_client.py

@@ -56,7 +56,16 @@ def get_client(*, public_key: Optional[str] = None) -> Langfuse:
 
             if len(active_instances) == 1:
                 # Only one client exists, safe to use without specifying key
-                return Langfuse(public_key=public_key)
+                instance = list(active_instances.values())[0]
+
+                # Initialize with the credentials bound to the instance
+                # This is important if the original instance was instantiated
+                # via constructor arguments
+                return Langfuse(
+                    public_key=instance.public_key,
+                    secret_key=instance.secret_key,
+                    host=instance.host,
+                )
 
             else:
                 # Multiple clients exist but no key specified - disable tracing

langfuse/_client/resource_manager.py

@@ -136,6 +136,8 @@ def _initialize_instance(
         mask: Optional[MaskFunction] = None,
     ):
         self.public_key = public_key
+        self.secret_key = secret_key
+        self.host = host
         self.mask = mask
 
         # OTEL Tracer

langfuse/_client/span.py

@@ -410,7 +410,7 @@ def _process_media_and_apply_mask(
             The processed and masked data
         """
         return self._mask_attribute(
-            data=self._process_media_in_attribute(data=data, span=span, field=field)
+            data=self._process_media_in_attribute(data=data, field=field)
         )
 
     def _mask_attribute(self, *, data):
@@ -441,7 +441,6 @@ def _process_media_in_attribute(
         self,
         *,
         data: Optional[Any] = None,
-        span: otel_trace_api.Span,
         field: Union[Literal["input"], Literal["output"], Literal["metadata"]],
     ):
         """Process any media content in the attribute data.
@@ -457,16 +456,17 @@ def _process_media_in_attribute(
         Returns:
             The data with any media content processed
         """
-        media_processed_attribute = (
-            self._langfuse_client._resources._media_manager._find_and_process_media(
-                data=data,
-                field=field,
-                trace_id=self.trace_id,
-                observation_id=self.id,
+        if self._langfuse_client._resources is not None:
+            return (
+                self._langfuse_client._resources._media_manager._find_and_process_media(
+                    data=data,
+                    field=field,
+                    trace_id=self.trace_id,
+                    observation_id=self.id,
+                )
             )
-        )
 
-        return media_processed_attribute
+        return data
 
 
 class LangfuseSpan(LangfuseSpanWrapper):

tests/test_otel.py

@@ -2355,7 +2355,7 @@ def mask_sensitive_data(data):
         # Since _process_media_in_attribute makes calls to media_manager
         original_process = span._process_media_in_attribute
 
-        def mock_process_media(*, data, span, field):
+        def mock_process_media(*, data, field):
             # Just return the data directly without processing
             return data