feat: Support Python 3.14 for reasoning engine.

PiperOrigin-RevId: 853820058

Author: shawn-yang-google

vertexai/_genai/types/common.py

@@ -5622,12 +5622,14 @@ class CreateAgentEngineConfig(_common.BaseModel):
       - If `source_packages` is specified, the agent framework will
         default to "custom".""",
     )
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]] = Field(
-        default=None,
-        description="""The Python version to be used for the Agent Engine.
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]] = (
+        Field(
+            default=None,
+            description="""The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """,
+        )
     )
     build_options: Optional[dict[str, list[str]]] = Field(
         default=None,
@@ -5755,10 +5757,10 @@ class CreateAgentEngineConfigDict(TypedDict, total=False):
       - If `source_packages` is specified, the agent framework will
         default to "custom"."""
 
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]]
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]]
     """The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """
 
     build_options: Optional[dict[str, list[str]]]
@@ -6393,12 +6395,14 @@ class UpdateAgentEngineConfig(_common.BaseModel):
       - If `source_packages` is specified, the agent framework will
         default to "custom".""",
     )
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]] = Field(
-        default=None,
-        description="""The Python version to be used for the Agent Engine.
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]] = (
+        Field(
+            default=None,
+            description="""The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """,
+        )
     )
     build_options: Optional[dict[str, list[str]]] = Field(
         default=None,
@@ -6531,10 +6535,10 @@ class UpdateAgentEngineConfigDict(TypedDict, total=False):
       - If `source_packages` is specified, the agent framework will
         default to "custom"."""
 
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]]
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]]
     """The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """
 
     build_options: Optional[dict[str, list[str]]]
@@ -13730,12 +13734,14 @@ class AgentEngineConfig(_common.BaseModel):
       - If `source_packages` is specified, the agent framework will
         default to "custom".""",
     )
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]] = Field(
-        default=None,
-        description="""The Python version to be used for the Agent Engine.
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]] = (
+        Field(
+            default=None,
+            description="""The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """,
+        )
     )
     build_options: Optional[dict[str, list[str]]] = Field(
         default=None,
@@ -13895,10 +13901,10 @@ class AgentEngineConfigDict(TypedDict, total=False):
       - If `source_packages` is specified, the agent framework will
         default to "custom"."""
 
-    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13"]]
+    python_version: Optional[Literal["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]]
     """The Python version to be used for the Agent Engine.
       If not specified, it will use the current Python version of the environment.
-      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13".
+      Supported versions: "3.9", "3.10", "3.11", "3.12", "3.13", "3.14".
       """
 
     build_options: Optional[dict[str, list[str]]]

vertexai/agent_engines/_agent_engines.py

@@ -53,7 +53,7 @@
 
 
 _LOGGER = _utils.LOGGER
-_SUPPORTED_PYTHON_VERSIONS = ("3.9", "3.10", "3.11", "3.12", "3.13")
+_SUPPORTED_PYTHON_VERSIONS = ("3.9", "3.10", "3.11", "3.12", "3.13", "3.14")
 _DEFAULT_GCS_DIR_NAME = "agent_engine"
 _BLOB_FILENAME = "agent_engine.pkl"
 _REQUIREMENTS_FILE = "requirements.txt"

vertexai/reasoning_engines/_reasoning_engines.py

@@ -48,7 +48,7 @@
 
 
 _LOGGER = base.Logger(__name__)
-_SUPPORTED_PYTHON_VERSIONS = ("3.9", "3.10", "3.11", "3.12", "3.13")
+_SUPPORTED_PYTHON_VERSIONS = ("3.9", "3.10", "3.11", "3.12", "3.13", "3.14")
 _DEFAULT_GCS_DIR_NAME = "reasoning_engine"
 _BLOB_FILENAME = "reasoning_engine.pkl"
 _REQUIREMENTS_FILE = "requirements.txt"