diff --git a/providers/common/ai/src/airflow/providers/common/ai/utils/decision.py b/providers/common/ai/src/airflow/providers/common/ai/utils/decision.py
index aaf1bc504a06e..6d7a94d49a3f7 100644
--- a/providers/common/ai/src/airflow/providers/common/ai/utils/decision.py
+++ b/providers/common/ai/src/airflow/providers/common/ai/utils/decision.py
@@ -99,8 +99,9 @@ def _choice_types() -> tuple[Any, Any]:
                 Choices as loaded_choices,
             )
         except ImportError:  # pydantic-ai < 2.46.0: build the same schema from an Enum instead
-            loaded_choice = loaded_choices = None
-        Choice, Choices = loaded_choice, loaded_choices
+            Choice = Choices = None
+        else:
+            Choice, Choices = loaded_choice, loaded_choices
     return Choice, Choices