fix: 'NoneType' object has no attribute '__dict__' (#978)

langfuse · Oct 25, 2024 · 7d15848 · 7d15848
1 parent 3d2655b
commit 7d15848
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/langfuse/openai.py b/langfuse/openai.py
@@ -519,7 +519,7 @@ def _get_langfuse_data_from_default_response(resource: OpenAiDefinition, respons
 
     usage = response.get("usage", None)
 
-    return model, completion, usage.__dict__ if _is_openai_v1() else usage
+    return model, completion, usage.__dict__ if _is_openai_v1() and usage is not None else usage
 
 
 def _is_openai_v1():