File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -285,7 +285,9 @@ def call_llm_and_track_usage(
285285
286286 event_properties = {
287287 "$ai_provider" : provider ,
288- "$ai_model" : kwargs .get ("model" ) or getattr (response , "model" , None ) or "unknown" ,
288+ "$ai_model" : kwargs .get ("model" )
289+ or getattr (response , "model" , None )
290+ or "unknown" ,
289291 "$ai_model_parameters" : get_model_params (kwargs ),
290292 "$ai_input" : with_privacy_mode (
291293 ph_client , posthog_privacy_mode , sanitized_messages
@@ -396,7 +398,9 @@ async def call_llm_and_track_usage_async(
396398
397399 event_properties = {
398400 "$ai_provider" : provider ,
399- "$ai_model" : kwargs .get ("model" ) or getattr (response , "model" , None ) or "unknown" ,
401+ "$ai_model" : kwargs .get ("model" )
402+ or getattr (response , "model" , None )
403+ or "unknown" ,
400404 "$ai_model_parameters" : get_model_params (kwargs ),
401405 "$ai_input" : with_privacy_mode (
402406 ph_client , posthog_privacy_mode , sanitized_messages
You can’t perform that action at this time.
0 commit comments