push

Author: hassiebp

langfuse/extract_model.py

@@ -53,9 +53,11 @@ def _extract_model_name(
             return kwargs.get("invocation_params").get("model_name")
 
         deployment_name = None
-        if serialized.get("kwargs").get("openai_api_version"):
-            deployment_name = serialized.get("kwargs").get("deployment_version")
         deployment_version = None
+
+        if serialized.get("kwargs").get("openai_api_version"):
+            deployment_version = serialized.get("kwargs").get("deployment_version")
+
         if serialized.get("kwargs").get("deployment_name"):
             deployment_name = serialized.get("kwargs").get("deployment_name")