google-cloud-agentplatform 1.165.1.dev0__py3-none-any.whl

agentplatform/_genai/runtime_revisions.py

@@ -0,0 +1,1257 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+# Code generated by the Google Gen AI SDK generator DO NOT EDIT.
+
+import functools
+import json
+import logging
+from typing import Any, AsyncIterator, Iterator, Optional, Union
+from urllib.parse import urlencode
+
+from google.genai import _api_module
+from google.genai import _common
+from google.genai._common import get_value_by_path as getv
+from google.genai._common import set_value_by_path as setv
+from google.genai.pagers import AsyncPager, Pager
+
+from . import _agent_engines_utils
+from . import types
+
+logger = logging.getLogger("agentplatform_genai.runtimerevisions")
+
+logger.setLevel(logging.INFO)
+
+
+def _DeleteAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _GetAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _GetDeleteAgentEngineRuntimeRevisionOperationParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+    if getv(from_object, ["operation_name"]) is not None:
+        setv(
+            to_object, ["_url", "operationName"], getv(from_object, ["operation_name"])
+        )
+
+    return to_object
+
+
+def _ListAgentEngineRuntimeRevisionsConfig_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+
+    if getv(from_object, ["page_size"]) is not None:
+        setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"]))
+
+    if getv(from_object, ["page_token"]) is not None:
+        setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"]))
+
+    if getv(from_object, ["filter"]) is not None:
+        setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"]))
+
+    return to_object
+
+
+def _ListAgentEngineRuntimeRevisionsRequestParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["config"]) is not None:
+        _ListAgentEngineRuntimeRevisionsConfig_to_vertex(
+            getv(from_object, ["config"]), to_object
+        )
+
+    return to_object
+
+
+def _QueryAgentEngineRuntimeRevisionConfig_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+
+    if getv(from_object, ["class_method"]) is not None:
+        setv(parent_object, ["classMethod"], getv(from_object, ["class_method"]))
+
+    if getv(from_object, ["input"]) is not None:
+        setv(parent_object, ["input"], getv(from_object, ["input"]))
+
+    if getv(from_object, ["include_all_fields"]) is not None:
+        setv(to_object, ["includeAllFields"], getv(from_object, ["include_all_fields"]))
+
+    return to_object
+
+
+def _QueryAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["config"]) is not None:
+        _QueryAgentEngineRuntimeRevisionConfig_to_vertex(
+            getv(from_object, ["config"]), to_object
+        )
+
+    return to_object
+
+
+class RuntimeRevisions(_api_module.BaseModule):
+
+    def _get(
+        self,
+        *,
+        name: str,
+        config: Optional[types.GetAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.ReasoningEngineRuntimeRevision:
+        """
+        Get an agent engine runtime revision instance.
+        """
+
+        parameter_model = types._GetAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _GetAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}".format_map(request_url_dict)
+            else:
+                path = "{name}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request("get", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.ReasoningEngineRuntimeRevision._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    def _list(
+        self,
+        *,
+        name: str,
+        config: Optional[types.ListAgentEngineRuntimeRevisionsConfigOrDict] = None,
+    ) -> types.ListReasoningEnginesRuntimeRevisionsResponse:
+        """
+        Lists reasoning engine runtime revisions.
+
+        Args:
+            name (str): Required. The name of the reasoning engine to list runtime revisions for. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`.
+            config (ListAgentEngineRuntimeRevisionsConfig):
+                Optional. Additional configurations for listing the reasoning engine runtime revisions.
+
+        Returns:
+            ListReasoningEnginesRuntimeRevisionsResponse: The requested reasoning engine runtime revisions.
+
+        """
+
+        parameter_model = types._ListAgentEngineRuntimeRevisionsRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _ListAgentEngineRuntimeRevisionsRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/runtimeRevisions".format_map(request_url_dict)
+            else:
+                path = "{name}/runtimeRevisions"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request("get", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = (
+            types.ListReasoningEnginesRuntimeRevisionsResponse._from_response(
+                response=response_dict,
+                kwargs=(
+                    {
+                        "config": {
+                            "response_schema": getattr(
+                                parameter_model.config, "response_schema", None
+                            ),
+                            "response_json_schema": getattr(
+                                parameter_model.config, "response_json_schema", None
+                            ),
+                            "include_all_fields": getattr(
+                                parameter_model.config, "include_all_fields", None
+                            ),
+                        }
+                    }
+                    if getattr(parameter_model, "config", None)
+                    else {}
+                ),
+            )
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    def _delete(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        """
+        Delete an Agent Engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to be deleted. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (DeleteAgentEngineRuntimeRevisionConfig):
+                Optional. Additional configurations for deleting the Agent Engine runtime revision.
+
+        Returns:
+            DeleteAgentEngineRuntimeRevisionOperation: The operation for deleting the Agent Engine runtime revision.
+
+        """
+
+        parameter_model = types._DeleteAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _DeleteAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}".format_map(request_url_dict)
+            else:
+                path = "{name}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request("delete", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.DeleteAgentEngineRuntimeRevisionOperation._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    def _get_delete_runtime_revision_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[
+            types.GetDeleteAgentEngineRuntimeRevisionOperationConfigOrDict
+        ] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        parameter_model = types._GetDeleteAgentEngineRuntimeRevisionOperationParameters(
+            operation_name=operation_name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = (
+                _GetDeleteAgentEngineRuntimeRevisionOperationParameters_to_vertex(
+                    parameter_model
+                )
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operationName}".format_map(request_url_dict)
+            else:
+                path = "{operationName}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request("get", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.DeleteAgentEngineRuntimeRevisionOperation._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    def _query(
+        self,
+        *,
+        name: str,
+        config: Optional[types.QueryAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.QueryReasoningEngineResponse:
+        """
+        Query an Agent Engine runtime revision.
+        """
+
+        parameter_model = types._QueryAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _QueryAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}:query".format_map(request_url_dict)
+            else:
+                path = "{name}:query"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request("post", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.QueryReasoningEngineResponse._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    def get(
+        self,
+        *,
+        name: str,
+        config: Optional[types.GetAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.AgentEngineRuntimeRevision:
+        """Gets an agent engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to get. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (GetAgentEngineRuntimeRevisionConfigOrDict):
+                Optional. Additional configurations for getting the Agent Engine runtime revision.
+
+        Returns:
+            AgentEngineRuntimeRevision: The requested Agent Engine runtime revision instance.
+        """
+        api_resource = self._get(name=name, config=config)
+        agent_engine_runtime_revision = types.AgentEngineRuntimeRevision(
+            api_client=self,
+            api_async_client=AsyncRuntimeRevisions(api_client_=self._api_client),
+            api_resource=api_resource,
+        )
+        if api_resource.spec:
+            self._register_api_methods(
+                agent_engine_runtime_revision=agent_engine_runtime_revision
+            )
+        return agent_engine_runtime_revision
+
+    def list(
+        self,
+        *,
+        name: str,
+        config: Optional[types.ListAgentEngineRuntimeRevisionsConfigOrDict] = None,
+    ) -> Iterator[types.AgentEngineRuntimeRevision]:
+        """Lists all reasoning engine runtime revision instances matching the given query.
+
+        Args:
+            name (str): Required. The name of the reasoning engine to list runtime revisions for. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`.
+            config (ListAgentEngineRuntimeRevisionsConfig):
+                Optional. Additional configurations for listing the reasoning engine runtime revisions.
+
+        Returns:
+           Iterable[AgentEngineRuntimeRevision]: An iterable of runtime revisions.
+        """
+        list_pager: Pager[types.ReasoningEngineRuntimeRevision] = Pager(
+            "reasoning_engine_runtime_revisions",
+            functools.partial(self._list, name=name),
+            self._list(name=name, config=config),
+            config,
+        )
+
+        return (
+            types.AgentEngineRuntimeRevision(
+                api_client=self,
+                api_async_client=AsyncRuntimeRevisions(api_client_=self._api_client),
+                api_resource=runtime_revision,
+            )
+            for runtime_revision in list_pager
+        )
+
+    def delete(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        """Delete an Agent Engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to be deleted. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (DeleteAgentEngineRuntimeRevisionConfig):
+                Optional. Additional configurations for deleting the Agent Engine runtime revision.
+
+        Returns:
+            DeleteAgentEngineRuntimeRevisionOperation: The operation for deleting the Agent Engine runtime revision.
+        """
+        if config is None:
+            config = types.DeleteAgentEngineRuntimeRevisionConfig()
+        elif isinstance(config, dict):
+            config = types.DeleteAgentEngineRuntimeRevisionConfig.model_validate(config)
+        operation = self._delete(
+            name=name,
+            config=config,
+        )
+        if config.wait_for_completion and not operation.done:
+            operation = _agent_engines_utils._await_operation(
+                operation_name=operation.name,
+                get_operation_fn=self._get_delete_runtime_revision_operation,
+                poll_interval_seconds=0.5,
+            )
+            if operation.error:
+                raise RuntimeError(
+                    f"Failed to delete runtime revision: {operation.error}"
+                )
+        return operation
+
+    def _register_api_methods(
+        self,
+        *,
+        agent_engine_runtime_revision: types.AgentEngineRuntimeRevision,
+    ) -> types.AgentEngineRuntimeRevision:
+        """Registers the API methods for the agent engine runtime revision."""
+        try:
+            _agent_engines_utils._register_api_methods_or_raise(
+                agent_engine=agent_engine_runtime_revision,
+                wrap_operation_fn={
+                    "": _agent_engines_utils._wrap_query_operation,  # type: ignore[dict-item]
+                    "async": _agent_engines_utils._wrap_async_query_operation,  # type: ignore[dict-item]
+                    "stream": _agent_engines_utils._wrap_stream_query_operation,  # type: ignore[dict-item]
+                    "async_stream": _agent_engines_utils._wrap_async_stream_query_operation,  # type: ignore[dict-item]
+                    "a2a_extension": _agent_engines_utils._wrap_a2a_operation,
+                },
+            )
+        except Exception as e:
+            logger.warning(
+                _agent_engines_utils._FAILED_TO_REGISTER_API_METHODS_WARNING_TEMPLATE, e
+            )
+        return agent_engine_runtime_revision
+
+    def _stream_query(
+        self,
+        *,
+        name: str,
+        config: Optional[types.QueryAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> Iterator[Any]:
+        """Streams the response of the agent engine."""
+        parameter_model = types._QueryAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+        request_dict = _QueryAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+            parameter_model
+        )
+        request_url_dict = request_dict.get("_url")
+        if request_url_dict:
+            path = "{name}:streamQuery?alt=sse".format_map(request_url_dict)
+        else:
+            path = "{name}:streamQuery?alt=sse"
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+        http_options = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+        for response in self._api_client.request_streamed(
+            "post", path, request_dict, http_options
+        ):
+            yield response
+
+    async def _async_stream_query(
+        self,
+        *,
+        name: str,
+        config: Optional[types.QueryAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> AsyncIterator[Any]:
+        """Streams the response of the agent engine."""
+        parameter_model = types._QueryAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+        request_dict = _QueryAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+            parameter_model
+        )
+        request_url_dict = request_dict.get("_url")
+        if request_url_dict:
+            path = "{name}:streamQuery?alt=sse".format_map(request_url_dict)
+        else:
+            path = "{name}:streamQuery?alt=sse"
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+        http_options = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+        async_iterator = await self._api_client.async_request_streamed(
+            "post", path, request_dict, http_options
+        )
+        async for response in async_iterator:
+            yield response
+
+
+class AsyncRuntimeRevisions(_api_module.BaseModule):
+
+    async def _get(
+        self,
+        *,
+        name: str,
+        config: Optional[types.GetAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.ReasoningEngineRuntimeRevision:
+        """
+        Get an agent engine runtime revision instance.
+        """
+
+        parameter_model = types._GetAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _GetAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}".format_map(request_url_dict)
+            else:
+                path = "{name}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "get", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.ReasoningEngineRuntimeRevision._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    async def _list(
+        self,
+        *,
+        name: str,
+        config: Optional[types.ListAgentEngineRuntimeRevisionsConfigOrDict] = None,
+    ) -> types.ListReasoningEnginesRuntimeRevisionsResponse:
+        """
+        Lists reasoning engine runtime revisions.
+
+        Args:
+            name (str): Required. The name of the reasoning engine to list runtime revisions for. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`.
+            config (ListAgentEngineRuntimeRevisionsConfig):
+                Optional. Additional configurations for listing the reasoning engine runtime revisions.
+
+        Returns:
+            ListReasoningEnginesRuntimeRevisionsResponse: The requested reasoning engine runtime revisions.
+
+        """
+
+        parameter_model = types._ListAgentEngineRuntimeRevisionsRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _ListAgentEngineRuntimeRevisionsRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/runtimeRevisions".format_map(request_url_dict)
+            else:
+                path = "{name}/runtimeRevisions"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "get", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = (
+            types.ListReasoningEnginesRuntimeRevisionsResponse._from_response(
+                response=response_dict,
+                kwargs=(
+                    {
+                        "config": {
+                            "response_schema": getattr(
+                                parameter_model.config, "response_schema", None
+                            ),
+                            "response_json_schema": getattr(
+                                parameter_model.config, "response_json_schema", None
+                            ),
+                            "include_all_fields": getattr(
+                                parameter_model.config, "include_all_fields", None
+                            ),
+                        }
+                    }
+                    if getattr(parameter_model, "config", None)
+                    else {}
+                ),
+            )
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    async def _delete(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        """
+        Delete an Agent Engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to be deleted. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (DeleteAgentEngineRuntimeRevisionConfig):
+                Optional. Additional configurations for deleting the Agent Engine runtime revision.
+
+        Returns:
+            DeleteAgentEngineRuntimeRevisionOperation: The operation for deleting the Agent Engine runtime revision.
+
+        """
+
+        parameter_model = types._DeleteAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _DeleteAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}".format_map(request_url_dict)
+            else:
+                path = "{name}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "delete", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.DeleteAgentEngineRuntimeRevisionOperation._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    async def _get_delete_runtime_revision_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[
+            types.GetDeleteAgentEngineRuntimeRevisionOperationConfigOrDict
+        ] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        parameter_model = types._GetDeleteAgentEngineRuntimeRevisionOperationParameters(
+            operation_name=operation_name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = (
+                _GetDeleteAgentEngineRuntimeRevisionOperationParameters_to_vertex(
+                    parameter_model
+                )
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operationName}".format_map(request_url_dict)
+            else:
+                path = "{operationName}"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "get", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.DeleteAgentEngineRuntimeRevisionOperation._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    async def _query(
+        self,
+        *,
+        name: str,
+        config: Optional[types.QueryAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.QueryReasoningEngineResponse:
+        """
+        Query an Agent Engine runtime revision.
+        """
+
+        parameter_model = types._QueryAgentEngineRuntimeRevisionRequestParameters(
+            name=name,
+            config=config,
+        )
+
+        request_url_dict: Optional[dict[str, str]]
+        if not self._api_client.vertexai:
+            raise ValueError(
+                "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode."
+            )
+        else:
+            request_dict = _QueryAgentEngineRuntimeRevisionRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}:query".format_map(request_url_dict)
+            else:
+                path = "{name}:query"
+
+        query_params = request_dict.get("_query")
+        if query_params:
+            path = f"{path}?{urlencode(query_params)}"
+        # TODO: remove the hack that pops config.
+        request_dict.pop("config", None)
+
+        http_options: Optional[types.HttpOptions] = None
+        if (
+            parameter_model.config is not None
+            and parameter_model.config.http_options is not None
+        ):
+            http_options = parameter_model.config.http_options
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "post", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.QueryReasoningEngineResponse._from_response(
+            response=response_dict,
+            kwargs=(
+                {
+                    "config": {
+                        "response_schema": getattr(
+                            parameter_model.config, "response_schema", None
+                        ),
+                        "response_json_schema": getattr(
+                            parameter_model.config, "response_json_schema", None
+                        ),
+                        "include_all_fields": getattr(
+                            parameter_model.config, "include_all_fields", None
+                        ),
+                    }
+                }
+                if getattr(parameter_model, "config", None)
+                else {}
+            ),
+        )
+
+        self._api_client._verify_response(return_value)
+        return return_value
+
+    async def get(
+        self,
+        *,
+        name: str,
+        config: Optional[types.GetAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.AgentEngineRuntimeRevision:
+        """Gets an agent engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to get. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (GetAgentEngineRuntimeRevisionConfigOrDict):
+                Optional. Additional configurations for getting the Agent Engine runtime revision.
+
+        Returns:
+            AgentEngineRuntimeRevision: The requested Agent Engine runtime revision instance.
+        """
+        api_resource = await self._get(name=name, config=config)
+        agent_engine_runtime_revision = types.AgentEngineRuntimeRevision(
+            api_client=self,
+            api_async_client=AsyncRuntimeRevisions(api_client_=self._api_client),
+            api_resource=api_resource,
+        )
+        if api_resource.spec:
+            self._register_api_methods(
+                agent_engine_runtime_revision=agent_engine_runtime_revision
+            )
+        return agent_engine_runtime_revision
+
+    async def list(
+        self,
+        *,
+        name: str,
+        config: Optional[types.ListAgentEngineRuntimeRevisionsConfigOrDict] = None,
+    ) -> AsyncIterator[types.AgentEngineRuntimeRevision]:
+        """Lists reasoning engine runtime revisions.
+
+        Args:
+            name (str): Required. The name of the reasoning engine to list runtime revisions for. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}`.
+            config (ListAgentEngineRuntimeRevisionsConfig):
+                Optional. Additional configurations for listing the reasoning engine runtime revisions.
+
+        Returns:
+           AsyncIterator[AgentEngineRuntimeRevision]: An async iterator of runtime revisions.
+        """
+        list_pager: AsyncPager[types.ReasoningEngineRuntimeRevision] = AsyncPager(
+            "reasoning_engine_runtime_revisions",
+            functools.partial(self._list, name=name),
+            await self._list(name=name, config=config),
+            config,
+        )
+
+        async for runtime_revision in list_pager:
+            yield types.AgentEngineRuntimeRevision(
+                api_client=self,
+                api_async_client=AsyncRuntimeRevisions(api_client_=self._api_client),
+                api_resource=runtime_revision,
+            )
+
+    async def delete(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteAgentEngineRuntimeRevisionConfigOrDict] = None,
+    ) -> types.DeleteAgentEngineRuntimeRevisionOperation:
+        """Delete an Agent Engine runtime revision.
+
+        Args:
+            name (str): Required. The name of the Agent Engine runtime revision to be deleted. Format:
+                `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/runtimeRevisions/{runtime_revision_id}`.
+            config (DeleteAgentEngineRuntimeRevisionConfig):
+                Optional. Additional configurations for deleting the Agent Engine runtime revision.
+
+        Returns:
+            DeleteAgentEngineRuntimeRevisionOperation: The operation for deleting the Agent Engine runtime revision.
+        """
+        if config is None:
+            config = types.DeleteAgentEngineRuntimeRevisionConfig()
+        elif isinstance(config, dict):
+            config = types.DeleteAgentEngineRuntimeRevisionConfig.model_validate(config)
+        operation = await self._delete(
+            name=name,
+            config=config,
+        )
+        if config.wait_for_completion and not operation.done:
+            operation = await _agent_engines_utils._await_async_operation(
+                operation_name=operation.name,
+                get_operation_fn=self._get_delete_runtime_revision_operation,
+                poll_interval_seconds=0.5,
+            )
+            if operation.error:
+                raise RuntimeError(
+                    f"Failed to delete runtime revision: {operation.error}"
+                )
+        return operation
+
+    def _register_api_methods(
+        self,
+        *,
+        agent_engine_runtime_revision: types.AgentEngineRuntimeRevision,
+    ) -> types.AgentEngineRuntimeRevision:
+        """Registers the API methods for the agent engine runtime revision."""
+        try:
+            _agent_engines_utils._register_api_methods_or_raise(
+                agent_engine=agent_engine_runtime_revision,
+                wrap_operation_fn={
+                    "": _agent_engines_utils._wrap_query_operation,  # type: ignore[dict-item]
+                    "async": _agent_engines_utils._wrap_async_query_operation,  # type: ignore[dict-item]
+                    "stream": _agent_engines_utils._wrap_stream_query_operation,  # type: ignore[dict-item]
+                    "async_stream": _agent_engines_utils._wrap_async_stream_query_operation,  # type: ignore[dict-item]
+                    "a2a_extension": _agent_engines_utils._wrap_a2a_operation,
+                },
+            )
+        except Exception as e:
+            logger.warning(
+                _agent_engines_utils._FAILED_TO_REGISTER_API_METHODS_WARNING_TEMPLATE, e
+            )
+        return agent_engine_runtime_revision