File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -172,7 +172,9 @@ def get_openai_args(self):
172172 # If OpenAI model distillation is enabled, we need to add the metadata to the kwargs
173173 # https://platform.openai.com/docs/guides/distillation
174174 if self .kwargs .get ("store" , False ):
175- self .kwargs ["metadata" ] = {} if self .args .get ("metadata" , None ) is None else self .args ["metadata" ]
175+ self .kwargs ["metadata" ] = (
176+ {} if self .args .get ("metadata" , None ) is None else self .args ["metadata" ]
177+ )
176178
177179 # OpenAI does not support non-string type values in metadata when using
178180 # model distillation feature
@@ -771,6 +773,7 @@ def initialize(self):
771773 enabled = openai .langfuse_enabled ,
772774 sdk_integration = "openai" ,
773775 sample_rate = openai .langfuse_sample_rate ,
776+ mask = openai .langfuse_mask ,
774777 )
775778
776779 return self ._langfuse
You can’t perform that action at this time.
0 commit comments