fix(masking): scope mask function to resources singleton

Author: hassiebp

langfuse/_client/client.py

@@ -162,7 +162,6 @@ def __init__(
     ):
         self._host = host or os.environ.get(LANGFUSE_HOST, "https://cloud.langfuse.com")
         self._environment = environment or os.environ.get(LANGFUSE_TRACING_ENVIRONMENT)
-        self._mask = mask
         self._project_id = None
         sample_rate = sample_rate or float(os.environ.get(LANGFUSE_SAMPLE_RATE, 1.0))
         if not 0.0 <= sample_rate <= 1.0:
@@ -219,7 +218,9 @@ def __init__(
             httpx_client=httpx_client,
             media_upload_thread_count=media_upload_thread_count,
             sample_rate=sample_rate,
+            mask=mask,
         )
+        self._mask = self._resources.mask
 
         self._otel_tracer = (
             self._resources.tracer

langfuse/_client/resource_manager.py

@@ -18,7 +18,7 @@
 import os
 import threading
 from queue import Full, Queue
-from typing import Dict, Optional, cast, Any
+from typing import Any, Dict, Optional, cast
 
 import httpx
 from opentelemetry import trace as otel_trace_api
@@ -43,6 +43,7 @@
 from langfuse._utils.request import LangfuseClient
 from langfuse.api.client import AsyncFernLangfuse, FernLangfuse
 from langfuse.logger import langfuse_logger
+from langfuse.types import MaskFunction
 
 from ..version import __version__ as langfuse_version
 
@@ -90,6 +91,7 @@ def __new__(
         httpx_client: Optional[httpx.Client] = None,
         media_upload_thread_count: Optional[int] = None,
         sample_rate: Optional[float] = None,
+        mask: Optional[MaskFunction] = None,
     ) -> "LangfuseResourceManager":
         if public_key in cls._instances:
             return cls._instances[public_key]
@@ -110,6 +112,7 @@ def __new__(
                     httpx_client=httpx_client,
                     media_upload_thread_count=media_upload_thread_count,
                     sample_rate=sample_rate,
+                    mask=mask,
                 )
 
                 cls._instances[public_key] = instance
@@ -130,8 +133,10 @@ def _initialize_instance(
         media_upload_thread_count: Optional[int] = None,
         httpx_client: Optional[httpx.Client] = None,
         sample_rate: Optional[float] = None,
+        mask: Optional[MaskFunction] = None,
     ):
         self.public_key = public_key
+        self.mask = mask
 
         # OTEL Tracer
         tracer_provider = _init_tracer_provider(