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

agentplatform/_genai/rag.py

@@ -0,0 +1,4961 @@
+# 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 json
+import logging
+import mimetypes
+import os
+from typing import Any, Optional, Union
+from urllib.parse import urlencode
+
+from google.genai import _api_module
+from google.genai import _common
+from google.genai import _extra_utils
+from google.genai import types as genai_types
+from google.genai._common import get_value_by_path as getv
+from google.genai._common import set_value_by_path as setv
+
+from . import _gcs_utils
+from . import _operations_utils
+from . import types
+
+logger = logging.getLogger("agentplatform_genai.rag")
+
+
+def _AskContextsConfig_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, ["tools"]) is not None:
+        setv(parent_object, ["tools"], getv(from_object, ["tools"]))
+
+    return to_object
+
+
+def _AskContextsRequestParameters_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, ["query"]) is not None:
+        setv(to_object, ["query"], getv(from_object, ["query"]))
+
+    if getv(from_object, ["config"]) is not None:
+        setv(
+            to_object,
+            ["config"],
+            _AskContextsConfig_to_vertex(getv(from_object, ["config"]), to_object),
+        )
+
+    return to_object
+
+
+def _CorpusOperation_from_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, ["name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["metadata"]) is not None:
+        setv(to_object, ["metadata"], getv(from_object, ["metadata"]))
+
+    if getv(from_object, ["done"]) is not None:
+        setv(to_object, ["done"], getv(from_object, ["done"]))
+
+    if getv(from_object, ["error"]) is not None:
+        setv(to_object, ["error"], getv(from_object, ["error"]))
+
+    if getv(from_object, ["response"]) is not None:
+        setv(
+            to_object,
+            ["response"],
+            _RagCorpus_from_vertex(getv(from_object, ["response"]), to_object),
+        )
+
+    return to_object
+
+
+def _CreateRagCorpusRequestParameters_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, ["rag_corpus"]) is not None:
+        setv(
+            to_object,
+            ["_self"],
+            _RagCorpus_to_vertex(getv(from_object, ["rag_corpus"]), to_object),
+        )
+
+    if getv(from_object, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    return to_object
+
+
+def _DeleteRagCorpusRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _DeleteRagFileRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _GetCorpusOperationParameters_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 _GetImportFilesOperationParameters_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", "operation_name"], getv(from_object, ["operation_name"])
+        )
+
+    return to_object
+
+
+def _GetRagConfigOperationParameters_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", "operation_name"], getv(from_object, ["operation_name"])
+        )
+
+    return to_object
+
+
+def _GetRagConfigRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    return to_object
+
+
+def _GetRagCorpusRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _GetRagFileRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _ImportRagFilesConfig_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, ["gcs_source"]) is not None:
+        setv(to_object, ["gcsSource"], getv(from_object, ["gcs_source"]))
+
+    if getv(from_object, ["global_max_embedding_requests_per_min"]) is not None:
+        setv(
+            to_object,
+            ["globalMaxEmbeddingRequestsPerMin"],
+            getv(from_object, ["global_max_embedding_requests_per_min"]),
+        )
+
+    if getv(from_object, ["google_drive_source"]) is not None:
+        setv(
+            to_object, ["googleDriveSource"], getv(from_object, ["google_drive_source"])
+        )
+
+    if getv(from_object, ["import_result_bigquery_sink"]) is not None:
+        setv(
+            to_object,
+            ["importResultBigquerySink"],
+            getv(from_object, ["import_result_bigquery_sink"]),
+        )
+
+    if getv(from_object, ["import_result_gcs_sink"]) is not None:
+        setv(
+            to_object,
+            ["importResultGcsSink"],
+            getv(from_object, ["import_result_gcs_sink"]),
+        )
+
+    if getv(from_object, ["jira_source"]) is not None:
+        setv(to_object, ["jiraSource"], getv(from_object, ["jira_source"]))
+
+    if getv(from_object, ["max_embedding_requests_per_min"]) is not None:
+        setv(
+            to_object,
+            ["maxEmbeddingRequestsPerMin"],
+            getv(from_object, ["max_embedding_requests_per_min"]),
+        )
+
+    if getv(from_object, ["partial_failure_bigquery_sink"]) is not None:
+        setv(
+            to_object,
+            ["partialFailureBigquerySink"],
+            getv(from_object, ["partial_failure_bigquery_sink"]),
+        )
+
+    if getv(from_object, ["partial_failure_gcs_sink"]) is not None:
+        setv(
+            to_object,
+            ["partialFailureGcsSink"],
+            getv(from_object, ["partial_failure_gcs_sink"]),
+        )
+
+    if getv(from_object, ["rag_file_chunking_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileChunkingConfig"],
+            getv(from_object, ["rag_file_chunking_config"]),
+        )
+
+    if getv(from_object, ["rag_file_metadata_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileMetadataConfig"],
+            getv(from_object, ["rag_file_metadata_config"]),
+        )
+
+    if getv(from_object, ["rag_file_parsing_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileParsingConfig"],
+            _RagFileParsingConfig_to_vertex(
+                getv(from_object, ["rag_file_parsing_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_file_transformation_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileTransformationConfig"],
+            getv(from_object, ["rag_file_transformation_config"]),
+        )
+
+    if getv(from_object, ["rebuild_ann_index"]) is not None:
+        setv(to_object, ["rebuildAnnIndex"], getv(from_object, ["rebuild_ann_index"]))
+
+    if getv(from_object, ["share_point_sources"]) is not None:
+        setv(
+            to_object, ["sharePointSources"], getv(from_object, ["share_point_sources"])
+        )
+
+    if getv(from_object, ["slack_source"]) is not None:
+        setv(to_object, ["slackSource"], getv(from_object, ["slack_source"]))
+
+    return to_object
+
+
+def _ImportRagFilesRequestParameters_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, ["import_rag_files_request"]) is not None:
+        setv(
+            to_object,
+            ["_self"],
+            _ImportRagFilesRequest_to_vertex(
+                getv(from_object, ["import_rag_files_request"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    return to_object
+
+
+def _ImportRagFilesRequest_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, ["import_rag_files_config"]) is not None:
+        setv(
+            to_object,
+            ["importRagFilesConfig"],
+            _ImportRagFilesConfig_to_vertex(
+                getv(from_object, ["import_rag_files_config"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _ListRagCorporaConfig_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"]))
+
+    return to_object
+
+
+def _ListRagCorporaRequestParameters_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, ["config"]) is not None:
+        setv(
+            to_object,
+            ["config"],
+            _ListRagCorporaConfig_to_vertex(getv(from_object, ["config"]), to_object),
+        )
+
+    return to_object
+
+
+def _ListRagCorporaResponse_from_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, ["sdkHttpResponse"]) is not None:
+        setv(to_object, ["sdk_http_response"], getv(from_object, ["sdkHttpResponse"]))
+
+    if getv(from_object, ["nextPageToken"]) is not None:
+        setv(to_object, ["next_page_token"], getv(from_object, ["nextPageToken"]))
+
+    if getv(from_object, ["ragCorpora"]) is not None:
+        setv(
+            to_object,
+            ["rag_corpora"],
+            [
+                _RagCorpus_from_vertex(item, to_object)
+                for item in getv(from_object, ["ragCorpora"])
+            ],
+        )
+
+    return to_object
+
+
+def _ListRagFilesConfig_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"]))
+
+    return to_object
+
+
+def _ListRagFilesRequestParameters_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, ["config"]) is not None:
+        setv(
+            to_object,
+            ["config"],
+            _ListRagFilesConfig_to_vertex(getv(from_object, ["config"]), to_object),
+        )
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    return to_object
+
+
+def _RagCorpusCorpusTypeConfigMemoryCorpus_from_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, ["llmParser"]) is not None:
+        setv(
+            to_object,
+            ["llm_parser"],
+            _RagFileParsingConfigLlmParser_from_vertex(
+                getv(from_object, ["llmParser"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagCorpusCorpusTypeConfigMemoryCorpus_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, ["llm_parser"]) is not None:
+        setv(
+            to_object,
+            ["llmParser"],
+            _RagFileParsingConfigLlmParser_to_vertex(
+                getv(from_object, ["llm_parser"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagCorpusCorpusTypeConfig_from_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, ["documentCorpus"]) is not None:
+        setv(to_object, ["document_corpus"], getv(from_object, ["documentCorpus"]))
+
+    if getv(from_object, ["memoryCorpus"]) is not None:
+        setv(
+            to_object,
+            ["memory_corpus"],
+            _RagCorpusCorpusTypeConfigMemoryCorpus_from_vertex(
+                getv(from_object, ["memoryCorpus"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagCorpusCorpusTypeConfig_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, ["document_corpus"]) is not None:
+        setv(to_object, ["documentCorpus"], getv(from_object, ["document_corpus"]))
+
+    if getv(from_object, ["memory_corpus"]) is not None:
+        setv(
+            to_object,
+            ["memoryCorpus"],
+            _RagCorpusCorpusTypeConfigMemoryCorpus_to_vertex(
+                getv(from_object, ["memory_corpus"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagCorpus_from_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, ["corpusStatus"]) is not None:
+        setv(to_object, ["corpus_status"], getv(from_object, ["corpusStatus"]))
+
+    if getv(from_object, ["corpusTypeConfig"]) is not None:
+        setv(
+            to_object,
+            ["corpus_type_config"],
+            _RagCorpusCorpusTypeConfig_from_vertex(
+                getv(from_object, ["corpusTypeConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["createTime"]) is not None:
+        setv(to_object, ["create_time"], getv(from_object, ["createTime"]))
+
+    if getv(from_object, ["description"]) is not None:
+        setv(to_object, ["description"], getv(from_object, ["description"]))
+
+    if getv(from_object, ["displayName"]) is not None:
+        setv(to_object, ["display_name"], getv(from_object, ["displayName"]))
+
+    if getv(from_object, ["encryptionSpec"]) is not None:
+        setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["ragEmbeddingModelConfig"]) is not None:
+        setv(
+            to_object,
+            ["rag_embedding_model_config"],
+            _RagEmbeddingModelConfig_from_vertex(
+                getv(from_object, ["ragEmbeddingModelConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["ragFilesCount"]) is not None:
+        setv(to_object, ["rag_files_count"], getv(from_object, ["ragFilesCount"]))
+
+    if getv(from_object, ["ragVectorDbConfig"]) is not None:
+        setv(
+            to_object,
+            ["rag_vector_db_config"],
+            _RagVectorDbConfig_from_vertex(
+                getv(from_object, ["ragVectorDbConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["satisfiesPzi"]) is not None:
+        setv(to_object, ["satisfies_pzi"], getv(from_object, ["satisfiesPzi"]))
+
+    if getv(from_object, ["satisfiesPzs"]) is not None:
+        setv(to_object, ["satisfies_pzs"], getv(from_object, ["satisfiesPzs"]))
+
+    if getv(from_object, ["updateTime"]) is not None:
+        setv(to_object, ["update_time"], getv(from_object, ["updateTime"]))
+
+    if getv(from_object, ["vectorDbConfig"]) is not None:
+        setv(
+            to_object,
+            ["vector_db_config"],
+            _RagVectorDbConfig_from_vertex(
+                getv(from_object, ["vectorDbConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertexAiSearchConfig"]) is not None:
+        setv(
+            to_object,
+            ["vertex_ai_search_config"],
+            _VertexAiSearchConfig_from_vertex(
+                getv(from_object, ["vertexAiSearchConfig"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagCorpus_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, ["corpus_status"]) is not None:
+        setv(to_object, ["corpusStatus"], getv(from_object, ["corpus_status"]))
+
+    if getv(from_object, ["corpus_type_config"]) is not None:
+        setv(
+            to_object,
+            ["corpusTypeConfig"],
+            _RagCorpusCorpusTypeConfig_to_vertex(
+                getv(from_object, ["corpus_type_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["create_time"]) is not None:
+        setv(to_object, ["createTime"], getv(from_object, ["create_time"]))
+
+    if getv(from_object, ["description"]) is not None:
+        setv(to_object, ["description"], getv(from_object, ["description"]))
+
+    if getv(from_object, ["display_name"]) is not None:
+        setv(to_object, ["displayName"], getv(from_object, ["display_name"]))
+
+    if getv(from_object, ["encryption_spec"]) is not None:
+        setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["rag_embedding_model_config"]) is not None:
+        setv(
+            to_object,
+            ["ragEmbeddingModelConfig"],
+            _RagEmbeddingModelConfig_to_vertex(
+                getv(from_object, ["rag_embedding_model_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_files_count"]) is not None:
+        setv(to_object, ["ragFilesCount"], getv(from_object, ["rag_files_count"]))
+
+    if getv(from_object, ["rag_vector_db_config"]) is not None:
+        setv(
+            to_object,
+            ["ragVectorDbConfig"],
+            _RagVectorDbConfig_to_vertex(
+                getv(from_object, ["rag_vector_db_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["satisfies_pzi"]) is not None:
+        setv(to_object, ["satisfiesPzi"], getv(from_object, ["satisfies_pzi"]))
+
+    if getv(from_object, ["satisfies_pzs"]) is not None:
+        setv(to_object, ["satisfiesPzs"], getv(from_object, ["satisfies_pzs"]))
+
+    if getv(from_object, ["update_time"]) is not None:
+        setv(to_object, ["updateTime"], getv(from_object, ["update_time"]))
+
+    if getv(from_object, ["vector_db_config"]) is not None:
+        setv(
+            to_object,
+            ["vectorDbConfig"],
+            _RagVectorDbConfig_to_vertex(
+                getv(from_object, ["vector_db_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertex_ai_search_config"]) is not None:
+        setv(
+            to_object,
+            ["vertexAiSearchConfig"],
+            _VertexAiSearchConfig_to_vertex(
+                getv(from_object, ["vertex_ai_search_config"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigHybridSearchConfig_from_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, ["denseEmbeddingModelPredictionEndpoint"]) is not None:
+        setv(
+            to_object,
+            ["dense_embedding_model_prediction_endpoint"],
+            _RagEmbeddingModelConfigVertexPredictionEndpoint_from_vertex(
+                getv(from_object, ["denseEmbeddingModelPredictionEndpoint"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["sparseEmbeddingConfig"]) is not None:
+        setv(
+            to_object,
+            ["sparse_embedding_config"],
+            _RagEmbeddingModelConfigSparseEmbeddingConfig_from_vertex(
+                getv(from_object, ["sparseEmbeddingConfig"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigHybridSearchConfig_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, ["dense_embedding_model_prediction_endpoint"]) is not None:
+        setv(
+            to_object,
+            ["denseEmbeddingModelPredictionEndpoint"],
+            _RagEmbeddingModelConfigVertexPredictionEndpoint_to_vertex(
+                getv(from_object, ["dense_embedding_model_prediction_endpoint"]),
+                to_object,
+            ),
+        )
+
+    if getv(from_object, ["sparse_embedding_config"]) is not None:
+        setv(
+            to_object,
+            ["sparseEmbeddingConfig"],
+            _RagEmbeddingModelConfigSparseEmbeddingConfig_to_vertex(
+                getv(from_object, ["sparse_embedding_config"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigSparseEmbeddingConfigBm25_from_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, ["b"]) is not None:
+        setv(to_object, ["b"], getv(from_object, ["b"]))
+
+    if getv(from_object, ["k1"]) is not None:
+        setv(to_object, ["k1"], getv(from_object, ["k1"]))
+
+    if getv(from_object, ["multilingual"]) is not None:
+        setv(to_object, ["multilingual"], getv(from_object, ["multilingual"]))
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigSparseEmbeddingConfigBm25_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, ["b"]) is not None:
+        setv(to_object, ["b"], getv(from_object, ["b"]))
+
+    if getv(from_object, ["k1"]) is not None:
+        setv(to_object, ["k1"], getv(from_object, ["k1"]))
+
+    if getv(from_object, ["multilingual"]) is not None:
+        setv(to_object, ["multilingual"], getv(from_object, ["multilingual"]))
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigSparseEmbeddingConfig_from_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, ["bm25"]) is not None:
+        setv(
+            to_object,
+            ["bm25"],
+            _RagEmbeddingModelConfigSparseEmbeddingConfigBm25_from_vertex(
+                getv(from_object, ["bm25"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigSparseEmbeddingConfig_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, ["bm25"]) is not None:
+        setv(
+            to_object,
+            ["bm25"],
+            _RagEmbeddingModelConfigSparseEmbeddingConfigBm25_to_vertex(
+                getv(from_object, ["bm25"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigVertexPredictionEndpoint_from_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, ["endpoint"]) is not None:
+        setv(to_object, ["endpoint"], getv(from_object, ["endpoint"]))
+
+    if getv(from_object, ["model"]) is not None:
+        setv(to_object, ["model"], getv(from_object, ["model"]))
+
+    if getv(from_object, ["modelVersionId"]) is not None:
+        setv(to_object, ["model_version_id"], getv(from_object, ["modelVersionId"]))
+
+    return to_object
+
+
+def _RagEmbeddingModelConfigVertexPredictionEndpoint_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, ["endpoint"]) is not None:
+        setv(to_object, ["endpoint"], getv(from_object, ["endpoint"]))
+
+    if getv(from_object, ["model"]) is not None:
+        setv(to_object, ["model"], getv(from_object, ["model"]))
+
+    if getv(from_object, ["model_version_id"]) is not None:
+        setv(to_object, ["modelVersionId"], getv(from_object, ["model_version_id"]))
+
+    return to_object
+
+
+def _RagEmbeddingModelConfig_from_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, ["hybridSearchConfig"]) is not None:
+        setv(
+            to_object,
+            ["hybrid_search_config"],
+            _RagEmbeddingModelConfigHybridSearchConfig_from_vertex(
+                getv(from_object, ["hybridSearchConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertexPredictionEndpoint"]) is not None:
+        setv(
+            to_object,
+            ["vertex_prediction_endpoint"],
+            _RagEmbeddingModelConfigVertexPredictionEndpoint_from_vertex(
+                getv(from_object, ["vertexPredictionEndpoint"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEmbeddingModelConfig_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, ["hybrid_search_config"]) is not None:
+        setv(
+            to_object,
+            ["hybridSearchConfig"],
+            _RagEmbeddingModelConfigHybridSearchConfig_to_vertex(
+                getv(from_object, ["hybrid_search_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertex_prediction_endpoint"]) is not None:
+        setv(
+            to_object,
+            ["vertexPredictionEndpoint"],
+            _RagEmbeddingModelConfigVertexPredictionEndpoint_to_vertex(
+                getv(from_object, ["vertex_prediction_endpoint"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEngineConfig_from_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, ["name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["ragManagedDbConfig"]) is not None:
+        setv(
+            to_object,
+            ["rag_managed_db_config"],
+            _RagManagedDbConfig_from_vertex(
+                getv(from_object, ["ragManagedDbConfig"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagEngineConfig_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, ["name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["rag_managed_db_config"]) is not None:
+        setv(
+            to_object,
+            ["ragManagedDbConfig"],
+            _RagManagedDbConfig_to_vertex(
+                getv(from_object, ["rag_managed_db_config"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagFileParsingConfigLlmParser_from_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, ["customParsingPrompt"]) is not None:
+        setv(
+            to_object,
+            ["custom_parsing_prompt"],
+            getv(from_object, ["customParsingPrompt"]),
+        )
+
+    if getv(from_object, ["globalMaxParsingRequestsPerMin"]) is not None:
+        setv(
+            to_object,
+            ["global_max_parsing_requests_per_min"],
+            getv(from_object, ["globalMaxParsingRequestsPerMin"]),
+        )
+
+    if getv(from_object, ["maxParsingRequestsPerMin"]) is not None:
+        setv(
+            to_object,
+            ["max_parsing_requests_per_min"],
+            getv(from_object, ["maxParsingRequestsPerMin"]),
+        )
+
+    if getv(from_object, ["modelName"]) is not None:
+        setv(to_object, ["model_name"], getv(from_object, ["modelName"]))
+
+    return to_object
+
+
+def _RagFileParsingConfigLlmParser_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, ["custom_parsing_prompt"]) is not None:
+        setv(
+            to_object,
+            ["customParsingPrompt"],
+            getv(from_object, ["custom_parsing_prompt"]),
+        )
+
+    if getv(from_object, ["global_max_parsing_requests_per_min"]) is not None:
+        setv(
+            to_object,
+            ["globalMaxParsingRequestsPerMin"],
+            getv(from_object, ["global_max_parsing_requests_per_min"]),
+        )
+
+    if getv(from_object, ["max_parsing_requests_per_min"]) is not None:
+        setv(
+            to_object,
+            ["maxParsingRequestsPerMin"],
+            getv(from_object, ["max_parsing_requests_per_min"]),
+        )
+
+    if getv(from_object, ["model_name"]) is not None:
+        setv(to_object, ["modelName"], getv(from_object, ["model_name"]))
+
+    return to_object
+
+
+def _RagFileParsingConfig_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, ["advanced_parser"]) is not None:
+        setv(to_object, ["advancedParser"], getv(from_object, ["advanced_parser"]))
+
+    if getv(from_object, ["layout_parser"]) is not None:
+        setv(to_object, ["layoutParser"], getv(from_object, ["layout_parser"]))
+
+    if getv(from_object, ["llm_parser"]) is not None:
+        setv(
+            to_object,
+            ["llmParser"],
+            _RagFileParsingConfigLlmParser_to_vertex(
+                getv(from_object, ["llm_parser"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["use_advanced_pdf_parsing"]) is not None:
+        setv(
+            to_object,
+            ["useAdvancedPdfParsing"],
+            getv(from_object, ["use_advanced_pdf_parsing"]),
+        )
+
+    return to_object
+
+
+def _RagManagedDbConfigSpanner_from_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, ["basic"]) is not None:
+        setv(to_object, ["basic"], getv(from_object, ["basic"]))
+
+    if getv(from_object, ["scaled"]) is not None:
+        setv(to_object, ["scaled"], getv(from_object, ["scaled"]))
+
+    if getv(from_object, ["unprovisioned"]) is not None:
+        setv(to_object, ["unprovisioned"], getv(from_object, ["unprovisioned"]))
+
+    return to_object
+
+
+def _RagManagedDbConfigSpanner_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, ["basic"]) is not None:
+        setv(to_object, ["basic"], getv(from_object, ["basic"]))
+
+    if getv(from_object, ["scaled"]) is not None:
+        setv(to_object, ["scaled"], getv(from_object, ["scaled"]))
+
+    if getv(from_object, ["unprovisioned"]) is not None:
+        setv(to_object, ["unprovisioned"], getv(from_object, ["unprovisioned"]))
+
+    return to_object
+
+
+def _RagManagedDbConfig_from_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, ["basic"]) is not None:
+        setv(to_object, ["basic"], getv(from_object, ["basic"]))
+
+    if getv(from_object, ["enterprise"]) is not None:
+        setv(to_object, ["enterprise"], getv(from_object, ["enterprise"]))
+
+    if getv(from_object, ["scaled"]) is not None:
+        setv(to_object, ["scaled"], getv(from_object, ["scaled"]))
+
+    if getv(from_object, ["serverless"]) is not None:
+        setv(to_object, ["serverless"], getv(from_object, ["serverless"]))
+
+    if getv(from_object, ["spanner"]) is not None:
+        setv(
+            to_object,
+            ["spanner"],
+            _RagManagedDbConfigSpanner_from_vertex(
+                getv(from_object, ["spanner"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["unprovisioned"]) is not None:
+        setv(to_object, ["unprovisioned"], getv(from_object, ["unprovisioned"]))
+
+    return to_object
+
+
+def _RagManagedDbConfig_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, ["basic"]) is not None:
+        setv(to_object, ["basic"], getv(from_object, ["basic"]))
+
+    if getv(from_object, ["enterprise"]) is not None:
+        setv(to_object, ["enterprise"], getv(from_object, ["enterprise"]))
+
+    if getv(from_object, ["scaled"]) is not None:
+        setv(to_object, ["scaled"], getv(from_object, ["scaled"]))
+
+    if getv(from_object, ["serverless"]) is not None:
+        setv(to_object, ["serverless"], getv(from_object, ["serverless"]))
+
+    if getv(from_object, ["spanner"]) is not None:
+        setv(
+            to_object,
+            ["spanner"],
+            _RagManagedDbConfigSpanner_to_vertex(
+                getv(from_object, ["spanner"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["unprovisioned"]) is not None:
+        setv(to_object, ["unprovisioned"], getv(from_object, ["unprovisioned"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigPinecone_from_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, ["indexName"]) is not None:
+        setv(to_object, ["index_name"], getv(from_object, ["indexName"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigPinecone_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, ["index_name"]) is not None:
+        setv(to_object, ["indexName"], getv(from_object, ["index_name"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDbANN_from_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, ["leafCount"]) is not None:
+        setv(to_object, ["leaf_count"], getv(from_object, ["leafCount"]))
+
+    if getv(from_object, ["treeDepth"]) is not None:
+        setv(to_object, ["tree_depth"], getv(from_object, ["treeDepth"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDbANN_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, ["leaf_count"]) is not None:
+        setv(to_object, ["leafCount"], getv(from_object, ["leaf_count"]))
+
+    if getv(from_object, ["tree_depth"]) is not None:
+        setv(to_object, ["treeDepth"], getv(from_object, ["tree_depth"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDbKNN_from_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDbKNN_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+    to_object: dict[str, Any] = {}
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDb_from_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, ["ann"]) is not None:
+        setv(
+            to_object,
+            ["ann"],
+            _RagVectorDbConfigRagManagedDbANN_from_vertex(
+                getv(from_object, ["ann"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["knn"]) is not None:
+        setv(
+            to_object,
+            ["knn"],
+            _RagVectorDbConfigRagManagedDbKNN_from_vertex(
+                getv(from_object, ["knn"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedDb_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, ["ann"]) is not None:
+        setv(
+            to_object,
+            ["ann"],
+            _RagVectorDbConfigRagManagedDbANN_to_vertex(
+                getv(from_object, ["ann"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["knn"]) is not None:
+        setv(
+            to_object,
+            ["knn"],
+            _RagVectorDbConfigRagManagedDbKNN_to_vertex(
+                getv(from_object, ["knn"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedVertexVectorSearch_from_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, ["collectionName"]) is not None:
+        setv(to_object, ["collection_name"], getv(from_object, ["collectionName"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigRagManagedVertexVectorSearch_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, ["collection_name"]) is not None:
+        setv(to_object, ["collectionName"], getv(from_object, ["collection_name"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigVertexFeatureStore_from_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, ["featureViewResourceName"]) is not None:
+        setv(
+            to_object,
+            ["feature_view_resource_name"],
+            getv(from_object, ["featureViewResourceName"]),
+        )
+
+    return to_object
+
+
+def _RagVectorDbConfigVertexFeatureStore_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, ["feature_view_resource_name"]) is not None:
+        setv(
+            to_object,
+            ["featureViewResourceName"],
+            getv(from_object, ["feature_view_resource_name"]),
+        )
+
+    return to_object
+
+
+def _RagVectorDbConfigVertexVectorSearch_from_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, ["index"]) is not None:
+        setv(to_object, ["index"], getv(from_object, ["index"]))
+
+    if getv(from_object, ["indexEndpoint"]) is not None:
+        setv(to_object, ["index_endpoint"], getv(from_object, ["indexEndpoint"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigVertexVectorSearch_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, ["index"]) is not None:
+        setv(to_object, ["index"], getv(from_object, ["index"]))
+
+    if getv(from_object, ["index_endpoint"]) is not None:
+        setv(to_object, ["indexEndpoint"], getv(from_object, ["index_endpoint"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigWeaviate_from_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, ["collectionName"]) is not None:
+        setv(to_object, ["collection_name"], getv(from_object, ["collectionName"]))
+
+    if getv(from_object, ["httpEndpoint"]) is not None:
+        setv(to_object, ["http_endpoint"], getv(from_object, ["httpEndpoint"]))
+
+    return to_object
+
+
+def _RagVectorDbConfigWeaviate_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, ["collection_name"]) is not None:
+        setv(to_object, ["collectionName"], getv(from_object, ["collection_name"]))
+
+    if getv(from_object, ["http_endpoint"]) is not None:
+        setv(to_object, ["httpEndpoint"], getv(from_object, ["http_endpoint"]))
+
+    return to_object
+
+
+def _RagVectorDbConfig_from_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, ["apiAuth"]) is not None:
+        setv(to_object, ["api_auth"], getv(from_object, ["apiAuth"]))
+
+    if getv(from_object, ["pinecone"]) is not None:
+        setv(
+            to_object,
+            ["pinecone"],
+            _RagVectorDbConfigPinecone_from_vertex(
+                getv(from_object, ["pinecone"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["ragEmbeddingModelConfig"]) is not None:
+        setv(
+            to_object,
+            ["rag_embedding_model_config"],
+            _RagEmbeddingModelConfig_from_vertex(
+                getv(from_object, ["ragEmbeddingModelConfig"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["ragManagedDb"]) is not None:
+        setv(
+            to_object,
+            ["rag_managed_db"],
+            _RagVectorDbConfigRagManagedDb_from_vertex(
+                getv(from_object, ["ragManagedDb"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["ragManagedVertexVectorSearch"]) is not None:
+        setv(
+            to_object,
+            ["rag_managed_vertex_vector_search"],
+            _RagVectorDbConfigRagManagedVertexVectorSearch_from_vertex(
+                getv(from_object, ["ragManagedVertexVectorSearch"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertexFeatureStore"]) is not None:
+        setv(
+            to_object,
+            ["vertex_feature_store"],
+            _RagVectorDbConfigVertexFeatureStore_from_vertex(
+                getv(from_object, ["vertexFeatureStore"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertexVectorSearch"]) is not None:
+        setv(
+            to_object,
+            ["vertex_vector_search"],
+            _RagVectorDbConfigVertexVectorSearch_from_vertex(
+                getv(from_object, ["vertexVectorSearch"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["weaviate"]) is not None:
+        setv(
+            to_object,
+            ["weaviate"],
+            _RagVectorDbConfigWeaviate_from_vertex(
+                getv(from_object, ["weaviate"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RagVectorDbConfig_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, ["api_auth"]) is not None:
+        setv(to_object, ["apiAuth"], getv(from_object, ["api_auth"]))
+
+    if getv(from_object, ["pinecone"]) is not None:
+        setv(
+            to_object,
+            ["pinecone"],
+            _RagVectorDbConfigPinecone_to_vertex(
+                getv(from_object, ["pinecone"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_embedding_model_config"]) is not None:
+        setv(
+            to_object,
+            ["ragEmbeddingModelConfig"],
+            _RagEmbeddingModelConfig_to_vertex(
+                getv(from_object, ["rag_embedding_model_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_managed_db"]) is not None:
+        setv(
+            to_object,
+            ["ragManagedDb"],
+            _RagVectorDbConfigRagManagedDb_to_vertex(
+                getv(from_object, ["rag_managed_db"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_managed_vertex_vector_search"]) is not None:
+        setv(
+            to_object,
+            ["ragManagedVertexVectorSearch"],
+            _RagVectorDbConfigRagManagedVertexVectorSearch_to_vertex(
+                getv(from_object, ["rag_managed_vertex_vector_search"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertex_feature_store"]) is not None:
+        setv(
+            to_object,
+            ["vertexFeatureStore"],
+            _RagVectorDbConfigVertexFeatureStore_to_vertex(
+                getv(from_object, ["vertex_feature_store"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["vertex_vector_search"]) is not None:
+        setv(
+            to_object,
+            ["vertexVectorSearch"],
+            _RagVectorDbConfigVertexVectorSearch_to_vertex(
+                getv(from_object, ["vertex_vector_search"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["weaviate"]) is not None:
+        setv(
+            to_object,
+            ["weaviate"],
+            _RagVectorDbConfigWeaviate_to_vertex(
+                getv(from_object, ["weaviate"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _RetrieveRagContextsRequestParameters_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, ["vertex_rag_store"]) is not None:
+        setv(to_object, ["vertexRagStore"], getv(from_object, ["vertex_rag_store"]))
+
+    if getv(from_object, ["query"]) is not None:
+        setv(to_object, ["query"], getv(from_object, ["query"]))
+
+    if getv(from_object, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    return to_object
+
+
+def _UpdateRagConfigRequestParameters_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, ["updated_config"]) is not None:
+        setv(
+            to_object,
+            ["_self"],
+            _RagEngineConfig_to_vertex(
+                getv(from_object, ["updated_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    return to_object
+
+
+def _UpdateRagCorpusRequestParameters_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, ["config"]) is not None:
+        setv(to_object, ["config"], getv(from_object, ["config"]))
+
+    if getv(from_object, ["name"]) is not None:
+        setv(to_object, ["_url", "name"], getv(from_object, ["name"]))
+
+    if getv(from_object, ["rag_corpus"]) is not None:
+        setv(
+            to_object,
+            ["_self"],
+            _RagCorpus_to_vertex(getv(from_object, ["rag_corpus"]), to_object),
+        )
+
+    return to_object
+
+
+def _UploadRagFileConfig_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, ["rag_file_chunking_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileChunkingConfig"],
+            getv(from_object, ["rag_file_chunking_config"]),
+        )
+
+    if getv(from_object, ["rag_file_metadata_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileMetadataConfig"],
+            getv(from_object, ["rag_file_metadata_config"]),
+        )
+
+    if getv(from_object, ["rag_file_parsing_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileParsingConfig"],
+            _RagFileParsingConfig_to_vertex(
+                getv(from_object, ["rag_file_parsing_config"]), to_object
+            ),
+        )
+
+    if getv(from_object, ["rag_file_transformation_config"]) is not None:
+        setv(
+            to_object,
+            ["ragFileTransformationConfig"],
+            getv(from_object, ["rag_file_transformation_config"]),
+        )
+
+    return to_object
+
+
+def _UploadRagFileParameters_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, ["rag_file"]) is not None:
+        setv(to_object, ["ragFile"], getv(from_object, ["rag_file"]))
+
+    if getv(from_object, ["upload_rag_file_config"]) is not None:
+        setv(
+            to_object,
+            ["uploadRagFileConfig"],
+            _UploadRagFileConfig_to_vertex(
+                getv(from_object, ["upload_rag_file_config"]), to_object
+            ),
+        )
+
+    return to_object
+
+
+def _VertexAiSearchConfig_from_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, ["servingConfig"]) is not None:
+        setv(to_object, ["serving_config"], getv(from_object, ["servingConfig"]))
+
+    return to_object
+
+
+def _VertexAiSearchConfig_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, ["serving_config"]) is not None:
+        setv(to_object, ["servingConfig"], getv(from_object, ["serving_config"]))
+
+    return to_object
+
+
+class Rag(_api_module.BaseModule):
+
+    def ask_contexts(
+        self,
+        *,
+        query: types.RagQueryOrDict,
+        config: Optional[types.AskContextsConfigOrDict] = None,
+    ) -> types.AskContextsResponse:
+        """
+        Agentic Retrieval Ask API for RAG.
+
+        Args:
+          query: The query to ask.
+          config: Optional configuration for the request.
+
+        Returns:
+          The AskContextsResponse.
+
+        """
+
+        parameter_model = types._AskContextsRequestParameters(
+            query=query,
+            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 = _AskContextsRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = ":askContexts".format_map(request_url_dict)
+            else:
+                path = ":askContexts"
+
+        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.AskContextsResponse._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 _create_corpus(
+        self,
+        *,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.CreateRagCorpusConfigOrDict] = None,
+    ) -> types.CreateRagCorpusOperation:
+        """
+        Creates a new Rag Corpus.
+        """
+
+        parameter_model = types._CreateRagCorpusRequestParameters(
+            rag_corpus=rag_corpus,
+            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 = _CreateRagCorpusRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragCorpora".format_map(request_url_dict)
+            else:
+                path = "ragCorpora"
+
+        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.CreateRagCorpusOperation._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_corpus_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetCorpusOperationConfigOrDict] = None,
+    ) -> types.CorpusOperation:
+        parameter_model = types._GetCorpusOperationParameters(
+            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 = _GetCorpusOperationParameters_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)
+
+        if self._api_client.vertexai:
+            response_dict = _CorpusOperation_from_vertex(response_dict)
+
+        return_value = types.CorpusOperation._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_corpus(
+        self, *, config: Optional[types.GetRagCorpusConfigOrDict] = None, name: str
+    ) -> types.RagCorpus:
+        """
+        Gets a RAG Corpus.
+
+        Args:
+          name: The name of the RagCorpus to retrieve.
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagCorpus.
+
+        """
+
+        parameter_model = types._GetRagCorpusRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _GetRagCorpusRequestParameters_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)
+
+        if self._api_client.vertexai:
+            response_dict = _RagCorpus_from_vertex(response_dict)
+
+        return_value = types.RagCorpus._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_corpora(
+        self, *, config: Optional[types.ListRagCorporaConfigOrDict] = None
+    ) -> types.ListRagCorporaResponse:
+        """
+        Lists RagCorpora for a project.
+
+        Args:
+          config: Optional configuration for listing corpora.
+
+        Returns:
+          A ListRagCorporaResponse containing the corpora.
+
+        """
+
+        parameter_model = types._ListRagCorporaRequestParameters(
+            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 = _ListRagCorporaRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragCorpora".format_map(request_url_dict)
+            else:
+                path = "ragCorpora"
+
+        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)
+
+        if self._api_client.vertexai:
+            response_dict = _ListRagCorporaResponse_from_vertex(response_dict)
+
+        return_value = types.ListRagCorporaResponse._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_file(
+        self, *, config: Optional[types.GetRagFileConfigOrDict] = None, name: str
+    ) -> types.RagFile:
+        """
+        Gets a RagFile.
+
+        Args:
+          name: The name of the RagFile to retrieve.
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagFile.
+
+        """
+
+        parameter_model = types._GetRagFileRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _GetRagFileRequestParameters_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.RagFile._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_files(
+        self, *, config: Optional[types.ListRagFilesConfigOrDict] = None, name: str
+    ) -> types.ListRagFilesResponse:
+        """
+        Lists RagFile instances within a RagCorpus.
+
+        Args:
+          name: The name of the RagCorpus to list files from.
+          config: Optional configuration for the request (e.g., pagination details).
+
+        Returns:
+          A ListRagFilesResponse containing the files.
+
+        """
+
+        parameter_model = types._ListRagFilesRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _ListRagFilesRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles"
+
+        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.ListRagFilesResponse._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_config(
+        self, *, config: Optional[types.GetRagConfigOrDict] = None
+    ) -> types.RagEngineConfig:
+        """
+        Gets the project-level RAG Engine Config.
+
+        Args:
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagEngineConfig.
+
+        """
+
+        parameter_model = types._GetRagConfigRequestParameters(
+            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 = _GetRagConfigRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragEngineConfig".format_map(request_url_dict)
+            else:
+                path = "ragEngineConfig"
+
+        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)
+
+        if self._api_client.vertexai:
+            response_dict = _RagEngineConfig_from_vertex(response_dict)
+
+        return_value = types.RagEngineConfig._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 _update_corpus(
+        self,
+        *,
+        config: Optional[types.UpdateRagCorpusConfigOrDict] = None,
+        name: Optional[str] = None,
+        rag_corpus: types.RagCorpusOrDict,
+    ) -> types.UpdateRagCorpusOperation:
+        """
+        Updates an existing Rag Corpus.
+        """
+
+        parameter_model = types._UpdateRagCorpusRequestParameters(
+            config=config,
+            name=name,
+            rag_corpus=rag_corpus,
+        )
+
+        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 = _UpdateRagCorpusRequestParameters_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("patch", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.UpdateRagCorpusOperation._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_corpus(
+        self,
+        *,
+        config: Optional[types.DeleteRagCorpusConfigOrDict] = None,
+        name: Optional[str] = None,
+    ) -> types.DeleteRagCorpusOperation:
+        """
+        Deletes a RAG Corpus.
+        """
+
+        parameter_model = types._DeleteRagCorpusRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _DeleteRagCorpusRequestParameters_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.DeleteRagCorpusOperation._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_file(
+        self, *, config: Optional[types.DeleteRagFileConfigOrDict] = None, name: str
+    ) -> types.DeleteRagFileOperation:
+        """
+        Deletes a RAG File.
+        """
+
+        parameter_model = types._DeleteRagFileRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _DeleteRagFileRequestParameters_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.DeleteRagFileOperation._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 _update_config(
+        self,
+        *,
+        updated_config: types.RagEngineConfigOrDict,
+        config: Optional[types.UpdateRagConfigOrDict] = None,
+    ) -> types.UpdateRagConfigOperation:
+        """
+        Updates a RAG Engine Config.
+        """
+
+        parameter_model = types._UpdateRagConfigRequestParameters(
+            updated_config=updated_config,
+            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 = _UpdateRagConfigRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragEngineConfig".format_map(request_url_dict)
+            else:
+                path = "ragEngineConfig"
+
+        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("patch", path, request_dict, http_options)
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.UpdateRagConfigOperation._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 retrieve_contexts(
+        self,
+        *,
+        vertex_rag_store: genai_types.VertexRagStoreOrDict,
+        query: types.RagQueryOrDict,
+        config: Optional[types.RetrieveContextsConfigOrDict] = None,
+    ) -> types.RetrieveContextsResponse:
+        """
+        Retrieves contexts from a Vertex RAG store based on a query.
+
+        Args:
+          vertex_rag_store: The Vertex RAG store to retrieve contexts from.
+          query: The query to retrieve contexts for.
+          config: Optional configuration for the request.
+
+        Returns:
+          The RetrieveContextsResponse containing the relevant contexts.
+
+        """
+
+        parameter_model = types._RetrieveRagContextsRequestParameters(
+            vertex_rag_store=vertex_rag_store,
+            query=query,
+            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 = _RetrieveRagContextsRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = ":retrieveContexts".format_map(request_url_dict)
+            else:
+                path = ":retrieveContexts"
+
+        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.RetrieveContextsResponse._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_rag_config_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetRagConfigOperationConfigOrDict] = None,
+    ) -> types.RagEngineConfigOperation:
+        parameter_model = types._GetRagConfigOperationParameters(
+            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 = _GetRagConfigOperationParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operation_name}".format_map(request_url_dict)
+            else:
+                path = "{operation_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.RagEngineConfigOperation._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 _import_files(
+        self,
+        *,
+        name: str,
+        import_rag_files_request: types.ImportRagFilesRequestOrDict,
+        config: Optional[types.ImportRagFilesRequestConfigOrDict] = None,
+    ) -> types.ImportRagFilesOperation:
+        """
+        Imports files into a RAG Corpus.
+        """
+
+        parameter_model = types._ImportRagFilesRequestParameters(
+            name=name,
+            import_rag_files_request=import_rag_files_request,
+            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 = _ImportRagFilesRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles:import".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles:import"
+
+        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.ImportRagFilesOperation._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_import_files_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetImportFilesOperationConfigOrDict] = None,
+    ) -> types.ImportRagFilesOperation:
+        parameter_model = types._GetImportFilesOperationParameters(
+            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 = _GetImportFilesOperationParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operation_name}".format_map(request_url_dict)
+            else:
+                path = "{operation_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.ImportRagFilesOperation._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 _upload_file(
+        self,
+        *,
+        name: str,
+        rag_file: types.RagFileOrDict,
+        upload_rag_file_config: Optional[types.UploadRagFileConfigOrDict] = None,
+        config: Optional[types.UploadRagFileRequestConfigOrDict] = None,
+    ) -> types.UploadRagFileResponse:
+        parameter_model = types._UploadRagFileParameters(
+            name=name,
+            rag_file=rag_file,
+            upload_rag_file_config=upload_rag_file_config,
+            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 = _UploadRagFileParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles:upload".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles:upload"
+
+        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.UploadRagFileResponse._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 create_corpus(
+        self,
+        *,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.CreateRagCorpusConfigOrDict] = None,
+    ) -> types.RagCorpus:
+        """
+        Creates a new RAG Corpus and waits for completion.
+
+        Args:
+          rag_corpus: The RagCorpus to create.
+          config: The configuration to use for the RagCorpus.
+
+        Returns:
+          The created RagCorpus.
+
+        Example usage:
+
+        ```
+        corpus = client.rag.create_corpus(
+            rag_corpus=types.RagCorpus(
+                display_name="test-corpus",
+                name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            ),
+        )
+        ```
+        """
+        operation = self._create_corpus(rag_corpus=rag_corpus, config=config)
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to create RagCorpus: {operation.error}"
+            )
+
+        return self.get_corpus(name=operation.response.name)
+
+    def delete_corpus(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteRagCorpusConfigOrDict] = None,
+    ) -> None:
+        """
+        Deletes a RAG Corpus and waits for the delete operation to complete.
+
+        Args:
+          name: The name of the RagCorpus to delete, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}`.
+          config: The configuration to use for the RagCorpus delete request.
+
+        Example usage:
+
+        ```
+        client.rag.delete_corpus(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+        )
+        ```
+        """
+
+        operation = self._delete_corpus(name=name, config=config)
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to delete RagCorpus: {operation.error}"
+            )
+
+        return None
+
+    def delete_file(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteRagFileConfigOrDict] = None,
+    ) -> None:
+        """
+        Deletes a file from a RAG Corpus and waits for the delete operation to complete.
+
+        Args:
+          name: The name of the RagFile to delete, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}/ragFiles/{file_id}`.
+          config: The configuration to use for the RagFile delete request.
+
+        Example usage:
+
+        ```
+        client.rag.delete_file(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789/ragFiles/987654321",
+        )
+        ```
+        """
+        operation = self._delete_file(name=name, config=config)
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to delete file from RagCorpus: {operation.error}"
+            )
+
+        return None
+
+    def update_corpus(
+        self,
+        *,
+        name: str,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.UpdateRagCorpusConfigOrDict] = None,
+    ) -> types.RagCorpus:
+        """
+        Updates a RAG Corpus and waits for completion.
+
+        Args:
+          name: The name of the RagCorpus to update, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}`.
+          rag_corpus: The RagCorpus to update.
+          config: The configuration to use for the RagCorpus update request.
+
+        Returns:
+          The updated RagCorpus.
+
+        Example usage:
+
+        ```
+        client.rag.update_corpus(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            rag_corpus=types.RagCorpus(
+                display_name="Updated corpus name",
+            ),
+        )
+        ```
+        """
+
+        operation = self._update_corpus(name=name, rag_corpus=rag_corpus, config=config)
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to update RagCorpus: {operation.error}"
+            )
+
+        return self.get_corpus(name=operation.response.name)
+
+    def update_config(
+        self,
+        *,
+        updated_config: types.RagEngineConfigOrDict,
+        request_config: Optional[types.UpdateRagConfigOrDict] = None,
+    ) -> types.RagEngineConfig:
+        """
+        Updates a RagEngineConfig and waits for completion.
+
+        Args:
+          updated_config: The RagEngineConfig to update.
+          request_config: The configuration to use for the RagEngineConfig update request.
+
+        Returns:
+          The updated RagEngineConfig.
+
+        Example usage:
+
+        ```
+        client.rag.update_config(
+            updated_config=types.RagEngineConfig(
+                name="projects/test-project/locations/us-central1/ragCorpora/123456789/ragEngineConfig/123456789",
+                rag_managed_db_config=types.RagManagedDbConfig(
+                    basic=types.RagManagedDbConfigBasic()
+                ),
+            ),
+        )
+        ```
+        """
+        operation = self._update_config(
+            updated_config=updated_config, config=request_config
+        )
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_rag_config_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to update RagEngineConfig: {operation.error}"
+            )
+
+        return self.get_config()
+
+    def import_files(
+        self,
+        *,
+        name: str,
+        import_config: types.ImportRagFilesConfigOrDict,
+        config: Optional[types.ImportRagFilesRequestConfigOrDict] = None,
+    ) -> types.ImportRagFilesResponse:
+        """
+        Imports files into a Rag Corpus. To get a list of all files on the corpus after calling import_files, call list_files.
+
+        Args:
+          name: The name of the Rag Corpus to import files into, format:
+            `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+          import_config: The configuration for importing files, including source files and processing options.
+          config: The configuration to use for the import request.
+
+        Returns:
+          The ImportRagFilesResponse from the import files operation.
+
+        Example usage:
+
+        ```
+        import_files_response = client.rag.import_files(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            import_config=types.ImportRagFilesConfig(
+                gcs_source=types.GcsSource(uris=["gs://test-bucket/test-file.pdf"]),
+            ),
+        )
+        ```
+        """
+        if isinstance(import_config, dict):
+            import_config = types.ImportRagFilesConfig.model_validate(import_config)
+
+        if import_config.gcs_source is not None:
+            for uri in import_config.gcs_source.uris:
+                if not _gcs_utils.GcsUtils(self._api_client)._verify_bucket_ownership(
+                    bucket_name=uri.split("/")[2],
+                    expected_project=self._api_client.project,
+                ):
+                    raise ValueError(
+                        f"Bucket {uri} does not belong to"
+                        f" project {self._api_client.project}."
+                    )
+
+        operation = self._import_files(
+            name=name,
+            import_rag_files_request=types.ImportRagFilesRequest(
+                import_rag_files_config=import_config
+            ),
+            config=config,
+        )
+
+        operation = _operations_utils.await_operation(
+            operation_name=operation.name,
+            get_operation_fn=self._get_import_files_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to import files into RagCorpus: {operation.error}"
+            )
+
+        return operation.response
+
+    def upload_file(
+        self,
+        *,
+        corpus_name: str,
+        path: str,
+        display_name: Optional[str] = None,
+        upload_rag_file_config: Optional[types.UploadRagFileConfigOrDict] = None,
+        request_config: Optional[types.UploadRagFileRequestConfigOrDict] = None,
+    ) -> types.RagFile:
+        """
+        Uploads a file to a RAG Corpus.
+
+        Args:
+          corpus_name: The name of the RAG Corpus to upload to.
+          path: The path to the file to upload.
+          display_name: Optional. The display name for the uploaded file. If not provided, a display name will be generated.
+          upload_rag_file_config: Optional. The configuration to use for the upload.
+          request_config: Optional. The configuration to use for the request.
+
+        Returns:
+          The uploaded RagFile.
+
+        Example usage:
+
+        ```
+        rag_file = client.rag.upload_file(
+            corpus_name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            path="/path/to/file.pdf",
+            display_name="test-file",
+            upload_rag_file_config=types.UploadRagFileConfig(
+                chunk_config=types.ChunkConfig(
+                    chunk_size=1024,
+                    chunk_overlap=128,
+                ),
+            ),
+        )
+        ```
+        """
+
+        if not display_name:
+            display_name = f"file_{_common.timestamped_unique_name()}"
+
+        rag_file = types.RagFile(display_name=display_name)
+
+        mime_type, _ = mimetypes.guess_type(path)
+
+        if mime_type is None:
+            mime_type = "application/octet-stream"
+
+        http_options, size_bytes, mime_type = _extra_utils.prepare_resumable_upload(
+            path,
+            user_http_options=request_config.http_options if request_config else None,
+            user_mime_type=mime_type,
+        )
+
+        current_api_version = self._api_client._http_options.api_version or "v1beta1"
+        upload_api_version = f"upload/{current_api_version}"
+
+        http_options.api_version = upload_api_version
+
+        parameter_model = types._UploadRagFileParameters(
+            name=corpus_name,
+            rag_file=rag_file,
+            upload_rag_file_config=upload_rag_file_config,
+        )
+        request_dict = _UploadRagFileParameters_to_vertex(parameter_model)
+
+        request_dict.pop("_url", None)
+        request_dict.pop("_query", None)
+
+        request_path = f"{corpus_name}/ragFiles:upload"
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = self._api_client.request(
+            "post",
+            request_path,
+            request_dict,
+            http_options,
+        )
+
+        if response.headers is None or (
+            "x-goog-upload-url" not in response.headers
+            and "X-Goog-Upload-URL" not in response.headers
+        ):
+            raise KeyError(
+                "Failed to create file. Upload URL was not returned from the create file request."
+            )
+
+        upload_url = response.headers.get(
+            "x-goog-upload-url", response.headers.get("X-Goog-Upload-URL")
+        )
+
+        fs_path = os.fspath(path)
+        return_file = self._api_client.upload_file(
+            fs_path, upload_url, size_bytes, http_options=http_options
+        )
+
+        rag_file_payload = return_file.json.get("ragFile") or return_file.json.get(
+            "rag_file", {}
+        )
+        return types.RagFile(**rag_file_payload)
+
+
+class AsyncRag(_api_module.BaseModule):
+
+    async def ask_contexts(
+        self,
+        *,
+        query: types.RagQueryOrDict,
+        config: Optional[types.AskContextsConfigOrDict] = None,
+    ) -> types.AskContextsResponse:
+        """
+        Agentic Retrieval Ask API for RAG.
+
+        Args:
+          query: The query to ask.
+          config: Optional configuration for the request.
+
+        Returns:
+          The AskContextsResponse.
+
+        """
+
+        parameter_model = types._AskContextsRequestParameters(
+            query=query,
+            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 = _AskContextsRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = ":askContexts".format_map(request_url_dict)
+            else:
+                path = ":askContexts"
+
+        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.AskContextsResponse._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 _create_corpus(
+        self,
+        *,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.CreateRagCorpusConfigOrDict] = None,
+    ) -> types.CreateRagCorpusOperation:
+        """
+        Creates a new Rag Corpus.
+        """
+
+        parameter_model = types._CreateRagCorpusRequestParameters(
+            rag_corpus=rag_corpus,
+            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 = _CreateRagCorpusRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragCorpora".format_map(request_url_dict)
+            else:
+                path = "ragCorpora"
+
+        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.CreateRagCorpusOperation._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_corpus_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetCorpusOperationConfigOrDict] = None,
+    ) -> types.CorpusOperation:
+        parameter_model = types._GetCorpusOperationParameters(
+            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 = _GetCorpusOperationParameters_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)
+
+        if self._api_client.vertexai:
+            response_dict = _CorpusOperation_from_vertex(response_dict)
+
+        return_value = types.CorpusOperation._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_corpus(
+        self, *, config: Optional[types.GetRagCorpusConfigOrDict] = None, name: str
+    ) -> types.RagCorpus:
+        """
+        Gets a RAG Corpus.
+
+        Args:
+          name: The name of the RagCorpus to retrieve.
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagCorpus.
+
+        """
+
+        parameter_model = types._GetRagCorpusRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _GetRagCorpusRequestParameters_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)
+
+        if self._api_client.vertexai:
+            response_dict = _RagCorpus_from_vertex(response_dict)
+
+        return_value = types.RagCorpus._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_corpora(
+        self, *, config: Optional[types.ListRagCorporaConfigOrDict] = None
+    ) -> types.ListRagCorporaResponse:
+        """
+        Lists RagCorpora for a project.
+
+        Args:
+          config: Optional configuration for listing corpora.
+
+        Returns:
+          A ListRagCorporaResponse containing the corpora.
+
+        """
+
+        parameter_model = types._ListRagCorporaRequestParameters(
+            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 = _ListRagCorporaRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragCorpora".format_map(request_url_dict)
+            else:
+                path = "ragCorpora"
+
+        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)
+
+        if self._api_client.vertexai:
+            response_dict = _ListRagCorporaResponse_from_vertex(response_dict)
+
+        return_value = types.ListRagCorporaResponse._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_file(
+        self, *, config: Optional[types.GetRagFileConfigOrDict] = None, name: str
+    ) -> types.RagFile:
+        """
+        Gets a RagFile.
+
+        Args:
+          name: The name of the RagFile to retrieve.
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagFile.
+
+        """
+
+        parameter_model = types._GetRagFileRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _GetRagFileRequestParameters_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.RagFile._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_files(
+        self, *, config: Optional[types.ListRagFilesConfigOrDict] = None, name: str
+    ) -> types.ListRagFilesResponse:
+        """
+        Lists RagFile instances within a RagCorpus.
+
+        Args:
+          name: The name of the RagCorpus to list files from.
+          config: Optional configuration for the request (e.g., pagination details).
+
+        Returns:
+          A ListRagFilesResponse containing the files.
+
+        """
+
+        parameter_model = types._ListRagFilesRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _ListRagFilesRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles"
+
+        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.ListRagFilesResponse._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_config(
+        self, *, config: Optional[types.GetRagConfigOrDict] = None
+    ) -> types.RagEngineConfig:
+        """
+        Gets the project-level RAG Engine Config.
+
+        Args:
+          config: Optional configuration for the request.
+
+        Returns:
+          The requested RagEngineConfig.
+
+        """
+
+        parameter_model = types._GetRagConfigRequestParameters(
+            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 = _GetRagConfigRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragEngineConfig".format_map(request_url_dict)
+            else:
+                path = "ragEngineConfig"
+
+        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)
+
+        if self._api_client.vertexai:
+            response_dict = _RagEngineConfig_from_vertex(response_dict)
+
+        return_value = types.RagEngineConfig._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 _update_corpus(
+        self,
+        *,
+        config: Optional[types.UpdateRagCorpusConfigOrDict] = None,
+        name: Optional[str] = None,
+        rag_corpus: types.RagCorpusOrDict,
+    ) -> types.UpdateRagCorpusOperation:
+        """
+        Updates an existing Rag Corpus.
+        """
+
+        parameter_model = types._UpdateRagCorpusRequestParameters(
+            config=config,
+            name=name,
+            rag_corpus=rag_corpus,
+        )
+
+        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 = _UpdateRagCorpusRequestParameters_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(
+            "patch", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.UpdateRagCorpusOperation._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_corpus(
+        self,
+        *,
+        config: Optional[types.DeleteRagCorpusConfigOrDict] = None,
+        name: Optional[str] = None,
+    ) -> types.DeleteRagCorpusOperation:
+        """
+        Deletes a RAG Corpus.
+        """
+
+        parameter_model = types._DeleteRagCorpusRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _DeleteRagCorpusRequestParameters_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.DeleteRagCorpusOperation._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_file(
+        self, *, config: Optional[types.DeleteRagFileConfigOrDict] = None, name: str
+    ) -> types.DeleteRagFileOperation:
+        """
+        Deletes a RAG File.
+        """
+
+        parameter_model = types._DeleteRagFileRequestParameters(
+            config=config,
+            name=name,
+        )
+
+        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 = _DeleteRagFileRequestParameters_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.DeleteRagFileOperation._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 _update_config(
+        self,
+        *,
+        updated_config: types.RagEngineConfigOrDict,
+        config: Optional[types.UpdateRagConfigOrDict] = None,
+    ) -> types.UpdateRagConfigOperation:
+        """
+        Updates a RAG Engine Config.
+        """
+
+        parameter_model = types._UpdateRagConfigRequestParameters(
+            updated_config=updated_config,
+            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 = _UpdateRagConfigRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "ragEngineConfig".format_map(request_url_dict)
+            else:
+                path = "ragEngineConfig"
+
+        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(
+            "patch", path, request_dict, http_options
+        )
+
+        response_dict = {} if not response.body else json.loads(response.body)
+
+        return_value = types.UpdateRagConfigOperation._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 retrieve_contexts(
+        self,
+        *,
+        vertex_rag_store: genai_types.VertexRagStoreOrDict,
+        query: types.RagQueryOrDict,
+        config: Optional[types.RetrieveContextsConfigOrDict] = None,
+    ) -> types.RetrieveContextsResponse:
+        """
+        Retrieves contexts from a Vertex RAG store based on a query.
+
+        Args:
+          vertex_rag_store: The Vertex RAG store to retrieve contexts from.
+          query: The query to retrieve contexts for.
+          config: Optional configuration for the request.
+
+        Returns:
+          The RetrieveContextsResponse containing the relevant contexts.
+
+        """
+
+        parameter_model = types._RetrieveRagContextsRequestParameters(
+            vertex_rag_store=vertex_rag_store,
+            query=query,
+            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 = _RetrieveRagContextsRequestParameters_to_vertex(
+                parameter_model
+            )
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = ":retrieveContexts".format_map(request_url_dict)
+            else:
+                path = ":retrieveContexts"
+
+        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.RetrieveContextsResponse._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_rag_config_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetRagConfigOperationConfigOrDict] = None,
+    ) -> types.RagEngineConfigOperation:
+        parameter_model = types._GetRagConfigOperationParameters(
+            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 = _GetRagConfigOperationParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operation_name}".format_map(request_url_dict)
+            else:
+                path = "{operation_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.RagEngineConfigOperation._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 _import_files(
+        self,
+        *,
+        name: str,
+        import_rag_files_request: types.ImportRagFilesRequestOrDict,
+        config: Optional[types.ImportRagFilesRequestConfigOrDict] = None,
+    ) -> types.ImportRagFilesOperation:
+        """
+        Imports files into a RAG Corpus.
+        """
+
+        parameter_model = types._ImportRagFilesRequestParameters(
+            name=name,
+            import_rag_files_request=import_rag_files_request,
+            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 = _ImportRagFilesRequestParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles:import".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles:import"
+
+        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.ImportRagFilesOperation._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_import_files_operation(
+        self,
+        *,
+        operation_name: str,
+        config: Optional[types.GetImportFilesOperationConfigOrDict] = None,
+    ) -> types.ImportRagFilesOperation:
+        parameter_model = types._GetImportFilesOperationParameters(
+            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 = _GetImportFilesOperationParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{operation_name}".format_map(request_url_dict)
+            else:
+                path = "{operation_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.ImportRagFilesOperation._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 _upload_file(
+        self,
+        *,
+        name: str,
+        rag_file: types.RagFileOrDict,
+        upload_rag_file_config: Optional[types.UploadRagFileConfigOrDict] = None,
+        config: Optional[types.UploadRagFileRequestConfigOrDict] = None,
+    ) -> types.UploadRagFileResponse:
+        parameter_model = types._UploadRagFileParameters(
+            name=name,
+            rag_file=rag_file,
+            upload_rag_file_config=upload_rag_file_config,
+            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 = _UploadRagFileParameters_to_vertex(parameter_model)
+            request_url_dict = request_dict.get("_url")
+            if request_url_dict:
+                path = "{name}/ragFiles:upload".format_map(request_url_dict)
+            else:
+                path = "{name}/ragFiles:upload"
+
+        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.UploadRagFileResponse._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 create_corpus(
+        self,
+        *,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.CreateRagCorpusConfigOrDict] = None,
+    ) -> types.RagCorpus:
+        """
+        Creates a new RAG Corpus and waits for completion asynchronously.
+
+        Args:
+          rag_corpus: The RagCorpus to create.
+          config: The configuration to use for the RagCorpus.
+
+        Returns:
+          The created RagCorpus.
+
+        Example usage:
+
+        ```
+        corpus = await client.aio.rag.create_corpus(
+            rag_corpus=types.RagCorpus(
+                display_name="test-corpus",
+                name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            ),
+        )
+        ```
+        """
+        operation = await self._create_corpus(rag_corpus=rag_corpus, config=config)
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to create RagCorpus: {operation.error}"
+            )
+
+        return await self.get_corpus(name=operation.response.name)
+
+    async def delete_corpus(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteRagCorpusConfigOrDict] = None,
+    ) -> None:
+        """
+        Deletes a RAG Corpus and waits for the delete operation to complete asynchronously.
+
+        Args:
+          name: The name of the RagCorpus to delete, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}`.
+          config: The configuration to use for the RagCorpus delete request.
+
+        Example usage:
+
+        ```
+        await client.aio.rag.delete_corpus(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+        )
+        ```
+        """
+        operation = await self._delete_corpus(name=name, config=config)
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to delete RagCorpus: {operation.error}"
+            )
+
+        return None
+
+    async def delete_file(
+        self,
+        *,
+        name: str,
+        config: Optional[types.DeleteRagFileConfigOrDict] = None,
+    ) -> None:
+        """
+        Deletes a file from a RAG Corpus and waits for the delete operation to complete.
+
+        Args:
+          name: The name of the RagFile to delete, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}/ragFiles/{file_id}`.
+          config: The configuration to use for the RagFile delete request.
+
+        Example usage:
+
+        ```
+        await client.aio.rag.delete_file(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789/ragFiles/987654321",
+        )
+        ```
+        """
+        operation = await self._delete_file(name=name, config=config)
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to delete file from RagCorpus: {operation.error}"
+            )
+
+        return None
+
+    async def update_corpus(
+        self,
+        *,
+        name: str,
+        rag_corpus: types.RagCorpusOrDict,
+        config: Optional[types.UpdateRagCorpusConfigOrDict] = None,
+    ) -> types.RagCorpus:
+        """
+        Updates a RAG Corpus and waits for completion asynchronously.
+
+        Args:
+          name: The name of the RagCorpus to update, formatted as
+            `projects/{project}/locations/{location}/ragCorpora/{corpus_id}`.
+          rag_corpus: The RagCorpus to update.
+          config: The configuration to use for the RagCorpus update request.
+
+        Returns:
+          The updated RagCorpus.
+
+        Example usage:
+
+        ```
+        await client.aio.rag.update_corpus(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            rag_corpus=types.RagCorpus(
+                display_name="Updated corpus name",
+            ),
+        )
+        ```
+        """
+        operation = await self._update_corpus(
+            name=name, rag_corpus=rag_corpus, config=config
+        )
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_corpus_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to update RagCorpus: {operation.error}"
+            )
+
+        return await self.get_corpus(name=operation.response.name)
+
+    async def update_config(
+        self,
+        *,
+        updated_config: types.RagEngineConfigOrDict,
+        request_config: Optional[types.UpdateRagConfigOrDict] = None,
+    ) -> types.RagEngineConfig:
+        """
+        Updates a RagEngineConfig and waits for completion asynchronously.
+
+        Args:
+          updated_config: The RagEngineConfig to update.
+          request_config: The configuration to use for the RagEngineConfig update request.
+
+        Returns:
+          The updated RagEngineConfig.
+
+        Example usage:
+
+        ```
+        await client.aio.rag.update_config(
+            updated_config=types.RagEngineConfig(
+                name="projects/test-project/locations/us-central1/ragCorpora/123456789/ragEngineConfig/123456789",
+                rag_managed_db_config=types.RagManagedDbConfig(
+                    basic=types.RagManagedDbConfigBasic()
+                )
+            ),
+        )
+        ```
+        """
+        operation = await self._update_config(
+            updated_config=updated_config, config=request_config
+        )
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_rag_config_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to update RagEngineConfig: {operation.error}"
+            )
+
+        return await self.get_config()
+
+    async def import_files(
+        self,
+        *,
+        name: str,
+        import_config: types.ImportRagFilesConfigOrDict,
+        config: Optional[types.ImportRagFilesRequestConfigOrDict] = None,
+    ) -> types.ImportRagFilesResponse:
+        """
+        Imports files into a Rag Corpus. To get a list of all files on the corpus after calling import_files, call list_files.
+
+        Example usage:
+
+        ```
+        import_files_response = await client.aio.rag.import_files(
+            name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            import_config=types.ImportRagFilesConfig(
+                gcs_source=types.GcsSource(uris=["gs://test-bucket/test-file.pdf"]),
+            ),
+        )
+        ```
+
+        Args:
+          name: The name of the Rag Corpus to import files into, format:
+            `projects/{project}/locations/{location}/ragCorpora/{rag_corpus_id}`
+          import_config: The configuration for importing files, including source files and processing options.
+          config: The configuration to use for the import request.
+
+        Returns:
+          The ImportRagFilesResponse from the import files operation.
+        """
+
+        if isinstance(import_config, dict):
+            import_config = types.ImportRagFilesConfig.model_validate(import_config)
+
+        if import_config.gcs_source is not None:
+            for uri in import_config.gcs_source.uris:
+                if not _gcs_utils.GcsUtils(self._api_client)._verify_bucket_ownership(
+                    bucket_name=uri.split("/")[2],
+                    expected_project=self._api_client.project,
+                ):
+                    raise ValueError(
+                        f"Bucket {uri} does not belong to"
+                        f" project {self._api_client.project}."
+                    )
+
+        operation = await self._import_files(
+            name=name,
+            import_rag_files_request=types.ImportRagFilesRequest(
+                import_rag_files_config=import_config
+            ),
+            config=config,
+        )
+
+        operation = await _operations_utils.await_operation_async(
+            operation_name=operation.name,
+            get_operation_fn=self._get_import_files_operation,
+        )
+
+        if operation.error:
+            raise RuntimeError(
+                f"Operation {operation.name} failed to import files into RagCorpus: {operation.error}"
+            )
+
+        return operation.response
+
+    async def upload_file(
+        self,
+        *,
+        corpus_name: str,
+        path: str,
+        display_name: Optional[str] = None,
+        upload_rag_file_config: Optional[types.UploadRagFileConfigOrDict] = None,
+        request_config: Optional[types.UploadRagFileRequestConfigOrDict] = None,
+    ) -> types.RagFile:
+        """
+        Uploads a file to a RAG Corpus.
+
+        Args:
+          corpus_name: The name of the RAG Corpus to upload to.
+          path: The path to the file to upload.
+          display_name: Optional. The display name for the uploaded file. If not provided, a display name will be generated.
+          upload_rag_file_config: Optional. The configuration to use for the upload.
+          request_config: Optional. The configuration to use for the request.
+
+        Returns:
+          The uploaded RagFile.
+
+        Example usage:
+
+        ```
+        rag_file = await client.aio.rag.upload_file(
+            corpus_name="projects/test-project/locations/us-central1/ragCorpora/123456789",
+            path="/path/to/file.pdf",
+            display_name="test-file",
+            upload_rag_file_config=types.UploadRagFileConfig(
+                chunk_config=types.ChunkConfig(
+                    chunk_size=1024,
+                    chunk_overlap=128,
+                ),
+            ),
+        )
+        ```
+        """
+
+        if not display_name:
+            display_name = f"file_{_common.timestamped_unique_name()}"
+
+        rag_file = types.RagFile(display_name=display_name)
+
+        mime_type, _ = mimetypes.guess_type(path)
+
+        if mime_type is None:
+            mime_type = "application/octet-stream"
+
+        http_options, size_bytes, mime_type = _extra_utils.prepare_resumable_upload(
+            path,
+            user_http_options=request_config.http_options if request_config else None,
+            user_mime_type=mime_type,
+        )
+
+        current_api_version = self._api_client._http_options.api_version or "v1beta1"
+        upload_api_version = f"upload/{current_api_version}"
+
+        http_options.api_version = upload_api_version
+
+        parameter_model = types._UploadRagFileParameters(
+            name=corpus_name,
+            rag_file=rag_file,
+            upload_rag_file_config=upload_rag_file_config,
+        )
+        request_dict = _UploadRagFileParameters_to_vertex(parameter_model)
+
+        request_dict.pop("_url", None)
+        request_dict.pop("_query", None)
+
+        request_path = f"{corpus_name}/ragFiles:upload"
+
+        request_dict = _common.convert_to_dict(request_dict)
+        request_dict = _common.encode_unserializable_types(request_dict)
+
+        response = await self._api_client.async_request(
+            "post",
+            request_path,
+            request_dict,
+            http_options,
+        )
+
+        if response.headers is None or (
+            "x-goog-upload-url" not in response.headers
+            and "X-Goog-Upload-URL" not in response.headers
+        ):
+            raise KeyError(
+                "Failed to create file. Upload URL was not returned from the create file request."
+            )
+
+        upload_url = response.headers.get(
+            "x-goog-upload-url", response.headers.get("X-Goog-Upload-URL")
+        )
+
+        fs_path = os.fspath(path)
+        return_file = await self._api_client.async_upload_file(
+            fs_path, upload_url, size_bytes, http_options=http_options
+        )
+
+        rag_file_payload = return_file.json.get("ragFile") or return_file.json.get(
+            "rag_file", {}
+        )
+        return types.RagFile(**rag_file_payload)