diff --git a/langfuse/openai.py b/langfuse/openai.py index 31206d77e..59e7e6c2a 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -172,7 +172,9 @@ def get_openai_args(self): # If OpenAI model distillation is enabled, we need to add the metadata to the kwargs # https://platform.openai.com/docs/guides/distillation if self.kwargs.get("store", False): - self.kwargs["metadata"] = {} if self.args.get("metadata", None) is None else self.args["metadata"] + self.kwargs["metadata"] = ( + {} if self.args.get("metadata", None) is None else self.args["metadata"] + ) # OpenAI does not support non-string type values in metadata when using # model distillation feature @@ -771,6 +773,7 @@ def initialize(self): enabled=openai.langfuse_enabled, sdk_integration="openai", sample_rate=openai.langfuse_sample_rate, + mask=openai.langfuse_mask, ) return self._langfuse