Fix .get not existing on Response - openai wrapper

bitterpanda63 · bitterpanda63 · commit 1344621b2f13 · 2025-06-10T16:42:06.000+02:00
diff --git a/aikido_zen/sinks/openai.py b/aikido_zen/sinks/openai.py
@@ -10,7 +10,7 @@ def _create(func, instance, args, kwargs, return_value):
 
     on_ai_call(
         provider="openai",
-        model=return_value.get("model", ""),
+        model=return_value.model,
         input_tokens=return_value.usage.input_tokens,
         output_tokens=return_value.usage.output_tokens,
     )

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ def _create(func, instance, args, kwargs, return_value):`
`10`	`10`
`11`	`11`	`on_ai_call(`
`12`	`12`	`provider="openai",`
`13`		`- model=return_value.get("model", ""),`
	`13`	`+ model=return_value.model,`
`14`	`14`	`input_tokens=return_value.usage.input_tokens,`
`15`	`15`	`output_tokens=return_value.usage.output_tokens,`
`16`	`16`	`)`