From 163e726df4bd35a70936918885af6e627c51e45c Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Fri, 16 Oct 2026 10:21:04 -0700 Subject: [PATCH] refactor: consolidate Python SDK converters into single _converters module PiperOrigin-RevId: 1000940190 --- agentplatform/_genai/_converters.py | 7537 +++++++++++++++++++++ agentplatform/_genai/datasets.py | 291 +- agentplatform/_genai/endpoints.py | 127 +- agentplatform/_genai/evals.py | 1409 +--- agentplatform/_genai/example_stores.py | 306 +- agentplatform/_genai/feedback_contexts.py | 102 +- agentplatform/_genai/feedback_entries.py | 240 +- agentplatform/_genai/memories.py | 543 +- agentplatform/_genai/memory_banks.py | 673 +- agentplatform/_genai/memory_revisions.py | 56 +- agentplatform/_genai/model_garden.py | 511 +- agentplatform/_genai/prompt_optimizer.py | 199 +- agentplatform/_genai/prompts.py | 521 +- agentplatform/_genai/rag.py | 1763 +---- agentplatform/_genai/runtime_revisions.py | 140 +- agentplatform/_genai/runtimes.py | 800 +-- agentplatform/_genai/sandbox_snapshots.py | 156 +- agentplatform/_genai/sandbox_templates.py | 174 +- agentplatform/_genai/sandboxes.py | 200 +- agentplatform/_genai/serving_profiles.py | 170 +- agentplatform/_genai/session_events.py | 115 +- agentplatform/_genai/sessions.py | 189 +- agentplatform/_genai/skill_revisions.py | 66 +- agentplatform/_genai/skills.py | 239 +- 24 files changed, 8537 insertions(+), 7990 deletions(-) create mode 100644 agentplatform/_genai/_converters.py diff --git a/agentplatform/_genai/_converters.py b/agentplatform/_genai/_converters.py new file mode 100644 index 0000000000..f1093d165e --- /dev/null +++ b/agentplatform/_genai/_converters.py @@ -0,0 +1,7537 @@ +# 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. + +from typing import Any, Optional, Union + +from google.genai._common import get_value_by_path as getv +from google.genai._common import set_value_by_path as setv + +from . import _transformers as t + + +def _AppendRuntimeSessionEventConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["content"]) is not None: + setv(parent_object, ["content"], getv(from_object, ["content"])) + + if getv(from_object, ["actions"]) is not None: + setv(parent_object, ["actions"], getv(from_object, ["actions"])) + + if getv(from_object, ["error_code"]) is not None: + setv(parent_object, ["errorCode"], getv(from_object, ["error_code"])) + + if getv(from_object, ["error_message"]) is not None: + setv(parent_object, ["errorMessage"], getv(from_object, ["error_message"])) + + if getv(from_object, ["event_metadata"]) is not None: + setv(parent_object, ["eventMetadata"], getv(from_object, ["event_metadata"])) + + if getv(from_object, ["raw_event"]) is not None: + setv(parent_object, ["rawEvent"], getv(from_object, ["raw_event"])) + + return to_object + + +def _AppendRuntimeSessionEventRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["author"]) is not None: + setv(to_object, ["author"], getv(from_object, ["author"])) + + if getv(from_object, ["invocation_id"]) is not None: + setv(to_object, ["invocationId"], getv(from_object, ["invocation_id"])) + + if getv(from_object, ["timestamp"]) is not None: + setv(to_object, ["timestamp"], getv(from_object, ["timestamp"])) + + if getv(from_object, ["config"]) is not None: + _AppendRuntimeSessionEventConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _AskContextsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _AssembleDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["gemini_request_read_config"]) is not None: + setv( + to_object, + ["geminiRequestReadConfig"], + getv(from_object, ["gemini_request_read_config"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _AssessDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["gemini_request_read_config"]) is not None: + setv( + to_object, + ["geminiRequestReadConfig"], + getv(from_object, ["gemini_request_read_config"]), + ) + + if getv(from_object, ["tuning_resource_usage_assessment_config"]) is not None: + setv( + to_object, + ["tuningResourceUsageAssessmentConfig"], + getv(from_object, ["tuning_resource_usage_assessment_config"]), + ) + + if getv(from_object, ["tuning_validation_assessment_config"]) is not None: + setv( + to_object, + ["tuningValidationAssessmentConfig"], + getv(from_object, ["tuning_validation_assessment_config"]), + ) + + if ( + getv(from_object, ["batch_prediction_resource_usage_assessment_config"]) + is not None + ): + setv( + to_object, + ["batchPredictionResourceUsageAssessmentConfig"], + getv(from_object, ["batch_prediction_resource_usage_assessment_config"]), + ) + + if getv(from_object, ["batch_prediction_validation_assessment_config"]) is not None: + setv( + to_object, + ["batchPredictionValidationAssessmentConfig"], + getv(from_object, ["batch_prediction_validation_assessment_config"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CancelQueryJobRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["operation_name"]) is not None: + setv(parent_object, ["operationName"], getv(from_object, ["operation_name"])) + + return to_object + + +def _CancelQueryJobRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _CancelQueryJobRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _CheckQueryJobResult_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(parent_object, ["operationName"]) is not None: + setv(to_object, ["operation_name"], getv(parent_object, ["operationName"])) + + if getv(parent_object, ["outputGcsUri"]) is not None: + setv(to_object, ["output_gcs_uri"], getv(parent_object, ["outputGcsUri"])) + + if getv(parent_object, ["status"]) is not None: + setv(to_object, ["status"], getv(parent_object, ["status"])) + + if getv(parent_object, ["result"]) is not None: + setv(to_object, ["result"], getv(parent_object, ["result"])) + + return to_object + + +def _CheckQueryJobRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["retrieve_result"]) is not None: + setv(parent_object, ["retrieveResult"], getv(from_object, ["retrieve_result"])) + + return to_object + + +def _CheckQueryJobRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _CheckQueryJobRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _CorpusOperation_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _CreateBatchPredictionJobConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["input_config"]) is not None: + setv( + parent_object, + ["inputConfig"], + _ModelGardenBatchJobSource_to_vertex( + getv(from_object, ["input_config"]), to_object, root_object + ), + ) + + if getv(from_object, ["output_config"]) is not None: + setv( + parent_object, + ["outputConfig"], + _ModelGardenBatchJobDestination_to_vertex( + getv(from_object, ["output_config"]), to_object, root_object + ), + ) + + if getv(from_object, ["dedicated_resources"]) is not None: + setv( + parent_object, + ["dedicatedResources"], + getv(from_object, ["dedicated_resources"]), + ) + + return to_object + + +def _CreateBatchPredictionJobParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["model"]) is not None: + setv(to_object, ["model"], getv(from_object, ["model"])) + + if getv(from_object, ["config"]) is not None: + _CreateBatchPredictionJobConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["metadata_schema_uri"]) is not None: + setv( + to_object, ["metadataSchemaUri"], getv(from_object, ["metadata_schema_uri"]) + ) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["model_reference"]) is not None: + setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateDatasetVersionParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["dataset_name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["dataset_name"])) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["model_reference"]) is not None: + setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) + + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateEvaluationExperimentParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["merge_strategy"]) is not None: + setv(to_object, ["mergeStrategy"], getv(from_object, ["merge_strategy"])) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateEvaluationItemParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["evaluation_item_type"]) is not None: + setv( + to_object, + ["evaluationItemType"], + getv(from_object, ["evaluation_item_type"]), + ) + + if getv(from_object, ["gcs_uri"]) is not None: + setv(to_object, ["gcsUri"], getv(from_object, ["gcs_uri"])) + + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateEvaluationMetricParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["metric"]) is not None: + setv( + to_object, + ["metric"], + t.t_metric_for_registry(getv(from_object, ["metric"])), + ) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateEvaluationRunParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["data_source"]) is not None: + setv(to_object, ["dataSource"], getv(from_object, ["data_source"])) + + if getv(from_object, ["evaluation_config"]) is not None: + setv( + to_object, + ["evaluationConfig"], + _EvaluationRunConfig_to_vertex( + getv(from_object, ["evaluation_config"]), to_object, root_object + ), + ) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["inference_configs"]) is not None: + setv( + to_object, + ["inferenceConfigs"], + { + k: _EvaluationRunInferenceConfig_to_vertex(v, to_object, root_object) + for k, v in getv(from_object, ["inference_configs"]).items() + }, + ) + + if getv(from_object, ["analysis_configs"]) is not None: + setv( + to_object, + ["analysisConfigs"], + [item for item in getv(from_object, ["analysis_configs"])], + ) + + if getv(from_object, ["evaluation_experiment"]) is not None: + setv( + to_object, + ["evaluationExperiment"], + getv(from_object, ["evaluation_experiment"]), + ) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateEvaluationSetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["evaluation_items"]) is not None: + setv(to_object, ["evaluationItems"], getv(from_object, ["evaluation_items"])) + + 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, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateExampleStoreConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["description"]) is not None: + setv( + to_object, + ["_query", "exampleStore", "description"], + getv(from_object, ["description"]), + ) + + if getv(from_object, ["vertex_embedding_model"]) is not None: + setv( + to_object, + ["_query", "exampleStore", "exampleStoreConfig", "vertexEmbeddingModel"], + getv(from_object, ["vertex_embedding_model"]), + ) + + return to_object + + +def _CreateExampleStoreParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["display_name"]) is not None: + setv( + to_object, + ["_query", "exampleStore", "displayName"], + getv(from_object, ["display_name"]), + ) + + if getv(from_object, ["config"]) is not None: + _CreateExampleStoreConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateMemoryBankConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + return to_object + + +def _CreateMemoryBankRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["memory_bank_config"]) is not None: + setv( + to_object, + ["context_spec", "memory_bank_config"], + _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( + getv(from_object, ["memory_bank_config"]), to_object, root_object + ), + ) + + if getv(from_object, ["config"]) is not None: + _CreateMemoryBankConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateMemoryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["fact"]) is not None: + setv(to_object, ["fact"], getv(from_object, ["fact"])) + + if getv(from_object, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["config"]) is not None: + _MemoryConfig_to_vertex(getv(from_object, ["config"]), to_object, root_object) + + return to_object + + +def _CreateMultimodalDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["metadata_schema_uri"]) is not None: + setv( + to_object, ["metadataSchemaUri"], getv(from_object, ["metadata_schema_uri"]) + ) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateRagCorpusRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CreateRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["spec"]) is not None: + setv(parent_object, ["spec"], getv(from_object, ["spec"])) + + if getv(from_object, ["context_spec"]) is not None: + setv( + parent_object, + ["contextSpec"], + _ReasoningEngineContextSpec_to_vertex( + getv(from_object, ["context_spec"]), to_object, root_object + ), + ) + + if getv(from_object, ["psc_interface_config"]) is not None: + setv( + parent_object, + ["pscInterfaceConfig"], + getv(from_object, ["psc_interface_config"]), + ) + + if getv(from_object, ["agent_gateway_config"]) is not None: + setv( + parent_object, + ["agentGatewayConfig"], + getv(from_object, ["agent_gateway_config"]), + ) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["labels"]) is not None: + setv(parent_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["source_packages"]) is not None: + setv(parent_object, ["sourcePackages"], getv(from_object, ["source_packages"])) + + if getv(from_object, ["entrypoint_module"]) is not None: + setv( + parent_object, + ["entrypointModule"], + getv(from_object, ["entrypoint_module"]), + ) + + if getv(from_object, ["entrypoint_object"]) is not None: + setv( + parent_object, + ["entrypointObject"], + getv(from_object, ["entrypoint_object"]), + ) + + if getv(from_object, ["requirements_file"]) is not None: + setv( + parent_object, + ["requirementsFile"], + getv(from_object, ["requirements_file"]), + ) + + if getv(from_object, ["agent_framework"]) is not None: + setv(parent_object, ["agentFramework"], getv(from_object, ["agent_framework"])) + + if getv(from_object, ["python_version"]) is not None: + setv(parent_object, ["pythonVersion"], getv(from_object, ["python_version"])) + + return to_object + + +def _CreateRuntimeFeedbackEntryConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["feedback_labels"]) is not None: + setv(parent_object, ["feedbackLabels"], getv(from_object, ["feedback_labels"])) + + if getv(from_object, ["feedback_text"]) is not None: + setv(parent_object, ["feedbackText"], getv(from_object, ["feedback_text"])) + + if getv(from_object, ["user_id"]) is not None: + setv(parent_object, ["userId"], getv(from_object, ["user_id"])) + + if getv(from_object, ["source"]) is not None: + setv(parent_object, ["source"], getv(from_object, ["source"])) + + if getv(from_object, ["custom_metadata"]) is not None: + setv(parent_object, ["customMetadata"], getv(from_object, ["custom_metadata"])) + + return to_object + + +def _CreateRuntimeFeedbackEntryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["name"])) + + if getv(from_object, ["feedback_type"]) is not None: + setv(to_object, ["feedbackType"], getv(from_object, ["feedback_type"])) + + if getv(from_object, ["session_id"]) is not None: + setv(to_object, ["sessionId"], getv(from_object, ["session_id"])) + + if getv(from_object, ["event_id"]) is not None: + setv(to_object, ["eventId"], getv(from_object, ["event_id"])) + + if getv(from_object, ["config"]) is not None: + _CreateRuntimeFeedbackEntryConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + _CreateRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateRuntimeSandboxConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + if getv(from_object, ["sandbox_environment_template"]) is not None: + setv( + parent_object, + ["sandboxEnvironmentTemplate"], + getv(from_object, ["sandbox_environment_template"]), + ) + + if getv(from_object, ["sandbox_environment_snapshot"]) is not None: + setv( + parent_object, + ["sandboxEnvironmentSnapshot"], + getv(from_object, ["sandbox_environment_snapshot"]), + ) + + if getv(from_object, ["owner"]) is not None: + setv(parent_object, ["owner"], getv(from_object, ["owner"])) + + return to_object + + +def _CreateRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["spec"]) is not None: + setv(to_object, ["spec"], getv(from_object, ["spec"])) + + if getv(from_object, ["config"]) is not None: + _CreateRuntimeSandboxConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateRuntimeSandboxSnapshotConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["owner"]) is not None: + setv(parent_object, ["owner"], getv(from_object, ["owner"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + return to_object + + +def _CreateRuntimeSessionConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["session_state"]) is not None: + setv(parent_object, ["sessionState"], getv(from_object, ["session_state"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + if getv(from_object, ["expire_time"]) is not None: + setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) + + if getv(from_object, ["labels"]) is not None: + setv(parent_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["session_id"]) is not None: + setv(parent_object, ["_query", "sessionId"], getv(from_object, ["session_id"])) + + return to_object + + +def _CreateRuntimeSessionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["user_id"]) is not None: + setv(to_object, ["userId"], getv(from_object, ["user_id"])) + + if getv(from_object, ["config"]) is not None: + _CreateRuntimeSessionConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateSandboxEnvironmentSnapshotRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["source_sandbox_environment_name"]) is not None: + setv( + to_object, + ["_url", "name"], + getv(from_object, ["source_sandbox_environment_name"]), + ) + + if getv(from_object, ["config"]) is not None: + _CreateRuntimeSandboxSnapshotConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateSandboxEnvironmentTemplateConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["custom_container_environment"]) is not None: + setv( + parent_object, + ["customContainerEnvironment"], + getv(from_object, ["custom_container_environment"]), + ) + + if getv(from_object, ["default_container_environment"]) is not None: + setv( + parent_object, + ["defaultContainerEnvironment"], + getv(from_object, ["default_container_environment"]), + ) + + if getv(from_object, ["egress_control_config"]) is not None: + setv( + parent_object, + ["egressControlConfig"], + getv(from_object, ["egress_control_config"]), + ) + + if getv(from_object, ["ingress_control_config"]) is not None: + setv( + parent_object, + ["ingressControlConfig"], + getv(from_object, ["ingress_control_config"]), + ) + + return to_object + + +def _CreateSandboxEnvironmentTemplateRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["config"]) is not None: + _CreateSandboxEnvironmentTemplateConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateServingProfileConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + return to_object + + +def _CreateServingProfileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["serving_profile_id"]) is not None: + setv( + to_object, + ["_query", "servingProfileId"], + getv(from_object, ["serving_profile_id"]), + ) + + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["cmek_config"]) is not None: + setv(to_object, ["cmekConfig"], getv(from_object, ["cmek_config"])) + + if getv(from_object, ["config"]) is not None: + _CreateServingProfileConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CreateSkillConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["zipped_filesystem"]) is not None: + setv( + parent_object, + ["zippedFilesystem"], + getv(from_object, ["zipped_filesystem"]), + ) + + return to_object + + +def _CreateSkillRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["skill_id"]) is not None: + setv(to_object, ["_query", "skillId"], getv(from_object, ["skill_id"])) + + if getv(from_object, ["config"]) is not None: + _CreateSkillConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _CustomCodeExecutionSpec_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["evaluationFunction"]) is not None: + setv( + to_object, + ["evaluation_function"], + getv(from_object, ["evaluationFunction"]), + ) + + if getv(from_object, ["evaluation_function"]) is not None: + setv( + to_object, + ["remote_custom_function"], + getv(from_object, ["evaluation_function"]), + ) + + if getv(from_object, ["codeExecutionRegion"]) is not None: + setv( + to_object, + ["code_execution_region"], + getv(from_object, ["codeExecutionRegion"]), + ) + + return to_object + + +def _CustomCodeExecutionSpec_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["evaluation_function"]) is not None: + setv( + to_object, + ["evaluationFunction"], + getv(from_object, ["evaluation_function"]), + ) + + if getv(from_object, ["remote_custom_function"]) is not None: + setv( + to_object, + ["evaluation_function"], + getv(from_object, ["remote_custom_function"]), + ) + + if getv(from_object, ["code_execution_region"]) is not None: + setv( + to_object, + ["codeExecutionRegion"], + getv(from_object, ["code_execution_region"]), + ) + + return to_object + + +def _CustomJobParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["custom_job"]) is not None: + setv( + parent_object, + ["customJob"], + _CustomJob_to_vertex( + getv(from_object, ["custom_job"]), to_object, root_object + ), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _CustomJob_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(parent_object, ["displayName"]) is not None: + setv(to_object, ["display_name"], getv(parent_object, ["displayName"])) + + if getv(parent_object, ["jobSpec"]) is not None: + setv(to_object, ["job_spec"], getv(parent_object, ["jobSpec"])) + + if getv(parent_object, ["encryptionSpec"]) is not None: + setv(to_object, ["encryption_spec"], getv(parent_object, ["encryptionSpec"])) + + if getv(from_object, ["state"]) is not None: + setv(to_object, ["state"], getv(from_object, ["state"])) + + if getv(parent_object, ["error"]) is not None: + setv(to_object, ["error"], getv(parent_object, ["error"])) + + if getv(from_object, ["createTime"]) is not None: + setv(to_object, ["create_time"], getv(from_object, ["createTime"])) + + if getv(from_object, ["endTime"]) is not None: + setv(to_object, ["end_time"], getv(from_object, ["endTime"])) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["name"]) is not None: + setv(to_object, ["name"], getv(from_object, ["name"])) + + 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, ["startTime"]) is not None: + setv(to_object, ["start_time"], getv(from_object, ["startTime"])) + + if getv(from_object, ["updateTime"]) is not None: + setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) + + if getv(from_object, ["webAccessUris"]) is not None: + setv(to_object, ["web_access_uris"], getv(from_object, ["webAccessUris"])) + + return to_object + + +def _CustomJob_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["job_spec"]) is not None: + setv(parent_object, ["jobSpec"], getv(from_object, ["job_spec"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["state"]) is not None: + setv(to_object, ["state"], getv(from_object, ["state"])) + + if getv(from_object, ["error"]) is not None: + setv(parent_object, ["error"], getv(from_object, ["error"])) + + if getv(from_object, ["create_time"]) is not None: + setv(to_object, ["createTime"], getv(from_object, ["create_time"])) + + if getv(from_object, ["end_time"]) is not None: + setv(to_object, ["endTime"], getv(from_object, ["end_time"])) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["name"]) is not None: + setv(to_object, ["name"], getv(from_object, ["name"])) + + 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, ["start_time"]) is not None: + setv(to_object, ["startTime"], getv(from_object, ["start_time"])) + + if getv(from_object, ["update_time"]) is not None: + setv(to_object, ["updateTime"], getv(from_object, ["update_time"])) + + if getv(from_object, ["web_access_uris"]) is not None: + setv(to_object, ["webAccessUris"], getv(from_object, ["web_access_uris"])) + + return to_object + + +def _DeleteDatasetRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["prompt_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["prompt_id"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteEndpointRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "name"], + t.t_strict_endpoint(getv(from_object, ["name"])), + ) + + return to_object + + +def _DeleteEvaluationExperimentParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteEvaluationMetricParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric_resource_name"]) is not None: + setv( + to_object, + ["_url", "evaluation_metric"], + getv(from_object, ["metric_resource_name"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteEvaluationSetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteExampleStoreConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + return to_object + + +def _DeleteExampleStoreRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, ["_url", "name"], t.t_example_store(getv(from_object, ["name"])) + ) + + if getv(from_object, ["config"]) is not None: + _DeleteExampleStoreConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _DeleteMemoryBankRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["force"]) is not None: + setv(to_object, ["force"], getv(from_object, ["force"])) + + return to_object + + +def _DeleteMemoryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteMultimodalDatasetRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeletePromptVersionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["prompt_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["prompt_id"])) + + if getv(from_object, ["version_id"]) is not None: + setv(to_object, ["_url", "version_id"], getv(from_object, ["version_id"])) + + 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, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteRagFileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["force"]) is not None: + setv(to_object, ["force"], getv(from_object, ["force"])) + + return to_object + + +def _DeleteRuntimeRevisionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteRuntimeSessionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteServingProfileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeleteSkillRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _DeployRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["destination"]) is not None: + setv(to_object, ["_url", "destination"], getv(from_object, ["destination"])) + + if getv(from_object, ["publisher_model_name"]) is not None: + setv( + to_object, + ["publisherModelName"], + getv(from_object, ["publisher_model_name"]), + ) + + if getv(from_object, ["hugging_face_model_id"]) is not None: + setv( + to_object, + ["huggingFaceModelId"], + getv(from_object, ["hugging_face_model_id"]), + ) + + if getv(from_object, ["custom_model"]) is not None: + setv(to_object, ["customModel"], getv(from_object, ["custom_model"])) + + if getv(from_object, ["model_config_val"]) is not None: + setv(to_object, ["modelConfig"], getv(from_object, ["model_config_val"])) + + if getv(from_object, ["endpoint_config"]) is not None: + setv(to_object, ["endpointConfig"], getv(from_object, ["endpoint_config"])) + + if getv(from_object, ["deploy_config"]) is not None: + setv(to_object, ["deployConfig"], getv(from_object, ["deploy_config"])) + + return to_object + + +def _EvaluateInstancesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["bleu_input"]) is not None: + setv(to_object, ["bleuInput"], getv(from_object, ["bleu_input"])) + + if getv(from_object, ["exact_match_input"]) is not None: + setv(to_object, ["exactMatchInput"], getv(from_object, ["exact_match_input"])) + + if getv(from_object, ["rouge_input"]) is not None: + setv(to_object, ["rougeInput"], getv(from_object, ["rouge_input"])) + + if getv(from_object, ["pointwise_metric_input"]) is not None: + setv( + to_object, + ["pointwiseMetricInput"], + getv(from_object, ["pointwise_metric_input"]), + ) + + if getv(from_object, ["pairwise_metric_input"]) is not None: + setv( + to_object, + ["pairwiseMetricInput"], + getv(from_object, ["pairwise_metric_input"]), + ) + + if getv(from_object, ["tool_call_valid_input"]) is not None: + setv( + to_object, + ["toolCallValidInput"], + getv(from_object, ["tool_call_valid_input"]), + ) + + if getv(from_object, ["tool_name_match_input"]) is not None: + setv( + to_object, + ["toolNameMatchInput"], + getv(from_object, ["tool_name_match_input"]), + ) + + if getv(from_object, ["tool_parameter_key_match_input"]) is not None: + setv( + to_object, + ["toolParameterKeyMatchInput"], + getv(from_object, ["tool_parameter_key_match_input"]), + ) + + if getv(from_object, ["tool_parameter_kv_match_input"]) is not None: + setv( + to_object, + ["toolParameterKvMatchInput"], + getv(from_object, ["tool_parameter_kv_match_input"]), + ) + + if getv(from_object, ["rubric_based_metric_input"]) is not None: + setv( + to_object, + ["rubricBasedMetricInput"], + _RubricBasedMetricInput_to_vertex( + getv(from_object, ["rubric_based_metric_input"]), to_object, root_object + ), + ) + + if getv(from_object, ["autorater_config"]) is not None: + setv(to_object, ["autoraterConfig"], getv(from_object, ["autorater_config"])) + + if getv(from_object, ["metrics"]) is not None: + setv( + to_object, + ["metrics"], + [item for item in t.t_metrics(getv(from_object, ["metrics"]))], + ) + + if getv(from_object, ["instance"]) is not None: + setv( + to_object, + ["instance"], + _EvaluationInstance_to_vertex( + getv(from_object, ["instance"]), to_object, root_object + ), + ) + + if getv(from_object, ["metric_sources"]) is not None: + setv( + to_object, + ["metricSources"], + [ + item + for item in t.t_metric_sources(getv(from_object, ["metric_sources"])) + ], + ) + + if getv(from_object, ["allow_cross_region_model"]) is not None: + setv( + to_object, + ["allowCrossRegionModel"], + getv(from_object, ["allow_cross_region_model"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _EvaluationInstance_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["prompt"]) is not None: + setv(to_object, ["prompt"], getv(from_object, ["prompt"])) + + if getv(from_object, ["response"]) is not None: + setv(to_object, ["response"], getv(from_object, ["response"])) + + if getv(from_object, ["reference"]) is not None: + setv(to_object, ["reference"], getv(from_object, ["reference"])) + + if getv(from_object, ["other_data"]) is not None: + setv(to_object, ["otherData"], getv(from_object, ["other_data"])) + + if getv(from_object, ["agent_data"]) is not None: + setv(to_object, ["agent_eval_data"], getv(from_object, ["agent_data"])) + + if getv(from_object, ["rubric_groups"]) is not None: + setv(to_object, ["rubricGroups"], getv(from_object, ["rubric_groups"])) + + if getv(from_object, ["interactions_data_source"]) is not None: + setv( + to_object, + ["interactionsDataSource"], + getv(from_object, ["interactions_data_source"]), + ) + + return to_object + + +def _EvaluationMetric_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["displayName"]) is not None: + setv(to_object, ["display_name"], getv(from_object, ["displayName"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["metric"]) is not None: + setv( + to_object, + ["metric"], + _UnifiedMetric_from_vertex( + getv(from_object, ["metric"]), to_object, root_object + ), + ) + + if getv(from_object, ["encryptionSpec"]) is not None: + setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) + + return to_object + + +def _EvaluationRunConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metrics"]) is not None: + setv( + to_object, + ["metrics"], + [ + _EvaluationRunMetric_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["metrics"]) + ], + ) + + if getv(from_object, ["outputConfig"]) is not None: + setv(to_object, ["output_config"], getv(from_object, ["outputConfig"])) + + if getv(from_object, ["autoraterConfig"]) is not None: + setv(to_object, ["autorater_config"], getv(from_object, ["autoraterConfig"])) + + if getv(from_object, ["promptTemplate"]) is not None: + setv(to_object, ["prompt_template"], getv(from_object, ["promptTemplate"])) + + if getv(from_object, ["lossAnalysisConfig"]) is not None: + setv( + to_object, + ["loss_analysis_config"], + [item for item in getv(from_object, ["lossAnalysisConfig"])], + ) + + if getv(from_object, ["allowCrossRegionModel"]) is not None: + setv( + to_object, + ["allow_cross_region_model"], + getv(from_object, ["allowCrossRegionModel"]), + ) + + return to_object + + +def _EvaluationRunConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metrics"]) is not None: + setv( + to_object, + ["metrics"], + [ + _EvaluationRunMetric_to_vertex(item, to_object, root_object) + for item in getv(from_object, ["metrics"]) + ], + ) + + if getv(from_object, ["output_config"]) is not None: + setv(to_object, ["outputConfig"], getv(from_object, ["output_config"])) + + if getv(from_object, ["autorater_config"]) is not None: + setv(to_object, ["autoraterConfig"], getv(from_object, ["autorater_config"])) + + if getv(from_object, ["prompt_template"]) is not None: + setv(to_object, ["promptTemplate"], getv(from_object, ["prompt_template"])) + + if getv(from_object, ["loss_analysis_config"]) is not None: + setv( + to_object, + ["lossAnalysisConfig"], + [item for item in getv(from_object, ["loss_analysis_config"])], + ) + + if getv(from_object, ["allow_cross_region_model"]) is not None: + setv( + to_object, + ["allowCrossRegionModel"], + getv(from_object, ["allow_cross_region_model"]), + ) + + return to_object + + +def _EvaluationRunInferenceConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["agentConfig"]) is not None: + setv(to_object, ["agent_config"], getv(from_object, ["agentConfig"])) + + if getv(from_object, ["model"]) is not None: + setv(to_object, ["model"], getv(from_object, ["model"])) + + if getv(from_object, ["promptTemplate"]) is not None: + setv(to_object, ["prompt_template"], getv(from_object, ["promptTemplate"])) + + if getv(from_object, ["agentRunConfig"]) is not None: + setv(to_object, ["agent_run_config"], getv(from_object, ["agentRunConfig"])) + + if getv(from_object, ["agents"]) is not None: + setv(to_object, ["agent_configs"], getv(from_object, ["agents"])) + + return to_object + + +def _EvaluationRunInferenceConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["agent_config"]) is not None: + setv(to_object, ["agentConfig"], getv(from_object, ["agent_config"])) + + if getv(from_object, ["model"]) is not None: + setv(to_object, ["model"], getv(from_object, ["model"])) + + if getv(from_object, ["prompt_template"]) is not None: + setv(to_object, ["promptTemplate"], getv(from_object, ["prompt_template"])) + + if getv(from_object, ["agent_run_config"]) is not None: + setv(to_object, ["agentRunConfig"], getv(from_object, ["agent_run_config"])) + + if getv(from_object, ["agent_configs"]) is not None: + setv(to_object, ["agents"], getv(from_object, ["agent_configs"])) + + return to_object + + +def _EvaluationRunMetric_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric"]) is not None: + setv(to_object, ["metric"], getv(from_object, ["metric"])) + + if getv(from_object, ["metricResourceName"]) is not None: + setv( + to_object, + ["metric_resource_name"], + getv(from_object, ["metricResourceName"]), + ) + + if getv(from_object, ["metricConfig"]) is not None: + setv( + to_object, + ["metric_config"], + _UnifiedMetric_from_vertex( + getv(from_object, ["metricConfig"]), to_object, root_object + ), + ) + + return to_object + + +def _EvaluationRunMetric_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric"]) is not None: + setv(to_object, ["metric"], getv(from_object, ["metric"])) + + if getv(from_object, ["metric_resource_name"]) is not None: + setv( + to_object, + ["metricResourceName"], + getv(from_object, ["metric_resource_name"]), + ) + + if getv(from_object, ["metric_config"]) is not None: + setv( + to_object, + ["metricConfig"], + _UnifiedMetric_to_vertex( + getv(from_object, ["metric_config"]), to_object, root_object + ), + ) + + return to_object + + +def _EvaluationRun_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["displayName"]) is not None: + setv(to_object, ["display_name"], getv(from_object, ["displayName"])) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["createTime"]) is not None: + setv(to_object, ["create_time"], getv(from_object, ["createTime"])) + + if getv(from_object, ["completionTime"]) is not None: + setv(to_object, ["completion_time"], getv(from_object, ["completionTime"])) + + if getv(from_object, ["state"]) is not None: + setv(to_object, ["state"], getv(from_object, ["state"])) + + if getv(from_object, ["evaluationSetSnapshot"]) is not None: + setv( + to_object, + ["evaluation_set_snapshot"], + getv(from_object, ["evaluationSetSnapshot"]), + ) + + if getv(from_object, ["error"]) is not None: + setv(to_object, ["error"], getv(from_object, ["error"])) + + if getv(from_object, ["dataSource"]) is not None: + setv(to_object, ["data_source"], getv(from_object, ["dataSource"])) + + if getv(from_object, ["evaluationResults"]) is not None: + setv( + to_object, + ["evaluation_run_results"], + getv(from_object, ["evaluationResults"]), + ) + + if getv(from_object, ["evaluationConfig"]) is not None: + setv( + to_object, + ["evaluation_config"], + _EvaluationRunConfig_from_vertex( + getv(from_object, ["evaluationConfig"]), to_object, root_object + ), + ) + + if getv(from_object, ["inferenceConfigs"]) is not None: + setv( + to_object, + ["inference_configs"], + { + k: _EvaluationRunInferenceConfig_from_vertex(v, to_object, root_object) + for k, v in getv(from_object, ["inferenceConfigs"]).items() + }, + ) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["analysisConfigs"]) is not None: + setv( + to_object, + ["analysis_configs"], + [item for item in getv(from_object, ["analysisConfigs"])], + ) + + if getv(from_object, ["encryptionSpec"]) is not None: + setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) + + return to_object + + +def _ExecuteCodeRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["inputs"]) is not None: + setv(to_object, ["inputs"], [item for item in getv(from_object, ["inputs"])]) + + return to_object + + +def _ExportPublisherModelConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["destination"]) is not None: + setv(parent_object, ["destination"], getv(from_object, ["destination"])) + + return to_object + + +def _ExportPublisherModelRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ExportPublisherModelConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _FetchExamplesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["example_ids"]) is not None: + setv(parent_object, ["exampleIds"], getv(from_object, ["example_ids"])) + + if getv(from_object, ["stored_contents_example_filter"]) is not None: + setv( + parent_object, + ["storedContentsExampleFilter"], + getv(from_object, ["stored_contents_example_filter"]), + ) + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["pageToken"], getv(from_object, ["page_token"])) + + return to_object + + +def _FetchExamplesParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "exampleStore"], + t.t_example_store(getv(from_object, ["name"])), + ) + + if getv(from_object, ["config"]) is not None: + _FetchExamplesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _GenerateInstanceRubricsRequest_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["contents"]) is not None: + setv(to_object, ["contents"], getv(from_object, ["contents"])) + + if getv(from_object, ["predefined_rubric_generation_spec"]) is not None: + setv( + to_object, + ["predefinedRubricGenerationSpec"], + getv(from_object, ["predefined_rubric_generation_spec"]), + ) + + if getv(from_object, ["rubric_generation_spec"]) is not None: + setv( + to_object, + ["rubricGenerationSpec"], + getv(from_object, ["rubric_generation_spec"]), + ) + + if getv(from_object, ["metric_resource_name"]) is not None: + setv( + to_object, + ["metricResourceName"], + getv(from_object, ["metric_resource_name"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GenerateLossClustersParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["location"]) is not None: + setv(to_object, ["location"], getv(from_object, ["location"])) + + if getv(from_object, ["evaluation_set"]) is not None: + setv(to_object, ["evaluationSet"], getv(from_object, ["evaluation_set"])) + + if getv(from_object, ["inline_results"]) is not None: + setv( + to_object, + ["inlineResults", "evaluationResults"], + [ + item + for item in t.t_inline_results(getv(from_object, ["inline_results"])) + ], + ) + + if getv(from_object, ["configs"]) is not None: + setv(to_object, ["configs"], [item for item in getv(from_object, ["configs"])]) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GenerateMemoriesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["disable_consolidation"]) is not None: + setv( + parent_object, + ["disableConsolidation"], + getv(from_object, ["disable_consolidation"]), + ) + + if getv(from_object, ["revision_labels"]) is not None: + setv(parent_object, ["revisionLabels"], getv(from_object, ["revision_labels"])) + + if getv(from_object, ["revision_expire_time"]) is not None: + setv( + parent_object, + ["revisionExpireTime"], + getv(from_object, ["revision_expire_time"]), + ) + + if getv(from_object, ["revision_ttl"]) is not None: + setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) + + if getv(from_object, ["disable_memory_revisions"]) is not None: + setv( + parent_object, + ["disableMemoryRevisions"], + getv(from_object, ["disable_memory_revisions"]), + ) + + if getv(from_object, ["metadata"]) is not None: + setv( + parent_object, + ["metadata"], + {k: v for k, v in getv(from_object, ["metadata"]).items()}, + ) + + if getv(from_object, ["metadata_merge_strategy"]) is not None: + setv( + parent_object, + ["metadataMergeStrategy"], + getv(from_object, ["metadata_merge_strategy"]), + ) + + if getv(from_object, ["allowed_topics"]) is not None: + setv( + parent_object, + ["allowedTopics"], + [item for item in getv(from_object, ["allowed_topics"])], + ) + + return to_object + + +def _GenerateMemoriesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["vertex_session_source"]) is not None: + setv( + to_object, + ["vertexSessionSource"], + getv(from_object, ["vertex_session_source"]), + ) + + if getv(from_object, ["direct_contents_source"]) is not None: + setv( + to_object, + ["directContentsSource"], + getv(from_object, ["direct_contents_source"]), + ) + + if getv(from_object, ["direct_memories_source"]) is not None: + setv( + to_object, + ["directMemoriesSource"], + getv(from_object, ["direct_memories_source"]), + ) + + if getv(from_object, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["config"]) is not None: + _GenerateMemoriesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _GenerateUserScenariosParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["location"]) is not None: + setv(to_object, ["location"], getv(from_object, ["location"])) + + if getv(from_object, ["agents"]) is not None: + setv(to_object, ["agents"], getv(from_object, ["agents"])) + + if getv(from_object, ["root_agent_id"]) is not None: + setv(to_object, ["rootAgentId"], getv(from_object, ["root_agent_id"])) + + if getv(from_object, ["user_scenario_generation_config"]) is not None: + setv( + to_object, + ["userScenarioGenerationConfig"], + t.t_user_scenario_generation_config( + getv(from_object, ["user_scenario_generation_config"]) + ), + ) + + if getv(from_object, ["allow_cross_region_model"]) is not None: + setv( + to_object, + ["allowCrossRegionModel"], + getv(from_object, ["allow_cross_region_model"]), + ) + + if getv(from_object, ["gemini_agent_config"]) is not None: + setv( + to_object, ["geminiAgentConfig"], getv(from_object, ["gemini_agent_config"]) + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetBatchPredictionJobParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetCorpusOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetCustomJobParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetDatasetOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["operation_id"]) is not None: + setv(to_object, ["_url", "operation_id"], getv(from_object, ["operation_id"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetDatasetVersionParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["dataset_version_id"]) is not None: + setv( + to_object, + ["_url", "dataset_version_id"], + getv(from_object, ["dataset_version_id"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetDeleteRuntimeRevisionOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetDeployOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetEndpointOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetEndpointParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "name"], + t.t_strict_endpoint(getv(from_object, ["name"])), + ) + + return to_object + + +def _GetEvaluationExperimentParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetEvaluationItemParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetEvaluationMetricParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric_resource_name"]) is not None: + setv( + to_object, + ["_url", "evaluation_metric"], + getv(from_object, ["metric_resource_name"]), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetEvaluationRunParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetEvaluationSetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetExampleStoreOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetExampleStoreParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, ["_url", "name"], t.t_example_store(getv(from_object, ["name"])) + ) + + return to_object + + +def _GetExportPublisherModelOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetGenerateMemoriesOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetMemoryBankOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetMemoryBankRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetMemoryOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetMemoryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetMemoryRevisionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetMultimodalDatasetOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["operation_id"]) is not None: + setv(to_object, ["_url", "operation_id"], getv(from_object, ["operation_id"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetMultimodalDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetPublisherModelConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["hugging_face_token"]) is not None: + setv( + parent_object, + ["_query", "huggingFaceToken"], + getv(from_object, ["hugging_face_token"]), + ) + + if ( + getv(from_object, ["include_equivalent_model_garden_model_deployment_configs"]) + is not None + ): + setv( + parent_object, + ["_query", "includeEquivalentModelGardenModelDeploymentConfigs"], + getv( + from_object, + ["include_equivalent_model_garden_model_deployment_configs"], + ), + ) + + if getv(from_object, ["is_hugging_face_model"]) is not None: + setv( + parent_object, + ["_query", "isHuggingFaceModel"], + getv(from_object, ["is_hugging_face_model"]), + ) + + return to_object + + +def _GetPublisherModelRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _GetPublisherModelConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _GetRagConfigOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetRagFileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetRuntimeFeedbackContextOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeFeedbackContextRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetRuntimeFeedbackOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeFeedbackRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetRuntimeOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetRuntimeRevisionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetRuntimeSandboxOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetRuntimeSandboxSnapshotOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeSessionOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetRuntimeSessionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetSandboxEnvironmentTemplateOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetSandboxEnvironmentTemplateRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _GetServingProfileOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetServingProfileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetSkillOperationParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _GetSkillRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _GetSkillRevisionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _ImportEvaluationSetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["evaluation_set"]) is not None: + setv(to_object, ["evaluationSet"], getv(from_object, ["evaluation_set"])) + + if getv(from_object, ["gcs_destination"]) is not None: + setv(to_object, ["gcsDestination"], getv(from_object, ["gcs_destination"])) + + if getv(from_object, ["gcs_source"]) is not None: + setv(to_object, ["gcsSource"], getv(from_object, ["gcs_source"])) + + if getv(from_object, ["inline_source"]) is not None: + setv(to_object, ["inlineSource"], getv(from_object, ["inline_source"])) + + if getv(from_object, ["cloud_trace_source"]) is not None: + setv(to_object, ["cloudTraceSource"], getv(from_object, ["cloud_trace_source"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _ImportRagFilesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _IngestEventsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["force_flush"]) is not None: + setv(parent_object, ["forceFlush"], getv(from_object, ["force_flush"])) + + if getv(from_object, ["revision_labels"]) is not None: + setv(parent_object, ["revisionLabels"], getv(from_object, ["revision_labels"])) + + if getv(from_object, ["revision_expire_time"]) is not None: + setv( + parent_object, + ["revisionExpireTime"], + getv(from_object, ["revision_expire_time"]), + ) + + if getv(from_object, ["revision_ttl"]) is not None: + setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) + + if getv(from_object, ["disable_memory_revisions"]) is not None: + setv( + parent_object, + ["disableMemoryRevisions"], + getv(from_object, ["disable_memory_revisions"]), + ) + + if getv(from_object, ["metadata"]) is not None: + setv( + parent_object, + ["metadata"], + {k: v for k, v in getv(from_object, ["metadata"]).items()}, + ) + + if getv(from_object, ["metadata_merge_strategy"]) is not None: + setv( + parent_object, + ["metadataMergeStrategy"], + getv(from_object, ["metadata_merge_strategy"]), + ) + + return to_object + + +def _IngestEventsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["stream_id"]) is not None: + setv(to_object, ["streamId"], getv(from_object, ["stream_id"])) + + if getv(from_object, ["direct_contents_source"]) is not None: + setv( + to_object, + ["directContentsSource"], + getv(from_object, ["direct_contents_source"]), + ) + + if getv(from_object, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["generation_trigger_config"]) is not None: + setv( + to_object, + ["generationTriggerConfig"], + getv(from_object, ["generation_trigger_config"]), + ) + + if getv(from_object, ["config"]) is not None: + _IngestEventsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListDatasetVersionsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["read_mask"]) is not None: + setv(to_object, ["_url", "read_mask"], getv(from_object, ["read_mask"])) + + if getv(from_object, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListPromptsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListDatasetsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListPromptsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListEvaluationExperimentsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["order_by"]) is not None: + setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) + + return to_object + + +def _ListEvaluationExperimentsParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListEvaluationExperimentsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListEvaluationMetricsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["order_by"]) is not None: + setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) + + return to_object + + +def _ListEvaluationMetricsParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListEvaluationMetricsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListEvaluationMetricsResponse_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["evaluationMetrics"]) is not None: + setv( + to_object, + ["evaluation_metrics"], + [ + _EvaluationMetric_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["evaluationMetrics"]) + ], + ) + + return to_object + + +def _ListEvaluationSetsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["order_by"]) is not None: + setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) + + return to_object + + +def _ListEvaluationSetsParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListEvaluationSetsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListMemoriesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["order_by"]) is not None: + setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) + + return to_object + + +def _ListMemoriesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListMemoriesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListMemoryBanksConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _ListMemoryBanksRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + _ListMemoryBanksConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListMemoryRevisionsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListMemoryRevisionsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListMemoryRevisionsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListMultimodalDatasetsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListMultimodalDatasetsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListMultimodalDatasetsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListPromptsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListPublisherModelsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["list_all_versions"]) is not None: + setv( + parent_object, + ["_query", "listAllVersions"], + getv(from_object, ["list_all_versions"]), + ) + + return to_object + + +def _ListPublisherModelsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["config"]) is not None: + _ListPublisherModelsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRagCorporaConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _ListRagCorporaResponse_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListRagFilesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListReasoningEnginesResponse_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["reasoningEngines"]) is not None: + setv( + to_object, + ["reasoning_engines"], + [ + _ReasoningEngine_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["reasoningEngines"]) + ], + ) + + return to_object + + +def _ListRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListRuntimeFeedbackEntriesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["order_by"]) is not None: + setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) + + return to_object + + +def _ListRuntimeFeedbackEntriesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["config"]) is not None: + _ListRuntimeFeedbackEntriesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + _ListRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRuntimeRevisionsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListRuntimeRevisionsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListRuntimeRevisionsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRuntimeSandboxesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListRuntimeSandboxesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListRuntimeSandboxesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRuntimeSessionEventsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListRuntimeSessionEventsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListRuntimeSessionEventsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListRuntimeSessionsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListRuntimeSessionsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListRuntimeSessionsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListSandboxEnvironmentSnapshotsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListSandboxEnvironmentSnapshotsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListSandboxEnvironmentTemplatesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _ListSandboxEnvironmentTemplatesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _ListServingProfilesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _ListServingProfilesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListServingProfilesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListSkillRevisionsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _ListSkillRevisionsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListSkillRevisionsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ListSkillsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["page_size"]) is not None: + setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) + + if getv(from_object, ["page_token"]) is not None: + setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) + + return to_object + + +def _ListSkillsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _ListSkillsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _ManagedSemanticMemoryConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["generationConfig"]) is not None: + setv(to_object, ["generation_config"], getv(from_object, ["generationConfig"])) + + if getv(from_object, ["ttlConfig"]) is not None: + setv(to_object, ["ttl_config"], getv(from_object, ["ttlConfig"])) + + if getv(from_object, ["disableMemoryRevisions"]) is not None: + setv( + to_object, + ["disable_memory_revisions"], + getv(from_object, ["disableMemoryRevisions"]), + ) + + if getv(from_object, ["similaritySearchConfig"]) is not None: + setv( + to_object, + ["similarity_search_config"], + getv(from_object, ["similaritySearchConfig"]), + ) + + if getv(from_object, ["unstructuredMemoryConfigs"]) is not None: + setv( + to_object, + ["unstructured_memory_configs"], + [item for item in getv(from_object, ["unstructuredMemoryConfigs"])], + ) + + if getv(from_object, ["structuredMemoryConfigs"]) is not None: + setv( + to_object, + ["structured_memory_configs"], + [ + _StructuredMemoryConfig_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["structuredMemoryConfigs"]) + ], + ) + + return to_object + + +def _MemoryBankOperation_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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"], + _MemoryBank_from_vertex( + getv(from_object, ["response"]), to_object, root_object + ), + ) + + return to_object + + +def _MemoryBank_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["managedSemanticMemoryConfig"]) is not None: + setv( + to_object, + ["managed_semantic_memory_config"], + _ManagedSemanticMemoryConfig_from_vertex( + getv(from_object, ["managedSemanticMemoryConfig"]), + to_object, + root_object, + ), + ) + + if getv(from_object, ["displayName"]) is not None: + setv(to_object, ["display_name"], getv(from_object, ["displayName"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["createTime"]) is not None: + setv(to_object, ["create_time"], getv(from_object, ["createTime"])) + + if getv(from_object, ["updateTime"]) is not None: + setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) + + if getv(from_object, ["encryptionSpec"]) is not None: + setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) + + return to_object + + +def _MemoryConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + if getv(from_object, ["expire_time"]) is not None: + setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) + + if getv(from_object, ["revision_expire_time"]) is not None: + setv( + parent_object, + ["revisionExpireTime"], + getv(from_object, ["revision_expire_time"]), + ) + + if getv(from_object, ["revision_ttl"]) is not None: + setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) + + if getv(from_object, ["disable_memory_revisions"]) is not None: + setv( + parent_object, + ["disableMemoryRevisions"], + getv(from_object, ["disable_memory_revisions"]), + ) + + if getv(from_object, ["topics"]) is not None: + setv( + parent_object, ["topics"], [item for item in getv(from_object, ["topics"])] + ) + + if getv(from_object, ["metadata"]) is not None: + setv( + parent_object, + ["metadata"], + {k: v for k, v in getv(from_object, ["metadata"]).items()}, + ) + + if getv(from_object, ["memory_id"]) is not None: + setv(parent_object, ["_query", "memoryId"], getv(from_object, ["memory_id"])) + + return to_object + + +def _ModelGardenBatchJobDestination_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["predictionsFormat"]) is not None: + setv(to_object, ["format"], getv(from_object, ["predictionsFormat"])) + + if getv(from_object, ["gcsDestination", "outputUriPrefix"]) is not None: + setv( + to_object, + ["gcs_uri"], + getv(from_object, ["gcsDestination", "outputUriPrefix"]), + ) + + if getv(from_object, ["bigqueryDestination", "outputUri"]) is not None: + setv( + to_object, + ["bigquery_uri"], + getv(from_object, ["bigqueryDestination", "outputUri"]), + ) + + return to_object + + +def _ModelGardenBatchJobDestination_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["format"]) is not None: + setv(to_object, ["predictionsFormat"], getv(from_object, ["format"])) + + if getv(from_object, ["gcs_uri"]) is not None: + setv( + to_object, + ["gcsDestination", "outputUriPrefix"], + getv(from_object, ["gcs_uri"]), + ) + + if getv(from_object, ["bigquery_uri"]) is not None: + setv( + to_object, + ["bigqueryDestination", "outputUri"], + getv(from_object, ["bigquery_uri"]), + ) + + return to_object + + +def _ModelGardenBatchJobSource_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["instancesFormat"]) is not None: + setv(to_object, ["format"], getv(from_object, ["instancesFormat"])) + + if getv(from_object, ["gcsSource", "uris"]) is not None: + setv(to_object, ["gcs_uri"], getv(from_object, ["gcsSource", "uris"])) + + if getv(from_object, ["bigquerySource", "inputUri"]) is not None: + setv( + to_object, + ["bigquery_uri"], + getv(from_object, ["bigquerySource", "inputUri"]), + ) + + return to_object + + +def _ModelGardenBatchJobSource_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["format"]) is not None: + setv(to_object, ["instancesFormat"], getv(from_object, ["format"])) + + if getv(from_object, ["gcs_uri"]) is not None: + setv(to_object, ["gcsSource", "uris"], getv(from_object, ["gcs_uri"])) + + if getv(from_object, ["bigquery_uri"]) is not None: + setv( + to_object, + ["bigquerySource", "inputUri"], + getv(from_object, ["bigquery_uri"]), + ) + + return to_object + + +def _ModelGardenBatchPredictionJob_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["displayName"]) is not None: + setv(to_object, ["display_name"], getv(from_object, ["displayName"])) + + if getv(from_object, ["model"]) is not None: + setv(to_object, ["model"], getv(from_object, ["model"])) + + if getv(from_object, ["inputConfig"]) is not None: + setv( + to_object, + ["input_config"], + _ModelGardenBatchJobSource_from_vertex( + getv(from_object, ["inputConfig"]), to_object, root_object + ), + ) + + if getv(from_object, ["outputConfig"]) is not None: + setv( + to_object, + ["output_config"], + _ModelGardenBatchJobDestination_from_vertex( + getv(from_object, ["outputConfig"]), to_object, root_object + ), + ) + + if getv(from_object, ["outputInfo"]) is not None: + setv(to_object, ["output_info"], getv(from_object, ["outputInfo"])) + + if getv(from_object, ["state"]) is not None: + setv(to_object, ["state"], getv(from_object, ["state"])) + + if getv(from_object, ["error"]) is not None: + setv(to_object, ["error"], getv(from_object, ["error"])) + + if getv(from_object, ["createTime"]) is not None: + setv(to_object, ["create_time"], getv(from_object, ["createTime"])) + + if getv(from_object, ["startTime"]) is not None: + setv(to_object, ["start_time"], getv(from_object, ["startTime"])) + + if getv(from_object, ["endTime"]) is not None: + setv(to_object, ["end_time"], getv(from_object, ["endTime"])) + + if getv(from_object, ["updateTime"]) is not None: + setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) + + return to_object + + +def _OptimizeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["optimization_target"]) is not None: + setv( + parent_object, + ["optimizationTarget"], + getv(from_object, ["optimization_target"]), + ) + + return to_object + + +def _OptimizeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["content"]) is not None: + setv(to_object, ["content"], getv(from_object, ["content"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _OptimizeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _PauseRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _PredictConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["parameters"]) is not None: + setv(parent_object, ["parameters"], getv(from_object, ["parameters"])) + + return to_object + + +def _PredictParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "endpoint"], t.t_endpoint(getv(from_object, ["name"]))) + + if getv(from_object, ["instances"]) is not None: + setv(to_object, ["instances"], getv(from_object, ["instances"])) + + if getv(from_object, ["config"]) is not None: + _PredictConfig_to_vertex(getv(from_object, ["config"]), to_object, root_object) + + return to_object + + +def _PurgeMemoriesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["filter"]) is not None: + setv(to_object, ["filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["filter_groups"]) is not None: + setv( + to_object, + ["filterGroups"], + [item for item in getv(from_object, ["filter_groups"])], + ) + + if getv(from_object, ["force"]) is not None: + setv(to_object, ["force"], getv(from_object, ["force"])) + + return to_object + + +def _QueryRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["class_method"]) is not None: + setv(parent_object, ["classMethod"], getv(from_object, ["class_method"])) + + if getv(from_object, ["input"]) is not None: + setv(parent_object, ["input"], getv(from_object, ["input"])) + + if getv(from_object, ["include_all_fields"]) is not None: + setv(to_object, ["includeAllFields"], getv(from_object, ["include_all_fields"])) + + return to_object + + +def _QueryRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _QueryRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _QueryRuntimeRevisionConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["class_method"]) is not None: + setv(parent_object, ["classMethod"], getv(from_object, ["class_method"])) + + if getv(from_object, ["input"]) is not None: + setv(parent_object, ["input"], getv(from_object, ["input"])) + + if getv(from_object, ["include_all_fields"]) is not None: + setv(to_object, ["includeAllFields"], getv(from_object, ["include_all_fields"])) + + return to_object + + +def _QueryRuntimeRevisionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _QueryRuntimeRevisionConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _RagCorpusCorpusTypeConfigMemoryCorpus_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagCorpusCorpusTypeConfigMemoryCorpus_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagCorpusCorpusTypeConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagCorpusCorpusTypeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagCorpus_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, root_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, root_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, root_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, root_object + ), + ) + + return to_object + + +def _RagCorpus_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_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, root_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, root_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, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfigHybridSearchConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object, + ), + ) + + if getv(from_object, ["sparseEmbeddingConfig"]) is not None: + setv( + to_object, + ["sparse_embedding_config"], + _RagEmbeddingModelConfigSparseEmbeddingConfig_from_vertex( + getv(from_object, ["sparseEmbeddingConfig"]), to_object, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfigHybridSearchConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_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, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfigSparseEmbeddingConfigBm25_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfigSparseEmbeddingConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfigVertexPredictionEndpoint_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + if getv(from_object, ["vertexPredictionEndpoint"]) is not None: + setv( + to_object, + ["vertex_prediction_endpoint"], + _RagEmbeddingModelConfigVertexPredictionEndpoint_from_vertex( + getv(from_object, ["vertexPredictionEndpoint"]), to_object, root_object + ), + ) + + return to_object + + +def _RagEmbeddingModelConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object, + ), + ) + + return to_object + + +def _RagEngineConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagEngineConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _RagFileParsingConfigLlmParser_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + if getv(from_object, ["knn"]) is not None: + setv( + to_object, + ["knn"], + _RagVectorDbConfigRagManagedDbKNN_from_vertex( + getv(from_object, ["knn"]), to_object, root_object + ), + ) + + return to_object + + +def _RagVectorDbConfigRagManagedDb_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + if getv(from_object, ["knn"]) is not None: + setv( + to_object, + ["knn"], + _RagVectorDbConfigRagManagedDbKNN_to_vertex( + getv(from_object, ["knn"]), to_object, root_object + ), + ) + + return to_object + + +def _RagVectorDbConfigRagManagedVertexVectorSearch_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, root_object + ), + ) + + if getv(from_object, ["ragManagedDb"]) is not None: + setv( + to_object, + ["rag_managed_db"], + _RagVectorDbConfigRagManagedDb_from_vertex( + getv(from_object, ["ragManagedDb"]), to_object, root_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, + root_object, + ), + ) + + if getv(from_object, ["vertexFeatureStore"]) is not None: + setv( + to_object, + ["vertex_feature_store"], + _RagVectorDbConfigVertexFeatureStore_from_vertex( + getv(from_object, ["vertexFeatureStore"]), to_object, root_object + ), + ) + + if getv(from_object, ["vertexVectorSearch"]) is not None: + setv( + to_object, + ["vertex_vector_search"], + _RagVectorDbConfigVertexVectorSearch_from_vertex( + getv(from_object, ["vertexVectorSearch"]), to_object, root_object + ), + ) + + if getv(from_object, ["weaviate"]) is not None: + setv( + to_object, + ["weaviate"], + _RagVectorDbConfigWeaviate_from_vertex( + getv(from_object, ["weaviate"]), to_object, root_object + ), + ) + + return to_object + + +def _RagVectorDbConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_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, root_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, + root_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, root_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, root_object + ), + ) + + if getv(from_object, ["weaviate"]) is not None: + setv( + to_object, + ["weaviate"], + _RagVectorDbConfigWeaviate_to_vertex( + getv(from_object, ["weaviate"]), to_object, root_object + ), + ) + + return to_object + + +def _ReasoningEngineContextSpecMemoryBankConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["customizationConfigs"]) is not None: + setv( + to_object, + ["customization_configs"], + [item for item in getv(from_object, ["customizationConfigs"])], + ) + + if getv(from_object, ["disableMemoryRevisions"]) is not None: + setv( + to_object, + ["disable_memory_revisions"], + getv(from_object, ["disableMemoryRevisions"]), + ) + + if getv(from_object, ["generationConfig"]) is not None: + setv(to_object, ["generation_config"], getv(from_object, ["generationConfig"])) + + if getv(from_object, ["similaritySearchConfig"]) is not None: + setv( + to_object, + ["similarity_search_config"], + getv(from_object, ["similaritySearchConfig"]), + ) + + if getv(from_object, ["ttlConfig"]) is not None: + setv(to_object, ["ttl_config"], getv(from_object, ["ttlConfig"])) + + if getv(from_object, ["structuredMemoryConfigs"]) is not None: + setv( + to_object, + ["structured_memory_configs"], + [ + _StructuredMemoryConfig_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["structuredMemoryConfigs"]) + ], + ) + + return to_object + + +def _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["customization_configs"]) is not None: + setv( + to_object, + ["customizationConfigs"], + [item for item in getv(from_object, ["customization_configs"])], + ) + + if getv(from_object, ["disable_memory_revisions"]) is not None: + setv( + to_object, + ["disableMemoryRevisions"], + getv(from_object, ["disable_memory_revisions"]), + ) + + if getv(from_object, ["generation_config"]) is not None: + setv(to_object, ["generationConfig"], getv(from_object, ["generation_config"])) + + if getv(from_object, ["similarity_search_config"]) is not None: + setv( + to_object, + ["similaritySearchConfig"], + getv(from_object, ["similarity_search_config"]), + ) + + if getv(from_object, ["ttl_config"]) is not None: + setv(to_object, ["ttlConfig"], getv(from_object, ["ttl_config"])) + + if getv(from_object, ["structured_memory_configs"]) is not None: + setv( + to_object, + ["structuredMemoryConfigs"], + [ + _StructuredMemoryConfig_to_vertex(item, to_object, root_object) + for item in getv(from_object, ["structured_memory_configs"]) + ], + ) + + return to_object + + +def _ReasoningEngineContextSpec_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["memoryBankConfig"]) is not None: + setv( + to_object, + ["memory_bank_config"], + _ReasoningEngineContextSpecMemoryBankConfig_from_vertex( + getv(from_object, ["memoryBankConfig"]), to_object, root_object + ), + ) + + return to_object + + +def _ReasoningEngineContextSpec_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["memory_bank_config"]) is not None: + setv( + to_object, + ["memoryBankConfig"], + _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( + getv(from_object, ["memory_bank_config"]), to_object, root_object + ), + ) + + return to_object + + +def _ReasoningEngine_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["encryptionSpec"]) is not None: + setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) + + if getv(from_object, ["contextSpec"]) is not None: + setv( + to_object, + ["context_spec"], + _ReasoningEngineContextSpec_from_vertex( + getv(from_object, ["contextSpec"]), to_object, root_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, ["etag"]) is not None: + setv(to_object, ["etag"], getv(from_object, ["etag"])) + + if getv(from_object, ["labels"]) is not None: + setv(to_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["name"]) is not None: + setv(to_object, ["name"], getv(from_object, ["name"])) + + if getv(from_object, ["spec"]) is not None: + setv(to_object, ["spec"], getv(from_object, ["spec"])) + + if getv(from_object, ["updateTime"]) is not None: + setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) + + if getv(from_object, ["trafficConfig"]) is not None: + setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"])) + + if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None: + setv( + to_object, + ["revision_garbage_collection_strategy"], + getv(from_object, ["revisionGarbageCollectionStrategy"]), + ) + + return to_object + + +def _RecommendSpecConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["check_machine_availability"]) is not None: + setv( + parent_object, + ["checkMachineAvailability"], + getv(from_object, ["check_machine_availability"]), + ) + + if getv(from_object, ["check_user_quota"]) is not None: + setv(parent_object, ["checkUserQuota"], getv(from_object, ["check_user_quota"])) + + return to_object + + +def _RecommendSpecRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["parent"]) is not None: + setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) + + if getv(from_object, ["gcs_uri"]) is not None: + setv(to_object, ["gcsUri"], getv(from_object, ["gcs_uri"])) + + if getv(from_object, ["config"]) is not None: + _RecommendSpecConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _RemoveExamplesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["example_ids"]) is not None: + setv(parent_object, ["exampleIds"], getv(from_object, ["example_ids"])) + + if getv(from_object, ["stored_contents_example_filter"]) is not None: + setv( + parent_object, + ["storedContentsExampleFilter"], + getv(from_object, ["stored_contents_example_filter"]), + ) + + return to_object + + +def _RemoveExamplesParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "exampleStore"], + t.t_example_store(getv(from_object, ["name"])), + ) + + if getv(from_object, ["config"]) is not None: + _RemoveExamplesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _RestoreVersionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["version_id"]) is not None: + setv(to_object, ["_url", "version_id"], getv(from_object, ["version_id"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _ResumeRuntimeSandboxRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + return to_object + + +def _RetrieveMemoriesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["filter"]) is not None: + setv(parent_object, ["filter"], getv(from_object, ["filter"])) + + if getv(from_object, ["filter_groups"]) is not None: + setv( + parent_object, + ["filterGroups"], + [item for item in getv(from_object, ["filter_groups"])], + ) + + if getv(from_object, ["memory_types"]) is not None: + setv(parent_object, ["memoryTypes"], getv(from_object, ["memory_types"])) + + return to_object + + +def _RetrieveMemoriesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["similarity_search_params"]) is not None: + setv( + to_object, + ["similaritySearchParams"], + getv(from_object, ["similarity_search_params"]), + ) + + if getv(from_object, ["simple_retrieval_params"]) is not None: + setv( + to_object, + ["simpleRetrievalParams"], + getv(from_object, ["simple_retrieval_params"]), + ) + + if getv(from_object, ["config"]) is not None: + _RetrieveMemoriesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _RetrieveMemoryProfilesRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + return to_object + + +def _RetrieveRagContextsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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 _RetrieveSkillsConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["top_k"]) is not None: + setv(parent_object, ["_query", "topK"], getv(from_object, ["top_k"])) + + return to_object + + +def _RetrieveSkillsRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["query"]) is not None: + setv(to_object, ["_query", "query"], getv(from_object, ["query"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _RetrieveSkillsConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _RollbackMemoryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["target_revision_id"]) is not None: + setv(to_object, ["targetRevisionId"], getv(from_object, ["target_revision_id"])) + + return to_object + + +def _RubricBasedMetricInput_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric_spec"]) is not None: + setv( + to_object, + ["metricSpec"], + _RubricBasedMetricSpec_to_vertex( + getv(from_object, ["metric_spec"]), to_object, root_object + ), + ) + + if getv(from_object, ["instance"]) is not None: + setv(to_object, ["instance"], getv(from_object, ["instance"])) + + return to_object + + +def _RubricBasedMetricSpec_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["metric_prompt_template"]) is not None: + setv( + to_object, + ["metricPromptTemplate"], + getv(from_object, ["metric_prompt_template"]), + ) + + if getv(from_object, ["judge_autorater_config"]) is not None: + setv( + to_object, + ["judgeAutoraterConfig"], + getv(from_object, ["judge_autorater_config"]), + ) + + if getv(from_object, ["inline_rubrics"]) is not None: + setv( + to_object, + ["inline_rubrics", "rubrics"], + getv(from_object, ["inline_rubrics"]), + ) + + if getv(from_object, ["rubric_group_key"]) is not None: + setv(to_object, ["rubricGroupKey"], getv(from_object, ["rubric_group_key"])) + + if getv(from_object, ["rubric_generation_spec"]) is not None: + setv( + to_object, + ["rubricGenerationSpec"], + getv(from_object, ["rubric_generation_spec"]), + ) + + return to_object + + +def _RunQueryJobRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["input_gcs_uri"]) is not None: + setv(parent_object, ["inputGcsUri"], getv(from_object, ["input_gcs_uri"])) + + if getv(from_object, ["output_gcs_uri"]) is not None: + setv(parent_object, ["outputGcsUri"], getv(from_object, ["output_gcs_uri"])) + + return to_object + + +def _RunQueryJobRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + setv( + to_object, + ["config"], + _RunQueryJobRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ), + ) + + return to_object + + +def _RuntimeOperation_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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"], + _ReasoningEngine_from_vertex( + getv(from_object, ["response"]), to_object, root_object + ), + ) + + return to_object + + +def _SearchExamplesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["top_k"]) is not None: + setv(parent_object, ["topK"], getv(from_object, ["top_k"])) + + return to_object + + +def _SearchExamplesParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "exampleStore"], + t.t_example_store(getv(from_object, ["name"])), + ) + + if getv(from_object, ["stored_contents_example_parameters"]) is not None: + setv( + to_object, + ["storedContentsExampleParameters"], + getv(from_object, ["stored_contents_example_parameters"]), + ) + + if getv(from_object, ["config"]) is not None: + _SearchExamplesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _StructuredMemoryConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["schemaConfigs"]) is not None: + setv( + to_object, + ["schema_configs"], + [ + _StructuredMemorySchemaConfig_from_vertex(item, to_object, root_object) + for item in getv(from_object, ["schemaConfigs"]) + ], + ) + + if getv(from_object, ["scopeKeys"]) is not None: + setv(to_object, ["scope_keys"], getv(from_object, ["scopeKeys"])) + + return to_object + + +def _StructuredMemoryConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["schema_configs"]) is not None: + setv( + to_object, + ["schemaConfigs"], + [ + _StructuredMemorySchemaConfig_to_vertex(item, to_object, root_object) + for item in getv(from_object, ["schema_configs"]) + ], + ) + + if getv(from_object, ["scope_keys"]) is not None: + setv(to_object, ["scopeKeys"], getv(from_object, ["scope_keys"])) + + return to_object + + +def _StructuredMemorySchemaConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["schema"]) is not None: + setv(to_object, ["memory_schema"], getv(from_object, ["schema"])) + + if getv(from_object, ["id"]) is not None: + setv(to_object, ["id"], getv(from_object, ["id"])) + + if getv(from_object, ["memoryType"]) is not None: + setv(to_object, ["memory_type"], getv(from_object, ["memoryType"])) + + return to_object + + +def _StructuredMemorySchemaConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["memory_schema"]) is not None: + setv(to_object, ["schema"], getv(from_object, ["memory_schema"])) + + if getv(from_object, ["id"]) is not None: + setv(to_object, ["id"], getv(from_object, ["id"])) + + if getv(from_object, ["memory_type"]) is not None: + setv(to_object, ["memoryType"], getv(from_object, ["memory_type"])) + + return to_object + + +def _UndeployModelRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "endpoint"], + t.t_strict_endpoint(getv(from_object, ["name"])), + ) + + if getv(from_object, ["deployed_model_id"]) is not None: + setv(to_object, ["deployed_model_id"], getv(from_object, ["deployed_model_id"])) + + return to_object + + +def _UnifiedMetric_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["bleuSpec"]) is not None: + setv(to_object, ["bleu_spec"], getv(from_object, ["bleuSpec"])) + + if getv(from_object, ["rougeSpec"]) is not None: + setv(to_object, ["rouge_spec"], getv(from_object, ["rougeSpec"])) + + if getv(from_object, ["pointwiseMetricSpec"]) is not None: + setv( + to_object, + ["pointwise_metric_spec"], + getv(from_object, ["pointwiseMetricSpec"]), + ) + + if getv(from_object, ["llmBasedMetricSpec"]) is not None: + setv( + to_object, + ["llm_based_metric_spec"], + getv(from_object, ["llmBasedMetricSpec"]), + ) + + if getv(from_object, ["customCodeExecutionSpec"]) is not None: + setv( + to_object, + ["custom_code_execution_spec"], + _CustomCodeExecutionSpec_from_vertex( + getv(from_object, ["customCodeExecutionSpec"]), to_object, root_object + ), + ) + + if getv(from_object, ["predefinedMetricSpec"]) is not None: + setv( + to_object, + ["predefined_metric_spec"], + getv(from_object, ["predefinedMetricSpec"]), + ) + + if getv(from_object, ["computationBasedMetricSpec"]) is not None: + setv( + to_object, + ["computation_based_metric_spec"], + getv(from_object, ["computationBasedMetricSpec"]), + ) + + return to_object + + +def _UnifiedMetric_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["bleu_spec"]) is not None: + setv(to_object, ["bleuSpec"], getv(from_object, ["bleu_spec"])) + + if getv(from_object, ["rouge_spec"]) is not None: + setv(to_object, ["rougeSpec"], getv(from_object, ["rouge_spec"])) + + if getv(from_object, ["pointwise_metric_spec"]) is not None: + setv( + to_object, + ["pointwiseMetricSpec"], + getv(from_object, ["pointwise_metric_spec"]), + ) + + if getv(from_object, ["llm_based_metric_spec"]) is not None: + setv( + to_object, + ["llmBasedMetricSpec"], + getv(from_object, ["llm_based_metric_spec"]), + ) + + if getv(from_object, ["custom_code_execution_spec"]) is not None: + setv( + to_object, + ["customCodeExecutionSpec"], + _CustomCodeExecutionSpec_to_vertex( + getv(from_object, ["custom_code_execution_spec"]), + to_object, + root_object, + ), + ) + + if getv(from_object, ["predefined_metric_spec"]) is not None: + setv( + to_object, + ["predefinedMetricSpec"], + getv(from_object, ["predefined_metric_spec"]), + ) + + if getv(from_object, ["computation_based_metric_spec"]) is not None: + setv( + to_object, + ["computationBasedMetricSpec"], + getv(from_object, ["computation_based_metric_spec"]), + ) + + return to_object + + +def _UpdateDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["dataset_id"]) is not None: + setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) + + if getv(from_object, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["model_reference"]) is not None: + setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _UpdateEvaluationExperimentConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["labels"]) is not None: + setv(parent_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["merge_strategy"]) is not None: + setv(parent_object, ["mergeStrategy"], getv(from_object, ["merge_strategy"])) + + if getv(from_object, ["metadata"]) is not None: + setv(parent_object, ["metadata"], getv(from_object, ["metadata"])) + + return to_object + + +def _UpdateEvaluationExperimentParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateEvaluationExperimentConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateMemoryBankConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + return to_object + + +def _UpdateMemoryBankRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["memory_bank_config"]) is not None: + setv( + to_object, + ["context_spec", "memory_bank_config"], + _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( + getv(from_object, ["memory_bank_config"]), to_object, root_object + ), + ) + + if getv(from_object, ["config"]) is not None: + _UpdateMemoryBankConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateMemoryConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + if getv(from_object, ["expire_time"]) is not None: + setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) + + if getv(from_object, ["revision_expire_time"]) is not None: + setv( + parent_object, + ["revisionExpireTime"], + getv(from_object, ["revision_expire_time"]), + ) + + if getv(from_object, ["revision_ttl"]) is not None: + setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) + + if getv(from_object, ["disable_memory_revisions"]) is not None: + setv( + parent_object, + ["disableMemoryRevisions"], + getv(from_object, ["disable_memory_revisions"]), + ) + + if getv(from_object, ["topics"]) is not None: + setv( + parent_object, ["topics"], [item for item in getv(from_object, ["topics"])] + ) + + if getv(from_object, ["metadata"]) is not None: + setv( + parent_object, + ["metadata"], + {k: v for k, v in getv(from_object, ["metadata"]).items()}, + ) + + if getv(from_object, ["memory_id"]) is not None: + setv(parent_object, ["_query", "memoryId"], getv(from_object, ["memory_id"])) + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + return to_object + + +def _UpdateMemoryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["fact"]) is not None: + setv(to_object, ["fact"], getv(from_object, ["fact"])) + + if getv(from_object, ["scope"]) is not None: + setv(to_object, ["scope"], getv(from_object, ["scope"])) + + if getv(from_object, ["config"]) is not None: + _UpdateMemoryConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateMultimodalDatasetParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["display_name"]) is not None: + setv(to_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["metadata"]) is not None: + setv(to_object, ["metadata"], getv(from_object, ["metadata"])) + + if getv(from_object, ["description"]) is not None: + setv(to_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + 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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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_corpus"]) is not None: + setv( + to_object, + ["_self"], + _RagCorpus_to_vertex( + getv(from_object, ["rag_corpus"]), to_object, root_object + ), + ) + + if getv(from_object, ["config"]) is not None: + setv(to_object, ["config"], getv(from_object, ["config"])) + + return to_object + + +def _UpdateRuntimeConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["spec"]) is not None: + setv(parent_object, ["spec"], getv(from_object, ["spec"])) + + if getv(from_object, ["context_spec"]) is not None: + setv( + parent_object, + ["contextSpec"], + _ReasoningEngineContextSpec_to_vertex( + getv(from_object, ["context_spec"]), to_object, root_object + ), + ) + + if getv(from_object, ["psc_interface_config"]) is not None: + setv( + parent_object, + ["pscInterfaceConfig"], + getv(from_object, ["psc_interface_config"]), + ) + + if getv(from_object, ["agent_gateway_config"]) is not None: + setv( + parent_object, + ["agentGatewayConfig"], + getv(from_object, ["agent_gateway_config"]), + ) + + if getv(from_object, ["encryption_spec"]) is not None: + setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) + + if getv(from_object, ["labels"]) is not None: + setv(parent_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["source_packages"]) is not None: + setv(parent_object, ["sourcePackages"], getv(from_object, ["source_packages"])) + + if getv(from_object, ["entrypoint_module"]) is not None: + setv( + parent_object, + ["entrypointModule"], + getv(from_object, ["entrypoint_module"]), + ) + + if getv(from_object, ["entrypoint_object"]) is not None: + setv( + parent_object, + ["entrypointObject"], + getv(from_object, ["entrypoint_object"]), + ) + + if getv(from_object, ["requirements_file"]) is not None: + setv( + parent_object, + ["requirementsFile"], + getv(from_object, ["requirements_file"]), + ) + + if getv(from_object, ["agent_framework"]) is not None: + setv(parent_object, ["agentFramework"], getv(from_object, ["agent_framework"])) + + if getv(from_object, ["python_version"]) is not None: + setv(parent_object, ["pythonVersion"], getv(from_object, ["python_version"])) + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + if getv(from_object, ["traffic_config"]) is not None: + setv(parent_object, ["trafficConfig"], getv(from_object, ["traffic_config"])) + + return to_object + + +def _UpdateRuntimeFeedbackContextConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + return to_object + + +def _UpdateRuntimeFeedbackContextRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, ["context_events"]) is not None: + setv( + to_object, + ["contextEvents"], + [item for item in getv(from_object, ["context_events"])], + ) + + if getv(from_object, ["config"]) is not None: + _UpdateRuntimeFeedbackContextConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateRuntimeFeedbackEntryConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + if getv(from_object, ["feedback_type"]) is not None: + setv(parent_object, ["feedbackType"], getv(from_object, ["feedback_type"])) + + if getv(from_object, ["session_id"]) is not None: + setv(parent_object, ["sessionId"], getv(from_object, ["session_id"])) + + if getv(from_object, ["event_id"]) is not None: + setv(parent_object, ["eventId"], getv(from_object, ["event_id"])) + + if getv(from_object, ["feedback_labels"]) is not None: + setv(parent_object, ["feedbackLabels"], getv(from_object, ["feedback_labels"])) + + if getv(from_object, ["feedback_text"]) is not None: + setv(parent_object, ["feedbackText"], getv(from_object, ["feedback_text"])) + + if getv(from_object, ["user_id"]) is not None: + setv(parent_object, ["userId"], getv(from_object, ["user_id"])) + + if getv(from_object, ["source"]) is not None: + setv(parent_object, ["source"], getv(from_object, ["source"])) + + if getv(from_object, ["custom_metadata"]) is not None: + setv(parent_object, ["customMetadata"], getv(from_object, ["custom_metadata"])) + + return to_object + + +def _UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateRuntimeFeedbackEntryConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateRuntimeRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateRuntimeConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateRuntimeSessionConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["session_state"]) is not None: + setv(parent_object, ["sessionState"], getv(from_object, ["session_state"])) + + if getv(from_object, ["ttl"]) is not None: + setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) + + if getv(from_object, ["expire_time"]) is not None: + setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) + + if getv(from_object, ["labels"]) is not None: + setv(parent_object, ["labels"], getv(from_object, ["labels"])) + + if getv(from_object, ["session_id"]) is not None: + setv(parent_object, ["_query", "sessionId"], getv(from_object, ["session_id"])) + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + if getv(from_object, ["user_id"]) is not None: + setv(parent_object, ["userId"], getv(from_object, ["user_id"])) + + return to_object + + +def _UpdateRuntimeSessionRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateRuntimeSessionConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateServingProfileConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + return to_object + + +def _UpdateServingProfileRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateServingProfileConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UpdateSkillConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["display_name"]) is not None: + setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) + + if getv(from_object, ["description"]) is not None: + setv(parent_object, ["description"], getv(from_object, ["description"])) + + if getv(from_object, ["zipped_filesystem"]) is not None: + setv( + parent_object, + ["zippedFilesystem"], + getv(from_object, ["zipped_filesystem"]), + ) + + if getv(from_object, ["update_mask"]) is not None: + setv( + parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) + ) + + return to_object + + +def _UpdateSkillRequestParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv(to_object, ["_url", "name"], getv(from_object, ["name"])) + + if getv(from_object, ["config"]) is not None: + _UpdateSkillConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _UploadRagFileConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_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, + root_object: Optional[Union[dict[str, Any], object]] = 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, root_object + ), + ) + + return to_object + + +def _UpsertExamplesConfig_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + + if getv(from_object, ["overwrite"]) is not None: + setv(parent_object, ["overwrite"], getv(from_object, ["overwrite"])) + + return to_object + + +def _UpsertExamplesParameters_to_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ["name"]) is not None: + setv( + to_object, + ["_url", "exampleStore"], + t.t_example_store(getv(from_object, ["name"])), + ) + + if getv(from_object, ["examples"]) is not None: + setv( + to_object, ["examples"], [item for item in getv(from_object, ["examples"])] + ) + + if getv(from_object, ["config"]) is not None: + _UpsertExamplesConfig_to_vertex( + getv(from_object, ["config"]), to_object, root_object + ) + + return to_object + + +def _VertexAiSearchConfig_from_vertex( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, + root_object: Optional[Union[dict[str, Any], object]] = 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, + root_object: Optional[Union[dict[str, Any], object]] = 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 diff --git a/agentplatform/_genai/datasets.py b/agentplatform/_genai/datasets.py index 8865685eef..d2faa93b20 100644 --- a/agentplatform/_genai/datasets.py +++ b/agentplatform/_genai/datasets.py @@ -19,232 +19,21 @@ import json import logging import time -from typing import Any, Optional, Union +from typing import Any, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common 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 import pandas as pd +from . import _converters from . import _datasets_utils from . import types logger = logging.getLogger("agentplatform_genai.datasets") -def _AssembleDatasetParameters_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, ["gemini_request_read_config"]) is not None: - setv( - to_object, - ["geminiRequestReadConfig"], - getv(from_object, ["gemini_request_read_config"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _AssessDatasetParameters_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, ["gemini_request_read_config"]) is not None: - setv( - to_object, - ["geminiRequestReadConfig"], - getv(from_object, ["gemini_request_read_config"]), - ) - - if getv(from_object, ["tuning_resource_usage_assessment_config"]) is not None: - setv( - to_object, - ["tuningResourceUsageAssessmentConfig"], - getv(from_object, ["tuning_resource_usage_assessment_config"]), - ) - - if getv(from_object, ["tuning_validation_assessment_config"]) is not None: - setv( - to_object, - ["tuningValidationAssessmentConfig"], - getv(from_object, ["tuning_validation_assessment_config"]), - ) - - if ( - getv(from_object, ["batch_prediction_resource_usage_assessment_config"]) - is not None - ): - setv( - to_object, - ["batchPredictionResourceUsageAssessmentConfig"], - getv(from_object, ["batch_prediction_resource_usage_assessment_config"]), - ) - - if getv(from_object, ["batch_prediction_validation_assessment_config"]) is not None: - setv( - to_object, - ["batchPredictionValidationAssessmentConfig"], - getv(from_object, ["batch_prediction_validation_assessment_config"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateMultimodalDatasetParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["metadata_schema_uri"]) is not None: - setv( - to_object, ["metadataSchemaUri"], getv(from_object, ["metadata_schema_uri"]) - ) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _DeleteMultimodalDatasetRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetMultimodalDatasetOperationParameters_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, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["operation_id"]) is not None: - setv(to_object, ["_url", "operation_id"], getv(from_object, ["operation_id"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetMultimodalDatasetParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListMultimodalDatasetsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListMultimodalDatasetsRequestParameters_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"], - _ListMultimodalDatasetsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _UpdateMultimodalDatasetParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - class Datasets(_api_module.BaseModule): def _assemble_multimodal_dataset( @@ -272,7 +61,9 @@ def _assemble_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AssembleDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._AssembleDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:assemble".format_map(request_url_dict) @@ -364,7 +155,9 @@ def _assess_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AssessDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._AssessDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:assess".format_map(request_url_dict) @@ -446,7 +239,9 @@ def _create_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -515,8 +310,10 @@ def _delete_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMultimodalDatasetRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteMultimodalDatasetRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -589,7 +386,9 @@ def _get_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._GetMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -663,8 +462,10 @@ def _get_multimodal_dataset_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMultimodalDatasetOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetMultimodalDatasetOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -735,8 +536,10 @@ def _list_multimodal_datasets( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMultimodalDatasetsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListMultimodalDatasetsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -817,7 +620,9 @@ def _update_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1664,7 +1469,9 @@ async def _assemble_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AssembleDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._AssembleDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:assemble".format_map(request_url_dict) @@ -1758,7 +1565,9 @@ async def _assess_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AssessDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._AssessDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:assess".format_map(request_url_dict) @@ -1842,7 +1651,9 @@ async def _create_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -1913,8 +1724,10 @@ async def _delete_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMultimodalDatasetRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteMultimodalDatasetRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1989,7 +1802,9 @@ async def _get_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._GetMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2065,8 +1880,10 @@ async def _get_multimodal_dataset_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMultimodalDatasetOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetMultimodalDatasetOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2139,8 +1956,10 @@ async def _list_multimodal_datasets( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMultimodalDatasetsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListMultimodalDatasetsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2223,7 +2042,9 @@ async def _update_multimodal_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMultimodalDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) diff --git a/agentplatform/_genai/endpoints.py b/agentplatform/_genai/endpoints.py index a82216219e..8886a3c4ad 100644 --- a/agentplatform/_genai/endpoints.py +++ b/agentplatform/_genai/endpoints.py @@ -17,15 +17,14 @@ import json import logging -from typing import Any, Optional, Union +from typing import Any, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common 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 _converters from . import _operations_utils from . import _transformers as t from . import types @@ -33,96 +32,6 @@ logger = logging.getLogger("agentplatform_genai.endpoints") -def _DeleteEndpointRequestParameters_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"], - t.t_strict_endpoint(getv(from_object, ["name"])), - ) - - return to_object - - -def _GetEndpointOperationParameters_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 _GetEndpointParameters_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"], - t.t_strict_endpoint(getv(from_object, ["name"])), - ) - - return to_object - - -def _PredictConfig_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, ["parameters"]) is not None: - setv(parent_object, ["parameters"], getv(from_object, ["parameters"])) - - return to_object - - -def _PredictParameters_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", "endpoint"], t.t_endpoint(getv(from_object, ["name"]))) - - if getv(from_object, ["instances"]) is not None: - setv(to_object, ["instances"], getv(from_object, ["instances"])) - - if getv(from_object, ["config"]) is not None: - _PredictConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _UndeployModelRequestParameters_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", "endpoint"], - t.t_strict_endpoint(getv(from_object, ["name"])), - ) - - if getv(from_object, ["deployed_model_id"]) is not None: - setv(to_object, ["deployed_model_id"], getv(from_object, ["deployed_model_id"])) - - return to_object - - class Endpoints(_api_module.BaseModule): """Class for managing Endpoints for prediction, undeployment, and deletion.""" @@ -150,7 +59,9 @@ def _undeploy( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UndeployModelRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UndeployModelRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{endpoint}:undeployModel".format_map(request_url_dict) @@ -224,7 +135,7 @@ def _predict( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PredictParameters_to_vertex(parameter_model) + request_dict = _converters._PredictParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{endpoint}:predict".format_map(request_url_dict) @@ -293,7 +204,9 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEndpointRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEndpointRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -362,7 +275,7 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEndpointParameters_to_vertex(parameter_model) + request_dict = _converters._GetEndpointParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -430,7 +343,9 @@ def _get_endpoint_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEndpointOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetEndpointOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -658,7 +573,9 @@ async def _undeploy( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UndeployModelRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UndeployModelRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{endpoint}:undeployModel".format_map(request_url_dict) @@ -734,7 +651,7 @@ async def _predict( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PredictParameters_to_vertex(parameter_model) + request_dict = _converters._PredictParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{endpoint}:predict".format_map(request_url_dict) @@ -805,7 +722,9 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEndpointRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEndpointRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -876,7 +795,7 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEndpointParameters_to_vertex(parameter_model) + request_dict = _converters._GetEndpointParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -946,7 +865,9 @@ async def _get_endpoint_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEndpointOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetEndpointOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) diff --git a/agentplatform/_genai/evals.py b/agentplatform/_genai/evals.py index 67405c5bbc..19cf6ae806 100644 --- a/agentplatform/_genai/evals.py +++ b/agentplatform/_genai/evals.py @@ -24,13 +24,11 @@ from google.genai import _api_module from google.genai import _common 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 import pandas as pd +from . import _converters from . import _evals_common from . import _evals_utils -from . import _transformers as t from . import types from .types import evals as evals_types @@ -43,1231 +41,6 @@ logger = logging.getLogger("agentplatform_genai.evals") -def _CreateEvaluationExperimentParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["merge_strategy"]) is not None: - setv(to_object, ["mergeStrategy"], getv(from_object, ["merge_strategy"])) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateEvaluationItemParameters_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, ["evaluation_item_type"]) is not None: - setv( - to_object, - ["evaluationItemType"], - getv(from_object, ["evaluation_item_type"]), - ) - - if getv(from_object, ["gcs_uri"]) is not None: - setv(to_object, ["gcsUri"], getv(from_object, ["gcs_uri"])) - - if getv(from_object, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateEvaluationMetricParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["metric"]) is not None: - setv( - to_object, - ["metric"], - t.t_metric_for_registry(getv(from_object, ["metric"])), - ) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateEvaluationRunParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["data_source"]) is not None: - setv(to_object, ["dataSource"], getv(from_object, ["data_source"])) - - if getv(from_object, ["evaluation_config"]) is not None: - setv( - to_object, - ["evaluationConfig"], - _EvaluationRunConfig_to_vertex( - getv(from_object, ["evaluation_config"]), to_object - ), - ) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["inference_configs"]) is not None: - setv( - to_object, - ["inferenceConfigs"], - { - k: _EvaluationRunInferenceConfig_to_vertex(v, to_object) - for k, v in getv(from_object, ["inference_configs"]).items() - }, - ) - - if getv(from_object, ["analysis_configs"]) is not None: - setv( - to_object, - ["analysisConfigs"], - [item for item in getv(from_object, ["analysis_configs"])], - ) - - if getv(from_object, ["evaluation_experiment"]) is not None: - setv( - to_object, - ["evaluationExperiment"], - getv(from_object, ["evaluation_experiment"]), - ) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateEvaluationSetParameters_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, ["evaluation_items"]) is not None: - setv(to_object, ["evaluationItems"], getv(from_object, ["evaluation_items"])) - - 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, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CustomCodeExecutionSpec_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, ["evaluationFunction"]) is not None: - setv( - to_object, - ["evaluation_function"], - getv(from_object, ["evaluationFunction"]), - ) - - if getv(from_object, ["evaluation_function"]) is not None: - setv( - to_object, - ["remote_custom_function"], - getv(from_object, ["evaluation_function"]), - ) - - if getv(from_object, ["codeExecutionRegion"]) is not None: - setv( - to_object, - ["code_execution_region"], - getv(from_object, ["codeExecutionRegion"]), - ) - - return to_object - - -def _CustomCodeExecutionSpec_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, ["evaluation_function"]) is not None: - setv( - to_object, - ["evaluationFunction"], - getv(from_object, ["evaluation_function"]), - ) - - if getv(from_object, ["remote_custom_function"]) is not None: - setv( - to_object, - ["evaluation_function"], - getv(from_object, ["remote_custom_function"]), - ) - - if getv(from_object, ["code_execution_region"]) is not None: - setv( - to_object, - ["codeExecutionRegion"], - getv(from_object, ["code_execution_region"]), - ) - - return to_object - - -def _DeleteEvaluationExperimentParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _DeleteEvaluationMetricParameters_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, ["metric_resource_name"]) is not None: - setv( - to_object, - ["_url", "evaluation_metric"], - getv(from_object, ["metric_resource_name"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _DeleteEvaluationSetParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _EvaluateInstancesRequestParameters_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, ["bleu_input"]) is not None: - setv(to_object, ["bleuInput"], getv(from_object, ["bleu_input"])) - - if getv(from_object, ["exact_match_input"]) is not None: - setv(to_object, ["exactMatchInput"], getv(from_object, ["exact_match_input"])) - - if getv(from_object, ["rouge_input"]) is not None: - setv(to_object, ["rougeInput"], getv(from_object, ["rouge_input"])) - - if getv(from_object, ["pointwise_metric_input"]) is not None: - setv( - to_object, - ["pointwiseMetricInput"], - getv(from_object, ["pointwise_metric_input"]), - ) - - if getv(from_object, ["pairwise_metric_input"]) is not None: - setv( - to_object, - ["pairwiseMetricInput"], - getv(from_object, ["pairwise_metric_input"]), - ) - - if getv(from_object, ["tool_call_valid_input"]) is not None: - setv( - to_object, - ["toolCallValidInput"], - getv(from_object, ["tool_call_valid_input"]), - ) - - if getv(from_object, ["tool_name_match_input"]) is not None: - setv( - to_object, - ["toolNameMatchInput"], - getv(from_object, ["tool_name_match_input"]), - ) - - if getv(from_object, ["tool_parameter_key_match_input"]) is not None: - setv( - to_object, - ["toolParameterKeyMatchInput"], - getv(from_object, ["tool_parameter_key_match_input"]), - ) - - if getv(from_object, ["tool_parameter_kv_match_input"]) is not None: - setv( - to_object, - ["toolParameterKvMatchInput"], - getv(from_object, ["tool_parameter_kv_match_input"]), - ) - - if getv(from_object, ["rubric_based_metric_input"]) is not None: - setv( - to_object, - ["rubricBasedMetricInput"], - _RubricBasedMetricInput_to_vertex( - getv(from_object, ["rubric_based_metric_input"]), to_object - ), - ) - - if getv(from_object, ["autorater_config"]) is not None: - setv(to_object, ["autoraterConfig"], getv(from_object, ["autorater_config"])) - - if getv(from_object, ["metrics"]) is not None: - setv( - to_object, - ["metrics"], - [item for item in t.t_metrics(getv(from_object, ["metrics"]))], - ) - - if getv(from_object, ["instance"]) is not None: - setv( - to_object, - ["instance"], - _EvaluationInstance_to_vertex(getv(from_object, ["instance"]), to_object), - ) - - if getv(from_object, ["metric_sources"]) is not None: - setv( - to_object, - ["metricSources"], - [ - item - for item in t.t_metric_sources(getv(from_object, ["metric_sources"])) - ], - ) - - if getv(from_object, ["allow_cross_region_model"]) is not None: - setv( - to_object, - ["allowCrossRegionModel"], - getv(from_object, ["allow_cross_region_model"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _EvaluationInstance_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, ["prompt"]) is not None: - setv(to_object, ["prompt"], getv(from_object, ["prompt"])) - - if getv(from_object, ["response"]) is not None: - setv(to_object, ["response"], getv(from_object, ["response"])) - - if getv(from_object, ["reference"]) is not None: - setv(to_object, ["reference"], getv(from_object, ["reference"])) - - if getv(from_object, ["other_data"]) is not None: - setv(to_object, ["otherData"], getv(from_object, ["other_data"])) - - if getv(from_object, ["agent_data"]) is not None: - setv(to_object, ["agent_eval_data"], getv(from_object, ["agent_data"])) - - if getv(from_object, ["rubric_groups"]) is not None: - setv(to_object, ["rubricGroups"], getv(from_object, ["rubric_groups"])) - - if getv(from_object, ["interactions_data_source"]) is not None: - setv( - to_object, - ["interactionsDataSource"], - getv(from_object, ["interactions_data_source"]), - ) - - return to_object - - -def _EvaluationMetric_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, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(from_object, ["displayName"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["metric"]) is not None: - setv( - to_object, - ["metric"], - _UnifiedMetric_from_vertex(getv(from_object, ["metric"]), to_object), - ) - - if getv(from_object, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) - - return to_object - - -def _EvaluationRunConfig_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, ["metrics"]) is not None: - setv( - to_object, - ["metrics"], - [ - _EvaluationRunMetric_from_vertex(item, to_object) - for item in getv(from_object, ["metrics"]) - ], - ) - - if getv(from_object, ["outputConfig"]) is not None: - setv(to_object, ["output_config"], getv(from_object, ["outputConfig"])) - - if getv(from_object, ["autoraterConfig"]) is not None: - setv(to_object, ["autorater_config"], getv(from_object, ["autoraterConfig"])) - - if getv(from_object, ["promptTemplate"]) is not None: - setv(to_object, ["prompt_template"], getv(from_object, ["promptTemplate"])) - - if getv(from_object, ["lossAnalysisConfig"]) is not None: - setv( - to_object, - ["loss_analysis_config"], - [item for item in getv(from_object, ["lossAnalysisConfig"])], - ) - - if getv(from_object, ["allowCrossRegionModel"]) is not None: - setv( - to_object, - ["allow_cross_region_model"], - getv(from_object, ["allowCrossRegionModel"]), - ) - - return to_object - - -def _EvaluationRunConfig_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, ["metrics"]) is not None: - setv( - to_object, - ["metrics"], - [ - _EvaluationRunMetric_to_vertex(item, to_object) - for item in getv(from_object, ["metrics"]) - ], - ) - - if getv(from_object, ["output_config"]) is not None: - setv(to_object, ["outputConfig"], getv(from_object, ["output_config"])) - - if getv(from_object, ["autorater_config"]) is not None: - setv(to_object, ["autoraterConfig"], getv(from_object, ["autorater_config"])) - - if getv(from_object, ["prompt_template"]) is not None: - setv(to_object, ["promptTemplate"], getv(from_object, ["prompt_template"])) - - if getv(from_object, ["loss_analysis_config"]) is not None: - setv( - to_object, - ["lossAnalysisConfig"], - [item for item in getv(from_object, ["loss_analysis_config"])], - ) - - if getv(from_object, ["allow_cross_region_model"]) is not None: - setv( - to_object, - ["allowCrossRegionModel"], - getv(from_object, ["allow_cross_region_model"]), - ) - - return to_object - - -def _EvaluationRunInferenceConfig_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, ["agentConfig"]) is not None: - setv(to_object, ["agent_config"], getv(from_object, ["agentConfig"])) - - if getv(from_object, ["model"]) is not None: - setv(to_object, ["model"], getv(from_object, ["model"])) - - if getv(from_object, ["promptTemplate"]) is not None: - setv(to_object, ["prompt_template"], getv(from_object, ["promptTemplate"])) - - if getv(from_object, ["agentRunConfig"]) is not None: - setv(to_object, ["agent_run_config"], getv(from_object, ["agentRunConfig"])) - - if getv(from_object, ["agents"]) is not None: - setv(to_object, ["agent_configs"], getv(from_object, ["agents"])) - - return to_object - - -def _EvaluationRunInferenceConfig_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, ["agent_config"]) is not None: - setv(to_object, ["agentConfig"], getv(from_object, ["agent_config"])) - - if getv(from_object, ["model"]) is not None: - setv(to_object, ["model"], getv(from_object, ["model"])) - - if getv(from_object, ["prompt_template"]) is not None: - setv(to_object, ["promptTemplate"], getv(from_object, ["prompt_template"])) - - if getv(from_object, ["agent_run_config"]) is not None: - setv(to_object, ["agentRunConfig"], getv(from_object, ["agent_run_config"])) - - if getv(from_object, ["agent_configs"]) is not None: - setv(to_object, ["agents"], getv(from_object, ["agent_configs"])) - - return to_object - - -def _EvaluationRunMetric_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, ["metric"]) is not None: - setv(to_object, ["metric"], getv(from_object, ["metric"])) - - if getv(from_object, ["metricResourceName"]) is not None: - setv( - to_object, - ["metric_resource_name"], - getv(from_object, ["metricResourceName"]), - ) - - if getv(from_object, ["metricConfig"]) is not None: - setv( - to_object, - ["metric_config"], - _UnifiedMetric_from_vertex(getv(from_object, ["metricConfig"]), to_object), - ) - - return to_object - - -def _EvaluationRunMetric_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, ["metric"]) is not None: - setv(to_object, ["metric"], getv(from_object, ["metric"])) - - if getv(from_object, ["metric_resource_name"]) is not None: - setv( - to_object, - ["metricResourceName"], - getv(from_object, ["metric_resource_name"]), - ) - - if getv(from_object, ["metric_config"]) is not None: - setv( - to_object, - ["metricConfig"], - _UnifiedMetric_to_vertex(getv(from_object, ["metric_config"]), to_object), - ) - - return to_object - - -def _EvaluationRun_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, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(from_object, ["displayName"])) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["createTime"]) is not None: - setv(to_object, ["create_time"], getv(from_object, ["createTime"])) - - if getv(from_object, ["completionTime"]) is not None: - setv(to_object, ["completion_time"], getv(from_object, ["completionTime"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(from_object, ["evaluationSetSnapshot"]) is not None: - setv( - to_object, - ["evaluation_set_snapshot"], - getv(from_object, ["evaluationSetSnapshot"]), - ) - - if getv(from_object, ["error"]) is not None: - setv(to_object, ["error"], getv(from_object, ["error"])) - - if getv(from_object, ["dataSource"]) is not None: - setv(to_object, ["data_source"], getv(from_object, ["dataSource"])) - - if getv(from_object, ["evaluationResults"]) is not None: - setv( - to_object, - ["evaluation_run_results"], - getv(from_object, ["evaluationResults"]), - ) - - if getv(from_object, ["evaluationConfig"]) is not None: - setv( - to_object, - ["evaluation_config"], - _EvaluationRunConfig_from_vertex( - getv(from_object, ["evaluationConfig"]), to_object - ), - ) - - if getv(from_object, ["inferenceConfigs"]) is not None: - setv( - to_object, - ["inference_configs"], - { - k: _EvaluationRunInferenceConfig_from_vertex(v, to_object) - for k, v in getv(from_object, ["inferenceConfigs"]).items() - }, - ) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["analysisConfigs"]) is not None: - setv( - to_object, - ["analysis_configs"], - [item for item in getv(from_object, ["analysisConfigs"])], - ) - - if getv(from_object, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) - - return to_object - - -def _GenerateInstanceRubricsRequest_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, ["contents"]) is not None: - setv(to_object, ["contents"], getv(from_object, ["contents"])) - - if getv(from_object, ["predefined_rubric_generation_spec"]) is not None: - setv( - to_object, - ["predefinedRubricGenerationSpec"], - getv(from_object, ["predefined_rubric_generation_spec"]), - ) - - if getv(from_object, ["rubric_generation_spec"]) is not None: - setv( - to_object, - ["rubricGenerationSpec"], - getv(from_object, ["rubric_generation_spec"]), - ) - - if getv(from_object, ["metric_resource_name"]) is not None: - setv( - to_object, - ["metricResourceName"], - getv(from_object, ["metric_resource_name"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GenerateLossClustersParameters_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, ["location"]) is not None: - setv(to_object, ["location"], getv(from_object, ["location"])) - - if getv(from_object, ["evaluation_set"]) is not None: - setv(to_object, ["evaluationSet"], getv(from_object, ["evaluation_set"])) - - if getv(from_object, ["inline_results"]) is not None: - setv( - to_object, - ["inlineResults", "evaluationResults"], - [ - item - for item in t.t_inline_results(getv(from_object, ["inline_results"])) - ], - ) - - if getv(from_object, ["configs"]) is not None: - setv(to_object, ["configs"], [item for item in getv(from_object, ["configs"])]) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GenerateUserScenariosParameters_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, ["location"]) is not None: - setv(to_object, ["location"], getv(from_object, ["location"])) - - if getv(from_object, ["agents"]) is not None: - setv(to_object, ["agents"], getv(from_object, ["agents"])) - - if getv(from_object, ["root_agent_id"]) is not None: - setv(to_object, ["rootAgentId"], getv(from_object, ["root_agent_id"])) - - if getv(from_object, ["user_scenario_generation_config"]) is not None: - setv( - to_object, - ["userScenarioGenerationConfig"], - t.t_user_scenario_generation_config( - getv(from_object, ["user_scenario_generation_config"]) - ), - ) - - if getv(from_object, ["allow_cross_region_model"]) is not None: - setv( - to_object, - ["allowCrossRegionModel"], - getv(from_object, ["allow_cross_region_model"]), - ) - - if getv(from_object, ["gemini_agent_config"]) is not None: - setv( - to_object, ["geminiAgentConfig"], getv(from_object, ["gemini_agent_config"]) - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetEvaluationExperimentParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetEvaluationItemParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetEvaluationMetricParameters_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, ["metric_resource_name"]) is not None: - setv( - to_object, - ["_url", "evaluation_metric"], - getv(from_object, ["metric_resource_name"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetEvaluationRunParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetEvaluationSetParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ImportEvaluationSetParameters_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, ["evaluation_set"]) is not None: - setv(to_object, ["evaluationSet"], getv(from_object, ["evaluation_set"])) - - if getv(from_object, ["gcs_destination"]) is not None: - setv(to_object, ["gcsDestination"], getv(from_object, ["gcs_destination"])) - - if getv(from_object, ["gcs_source"]) is not None: - setv(to_object, ["gcsSource"], getv(from_object, ["gcs_source"])) - - if getv(from_object, ["inline_source"]) is not None: - setv(to_object, ["inlineSource"], getv(from_object, ["inline_source"])) - - if getv(from_object, ["cloud_trace_source"]) is not None: - setv(to_object, ["cloudTraceSource"], getv(from_object, ["cloud_trace_source"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListEvaluationExperimentsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["order_by"]) is not None: - setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) - - return to_object - - -def _ListEvaluationExperimentsParameters_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"], - _ListEvaluationExperimentsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _ListEvaluationMetricsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["order_by"]) is not None: - setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) - - return to_object - - -def _ListEvaluationMetricsParameters_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"], - _ListEvaluationMetricsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _ListEvaluationMetricsResponse_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, ["evaluationMetrics"]) is not None: - setv( - to_object, - ["evaluation_metrics"], - [ - _EvaluationMetric_from_vertex(item, to_object) - for item in getv(from_object, ["evaluationMetrics"]) - ], - ) - - return to_object - - -def _ListEvaluationSetsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["order_by"]) is not None: - setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) - - return to_object - - -def _ListEvaluationSetsParameters_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"], - _ListEvaluationSetsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _RubricBasedMetricInput_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, ["metric_spec"]) is not None: - setv( - to_object, - ["metricSpec"], - _RubricBasedMetricSpec_to_vertex( - getv(from_object, ["metric_spec"]), to_object - ), - ) - - if getv(from_object, ["instance"]) is not None: - setv(to_object, ["instance"], getv(from_object, ["instance"])) - - return to_object - - -def _RubricBasedMetricSpec_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, ["metric_prompt_template"]) is not None: - setv( - to_object, - ["metricPromptTemplate"], - getv(from_object, ["metric_prompt_template"]), - ) - - if getv(from_object, ["judge_autorater_config"]) is not None: - setv( - to_object, - ["judgeAutoraterConfig"], - getv(from_object, ["judge_autorater_config"]), - ) - - if getv(from_object, ["inline_rubrics"]) is not None: - setv( - to_object, - ["inline_rubrics", "rubrics"], - getv(from_object, ["inline_rubrics"]), - ) - - if getv(from_object, ["rubric_group_key"]) is not None: - setv(to_object, ["rubricGroupKey"], getv(from_object, ["rubric_group_key"])) - - if getv(from_object, ["rubric_generation_spec"]) is not None: - setv( - to_object, - ["rubricGenerationSpec"], - getv(from_object, ["rubric_generation_spec"]), - ) - - return to_object - - -def _UnifiedMetric_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, ["bleuSpec"]) is not None: - setv(to_object, ["bleu_spec"], getv(from_object, ["bleuSpec"])) - - if getv(from_object, ["rougeSpec"]) is not None: - setv(to_object, ["rouge_spec"], getv(from_object, ["rougeSpec"])) - - if getv(from_object, ["pointwiseMetricSpec"]) is not None: - setv( - to_object, - ["pointwise_metric_spec"], - getv(from_object, ["pointwiseMetricSpec"]), - ) - - if getv(from_object, ["llmBasedMetricSpec"]) is not None: - setv( - to_object, - ["llm_based_metric_spec"], - getv(from_object, ["llmBasedMetricSpec"]), - ) - - if getv(from_object, ["customCodeExecutionSpec"]) is not None: - setv( - to_object, - ["custom_code_execution_spec"], - _CustomCodeExecutionSpec_from_vertex( - getv(from_object, ["customCodeExecutionSpec"]), to_object - ), - ) - - if getv(from_object, ["predefinedMetricSpec"]) is not None: - setv( - to_object, - ["predefined_metric_spec"], - getv(from_object, ["predefinedMetricSpec"]), - ) - - if getv(from_object, ["computationBasedMetricSpec"]) is not None: - setv( - to_object, - ["computation_based_metric_spec"], - getv(from_object, ["computationBasedMetricSpec"]), - ) - - return to_object - - -def _UnifiedMetric_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, ["bleu_spec"]) is not None: - setv(to_object, ["bleuSpec"], getv(from_object, ["bleu_spec"])) - - if getv(from_object, ["rouge_spec"]) is not None: - setv(to_object, ["rougeSpec"], getv(from_object, ["rouge_spec"])) - - if getv(from_object, ["pointwise_metric_spec"]) is not None: - setv( - to_object, - ["pointwiseMetricSpec"], - getv(from_object, ["pointwise_metric_spec"]), - ) - - if getv(from_object, ["llm_based_metric_spec"]) is not None: - setv( - to_object, - ["llmBasedMetricSpec"], - getv(from_object, ["llm_based_metric_spec"]), - ) - - if getv(from_object, ["custom_code_execution_spec"]) is not None: - setv( - to_object, - ["customCodeExecutionSpec"], - _CustomCodeExecutionSpec_to_vertex( - getv(from_object, ["custom_code_execution_spec"]), to_object - ), - ) - - if getv(from_object, ["predefined_metric_spec"]) is not None: - setv( - to_object, - ["predefinedMetricSpec"], - getv(from_object, ["predefined_metric_spec"]), - ) - - if getv(from_object, ["computation_based_metric_spec"]) is not None: - setv( - to_object, - ["computationBasedMetricSpec"], - getv(from_object, ["computation_based_metric_spec"]), - ) - - return to_object - - -def _UpdateEvaluationExperimentConfig_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, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - if getv(from_object, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["labels"]) is not None: - setv(parent_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["merge_strategy"]) is not None: - setv(parent_object, ["mergeStrategy"], getv(from_object, ["merge_strategy"])) - - if getv(from_object, ["metadata"]) is not None: - setv(parent_object, ["metadata"], getv(from_object, ["metadata"])) - - return to_object - - -def _UpdateEvaluationExperimentParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateEvaluationExperimentConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class Evals(_api_module.BaseModule): def create_evaluation_experiment( @@ -1315,7 +88,7 @@ def create_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationExperimentParameters_to_vertex( + request_dict = _converters._CreateEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1393,7 +166,9 @@ def _create_evaluation_item( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationItemParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationItemParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationItems".format_map(request_url_dict) @@ -1471,7 +246,9 @@ def _create_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationMetrics".format_map(request_url_dict) @@ -1499,7 +276,7 @@ def _create_evaluation_metric( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationMetric_from_vertex(response_dict) + response_dict = _converters._EvaluationMetric_from_vertex(response_dict) return_value = types.EvaluationMetric._from_response( response=response_dict, @@ -1564,7 +341,9 @@ def _create_evaluation_run( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationRunParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationRunParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationRuns".format_map(request_url_dict) @@ -1592,7 +371,7 @@ def _create_evaluation_run( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationRun_from_vertex(response_dict) + response_dict = _converters._EvaluationRun_from_vertex(response_dict) return_value = types.EvaluationRun._from_response( response=response_dict, @@ -1643,7 +422,9 @@ def _create_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets".format_map(request_url_dict) @@ -1724,7 +505,7 @@ def delete_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationExperimentParameters_to_vertex( + request_dict = _converters._DeleteEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1798,7 +579,9 @@ def _delete_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{evaluation_metric}".format_map(request_url_dict) @@ -1870,7 +653,9 @@ def _delete_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets/{name}".format_map(request_url_dict) @@ -1974,7 +759,7 @@ def _evaluate_instances( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _EvaluateInstancesRequestParameters_to_vertex( + request_dict = _converters._EvaluateInstancesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2060,7 +845,9 @@ def _generate_user_scenarios( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateUserScenariosParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateUserScenariosParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateUserScenarios".format_map(request_url_dict) @@ -2138,7 +925,9 @@ def _generate_loss_clusters( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateLossClustersParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateLossClustersParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateLossClusters".format_map(request_url_dict) @@ -2218,7 +1007,9 @@ def _generate_rubrics( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateInstanceRubricsRequest_to_vertex(parameter_model) + request_dict = _converters._GenerateInstanceRubricsRequest_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateInstanceRubrics".format_map(request_url_dict) @@ -2307,7 +1098,9 @@ def get_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationExperimentParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationExperimentParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2379,7 +1172,9 @@ def _get_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{evaluation_metric}".format_map(request_url_dict) @@ -2407,7 +1202,7 @@ def _get_evaluation_metric( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationMetric_from_vertex(response_dict) + response_dict = _converters._EvaluationMetric_from_vertex(response_dict) return_value = types.EvaluationMetric._from_response( response=response_dict, @@ -2451,7 +1246,9 @@ def _get_evaluation_run( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationRunParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationRunParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationRuns/{name}".format_map(request_url_dict) @@ -2479,7 +1276,7 @@ def _get_evaluation_run( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationRun_from_vertex(response_dict) + response_dict = _converters._EvaluationRun_from_vertex(response_dict) return_value = types.EvaluationRun._from_response( response=response_dict, @@ -2523,7 +1320,9 @@ def _get_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets/{name}".format_map(request_url_dict) @@ -2592,7 +1391,9 @@ def _get_evaluation_item( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationItemParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationItemParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationItems/{name}".format_map(request_url_dict) @@ -2672,7 +1473,9 @@ def _import_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ImportEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._ImportEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets:import".format_map(request_url_dict) @@ -2757,7 +1560,7 @@ def list_evaluation_experiments( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationExperimentsParameters_to_vertex( + request_dict = _converters._ListEvaluationExperimentsParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2827,7 +1630,9 @@ def _list_evaluation_metrics( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationMetricsParameters_to_vertex(parameter_model) + request_dict = _converters._ListEvaluationMetricsParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationMetrics".format_map(request_url_dict) @@ -2855,7 +1660,9 @@ def _list_evaluation_metrics( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListEvaluationMetricsResponse_from_vertex(response_dict) + response_dict = _converters._ListEvaluationMetricsResponse_from_vertex( + response_dict + ) return_value = types.ListEvaluationMetricsResponse._from_response( response=response_dict, @@ -2898,7 +1705,9 @@ def _list_evaluation_sets( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationSetsParameters_to_vertex(parameter_model) + request_dict = _converters._ListEvaluationSetsParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets".format_map(request_url_dict) @@ -2980,7 +1789,7 @@ def update_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateEvaluationExperimentParameters_to_vertex( + request_dict = _converters._UpdateEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -4313,7 +3122,7 @@ async def create_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationExperimentParameters_to_vertex( + request_dict = _converters._CreateEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -4393,7 +3202,9 @@ async def _create_evaluation_item( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationItemParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationItemParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationItems".format_map(request_url_dict) @@ -4473,7 +3284,9 @@ async def _create_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationMetrics".format_map(request_url_dict) @@ -4503,7 +3316,7 @@ async def _create_evaluation_metric( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationMetric_from_vertex(response_dict) + response_dict = _converters._EvaluationMetric_from_vertex(response_dict) return_value = types.EvaluationMetric._from_response( response=response_dict, @@ -4568,7 +3381,9 @@ async def _create_evaluation_run( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationRunParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationRunParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationRuns".format_map(request_url_dict) @@ -4598,7 +3413,7 @@ async def _create_evaluation_run( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationRun_from_vertex(response_dict) + response_dict = _converters._EvaluationRun_from_vertex(response_dict) return_value = types.EvaluationRun._from_response( response=response_dict, @@ -4649,7 +3464,9 @@ async def _create_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets".format_map(request_url_dict) @@ -4732,7 +3549,7 @@ async def delete_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationExperimentParameters_to_vertex( + request_dict = _converters._DeleteEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -4808,7 +3625,9 @@ async def _delete_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{evaluation_metric}".format_map(request_url_dict) @@ -4882,7 +3701,9 @@ async def _delete_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets/{name}".format_map(request_url_dict) @@ -4988,7 +3809,7 @@ async def _evaluate_instances( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _EvaluateInstancesRequestParameters_to_vertex( + request_dict = _converters._EvaluateInstancesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -5076,7 +3897,9 @@ async def _generate_user_scenarios( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateUserScenariosParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateUserScenariosParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateUserScenarios".format_map(request_url_dict) @@ -5156,7 +3979,9 @@ async def _generate_loss_clusters( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateLossClustersParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateLossClustersParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateLossClusters".format_map(request_url_dict) @@ -5238,7 +4063,9 @@ async def _generate_rubrics( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateInstanceRubricsRequest_to_vertex(parameter_model) + request_dict = _converters._GenerateInstanceRubricsRequest_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":generateInstanceRubrics".format_map(request_url_dict) @@ -5329,7 +4156,9 @@ async def get_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationExperimentParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationExperimentParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -5403,7 +4232,9 @@ async def _get_evaluation_metric( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationMetricParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationMetricParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{evaluation_metric}".format_map(request_url_dict) @@ -5433,7 +4264,7 @@ async def _get_evaluation_metric( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationMetric_from_vertex(response_dict) + response_dict = _converters._EvaluationMetric_from_vertex(response_dict) return_value = types.EvaluationMetric._from_response( response=response_dict, @@ -5477,7 +4308,9 @@ async def _get_evaluation_run( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationRunParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationRunParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationRuns/{name}".format_map(request_url_dict) @@ -5507,7 +4340,7 @@ async def _get_evaluation_run( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _EvaluationRun_from_vertex(response_dict) + response_dict = _converters._EvaluationRun_from_vertex(response_dict) return_value = types.EvaluationRun._from_response( response=response_dict, @@ -5551,7 +4384,9 @@ async def _get_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets/{name}".format_map(request_url_dict) @@ -5622,7 +4457,9 @@ async def _get_evaluation_item( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetEvaluationItemParameters_to_vertex(parameter_model) + request_dict = _converters._GetEvaluationItemParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationItems/{name}".format_map(request_url_dict) @@ -5704,7 +4541,9 @@ async def _import_evaluation_set( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ImportEvaluationSetParameters_to_vertex(parameter_model) + request_dict = _converters._ImportEvaluationSetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets:import".format_map(request_url_dict) @@ -5791,7 +4630,7 @@ async def list_evaluation_experiments( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationExperimentsParameters_to_vertex( + request_dict = _converters._ListEvaluationExperimentsParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -5863,7 +4702,9 @@ async def _list_evaluation_metrics( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationMetricsParameters_to_vertex(parameter_model) + request_dict = _converters._ListEvaluationMetricsParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationMetrics".format_map(request_url_dict) @@ -5893,7 +4734,9 @@ async def _list_evaluation_metrics( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListEvaluationMetricsResponse_from_vertex(response_dict) + response_dict = _converters._ListEvaluationMetricsResponse_from_vertex( + response_dict + ) return_value = types.ListEvaluationMetricsResponse._from_response( response=response_dict, @@ -5936,7 +4779,9 @@ async def _list_evaluation_sets( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListEvaluationSetsParameters_to_vertex(parameter_model) + request_dict = _converters._ListEvaluationSetsParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "evaluationSets".format_map(request_url_dict) @@ -6020,7 +4865,7 @@ async def update_evaluation_experiment( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateEvaluationExperimentParameters_to_vertex( + request_dict = _converters._UpdateEvaluationExperimentParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/example_stores.py b/agentplatform/_genai/example_stores.py index 54c8754dba..40beda85ff 100644 --- a/agentplatform/_genai/example_stores.py +++ b/agentplatform/_genai/example_stores.py @@ -17,264 +17,18 @@ import json import logging -from typing import Any, Optional, Union +from typing import Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv -from . import _transformers as t +from . import _converters from . import types logger = logging.getLogger("agentplatform_genai.examplestores") -def _CreateExampleStoreConfig_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, ["description"]) is not None: - setv( - to_object, - ["_query", "exampleStore", "description"], - getv(from_object, ["description"]), - ) - - if getv(from_object, ["vertex_embedding_model"]) is not None: - setv( - to_object, - ["_query", "exampleStore", "exampleStoreConfig", "vertexEmbeddingModel"], - getv(from_object, ["vertex_embedding_model"]), - ) - - return to_object - - -def _CreateExampleStoreParameters_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, ["display_name"]) is not None: - setv( - to_object, - ["_query", "exampleStore", "displayName"], - getv(from_object, ["display_name"]), - ) - - if getv(from_object, ["config"]) is not None: - _CreateExampleStoreConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteExampleStoreConfig_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 _DeleteExampleStoreRequestParameters_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"], t.t_example_store(getv(from_object, ["name"])) - ) - - if getv(from_object, ["config"]) is not None: - _DeleteExampleStoreConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _FetchExamplesConfig_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, ["example_ids"]) is not None: - setv(parent_object, ["exampleIds"], getv(from_object, ["example_ids"])) - - if getv(from_object, ["stored_contents_example_filter"]) is not None: - setv( - parent_object, - ["storedContentsExampleFilter"], - getv(from_object, ["stored_contents_example_filter"]), - ) - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["pageToken"], getv(from_object, ["page_token"])) - - return to_object - - -def _FetchExamplesParameters_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", "exampleStore"], - t.t_example_store(getv(from_object, ["name"])), - ) - - if getv(from_object, ["config"]) is not None: - _FetchExamplesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _GetExampleStoreOperationParameters_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 _GetExampleStoreParameters_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"], t.t_example_store(getv(from_object, ["name"])) - ) - - return to_object - - -def _RemoveExamplesConfig_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, ["example_ids"]) is not None: - setv(parent_object, ["exampleIds"], getv(from_object, ["example_ids"])) - - if getv(from_object, ["stored_contents_example_filter"]) is not None: - setv( - parent_object, - ["storedContentsExampleFilter"], - getv(from_object, ["stored_contents_example_filter"]), - ) - - return to_object - - -def _RemoveExamplesParameters_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", "exampleStore"], - t.t_example_store(getv(from_object, ["name"])), - ) - - if getv(from_object, ["config"]) is not None: - _RemoveExamplesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _SearchExamplesConfig_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, ["top_k"]) is not None: - setv(parent_object, ["topK"], getv(from_object, ["top_k"])) - - return to_object - - -def _SearchExamplesParameters_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", "exampleStore"], - t.t_example_store(getv(from_object, ["name"])), - ) - - if getv(from_object, ["stored_contents_example_parameters"]) is not None: - setv( - to_object, - ["storedContentsExampleParameters"], - getv(from_object, ["stored_contents_example_parameters"]), - ) - - if getv(from_object, ["config"]) is not None: - _SearchExamplesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _UpsertExamplesConfig_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, ["overwrite"]) is not None: - setv(parent_object, ["overwrite"], getv(from_object, ["overwrite"])) - - return to_object - - -def _UpsertExamplesParameters_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", "exampleStore"], - t.t_example_store(getv(from_object, ["name"])), - ) - - if getv(from_object, ["examples"]) is not None: - setv( - to_object, ["examples"], [item for item in getv(from_object, ["examples"])] - ) - - if getv(from_object, ["config"]) is not None: - _UpsertExamplesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class ExampleStores(_api_module.BaseModule): """Class for managing Example Stores of few-shot examples.""" @@ -299,7 +53,9 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateExampleStoreParameters_to_vertex(parameter_model) + request_dict = _converters._CreateExampleStoreParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{parent}/exampleStores:create".format_map(request_url_dict) @@ -368,7 +124,9 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExampleStoreParameters_to_vertex(parameter_model) + request_dict = _converters._GetExampleStoreParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -440,7 +198,7 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteExampleStoreRequestParameters_to_vertex( + request_dict = _converters._DeleteExampleStoreRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -516,7 +274,9 @@ def upsert_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpsertExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._UpsertExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:upsertExamples".format_map(request_url_dict) @@ -592,7 +352,9 @@ def search_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _SearchExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._SearchExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:searchExamples".format_map(request_url_dict) @@ -661,7 +423,9 @@ def fetch_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _FetchExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._FetchExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:fetchExamples".format_map(request_url_dict) @@ -730,7 +494,9 @@ def remove_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RemoveExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._RemoveExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:removeExamples".format_map(request_url_dict) @@ -798,7 +564,7 @@ def _get_example_store_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExampleStoreOperationParameters_to_vertex( + request_dict = _converters._GetExampleStoreOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -876,7 +642,9 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateExampleStoreParameters_to_vertex(parameter_model) + request_dict = _converters._CreateExampleStoreParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{parent}/exampleStores:create".format_map(request_url_dict) @@ -947,7 +715,9 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExampleStoreParameters_to_vertex(parameter_model) + request_dict = _converters._GetExampleStoreParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1021,7 +791,7 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteExampleStoreRequestParameters_to_vertex( + request_dict = _converters._DeleteExampleStoreRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1099,7 +869,9 @@ async def upsert_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpsertExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._UpsertExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:upsertExamples".format_map(request_url_dict) @@ -1177,7 +949,9 @@ async def search_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _SearchExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._SearchExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:searchExamples".format_map(request_url_dict) @@ -1248,7 +1022,9 @@ async def fetch_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _FetchExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._FetchExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:fetchExamples".format_map(request_url_dict) @@ -1319,7 +1095,9 @@ async def remove_examples( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RemoveExamplesParameters_to_vertex(parameter_model) + request_dict = _converters._RemoveExamplesParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{exampleStore}:removeExamples".format_map(request_url_dict) @@ -1389,7 +1167,7 @@ async def _get_example_store_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExampleStoreOperationParameters_to_vertex( + request_dict = _converters._GetExampleStoreOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/feedback_contexts.py b/agentplatform/_genai/feedback_contexts.py index 8fd4d04850..86b0cc81c8 100644 --- a/agentplatform/_genai/feedback_contexts.py +++ b/agentplatform/_genai/feedback_contexts.py @@ -17,14 +17,13 @@ import json import logging -from typing import Any, Optional, Union +from typing import Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv +from . import _converters from . import _runtimes_utils from . import types @@ -33,67 +32,6 @@ logger.setLevel(logging.INFO) -def _GetRuntimeFeedbackContextOperationParameters_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 _GetRuntimeFeedbackContextRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _UpdateRuntimeFeedbackContextConfig_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, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - return to_object - - -def _UpdateRuntimeFeedbackContextRequestParameters_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, ["context_events"]) is not None: - setv( - to_object, - ["contextEvents"], - [item for item in getv(from_object, ["context_events"])], - ) - - if getv(from_object, ["config"]) is not None: - _UpdateRuntimeFeedbackContextConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class FeedbackContexts(_api_module.BaseModule): def _get( @@ -127,8 +65,10 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackContextRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeFeedbackContextRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -197,8 +137,10 @@ def _get_feedback_context_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackContextOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeFeedbackContextOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -273,8 +215,10 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeFeedbackContextRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._UpdateRuntimeFeedbackContextRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -427,8 +371,10 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackContextRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeFeedbackContextRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -499,8 +445,10 @@ async def _get_feedback_context_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackContextOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeFeedbackContextOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -577,8 +525,10 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeFeedbackContextRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._UpdateRuntimeFeedbackContextRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: diff --git a/agentplatform/_genai/feedback_entries.py b/agentplatform/_genai/feedback_entries.py index b214646ad0..9487aec3e5 100644 --- a/agentplatform/_genai/feedback_entries.py +++ b/agentplatform/_genai/feedback_entries.py @@ -20,15 +20,14 @@ import json import logging import typing -from typing import Any, AsyncIterator, Iterator, Optional, Union +from typing import AsyncIterator, Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _runtimes_utils from . import types @@ -43,181 +42,6 @@ logger.setLevel(logging.INFO) -def _CreateRuntimeFeedbackEntryConfig_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, ["feedback_labels"]) is not None: - setv(parent_object, ["feedbackLabels"], getv(from_object, ["feedback_labels"])) - - if getv(from_object, ["feedback_text"]) is not None: - setv(parent_object, ["feedbackText"], getv(from_object, ["feedback_text"])) - - if getv(from_object, ["user_id"]) is not None: - setv(parent_object, ["userId"], getv(from_object, ["user_id"])) - - if getv(from_object, ["source"]) is not None: - setv(parent_object, ["source"], getv(from_object, ["source"])) - - if getv(from_object, ["custom_metadata"]) is not None: - setv(parent_object, ["customMetadata"], getv(from_object, ["custom_metadata"])) - - return to_object - - -def _CreateRuntimeFeedbackEntryRequestParameters_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", "parent"], getv(from_object, ["name"])) - - if getv(from_object, ["feedback_type"]) is not None: - setv(to_object, ["feedbackType"], getv(from_object, ["feedback_type"])) - - if getv(from_object, ["session_id"]) is not None: - setv(to_object, ["sessionId"], getv(from_object, ["session_id"])) - - if getv(from_object, ["event_id"]) is not None: - setv(to_object, ["eventId"], getv(from_object, ["event_id"])) - - if getv(from_object, ["config"]) is not None: - _CreateRuntimeFeedbackEntryConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetRuntimeFeedbackOperationParameters_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 _GetRuntimeFeedbackRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListRuntimeFeedbackEntriesConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["order_by"]) is not None: - setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) - - return to_object - - -def _ListRuntimeFeedbackEntriesRequestParameters_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, ["parent"]) is not None: - setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["config"]) is not None: - _ListRuntimeFeedbackEntriesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _UpdateRuntimeFeedbackEntryConfig_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, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - if getv(from_object, ["feedback_type"]) is not None: - setv(parent_object, ["feedbackType"], getv(from_object, ["feedback_type"])) - - if getv(from_object, ["session_id"]) is not None: - setv(parent_object, ["sessionId"], getv(from_object, ["session_id"])) - - if getv(from_object, ["event_id"]) is not None: - setv(parent_object, ["eventId"], getv(from_object, ["event_id"])) - - if getv(from_object, ["feedback_labels"]) is not None: - setv(parent_object, ["feedbackLabels"], getv(from_object, ["feedback_labels"])) - - if getv(from_object, ["feedback_text"]) is not None: - setv(parent_object, ["feedbackText"], getv(from_object, ["feedback_text"])) - - if getv(from_object, ["user_id"]) is not None: - setv(parent_object, ["userId"], getv(from_object, ["user_id"])) - - if getv(from_object, ["source"]) is not None: - setv(parent_object, ["source"], getv(from_object, ["source"])) - - if getv(from_object, ["custom_metadata"]) is not None: - setv(parent_object, ["customMetadata"], getv(from_object, ["custom_metadata"])) - - return to_object - - -def _UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateRuntimeFeedbackEntryConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class FeedbackEntries(_api_module.BaseModule): def _create( @@ -243,8 +67,10 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._CreateRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -313,8 +139,10 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -397,7 +225,7 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -467,8 +295,10 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeFeedbackEntriesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListRuntimeFeedbackEntriesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -551,8 +381,10 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -621,7 +453,7 @@ def _get_feedback_entry_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -691,7 +523,7 @@ def _get_delete_feedback_entry_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -948,8 +780,10 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._CreateRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1020,8 +854,10 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1106,7 +942,7 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1178,8 +1014,10 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeFeedbackEntriesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListRuntimeFeedbackEntriesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1264,8 +1102,10 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._UpdateRuntimeFeedbackEntryRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1336,7 +1176,7 @@ async def _get_feedback_entry_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1408,7 +1248,7 @@ async def _get_delete_feedback_entry_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeFeedbackOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeFeedbackOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/memories.py b/agentplatform/_genai/memories.py index 6c660c14d2..8b2a248ae6 100644 --- a/agentplatform/_genai/memories.py +++ b/agentplatform/_genai/memories.py @@ -21,15 +21,14 @@ import json import logging import typing -from typing import Any, Iterator, List, Optional, Union +from typing import Iterator, List, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _memory_bank_utils from . import types @@ -42,440 +41,6 @@ logger = logging.getLogger("agentplatform_genai.memories") -def _CreateMemoryRequestParameters_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, ["fact"]) is not None: - setv(to_object, ["fact"], getv(from_object, ["fact"])) - - if getv(from_object, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["config"]) is not None: - _MemoryConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteMemoryRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GenerateMemoriesConfig_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, ["disable_consolidation"]) is not None: - setv( - parent_object, - ["disableConsolidation"], - getv(from_object, ["disable_consolidation"]), - ) - - if getv(from_object, ["revision_labels"]) is not None: - setv(parent_object, ["revisionLabels"], getv(from_object, ["revision_labels"])) - - if getv(from_object, ["revision_expire_time"]) is not None: - setv( - parent_object, - ["revisionExpireTime"], - getv(from_object, ["revision_expire_time"]), - ) - - if getv(from_object, ["revision_ttl"]) is not None: - setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - parent_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["metadata"]) is not None: - setv( - parent_object, - ["metadata"], - {k: v for k, v in getv(from_object, ["metadata"]).items()}, - ) - - if getv(from_object, ["metadata_merge_strategy"]) is not None: - setv( - parent_object, - ["metadataMergeStrategy"], - getv(from_object, ["metadata_merge_strategy"]), - ) - - if getv(from_object, ["allowed_topics"]) is not None: - setv( - parent_object, - ["allowedTopics"], - [item for item in getv(from_object, ["allowed_topics"])], - ) - - return to_object - - -def _GenerateMemoriesRequestParameters_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, ["vertex_session_source"]) is not None: - setv( - to_object, - ["vertexSessionSource"], - getv(from_object, ["vertex_session_source"]), - ) - - if getv(from_object, ["direct_contents_source"]) is not None: - setv( - to_object, - ["directContentsSource"], - getv(from_object, ["direct_contents_source"]), - ) - - if getv(from_object, ["direct_memories_source"]) is not None: - setv( - to_object, - ["directMemoriesSource"], - getv(from_object, ["direct_memories_source"]), - ) - - if getv(from_object, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["config"]) is not None: - _GenerateMemoriesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _GetGenerateMemoriesOperationParameters_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 _GetMemoryOperationParameters_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 _GetMemoryRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListMemoriesConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["order_by"]) is not None: - setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"])) - - return to_object - - -def _ListMemoriesRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListMemoriesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _MemoryConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - if getv(from_object, ["expire_time"]) is not None: - setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) - - if getv(from_object, ["revision_expire_time"]) is not None: - setv( - parent_object, - ["revisionExpireTime"], - getv(from_object, ["revision_expire_time"]), - ) - - if getv(from_object, ["revision_ttl"]) is not None: - setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - parent_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["topics"]) is not None: - setv( - parent_object, ["topics"], [item for item in getv(from_object, ["topics"])] - ) - - if getv(from_object, ["metadata"]) is not None: - setv( - parent_object, - ["metadata"], - {k: v for k, v in getv(from_object, ["metadata"]).items()}, - ) - - if getv(from_object, ["memory_id"]) is not None: - setv(parent_object, ["_query", "memoryId"], getv(from_object, ["memory_id"])) - - return to_object - - -def _PurgeMemoriesRequestParameters_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, ["filter"]) is not None: - setv(to_object, ["filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["filter_groups"]) is not None: - setv( - to_object, - ["filterGroups"], - [item for item in getv(from_object, ["filter_groups"])], - ) - - if getv(from_object, ["force"]) is not None: - setv(to_object, ["force"], getv(from_object, ["force"])) - - return to_object - - -def _RetrieveMemoriesConfig_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, ["filter"]) is not None: - setv(parent_object, ["filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["filter_groups"]) is not None: - setv( - parent_object, - ["filterGroups"], - [item for item in getv(from_object, ["filter_groups"])], - ) - - if getv(from_object, ["memory_types"]) is not None: - setv(parent_object, ["memoryTypes"], getv(from_object, ["memory_types"])) - - return to_object - - -def _RetrieveMemoriesRequestParameters_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, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["similarity_search_params"]) is not None: - setv( - to_object, - ["similaritySearchParams"], - getv(from_object, ["similarity_search_params"]), - ) - - if getv(from_object, ["simple_retrieval_params"]) is not None: - setv( - to_object, - ["simpleRetrievalParams"], - getv(from_object, ["simple_retrieval_params"]), - ) - - if getv(from_object, ["config"]) is not None: - _RetrieveMemoriesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _RetrieveMemoryProfilesRequestParameters_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, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - return to_object - - -def _RollbackMemoryRequestParameters_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, ["target_revision_id"]) is not None: - setv(to_object, ["targetRevisionId"], getv(from_object, ["target_revision_id"])) - - return to_object - - -def _UpdateMemoryConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - if getv(from_object, ["expire_time"]) is not None: - setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) - - if getv(from_object, ["revision_expire_time"]) is not None: - setv( - parent_object, - ["revisionExpireTime"], - getv(from_object, ["revision_expire_time"]), - ) - - if getv(from_object, ["revision_ttl"]) is not None: - setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - parent_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["topics"]) is not None: - setv( - parent_object, ["topics"], [item for item in getv(from_object, ["topics"])] - ) - - if getv(from_object, ["metadata"]) is not None: - setv( - parent_object, - ["metadata"], - {k: v for k, v in getv(from_object, ["metadata"]).items()}, - ) - - if getv(from_object, ["memory_id"]) is not None: - setv(parent_object, ["_query", "memoryId"], getv(from_object, ["memory_id"])) - - if getv(from_object, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - return to_object - - -def _UpdateMemoryRequestParameters_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, ["fact"]) is not None: - setv(to_object, ["fact"], getv(from_object, ["fact"])) - - if getv(from_object, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["config"]) is not None: - _UpdateMemoryConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class Memories(_api_module.BaseModule): def _create( @@ -503,7 +68,9 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories".format_map(request_url_dict) @@ -580,7 +147,9 @@ def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -666,7 +235,9 @@ def _generate( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:generate".format_map(request_url_dict) @@ -741,7 +312,9 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -810,7 +383,9 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories".format_map(request_url_dict) @@ -878,7 +453,9 @@ def _get_memory_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -946,8 +523,10 @@ def _get_generate_memories_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetGenerateMemoriesOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetGenerateMemoriesOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1030,7 +609,9 @@ def _retrieve( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RetrieveMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:retrieve".format_map(request_url_dict) @@ -1130,8 +711,10 @@ def retrieve_profiles( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveMemoryProfilesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._RetrieveMemoryProfilesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1206,7 +789,9 @@ def _rollback( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RollbackMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RollbackMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:rollback".format_map(request_url_dict) @@ -1282,7 +867,9 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1362,7 +949,9 @@ def _purge( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PurgeMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._PurgeMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:purge".format_map(request_url_dict) @@ -1734,7 +1323,9 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories".format_map(request_url_dict) @@ -1813,7 +1404,9 @@ async def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1901,7 +1494,9 @@ async def _generate( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GenerateMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GenerateMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:generate".format_map(request_url_dict) @@ -1978,7 +1573,9 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2049,7 +1646,9 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories".format_map(request_url_dict) @@ -2119,7 +1718,9 @@ async def _get_memory_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -2189,8 +1790,10 @@ async def _get_generate_memories_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetGenerateMemoriesOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetGenerateMemoriesOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2275,7 +1878,9 @@ async def _retrieve( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RetrieveMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:retrieve".format_map(request_url_dict) @@ -2377,8 +1982,10 @@ async def retrieve_profiles( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveMemoryProfilesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._RetrieveMemoryProfilesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2455,7 +2062,9 @@ async def _rollback( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RollbackMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RollbackMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:rollback".format_map(request_url_dict) @@ -2533,7 +2142,9 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMemoryRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMemoryRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2615,7 +2226,9 @@ async def _purge( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PurgeMemoriesRequestParameters_to_vertex(parameter_model) + request_dict = _converters._PurgeMemoriesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:purge".format_map(request_url_dict) diff --git a/agentplatform/_genai/memory_banks.py b/agentplatform/_genai/memory_banks.py index bddc4ea5ca..ef1de797f7 100644 --- a/agentplatform/_genai/memory_banks.py +++ b/agentplatform/_genai/memory_banks.py @@ -19,15 +19,14 @@ import json import logging import typing -from typing import Any, AsyncIterator, Iterator, Optional, Union +from typing import AsyncIterator, Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _memory_bank_utils from . import types @@ -40,594 +39,6 @@ logger = logging.getLogger("agentplatform_genai.memorybanks") -def _CreateMemoryBankConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - return to_object - - -def _CreateMemoryBankRequestParameters_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, ["memory_bank_config"]) is not None: - setv( - to_object, - ["context_spec", "memory_bank_config"], - _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( - getv(from_object, ["memory_bank_config"]), to_object - ), - ) - - if getv(from_object, ["config"]) is not None: - _CreateMemoryBankConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteMemoryBankRequestParameters_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, ["force"]) is not None: - setv(to_object, ["force"], getv(from_object, ["force"])) - - return to_object - - -def _GetMemoryBankOperationParameters_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 _GetMemoryBankRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _IngestEventsConfig_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, ["force_flush"]) is not None: - setv(parent_object, ["forceFlush"], getv(from_object, ["force_flush"])) - - if getv(from_object, ["revision_labels"]) is not None: - setv(parent_object, ["revisionLabels"], getv(from_object, ["revision_labels"])) - - if getv(from_object, ["revision_expire_time"]) is not None: - setv( - parent_object, - ["revisionExpireTime"], - getv(from_object, ["revision_expire_time"]), - ) - - if getv(from_object, ["revision_ttl"]) is not None: - setv(parent_object, ["revisionTtl"], getv(from_object, ["revision_ttl"])) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - parent_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["metadata"]) is not None: - setv( - parent_object, - ["metadata"], - {k: v for k, v in getv(from_object, ["metadata"]).items()}, - ) - - if getv(from_object, ["metadata_merge_strategy"]) is not None: - setv( - parent_object, - ["metadataMergeStrategy"], - getv(from_object, ["metadata_merge_strategy"]), - ) - - return to_object - - -def _IngestEventsRequestParameters_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, ["stream_id"]) is not None: - setv(to_object, ["streamId"], getv(from_object, ["stream_id"])) - - if getv(from_object, ["direct_contents_source"]) is not None: - setv( - to_object, - ["directContentsSource"], - getv(from_object, ["direct_contents_source"]), - ) - - if getv(from_object, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["generation_trigger_config"]) is not None: - setv( - to_object, - ["generationTriggerConfig"], - getv(from_object, ["generation_trigger_config"]), - ) - - if getv(from_object, ["config"]) is not None: - _IngestEventsConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _ListMemoryBanksConfig_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 _ListMemoryBanksRequestParameters_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: - _ListMemoryBanksConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _ListReasoningEnginesResponse_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, ["reasoningEngines"]) is not None: - setv( - to_object, - ["reasoning_engines"], - [ - _ReasoningEngine_from_vertex(item, to_object) - for item in getv(from_object, ["reasoningEngines"]) - ], - ) - - return to_object - - -def _ManagedSemanticMemoryConfig_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, ["generationConfig"]) is not None: - setv(to_object, ["generation_config"], getv(from_object, ["generationConfig"])) - - if getv(from_object, ["ttlConfig"]) is not None: - setv(to_object, ["ttl_config"], getv(from_object, ["ttlConfig"])) - - if getv(from_object, ["disableMemoryRevisions"]) is not None: - setv( - to_object, - ["disable_memory_revisions"], - getv(from_object, ["disableMemoryRevisions"]), - ) - - if getv(from_object, ["similaritySearchConfig"]) is not None: - setv( - to_object, - ["similarity_search_config"], - getv(from_object, ["similaritySearchConfig"]), - ) - - if getv(from_object, ["unstructuredMemoryConfigs"]) is not None: - setv( - to_object, - ["unstructured_memory_configs"], - [item for item in getv(from_object, ["unstructuredMemoryConfigs"])], - ) - - if getv(from_object, ["structuredMemoryConfigs"]) is not None: - setv( - to_object, - ["structured_memory_configs"], - [ - _StructuredMemoryConfig_from_vertex(item, to_object) - for item in getv(from_object, ["structuredMemoryConfigs"]) - ], - ) - - return to_object - - -def _MemoryBankOperation_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"], - _MemoryBank_from_vertex(getv(from_object, ["response"]), to_object), - ) - - return to_object - - -def _MemoryBank_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, ["managedSemanticMemoryConfig"]) is not None: - setv( - to_object, - ["managed_semantic_memory_config"], - _ManagedSemanticMemoryConfig_from_vertex( - getv(from_object, ["managedSemanticMemoryConfig"]), to_object - ), - ) - - if getv(from_object, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(from_object, ["displayName"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["createTime"]) is not None: - setv(to_object, ["create_time"], getv(from_object, ["createTime"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - if getv(from_object, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) - - return to_object - - -def _ReasoningEngineContextSpecMemoryBankConfig_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, ["customizationConfigs"]) is not None: - setv( - to_object, - ["customization_configs"], - [item for item in getv(from_object, ["customizationConfigs"])], - ) - - if getv(from_object, ["disableMemoryRevisions"]) is not None: - setv( - to_object, - ["disable_memory_revisions"], - getv(from_object, ["disableMemoryRevisions"]), - ) - - if getv(from_object, ["generationConfig"]) is not None: - setv(to_object, ["generation_config"], getv(from_object, ["generationConfig"])) - - if getv(from_object, ["similaritySearchConfig"]) is not None: - setv( - to_object, - ["similarity_search_config"], - getv(from_object, ["similaritySearchConfig"]), - ) - - if getv(from_object, ["ttlConfig"]) is not None: - setv(to_object, ["ttl_config"], getv(from_object, ["ttlConfig"])) - - if getv(from_object, ["structuredMemoryConfigs"]) is not None: - setv( - to_object, - ["structured_memory_configs"], - [ - _StructuredMemoryConfig_from_vertex(item, to_object) - for item in getv(from_object, ["structuredMemoryConfigs"]) - ], - ) - - return to_object - - -def _ReasoningEngineContextSpecMemoryBankConfig_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, ["customization_configs"]) is not None: - setv( - to_object, - ["customizationConfigs"], - [item for item in getv(from_object, ["customization_configs"])], - ) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - to_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["generation_config"]) is not None: - setv(to_object, ["generationConfig"], getv(from_object, ["generation_config"])) - - if getv(from_object, ["similarity_search_config"]) is not None: - setv( - to_object, - ["similaritySearchConfig"], - getv(from_object, ["similarity_search_config"]), - ) - - if getv(from_object, ["ttl_config"]) is not None: - setv(to_object, ["ttlConfig"], getv(from_object, ["ttl_config"])) - - if getv(from_object, ["structured_memory_configs"]) is not None: - setv( - to_object, - ["structuredMemoryConfigs"], - [ - _StructuredMemoryConfig_to_vertex(item, to_object) - for item in getv(from_object, ["structured_memory_configs"]) - ], - ) - - return to_object - - -def _ReasoningEngineContextSpec_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, ["memoryBankConfig"]) is not None: - setv( - to_object, - ["memory_bank_config"], - _ReasoningEngineContextSpecMemoryBankConfig_from_vertex( - getv(from_object, ["memoryBankConfig"]), to_object - ), - ) - - return to_object - - -def _ReasoningEngine_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, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) - - if getv(from_object, ["contextSpec"]) is not None: - setv( - to_object, - ["context_spec"], - _ReasoningEngineContextSpec_from_vertex( - getv(from_object, ["contextSpec"]), 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, ["etag"]) is not None: - setv(to_object, ["etag"], getv(from_object, ["etag"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - if getv(from_object, ["spec"]) is not None: - setv(to_object, ["spec"], getv(from_object, ["spec"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - if getv(from_object, ["trafficConfig"]) is not None: - setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"])) - - if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None: - setv( - to_object, - ["revision_garbage_collection_strategy"], - getv(from_object, ["revisionGarbageCollectionStrategy"]), - ) - - return to_object - - -def _StructuredMemoryConfig_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, ["schemaConfigs"]) is not None: - setv( - to_object, - ["schema_configs"], - [ - _StructuredMemorySchemaConfig_from_vertex(item, to_object) - for item in getv(from_object, ["schemaConfigs"]) - ], - ) - - if getv(from_object, ["scopeKeys"]) is not None: - setv(to_object, ["scope_keys"], getv(from_object, ["scopeKeys"])) - - return to_object - - -def _StructuredMemoryConfig_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, ["schema_configs"]) is not None: - setv( - to_object, - ["schemaConfigs"], - [ - _StructuredMemorySchemaConfig_to_vertex(item, to_object) - for item in getv(from_object, ["schema_configs"]) - ], - ) - - if getv(from_object, ["scope_keys"]) is not None: - setv(to_object, ["scopeKeys"], getv(from_object, ["scope_keys"])) - - return to_object - - -def _StructuredMemorySchemaConfig_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, ["schema"]) is not None: - setv(to_object, ["memory_schema"], getv(from_object, ["schema"])) - - if getv(from_object, ["id"]) is not None: - setv(to_object, ["id"], getv(from_object, ["id"])) - - if getv(from_object, ["memoryType"]) is not None: - setv(to_object, ["memory_type"], getv(from_object, ["memoryType"])) - - return to_object - - -def _StructuredMemorySchemaConfig_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, ["memory_schema"]) is not None: - setv(to_object, ["schema"], getv(from_object, ["memory_schema"])) - - if getv(from_object, ["id"]) is not None: - setv(to_object, ["id"], getv(from_object, ["id"])) - - if getv(from_object, ["memory_type"]) is not None: - setv(to_object, ["memoryType"], getv(from_object, ["memory_type"])) - - return to_object - - -def _UpdateMemoryBankConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - return to_object - - -def _UpdateMemoryBankRequestParameters_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, ["memory_bank_config"]) is not None: - setv( - to_object, - ["context_spec", "memory_bank_config"], - _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( - getv(from_object, ["memory_bank_config"]), to_object - ), - ) - - if getv(from_object, ["config"]) is not None: - _UpdateMemoryBankConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class MemoryBanks(_api_module.BaseModule): def _create( @@ -653,7 +64,9 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -681,7 +94,7 @@ def _create( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, @@ -732,7 +145,9 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -760,7 +175,7 @@ def _update( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, @@ -809,7 +224,9 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -878,7 +295,9 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -906,7 +325,7 @@ def _get( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ReasoningEngine_from_vertex(response_dict) + response_dict = _converters._ReasoningEngine_from_vertex(response_dict) return_value = types.ReasoningEngine._from_response( response=response_dict, @@ -965,7 +384,9 @@ def _ingest_events( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _IngestEventsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._IngestEventsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:ingestEvents".format_map(request_url_dict) @@ -1033,7 +454,9 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoryBanksRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListMemoryBanksRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -1061,7 +484,9 @@ def _list( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListReasoningEnginesResponse_from_vertex(response_dict) + response_dict = _converters._ListReasoningEnginesResponse_from_vertex( + response_dict + ) return_value = types.ListReasoningEnginesResponse._from_response( response=response_dict, @@ -1104,7 +529,9 @@ def _get_memory_bank_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryBankOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryBankOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -1132,7 +559,7 @@ def _get_memory_bank_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, @@ -1428,7 +855,9 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -1458,7 +887,7 @@ async def _create( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, @@ -1509,7 +938,9 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1539,7 +970,7 @@ async def _update( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, @@ -1588,7 +1019,9 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1659,7 +1092,9 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryBankRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryBankRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1689,7 +1124,7 @@ async def _get( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ReasoningEngine_from_vertex(response_dict) + response_dict = _converters._ReasoningEngine_from_vertex(response_dict) return_value = types.ReasoningEngine._from_response( response=response_dict, @@ -1748,7 +1183,9 @@ async def _ingest_events( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _IngestEventsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._IngestEventsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/memories:ingestEvents".format_map(request_url_dict) @@ -1818,7 +1255,9 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoryBanksRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListMemoryBanksRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -1848,7 +1287,9 @@ async def _list( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListReasoningEnginesResponse_from_vertex(response_dict) + response_dict = _converters._ListReasoningEnginesResponse_from_vertex( + response_dict + ) return_value = types.ListReasoningEnginesResponse._from_response( response=response_dict, @@ -1891,7 +1332,9 @@ async def _get_memory_bank_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryBankOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetMemoryBankOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -1921,7 +1364,7 @@ async def _get_memory_bank_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _MemoryBankOperation_from_vertex(response_dict) + response_dict = _converters._MemoryBankOperation_from_vertex(response_dict) return_value = types.MemoryBankOperation._from_response( response=response_dict, diff --git a/agentplatform/_genai/memory_revisions.py b/agentplatform/_genai/memory_revisions.py index 73917a973b..c4de734220 100644 --- a/agentplatform/_genai/memory_revisions.py +++ b/agentplatform/_genai/memory_revisions.py @@ -18,63 +18,19 @@ import functools import json import logging -from typing import Any, Iterator, Optional, Union +from typing import Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import types logger = logging.getLogger("agentplatform_genai.memoryrevisions") -def _GetMemoryRevisionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListMemoryRevisionsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListMemoryRevisionsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListMemoryRevisionsConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class MemoryRevisions(_api_module.BaseModule): def get( @@ -105,7 +61,7 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryRevisionRequestParameters_to_vertex( + request_dict = _converters._GetMemoryRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -189,7 +145,7 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoryRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListMemoryRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -298,7 +254,7 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetMemoryRevisionRequestParameters_to_vertex( + request_dict = _converters._GetMemoryRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -384,7 +340,7 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListMemoryRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListMemoryRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/model_garden.py b/agentplatform/_genai/model_garden.py index 70d8cbade7..e2732b55e8 100644 --- a/agentplatform/_genai/model_garden.py +++ b/agentplatform/_genai/model_garden.py @@ -23,447 +23,14 @@ from google.genai import _api_module from google.genai import _common 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 _converters from . import _operations_utils from . import types logger = logging.getLogger("agentplatform_genai.modelgarden") -def _CreateBatchPredictionJobConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["input_config"]) is not None: - setv( - parent_object, - ["inputConfig"], - _ModelGardenBatchJobSource_to_vertex( - getv(from_object, ["input_config"]), to_object - ), - ) - - if getv(from_object, ["output_config"]) is not None: - setv( - parent_object, - ["outputConfig"], - _ModelGardenBatchJobDestination_to_vertex( - getv(from_object, ["output_config"]), to_object - ), - ) - - if getv(from_object, ["dedicated_resources"]) is not None: - setv( - parent_object, - ["dedicatedResources"], - getv(from_object, ["dedicated_resources"]), - ) - - return to_object - - -def _CreateBatchPredictionJobParameters_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, ["parent"]) is not None: - setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["model"]) is not None: - setv(to_object, ["model"], getv(from_object, ["model"])) - - if getv(from_object, ["config"]) is not None: - _CreateBatchPredictionJobConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _DeployRequestParameters_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, ["destination"]) is not None: - setv(to_object, ["_url", "destination"], getv(from_object, ["destination"])) - - if getv(from_object, ["publisher_model_name"]) is not None: - setv( - to_object, - ["publisherModelName"], - getv(from_object, ["publisher_model_name"]), - ) - - if getv(from_object, ["hugging_face_model_id"]) is not None: - setv( - to_object, - ["huggingFaceModelId"], - getv(from_object, ["hugging_face_model_id"]), - ) - - if getv(from_object, ["custom_model"]) is not None: - setv(to_object, ["customModel"], getv(from_object, ["custom_model"])) - - if getv(from_object, ["model_config_val"]) is not None: - setv(to_object, ["modelConfig"], getv(from_object, ["model_config_val"])) - - if getv(from_object, ["endpoint_config"]) is not None: - setv(to_object, ["endpointConfig"], getv(from_object, ["endpoint_config"])) - - if getv(from_object, ["deploy_config"]) is not None: - setv(to_object, ["deployConfig"], getv(from_object, ["deploy_config"])) - - return to_object - - -def _ExportPublisherModelConfig_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, ["destination"]) is not None: - setv(parent_object, ["destination"], getv(from_object, ["destination"])) - - return to_object - - -def _ExportPublisherModelRequestParameters_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, ["parent"]) is not None: - setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ExportPublisherModelConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _GetBatchPredictionJobParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetDeployOperationParameters_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 _GetExportPublisherModelOperationParameters_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 _GetPublisherModelConfig_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, ["hugging_face_token"]) is not None: - setv( - parent_object, - ["_query", "huggingFaceToken"], - getv(from_object, ["hugging_face_token"]), - ) - - if ( - getv(from_object, ["include_equivalent_model_garden_model_deployment_configs"]) - is not None - ): - setv( - parent_object, - ["_query", "includeEquivalentModelGardenModelDeploymentConfigs"], - getv( - from_object, - ["include_equivalent_model_garden_model_deployment_configs"], - ), - ) - - if getv(from_object, ["is_hugging_face_model"]) is not None: - setv( - parent_object, - ["_query", "isHuggingFaceModel"], - getv(from_object, ["is_hugging_face_model"]), - ) - - return to_object - - -def _GetPublisherModelRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _GetPublisherModelConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _ListPublisherModelsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - if getv(from_object, ["list_all_versions"]) is not None: - setv( - parent_object, - ["_query", "listAllVersions"], - getv(from_object, ["list_all_versions"]), - ) - - return to_object - - -def _ListPublisherModelsRequestParameters_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, ["parent"]) is not None: - setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["config"]) is not None: - _ListPublisherModelsConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _ModelGardenBatchJobDestination_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, ["predictionsFormat"]) is not None: - setv(to_object, ["format"], getv(from_object, ["predictionsFormat"])) - - if getv(from_object, ["gcsDestination", "outputUriPrefix"]) is not None: - setv( - to_object, - ["gcs_uri"], - getv(from_object, ["gcsDestination", "outputUriPrefix"]), - ) - - if getv(from_object, ["bigqueryDestination", "outputUri"]) is not None: - setv( - to_object, - ["bigquery_uri"], - getv(from_object, ["bigqueryDestination", "outputUri"]), - ) - - return to_object - - -def _ModelGardenBatchJobDestination_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, ["format"]) is not None: - setv(to_object, ["predictionsFormat"], getv(from_object, ["format"])) - - if getv(from_object, ["gcs_uri"]) is not None: - setv( - to_object, - ["gcsDestination", "outputUriPrefix"], - getv(from_object, ["gcs_uri"]), - ) - - if getv(from_object, ["bigquery_uri"]) is not None: - setv( - to_object, - ["bigqueryDestination", "outputUri"], - getv(from_object, ["bigquery_uri"]), - ) - - return to_object - - -def _ModelGardenBatchJobSource_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, ["instancesFormat"]) is not None: - setv(to_object, ["format"], getv(from_object, ["instancesFormat"])) - - if getv(from_object, ["gcsSource", "uris"]) is not None: - setv(to_object, ["gcs_uri"], getv(from_object, ["gcsSource", "uris"])) - - if getv(from_object, ["bigquerySource", "inputUri"]) is not None: - setv( - to_object, - ["bigquery_uri"], - getv(from_object, ["bigquerySource", "inputUri"]), - ) - - return to_object - - -def _ModelGardenBatchJobSource_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, ["format"]) is not None: - setv(to_object, ["instancesFormat"], getv(from_object, ["format"])) - - if getv(from_object, ["gcs_uri"]) is not None: - setv(to_object, ["gcsSource", "uris"], getv(from_object, ["gcs_uri"])) - - if getv(from_object, ["bigquery_uri"]) is not None: - setv( - to_object, - ["bigquerySource", "inputUri"], - getv(from_object, ["bigquery_uri"]), - ) - - return to_object - - -def _ModelGardenBatchPredictionJob_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, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(from_object, ["displayName"])) - - if getv(from_object, ["model"]) is not None: - setv(to_object, ["model"], getv(from_object, ["model"])) - - if getv(from_object, ["inputConfig"]) is not None: - setv( - to_object, - ["input_config"], - _ModelGardenBatchJobSource_from_vertex( - getv(from_object, ["inputConfig"]), to_object - ), - ) - - if getv(from_object, ["outputConfig"]) is not None: - setv( - to_object, - ["output_config"], - _ModelGardenBatchJobDestination_from_vertex( - getv(from_object, ["outputConfig"]), to_object - ), - ) - - if getv(from_object, ["outputInfo"]) is not None: - setv(to_object, ["output_info"], getv(from_object, ["outputInfo"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(from_object, ["error"]) is not None: - setv(to_object, ["error"], getv(from_object, ["error"])) - - if getv(from_object, ["createTime"]) is not None: - setv(to_object, ["create_time"], getv(from_object, ["createTime"])) - - if getv(from_object, ["startTime"]) is not None: - setv(to_object, ["start_time"], getv(from_object, ["startTime"])) - - if getv(from_object, ["endTime"]) is not None: - setv(to_object, ["end_time"], getv(from_object, ["endTime"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - return to_object - - -def _RecommendSpecConfig_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, ["check_machine_availability"]) is not None: - setv( - parent_object, - ["checkMachineAvailability"], - getv(from_object, ["check_machine_availability"]), - ) - - if getv(from_object, ["check_user_quota"]) is not None: - setv(parent_object, ["checkUserQuota"], getv(from_object, ["check_user_quota"])) - - return to_object - - -def _RecommendSpecRequestParameters_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, ["parent"]) is not None: - setv(to_object, ["_url", "parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["gcs_uri"]) is not None: - setv(to_object, ["gcsUri"], getv(from_object, ["gcs_uri"])) - - if getv(from_object, ["config"]) is not None: - _RecommendSpecConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class ModelGarden(_api_module.BaseModule): """Model Garden module.""" @@ -488,7 +55,7 @@ def _list_publisher_models( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListPublisherModelsRequestParameters_to_vertex( + request_dict = _converters._ListPublisherModelsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -559,7 +126,7 @@ def _get_publisher_model( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetPublisherModelRequestParameters_to_vertex( + request_dict = _converters._GetPublisherModelRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -635,7 +202,9 @@ def _recommend_spec( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RecommendSpecRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RecommendSpecRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{parent}:recommendSpec".format_map(request_url_dict) @@ -709,7 +278,7 @@ def _export_publisher_model( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ExportPublisherModelRequestParameters_to_vertex( + request_dict = _converters._ExportPublisherModelRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -783,8 +352,10 @@ def get_export_publisher_model_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExportPublisherModelOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetExportPublisherModelOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -869,7 +440,9 @@ def _deploy( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeployRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeployRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{destination}:deploy".format_map(request_url_dict) @@ -941,7 +514,9 @@ def get_deploy_publisher_model_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDeployOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetDeployOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -1015,7 +590,7 @@ def _create_batch_prediction_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateBatchPredictionJobParameters_to_vertex( + request_dict = _converters._CreateBatchPredictionJobParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1045,7 +620,9 @@ def _create_batch_prediction_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ModelGardenBatchPredictionJob_from_vertex(response_dict) + response_dict = _converters._ModelGardenBatchPredictionJob_from_vertex( + response_dict + ) return_value = types.ModelGardenBatchPredictionJob._from_response( response=response_dict, @@ -1092,7 +669,9 @@ def get_batch_prediction_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetBatchPredictionJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetBatchPredictionJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1120,7 +699,9 @@ def get_batch_prediction_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ModelGardenBatchPredictionJob_from_vertex(response_dict) + response_dict = _converters._ModelGardenBatchPredictionJob_from_vertex( + response_dict + ) return_value = types.ModelGardenBatchPredictionJob._from_response( response=response_dict, @@ -2492,7 +2073,7 @@ async def _list_publisher_models( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListPublisherModelsRequestParameters_to_vertex( + request_dict = _converters._ListPublisherModelsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2565,7 +2146,7 @@ async def _get_publisher_model( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetPublisherModelRequestParameters_to_vertex( + request_dict = _converters._GetPublisherModelRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2643,7 +2224,9 @@ async def _recommend_spec( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RecommendSpecRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RecommendSpecRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{parent}:recommendSpec".format_map(request_url_dict) @@ -2719,7 +2302,7 @@ async def _export_publisher_model( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ExportPublisherModelRequestParameters_to_vertex( + request_dict = _converters._ExportPublisherModelRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2795,8 +2378,10 @@ async def get_export_publisher_model_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetExportPublisherModelOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetExportPublisherModelOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2883,7 +2468,9 @@ async def _deploy( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeployRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeployRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{destination}:deploy".format_map(request_url_dict) @@ -2957,7 +2544,9 @@ async def get_deploy_publisher_model_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDeployOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetDeployOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -3033,7 +2622,7 @@ async def _create_batch_prediction_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateBatchPredictionJobParameters_to_vertex( + request_dict = _converters._CreateBatchPredictionJobParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -3065,7 +2654,9 @@ async def _create_batch_prediction_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ModelGardenBatchPredictionJob_from_vertex(response_dict) + response_dict = _converters._ModelGardenBatchPredictionJob_from_vertex( + response_dict + ) return_value = types.ModelGardenBatchPredictionJob._from_response( response=response_dict, @@ -3112,7 +2703,9 @@ async def get_batch_prediction_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetBatchPredictionJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetBatchPredictionJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3142,7 +2735,9 @@ async def get_batch_prediction_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ModelGardenBatchPredictionJob_from_vertex(response_dict) + response_dict = _converters._ModelGardenBatchPredictionJob_from_vertex( + response_dict + ) return_value = types.ModelGardenBatchPredictionJob._from_response( response=response_dict, diff --git a/agentplatform/_genai/prompt_optimizer.py b/agentplatform/_genai/prompt_optimizer.py index 3aab5f5666..62325c6866 100644 --- a/agentplatform/_genai/prompt_optimizer.py +++ b/agentplatform/_genai/prompt_optimizer.py @@ -18,15 +18,14 @@ import json import logging import time -from typing import Any, Optional, Union +from typing import Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common 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 _converters from . import _logging_utils from . import _prompt_optimizer_utils from . import prompts @@ -35,172 +34,6 @@ logger = logging.getLogger("agentplatform_genai.promptoptimizer") -def _CustomJobParameters_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_job"]) is not None: - setv( - parent_object, - ["customJob"], - _CustomJob_to_vertex(getv(from_object, ["custom_job"]), to_object), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CustomJob_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(parent_object, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(parent_object, ["displayName"])) - - if getv(parent_object, ["jobSpec"]) is not None: - setv(to_object, ["job_spec"], getv(parent_object, ["jobSpec"])) - - if getv(parent_object, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(parent_object, ["encryptionSpec"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(parent_object, ["error"]) is not None: - setv(to_object, ["error"], getv(parent_object, ["error"])) - - if getv(from_object, ["createTime"]) is not None: - setv(to_object, ["create_time"], getv(from_object, ["createTime"])) - - if getv(from_object, ["endTime"]) is not None: - setv(to_object, ["end_time"], getv(from_object, ["endTime"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - 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, ["startTime"]) is not None: - setv(to_object, ["start_time"], getv(from_object, ["startTime"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - if getv(from_object, ["webAccessUris"]) is not None: - setv(to_object, ["web_access_uris"], getv(from_object, ["webAccessUris"])) - - return to_object - - -def _CustomJob_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["job_spec"]) is not None: - setv(parent_object, ["jobSpec"], getv(from_object, ["job_spec"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(from_object, ["error"]) is not None: - setv(parent_object, ["error"], getv(from_object, ["error"])) - - if getv(from_object, ["create_time"]) is not None: - setv(to_object, ["createTime"], getv(from_object, ["create_time"])) - - if getv(from_object, ["end_time"]) is not None: - setv(to_object, ["endTime"], getv(from_object, ["end_time"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - 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, ["start_time"]) is not None: - setv(to_object, ["startTime"], getv(from_object, ["start_time"])) - - if getv(from_object, ["update_time"]) is not None: - setv(to_object, ["updateTime"], getv(from_object, ["update_time"])) - - if getv(from_object, ["web_access_uris"]) is not None: - setv(to_object, ["webAccessUris"], getv(from_object, ["web_access_uris"])) - - return to_object - - -def _GetCustomJobParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _OptimizeConfig_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, ["optimization_target"]) is not None: - setv( - parent_object, - ["optimizationTarget"], - getv(from_object, ["optimization_target"]), - ) - - return to_object - - -def _OptimizeRequestParameters_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, ["content"]) is not None: - setv(to_object, ["content"], getv(from_object, ["content"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _OptimizeConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - class PromptOptimizer(_api_module.BaseModule): """Prompt Optimizer""" @@ -225,7 +58,9 @@ def _optimize_prompt( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._OptimizeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "tuningJobs:optimizePrompt".format_map(request_url_dict) @@ -297,7 +132,7 @@ def _create_custom_job_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs".format_map(request_url_dict) @@ -325,7 +160,7 @@ def _create_custom_job_resource( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -369,7 +204,9 @@ def _get_custom_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetCustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetCustomJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs/{name}".format_map(request_url_dict) @@ -397,7 +234,7 @@ def _get_custom_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -635,7 +472,9 @@ async def _optimize_prompt( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._OptimizeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "tuningJobs:optimizePrompt".format_map(request_url_dict) @@ -709,7 +548,7 @@ async def _create_custom_job_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs".format_map(request_url_dict) @@ -739,7 +578,7 @@ async def _create_custom_job_resource( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -783,7 +622,9 @@ async def _get_custom_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetCustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetCustomJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs/{name}".format_map(request_url_dict) @@ -813,7 +654,7 @@ async def _get_custom_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, diff --git a/agentplatform/_genai/prompts.py b/agentplatform/_genai/prompts.py index 43e1ac1944..0aba4c0ee1 100644 --- a/agentplatform/_genai/prompts.py +++ b/agentplatform/_genai/prompts.py @@ -20,18 +20,17 @@ import json import logging import time -from typing import Any, AsyncIterator, Iterator, Optional, Union +from typing import AsyncIterator, Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common from google.genai import operations 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 google.genai.pagers import AsyncPager, Pager from pydantic import ValidationError +from . import _converters from . import _prompt_management_utils from . import _prompt_optimizer_utils from . import types @@ -39,418 +38,6 @@ logger = logging.getLogger("agentplatform_genai.prompts") -def _CreateDatasetParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["metadata_schema_uri"]) is not None: - setv( - to_object, ["metadataSchemaUri"], getv(from_object, ["metadata_schema_uri"]) - ) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["model_reference"]) is not None: - setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CreateDatasetVersionParameters_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, ["dataset_name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["dataset_name"])) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["model_reference"]) is not None: - setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) - - if getv(from_object, ["parent"]) is not None: - setv(to_object, ["parent"], getv(from_object, ["parent"])) - - if getv(from_object, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CustomJobParameters_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_job"]) is not None: - setv( - parent_object, - ["customJob"], - _CustomJob_to_vertex(getv(from_object, ["custom_job"]), to_object), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _CustomJob_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(parent_object, ["displayName"]) is not None: - setv(to_object, ["display_name"], getv(parent_object, ["displayName"])) - - if getv(parent_object, ["jobSpec"]) is not None: - setv(to_object, ["job_spec"], getv(parent_object, ["jobSpec"])) - - if getv(parent_object, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(parent_object, ["encryptionSpec"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(parent_object, ["error"]) is not None: - setv(to_object, ["error"], getv(parent_object, ["error"])) - - if getv(from_object, ["createTime"]) is not None: - setv(to_object, ["create_time"], getv(from_object, ["createTime"])) - - if getv(from_object, ["endTime"]) is not None: - setv(to_object, ["end_time"], getv(from_object, ["endTime"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - 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, ["startTime"]) is not None: - setv(to_object, ["start_time"], getv(from_object, ["startTime"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - if getv(from_object, ["webAccessUris"]) is not None: - setv(to_object, ["web_access_uris"], getv(from_object, ["webAccessUris"])) - - return to_object - - -def _CustomJob_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["job_spec"]) is not None: - setv(parent_object, ["jobSpec"], getv(from_object, ["job_spec"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["state"]) is not None: - setv(to_object, ["state"], getv(from_object, ["state"])) - - if getv(from_object, ["error"]) is not None: - setv(parent_object, ["error"], getv(from_object, ["error"])) - - if getv(from_object, ["create_time"]) is not None: - setv(to_object, ["createTime"], getv(from_object, ["create_time"])) - - if getv(from_object, ["end_time"]) is not None: - setv(to_object, ["endTime"], getv(from_object, ["end_time"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - 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, ["start_time"]) is not None: - setv(to_object, ["startTime"], getv(from_object, ["start_time"])) - - if getv(from_object, ["update_time"]) is not None: - setv(to_object, ["updateTime"], getv(from_object, ["update_time"])) - - if getv(from_object, ["web_access_uris"]) is not None: - setv(to_object, ["webAccessUris"], getv(from_object, ["web_access_uris"])) - - return to_object - - -def _DeleteDatasetRequestParameters_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, ["prompt_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["prompt_id"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _DeletePromptVersionRequestParameters_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, ["prompt_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["prompt_id"])) - - if getv(from_object, ["version_id"]) is not None: - setv(to_object, ["_url", "version_id"], getv(from_object, ["version_id"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetCustomJobParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetDatasetOperationParameters_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, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["operation_id"]) is not None: - setv(to_object, ["_url", "operation_id"], getv(from_object, ["operation_id"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetDatasetParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _GetDatasetVersionParameters_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, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["dataset_version_id"]) is not None: - setv( - to_object, - ["_url", "dataset_version_id"], - getv(from_object, ["dataset_version_id"]), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListDatasetVersionsRequestParameters_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, ["read_mask"]) is not None: - setv(to_object, ["_url", "read_mask"], getv(from_object, ["read_mask"])) - - if getv(from_object, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListPromptsConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - -def _ListDatasetsRequestParameters_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"], - _ListPromptsConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - -def _ListPromptsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _OptimizeConfig_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, ["optimization_target"]) is not None: - setv( - parent_object, - ["optimizationTarget"], - getv(from_object, ["optimization_target"]), - ) - - return to_object - - -def _OptimizeRequestParameters_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, ["content"]) is not None: - setv(to_object, ["content"], getv(from_object, ["content"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _OptimizeConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - -def _RestoreVersionRequestParameters_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, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["version_id"]) is not None: - setv(to_object, ["_url", "version_id"], getv(from_object, ["version_id"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _UpdateDatasetParameters_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, ["dataset_id"]) is not None: - setv(to_object, ["_url", "dataset_id"], getv(from_object, ["dataset_id"])) - - if getv(from_object, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["metadata"]) is not None: - setv(to_object, ["metadata"], getv(from_object, ["metadata"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(to_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["model_reference"]) is not None: - setv(to_object, ["modelReference"], getv(from_object, ["model_reference"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - class Prompts(_api_module.BaseModule): def _create_dataset_resource( @@ -486,7 +73,9 @@ def _create_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -566,7 +155,9 @@ def _create_dataset_version_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateDatasetVersionParameters_to_vertex(parameter_model) + request_dict = _converters._CreateDatasetVersionParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{name}/datasetVersions".format_map(request_url_dict) @@ -638,7 +229,7 @@ def _get_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{name}".format_map(request_url_dict) @@ -712,7 +303,9 @@ def _get_dataset_version_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetVersionParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetVersionParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/datasetVersions/{dataset_version_id}".format_map( @@ -788,7 +381,9 @@ def _get_dataset_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/operations/{operation_id}".format_map( @@ -858,7 +453,9 @@ def _list_prompts( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListDatasetsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListDatasetsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -932,7 +529,7 @@ def _list_versions( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListDatasetVersionsRequestParameters_to_vertex( + request_dict = _converters._ListDatasetVersionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1001,7 +598,9 @@ def _delete_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteDatasetRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteDatasetRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}".format_map(request_url_dict) @@ -1071,7 +670,7 @@ def _delete_dataset_version( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeletePromptVersionRequestParameters_to_vertex( + request_dict = _converters._DeletePromptVersionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1149,7 +748,9 @@ def _restore_version( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RestoreVersionRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RestoreVersionRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/datasetVersions/{version_id}:restore".format_map( @@ -1235,7 +836,9 @@ def _update_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}".format_map(request_url_dict) @@ -1307,7 +910,7 @@ def _create_custom_job_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs".format_map(request_url_dict) @@ -1335,7 +938,7 @@ def _create_custom_job_resource( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -1379,7 +982,9 @@ def _get_custom_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetCustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetCustomJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs/{name}".format_map(request_url_dict) @@ -1407,7 +1012,7 @@ def _get_custom_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -1454,7 +1059,9 @@ def _optimize( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._OptimizeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "tuningJobs:optimizePrompt".format_map(request_url_dict) @@ -2523,7 +2130,9 @@ async def _create_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._CreateDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -2605,7 +2214,9 @@ async def _create_dataset_version_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateDatasetVersionParameters_to_vertex(parameter_model) + request_dict = _converters._CreateDatasetVersionParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{name}/datasetVersions".format_map(request_url_dict) @@ -2679,7 +2290,7 @@ async def _get_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{name}".format_map(request_url_dict) @@ -2755,7 +2366,9 @@ async def _get_dataset_version_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetVersionParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetVersionParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/datasetVersions/{dataset_version_id}".format_map( @@ -2833,7 +2446,9 @@ async def _get_dataset_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDatasetOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetDatasetOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/operations/{operation_id}".format_map( @@ -2905,7 +2520,9 @@ async def _list_prompts( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListDatasetsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListDatasetsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets".format_map(request_url_dict) @@ -2981,7 +2598,7 @@ async def _list_versions( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListDatasetVersionsRequestParameters_to_vertex( + request_dict = _converters._ListDatasetVersionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -3052,7 +2669,9 @@ async def _delete_dataset( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteDatasetRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteDatasetRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}".format_map(request_url_dict) @@ -3124,7 +2743,7 @@ async def _delete_dataset_version( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeletePromptVersionRequestParameters_to_vertex( + request_dict = _converters._DeletePromptVersionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -3204,7 +2823,9 @@ async def _restore_version( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RestoreVersionRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RestoreVersionRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}/datasetVersions/{version_id}:restore".format_map( @@ -3292,7 +2913,9 @@ async def _update_dataset_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateDatasetParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateDatasetParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "datasets/{dataset_id}".format_map(request_url_dict) @@ -3366,7 +2989,7 @@ async def _create_custom_job_resource( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs".format_map(request_url_dict) @@ -3396,7 +3019,7 @@ async def _create_custom_job_resource( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -3440,7 +3063,9 @@ async def _get_custom_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetCustomJobParameters_to_vertex(parameter_model) + request_dict = _converters._GetCustomJobParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "customJobs/{name}".format_map(request_url_dict) @@ -3470,7 +3095,7 @@ async def _get_custom_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CustomJob_from_vertex(response_dict) + response_dict = _converters._CustomJob_from_vertex(response_dict) return_value = types.CustomJob._from_response( response=response_dict, @@ -3517,7 +3142,9 @@ async def _optimize( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._OptimizeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "tuningJobs:optimizePrompt".format_map(request_url_dict) diff --git a/agentplatform/_genai/rag.py b/agentplatform/_genai/rag.py index 7414326f2c..e68bd0362f 100644 --- a/agentplatform/_genai/rag.py +++ b/agentplatform/_genai/rag.py @@ -19,16 +19,15 @@ import logging import mimetypes import os -from typing import Any, Optional, Union +from typing import Optional 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 _converters from . import _gcs_utils from . import _operations_utils from . import types @@ -36,1612 +35,6 @@ 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, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - 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, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - 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, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - 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, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - 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, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListRagFilesConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - 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, ["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), - ) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - 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( @@ -1673,7 +66,9 @@ def ask_contexts( "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_dict = _converters._AskContextsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":askContexts".format_map(request_url_dict) @@ -1745,7 +140,9 @@ def _create_corpus( "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_dict = _converters._CreateRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragCorpora".format_map(request_url_dict) @@ -1813,7 +210,9 @@ def _get_corpus_operation( "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_dict = _converters._GetCorpusOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -1841,7 +240,7 @@ def _get_corpus_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CorpusOperation_from_vertex(response_dict) + response_dict = _converters._CorpusOperation_from_vertex(response_dict) return_value = types.CorpusOperation._from_response( response=response_dict, @@ -1893,7 +292,9 @@ def get_corpus( "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_dict = _converters._GetRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1921,7 +322,7 @@ def get_corpus( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RagCorpus_from_vertex(response_dict) + response_dict = _converters._RagCorpus_from_vertex(response_dict) return_value = types.RagCorpus._from_response( response=response_dict, @@ -1971,7 +372,9 @@ def list_corpora( "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_dict = _converters._ListRagCorporaRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragCorpora".format_map(request_url_dict) @@ -1999,7 +402,9 @@ def list_corpora( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListRagCorporaResponse_from_vertex(response_dict) + response_dict = _converters._ListRagCorporaResponse_from_vertex( + response_dict + ) return_value = types.ListRagCorporaResponse._from_response( response=response_dict, @@ -2051,7 +456,9 @@ def get_file( "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_dict = _converters._GetRagFileRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2128,7 +535,9 @@ def list_files( "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_dict = _converters._ListRagFilesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/ragFiles".format_map(request_url_dict) @@ -2203,7 +612,9 @@ def get_config( "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_dict = _converters._GetRagConfigRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragEngineConfig".format_map(request_url_dict) @@ -2231,7 +642,7 @@ def get_config( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RagEngineConfig_from_vertex(response_dict) + response_dict = _converters._RagEngineConfig_from_vertex(response_dict) return_value = types.RagEngineConfig._from_response( response=response_dict, @@ -2280,7 +691,9 @@ def _update_corpus( "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_dict = _converters._UpdateRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2352,7 +765,9 @@ def _delete_corpus( "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_dict = _converters._DeleteRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2421,7 +836,9 @@ def _delete_file( "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_dict = _converters._DeleteRagFileRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -2493,7 +910,9 @@ def _update_config( "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_dict = _converters._UpdateRagConfigRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragEngineConfig".format_map(request_url_dict) @@ -2576,7 +995,7 @@ def retrieve_contexts( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveRagContextsRequestParameters_to_vertex( + request_dict = _converters._RetrieveRagContextsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2646,7 +1065,9 @@ def _get_rag_config_operation( "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_dict = _converters._GetRagConfigOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operation_name}".format_map(request_url_dict) @@ -2720,7 +1141,9 @@ def _import_files( "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_dict = _converters._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) @@ -2788,7 +1211,9 @@ def _get_import_files_operation( "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_dict = _converters._GetImportFilesOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operation_name}".format_map(request_url_dict) @@ -2860,7 +1285,9 @@ def _upload_file( "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_dict = _converters._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) @@ -3315,7 +1742,9 @@ async def ask_contexts( "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_dict = _converters._AskContextsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = ":askContexts".format_map(request_url_dict) @@ -3389,7 +1818,9 @@ async def _create_corpus( "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_dict = _converters._CreateRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragCorpora".format_map(request_url_dict) @@ -3459,7 +1890,9 @@ async def _get_corpus_operation( "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_dict = _converters._GetCorpusOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -3489,7 +1922,7 @@ async def _get_corpus_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CorpusOperation_from_vertex(response_dict) + response_dict = _converters._CorpusOperation_from_vertex(response_dict) return_value = types.CorpusOperation._from_response( response=response_dict, @@ -3541,7 +1974,9 @@ async def get_corpus( "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_dict = _converters._GetRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3571,7 +2006,7 @@ async def get_corpus( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RagCorpus_from_vertex(response_dict) + response_dict = _converters._RagCorpus_from_vertex(response_dict) return_value = types.RagCorpus._from_response( response=response_dict, @@ -3621,7 +2056,9 @@ async def list_corpora( "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_dict = _converters._ListRagCorporaRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragCorpora".format_map(request_url_dict) @@ -3651,7 +2088,9 @@ async def list_corpora( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ListRagCorporaResponse_from_vertex(response_dict) + response_dict = _converters._ListRagCorporaResponse_from_vertex( + response_dict + ) return_value = types.ListRagCorporaResponse._from_response( response=response_dict, @@ -3703,7 +2142,9 @@ async def get_file( "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_dict = _converters._GetRagFileRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3782,7 +2223,9 @@ async def list_files( "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_dict = _converters._ListRagFilesRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}/ragFiles".format_map(request_url_dict) @@ -3859,7 +2302,9 @@ async def get_config( "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_dict = _converters._GetRagConfigRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragEngineConfig".format_map(request_url_dict) @@ -3889,7 +2334,7 @@ async def get_config( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RagEngineConfig_from_vertex(response_dict) + response_dict = _converters._RagEngineConfig_from_vertex(response_dict) return_value = types.RagEngineConfig._from_response( response=response_dict, @@ -3938,7 +2383,9 @@ async def _update_corpus( "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_dict = _converters._UpdateRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -4012,7 +2459,9 @@ async def _delete_corpus( "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_dict = _converters._DeleteRagCorpusRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -4083,7 +2532,9 @@ async def _delete_file( "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_dict = _converters._DeleteRagFileRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -4157,7 +2608,9 @@ async def _update_config( "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_dict = _converters._UpdateRagConfigRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "ragEngineConfig".format_map(request_url_dict) @@ -4242,7 +2695,7 @@ async def retrieve_contexts( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveRagContextsRequestParameters_to_vertex( + request_dict = _converters._RetrieveRagContextsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -4314,7 +2767,9 @@ async def _get_rag_config_operation( "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_dict = _converters._GetRagConfigOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operation_name}".format_map(request_url_dict) @@ -4390,7 +2845,9 @@ async def _import_files( "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_dict = _converters._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) @@ -4460,7 +2917,9 @@ async def _get_import_files_operation( "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_dict = _converters._GetImportFilesOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operation_name}".format_map(request_url_dict) @@ -4534,7 +2993,9 @@ async def _upload_file( "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_dict = _converters._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) diff --git a/agentplatform/_genai/runtime_revisions.py b/agentplatform/_genai/runtime_revisions.py index bb7476b159..c0197faef5 100644 --- a/agentplatform/_genai/runtime_revisions.py +++ b/agentplatform/_genai/runtime_revisions.py @@ -18,15 +18,14 @@ import functools import json import logging -from typing import Any, AsyncIterator, Iterator, Optional, Union +from typing import Any, AsyncIterator, Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _runtimes_utils from . import types @@ -35,105 +34,6 @@ logger.setLevel(logging.INFO) -def _DeleteRuntimeRevisionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetDeleteRuntimeRevisionOperationParameters_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 _GetRuntimeRevisionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListRuntimeRevisionsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListRuntimeRevisionsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListRuntimeRevisionsConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _QueryRuntimeRevisionConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["class_method"]) is not None: - setv(parent_object, ["classMethod"], getv(from_object, ["class_method"])) - - if getv(from_object, ["input"]) is not None: - setv(parent_object, ["input"], getv(from_object, ["input"])) - - if getv(from_object, ["include_all_fields"]) is not None: - setv(to_object, ["includeAllFields"], getv(from_object, ["include_all_fields"])) - - return to_object - - -def _QueryRuntimeRevisionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _QueryRuntimeRevisionConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class RuntimeRevisions(_api_module.BaseModule): def _get( @@ -157,7 +57,7 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeRevisionRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -241,7 +141,7 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -327,8 +227,10 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeRevisionRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteRuntimeRevisionRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -397,8 +299,10 @@ def _get_delete_runtime_revision_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDeleteRuntimeRevisionOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetDeleteRuntimeRevisionOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -471,7 +375,7 @@ def _query( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _QueryRuntimeRevisionRequestParameters_to_vertex( + request_dict = _converters._QueryRuntimeRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -741,7 +645,7 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeRevisionRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -827,7 +731,7 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -915,8 +819,10 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeRevisionRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._DeleteRuntimeRevisionRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -987,8 +893,10 @@ async def _get_delete_runtime_revision_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetDeleteRuntimeRevisionOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetDeleteRuntimeRevisionOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1063,7 +971,7 @@ async def _query( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _QueryRuntimeRevisionRequestParameters_to_vertex( + request_dict = _converters._QueryRuntimeRevisionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/runtimes.py b/agentplatform/_genai/runtimes.py index 479badf162..9168fa7bdd 100644 --- a/agentplatform/_genai/runtimes.py +++ b/agentplatform/_genai/runtimes.py @@ -26,10 +26,9 @@ from google.genai import _api_module from google.genai import _common 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 google.genai.pagers import Pager +from . import _converters from . import _runtimes_utils from . import types @@ -44,693 +43,6 @@ logger.setLevel(logging.INFO) -def _CancelQueryJobRuntimeConfig_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(parent_object, ["operationName"], getv(from_object, ["operation_name"])) - - return to_object - - -def _CancelQueryJobRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CancelQueryJobRuntimeConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _CheckQueryJobResult_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(parent_object, ["operationName"]) is not None: - setv(to_object, ["operation_name"], getv(parent_object, ["operationName"])) - - if getv(parent_object, ["outputGcsUri"]) is not None: - setv(to_object, ["output_gcs_uri"], getv(parent_object, ["outputGcsUri"])) - - if getv(parent_object, ["status"]) is not None: - setv(to_object, ["status"], getv(parent_object, ["status"])) - - if getv(parent_object, ["result"]) is not None: - setv(to_object, ["result"], getv(parent_object, ["result"])) - - return to_object - - -def _CheckQueryJobRuntimeConfig_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, ["retrieve_result"]) is not None: - setv(parent_object, ["retrieveResult"], getv(from_object, ["retrieve_result"])) - - return to_object - - -def _CheckQueryJobRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CheckQueryJobRuntimeConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _CreateRuntimeConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["spec"]) is not None: - setv(parent_object, ["spec"], getv(from_object, ["spec"])) - - if getv(from_object, ["context_spec"]) is not None: - setv( - parent_object, - ["contextSpec"], - _ReasoningEngineContextSpec_to_vertex( - getv(from_object, ["context_spec"]), to_object - ), - ) - - if getv(from_object, ["psc_interface_config"]) is not None: - setv( - parent_object, - ["pscInterfaceConfig"], - getv(from_object, ["psc_interface_config"]), - ) - - if getv(from_object, ["agent_gateway_config"]) is not None: - setv( - parent_object, - ["agentGatewayConfig"], - getv(from_object, ["agent_gateway_config"]), - ) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["labels"]) is not None: - setv(parent_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["source_packages"]) is not None: - setv(parent_object, ["sourcePackages"], getv(from_object, ["source_packages"])) - - if getv(from_object, ["entrypoint_module"]) is not None: - setv( - parent_object, - ["entrypointModule"], - getv(from_object, ["entrypoint_module"]), - ) - - if getv(from_object, ["entrypoint_object"]) is not None: - setv( - parent_object, - ["entrypointObject"], - getv(from_object, ["entrypoint_object"]), - ) - - if getv(from_object, ["requirements_file"]) is not None: - setv( - parent_object, - ["requirementsFile"], - getv(from_object, ["requirements_file"]), - ) - - if getv(from_object, ["agent_framework"]) is not None: - setv(parent_object, ["agentFramework"], getv(from_object, ["agent_framework"])) - - if getv(from_object, ["python_version"]) is not None: - setv(parent_object, ["pythonVersion"], getv(from_object, ["python_version"])) - - return to_object - - -def _CreateRuntimeRequestParameters_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: - _CreateRuntimeConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteRuntimeRequestParameters_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, ["force"]) is not None: - setv(to_object, ["force"], getv(from_object, ["force"])) - - return to_object - - -def _GetRuntimeOperationParameters_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 _GetRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListRuntimeConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListRuntimeRequestParameters_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: - _ListRuntimeConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _QueryRuntimeConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["class_method"]) is not None: - setv(parent_object, ["classMethod"], getv(from_object, ["class_method"])) - - if getv(from_object, ["input"]) is not None: - setv(parent_object, ["input"], getv(from_object, ["input"])) - - if getv(from_object, ["include_all_fields"]) is not None: - setv(to_object, ["includeAllFields"], getv(from_object, ["include_all_fields"])) - - return to_object - - -def _QueryRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _QueryRuntimeConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _ReasoningEngineContextSpecMemoryBankConfig_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, ["customizationConfigs"]) is not None: - setv( - to_object, - ["customization_configs"], - [item for item in getv(from_object, ["customizationConfigs"])], - ) - - if getv(from_object, ["disableMemoryRevisions"]) is not None: - setv( - to_object, - ["disable_memory_revisions"], - getv(from_object, ["disableMemoryRevisions"]), - ) - - if getv(from_object, ["generationConfig"]) is not None: - setv(to_object, ["generation_config"], getv(from_object, ["generationConfig"])) - - if getv(from_object, ["similaritySearchConfig"]) is not None: - setv( - to_object, - ["similarity_search_config"], - getv(from_object, ["similaritySearchConfig"]), - ) - - if getv(from_object, ["ttlConfig"]) is not None: - setv(to_object, ["ttl_config"], getv(from_object, ["ttlConfig"])) - - if getv(from_object, ["structuredMemoryConfigs"]) is not None: - setv( - to_object, - ["structured_memory_configs"], - [ - _StructuredMemoryConfig_from_vertex(item, to_object) - for item in getv(from_object, ["structuredMemoryConfigs"]) - ], - ) - - return to_object - - -def _ReasoningEngineContextSpecMemoryBankConfig_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, ["customization_configs"]) is not None: - setv( - to_object, - ["customizationConfigs"], - [item for item in getv(from_object, ["customization_configs"])], - ) - - if getv(from_object, ["disable_memory_revisions"]) is not None: - setv( - to_object, - ["disableMemoryRevisions"], - getv(from_object, ["disable_memory_revisions"]), - ) - - if getv(from_object, ["generation_config"]) is not None: - setv(to_object, ["generationConfig"], getv(from_object, ["generation_config"])) - - if getv(from_object, ["similarity_search_config"]) is not None: - setv( - to_object, - ["similaritySearchConfig"], - getv(from_object, ["similarity_search_config"]), - ) - - if getv(from_object, ["ttl_config"]) is not None: - setv(to_object, ["ttlConfig"], getv(from_object, ["ttl_config"])) - - if getv(from_object, ["structured_memory_configs"]) is not None: - setv( - to_object, - ["structuredMemoryConfigs"], - [ - _StructuredMemoryConfig_to_vertex(item, to_object) - for item in getv(from_object, ["structured_memory_configs"]) - ], - ) - - return to_object - - -def _ReasoningEngineContextSpec_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, ["memoryBankConfig"]) is not None: - setv( - to_object, - ["memory_bank_config"], - _ReasoningEngineContextSpecMemoryBankConfig_from_vertex( - getv(from_object, ["memoryBankConfig"]), to_object - ), - ) - - return to_object - - -def _ReasoningEngineContextSpec_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, ["memory_bank_config"]) is not None: - setv( - to_object, - ["memoryBankConfig"], - _ReasoningEngineContextSpecMemoryBankConfig_to_vertex( - getv(from_object, ["memory_bank_config"]), to_object - ), - ) - - return to_object - - -def _ReasoningEngine_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, ["encryptionSpec"]) is not None: - setv(to_object, ["encryption_spec"], getv(from_object, ["encryptionSpec"])) - - if getv(from_object, ["contextSpec"]) is not None: - setv( - to_object, - ["context_spec"], - _ReasoningEngineContextSpec_from_vertex( - getv(from_object, ["contextSpec"]), 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, ["etag"]) is not None: - setv(to_object, ["etag"], getv(from_object, ["etag"])) - - if getv(from_object, ["labels"]) is not None: - setv(to_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["name"]) is not None: - setv(to_object, ["name"], getv(from_object, ["name"])) - - if getv(from_object, ["spec"]) is not None: - setv(to_object, ["spec"], getv(from_object, ["spec"])) - - if getv(from_object, ["updateTime"]) is not None: - setv(to_object, ["update_time"], getv(from_object, ["updateTime"])) - - if getv(from_object, ["trafficConfig"]) is not None: - setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"])) - - if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None: - setv( - to_object, - ["revision_garbage_collection_strategy"], - getv(from_object, ["revisionGarbageCollectionStrategy"]), - ) - - return to_object - - -def _RunQueryJobRuntimeConfig_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, ["input_gcs_uri"]) is not None: - setv(parent_object, ["inputGcsUri"], getv(from_object, ["input_gcs_uri"])) - - if getv(from_object, ["output_gcs_uri"]) is not None: - setv(parent_object, ["outputGcsUri"], getv(from_object, ["output_gcs_uri"])) - - return to_object - - -def _RunQueryJobRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _RunQueryJobRuntimeConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _RuntimeOperation_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"], - _ReasoningEngine_from_vertex(getv(from_object, ["response"]), to_object), - ) - - return to_object - - -def _StructuredMemoryConfig_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, ["schemaConfigs"]) is not None: - setv( - to_object, - ["schema_configs"], - [ - _StructuredMemorySchemaConfig_from_vertex(item, to_object) - for item in getv(from_object, ["schemaConfigs"]) - ], - ) - - if getv(from_object, ["scopeKeys"]) is not None: - setv(to_object, ["scope_keys"], getv(from_object, ["scopeKeys"])) - - return to_object - - -def _StructuredMemoryConfig_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, ["schema_configs"]) is not None: - setv( - to_object, - ["schemaConfigs"], - [ - _StructuredMemorySchemaConfig_to_vertex(item, to_object) - for item in getv(from_object, ["schema_configs"]) - ], - ) - - if getv(from_object, ["scope_keys"]) is not None: - setv(to_object, ["scopeKeys"], getv(from_object, ["scope_keys"])) - - return to_object - - -def _StructuredMemorySchemaConfig_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, ["schema"]) is not None: - setv(to_object, ["memory_schema"], getv(from_object, ["schema"])) - - if getv(from_object, ["id"]) is not None: - setv(to_object, ["id"], getv(from_object, ["id"])) - - if getv(from_object, ["memoryType"]) is not None: - setv(to_object, ["memory_type"], getv(from_object, ["memoryType"])) - - return to_object - - -def _StructuredMemorySchemaConfig_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, ["memory_schema"]) is not None: - setv(to_object, ["schema"], getv(from_object, ["memory_schema"])) - - if getv(from_object, ["id"]) is not None: - setv(to_object, ["id"], getv(from_object, ["id"])) - - if getv(from_object, ["memory_type"]) is not None: - setv(to_object, ["memoryType"], getv(from_object, ["memory_type"])) - - return to_object - - -def _UpdateRuntimeConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["spec"]) is not None: - setv(parent_object, ["spec"], getv(from_object, ["spec"])) - - if getv(from_object, ["context_spec"]) is not None: - setv( - parent_object, - ["contextSpec"], - _ReasoningEngineContextSpec_to_vertex( - getv(from_object, ["context_spec"]), to_object - ), - ) - - if getv(from_object, ["psc_interface_config"]) is not None: - setv( - parent_object, - ["pscInterfaceConfig"], - getv(from_object, ["psc_interface_config"]), - ) - - if getv(from_object, ["agent_gateway_config"]) is not None: - setv( - parent_object, - ["agentGatewayConfig"], - getv(from_object, ["agent_gateway_config"]), - ) - - if getv(from_object, ["encryption_spec"]) is not None: - setv(parent_object, ["encryptionSpec"], getv(from_object, ["encryption_spec"])) - - if getv(from_object, ["labels"]) is not None: - setv(parent_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["source_packages"]) is not None: - setv(parent_object, ["sourcePackages"], getv(from_object, ["source_packages"])) - - if getv(from_object, ["entrypoint_module"]) is not None: - setv( - parent_object, - ["entrypointModule"], - getv(from_object, ["entrypoint_module"]), - ) - - if getv(from_object, ["entrypoint_object"]) is not None: - setv( - parent_object, - ["entrypointObject"], - getv(from_object, ["entrypoint_object"]), - ) - - if getv(from_object, ["requirements_file"]) is not None: - setv( - parent_object, - ["requirementsFile"], - getv(from_object, ["requirements_file"]), - ) - - if getv(from_object, ["agent_framework"]) is not None: - setv(parent_object, ["agentFramework"], getv(from_object, ["agent_framework"])) - - if getv(from_object, ["python_version"]) is not None: - setv(parent_object, ["pythonVersion"], getv(from_object, ["python_version"])) - - if getv(from_object, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - if getv(from_object, ["traffic_config"]) is not None: - setv(parent_object, ["trafficConfig"], getv(from_object, ["traffic_config"])) - - return to_object - - -def _UpdateRuntimeRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateRuntimeConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class Runtimes(_api_module.BaseModule): def cancel_query_job( @@ -761,8 +73,10 @@ def cancel_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CancelQueryJobRuntimeRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._CancelQueryJobRuntimeRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -835,7 +149,7 @@ def _check_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CheckQueryJobRuntimeRequestParameters_to_vertex( + request_dict = _converters._CheckQueryJobRuntimeRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -865,7 +179,7 @@ def _check_query_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CheckQueryJobResult_from_vertex(response_dict) + response_dict = _converters._CheckQueryJobResult_from_vertex(response_dict) return_value = types.CheckQueryJobResult._from_response( response=response_dict, @@ -912,7 +226,7 @@ def _run_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RunQueryJobRuntimeRequestParameters_to_vertex( + request_dict = _converters._RunQueryJobRuntimeRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -942,7 +256,7 @@ def _run_query_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -985,7 +299,9 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -1013,7 +329,7 @@ def _create( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -1075,7 +391,9 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1144,7 +462,9 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1172,7 +492,7 @@ def _get( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ReasoningEngine_from_vertex(response_dict) + response_dict = _converters._ReasoningEngine_from_vertex(response_dict) return_value = types.ReasoningEngine._from_response( response=response_dict, @@ -1215,7 +535,9 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -1242,6 +564,11 @@ def _list( response_dict = {} if not response.body else json.loads(response.body) + if self._api_client.vertexai: + response_dict = _converters._ListReasoningEnginesResponse_from_vertex( + response_dict + ) + return_value = types.ListReasoningEnginesResponse._from_response( response=response_dict, kwargs=( @@ -1283,7 +610,9 @@ def _get_agent_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetRuntimeOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -1311,7 +640,7 @@ def _get_agent_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -1355,7 +684,9 @@ def _query( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _QueryRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._QueryRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:query".format_map(request_url_dict) @@ -1424,7 +755,9 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1452,7 +785,7 @@ def _update( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -2875,8 +2208,10 @@ async def cancel_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CancelQueryJobRuntimeRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._CancelQueryJobRuntimeRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -2951,7 +2286,7 @@ async def _check_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CheckQueryJobRuntimeRequestParameters_to_vertex( + request_dict = _converters._CheckQueryJobRuntimeRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2983,7 +2318,7 @@ async def _check_query_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _CheckQueryJobResult_from_vertex(response_dict) + response_dict = _converters._CheckQueryJobResult_from_vertex(response_dict) return_value = types.CheckQueryJobResult._from_response( response=response_dict, @@ -3030,7 +2365,7 @@ async def _run_query_job( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RunQueryJobRuntimeRequestParameters_to_vertex( + request_dict = _converters._RunQueryJobRuntimeRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -3062,7 +2397,7 @@ async def _run_query_job( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -3105,7 +2440,9 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -3135,7 +2472,7 @@ async def _create( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -3197,7 +2534,9 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3268,7 +2607,9 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3298,7 +2639,7 @@ async def _get( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _ReasoningEngine_from_vertex(response_dict) + response_dict = _converters._ReasoningEngine_from_vertex(response_dict) return_value = types.ReasoningEngine._from_response( response=response_dict, @@ -3341,7 +2682,9 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "reasoningEngines".format_map(request_url_dict) @@ -3370,6 +2713,11 @@ async def _list( response_dict = {} if not response.body else json.loads(response.body) + if self._api_client.vertexai: + response_dict = _converters._ListReasoningEnginesResponse_from_vertex( + response_dict + ) + return_value = types.ListReasoningEnginesResponse._from_response( response=response_dict, kwargs=( @@ -3411,7 +2759,9 @@ async def _get_agent_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetRuntimeOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -3441,7 +2791,7 @@ async def _get_agent_operation( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, @@ -3485,7 +2835,9 @@ async def _query( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _QueryRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._QueryRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}:query".format_map(request_url_dict) @@ -3556,7 +2908,9 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateRuntimeRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -3586,7 +2940,7 @@ async def _update( response_dict = {} if not response.body else json.loads(response.body) if self._api_client.vertexai: - response_dict = _RuntimeOperation_from_vertex(response_dict) + response_dict = _converters._RuntimeOperation_from_vertex(response_dict) return_value = types.RuntimeOperation._from_response( response=response_dict, diff --git a/agentplatform/_genai/sandbox_snapshots.py b/agentplatform/_genai/sandbox_snapshots.py index 70390747db..f285f3686a 100644 --- a/agentplatform/_genai/sandbox_snapshots.py +++ b/agentplatform/_genai/sandbox_snapshots.py @@ -18,15 +18,14 @@ import functools import json import logging -from typing import Any, Iterator, Optional, Union +from typing import Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import Pager +from . import _converters from . import _runtimes_utils from . import types @@ -35,113 +34,6 @@ logger.setLevel(logging.INFO) -def _CreateRuntimeSandboxSnapshotConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["owner"]) is not None: - setv(parent_object, ["owner"], getv(from_object, ["owner"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - return to_object - - -def _CreateSandboxEnvironmentSnapshotRequestParameters_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, ["source_sandbox_environment_name"]) is not None: - setv( - to_object, - ["_url", "name"], - getv(from_object, ["source_sandbox_environment_name"]), - ) - - if getv(from_object, ["config"]) is not None: - _CreateRuntimeSandboxSnapshotConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetRuntimeSandboxSnapshotOperationParameters_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 _GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListSandboxEnvironmentSnapshotsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListSandboxEnvironmentSnapshotsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class SandboxSnapshots(_api_module.BaseModule): """Sandbox environment snapshot commands.""" @@ -174,7 +66,7 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSandboxEnvironmentSnapshotRequestParameters_to_vertex( + request_dict = _converters._CreateSandboxEnvironmentSnapshotRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -249,7 +141,7 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( + request_dict = _converters._DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -324,8 +216,10 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -399,8 +293,10 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -469,8 +365,10 @@ def get_sandbox_snapshot_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxSnapshotOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeSandboxSnapshotOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -656,7 +554,7 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSandboxEnvironmentSnapshotRequestParameters_to_vertex( + request_dict = _converters._CreateSandboxEnvironmentSnapshotRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -733,7 +631,7 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( + request_dict = _converters._DeleteSandboxEnvironmentSnapshotRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -810,8 +708,10 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentSnapshotRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -887,8 +787,10 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListSandboxEnvironmentSnapshotsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -959,8 +861,10 @@ async def get_sandbox_snapshot_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxSnapshotOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetRuntimeSandboxSnapshotOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: diff --git a/agentplatform/_genai/sandbox_templates.py b/agentplatform/_genai/sandbox_templates.py index 66d5767824..72e79cce6b 100644 --- a/agentplatform/_genai/sandbox_templates.py +++ b/agentplatform/_genai/sandbox_templates.py @@ -18,15 +18,14 @@ import functools import json import logging -from typing import Any, Iterator, Optional, Union +from typing import Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import Pager +from . import _converters from . import _runtimes_utils from . import types @@ -35,131 +34,6 @@ logger.setLevel(logging.INFO) -def _CreateSandboxEnvironmentTemplateConfig_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_container_environment"]) is not None: - setv( - parent_object, - ["customContainerEnvironment"], - getv(from_object, ["custom_container_environment"]), - ) - - if getv(from_object, ["default_container_environment"]) is not None: - setv( - parent_object, - ["defaultContainerEnvironment"], - getv(from_object, ["default_container_environment"]), - ) - - if getv(from_object, ["egress_control_config"]) is not None: - setv( - parent_object, - ["egressControlConfig"], - getv(from_object, ["egress_control_config"]), - ) - - if getv(from_object, ["ingress_control_config"]) is not None: - setv( - parent_object, - ["ingressControlConfig"], - getv(from_object, ["ingress_control_config"]), - ) - - return to_object - - -def _CreateSandboxEnvironmentTemplateRequestParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["config"]) is not None: - _CreateSandboxEnvironmentTemplateConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetSandboxEnvironmentTemplateOperationParameters_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 _GetSandboxEnvironmentTemplateRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListSandboxEnvironmentTemplatesConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListSandboxEnvironmentTemplatesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class SandboxTemplates(_api_module.BaseModule): """Sandbox environment templates commands.""" @@ -196,7 +70,7 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSandboxEnvironmentTemplateRequestParameters_to_vertex( + request_dict = _converters._CreateSandboxEnvironmentTemplateRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -278,7 +152,7 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( + request_dict = _converters._DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -364,8 +238,10 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentTemplateRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentTemplateRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -446,8 +322,10 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -516,8 +394,10 @@ def get_sandbox_environment_template_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentTemplateOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentTemplateOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -711,7 +591,7 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSandboxEnvironmentTemplateRequestParameters_to_vertex( + request_dict = _converters._CreateSandboxEnvironmentTemplateRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -795,7 +675,7 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( + request_dict = _converters._DeleteSandboxEnvironmentTemplateRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -883,8 +763,10 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentTemplateRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentTemplateRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -967,8 +849,10 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListSandboxEnvironmentTemplatesRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1039,8 +923,10 @@ async def get_sandbox_environment_template_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSandboxEnvironmentTemplateOperationParameters_to_vertex( - parameter_model + request_dict = ( + _converters._GetSandboxEnvironmentTemplateOperationParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: diff --git a/agentplatform/_genai/sandboxes.py b/agentplatform/_genai/sandboxes.py index 649cf3a743..d67ad4ae0c 100644 --- a/agentplatform/_genai/sandboxes.py +++ b/agentplatform/_genai/sandboxes.py @@ -22,7 +22,7 @@ import mimetypes import secrets import time -from typing import Any, Iterator, Optional, Union +from typing import Any, Iterator, Optional from urllib.parse import urlencode from google import auth as google_auth @@ -30,11 +30,10 @@ from google.genai import _api_module from google.genai import _common 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 google.genai.pagers import Pager import requests +from . import _converters from . import _runtimes_utils from . import types @@ -43,161 +42,6 @@ logger.setLevel(logging.INFO) -def _CreateRuntimeSandboxConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - if getv(from_object, ["sandbox_environment_template"]) is not None: - setv( - parent_object, - ["sandboxEnvironmentTemplate"], - getv(from_object, ["sandbox_environment_template"]), - ) - - if getv(from_object, ["sandbox_environment_snapshot"]) is not None: - setv( - parent_object, - ["sandboxEnvironmentSnapshot"], - getv(from_object, ["sandbox_environment_snapshot"]), - ) - - if getv(from_object, ["owner"]) is not None: - setv(parent_object, ["owner"], getv(from_object, ["owner"])) - - return to_object - - -def _CreateRuntimeSandboxRequestParameters_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, ["spec"]) is not None: - setv(to_object, ["spec"], getv(from_object, ["spec"])) - - if getv(from_object, ["config"]) is not None: - _CreateRuntimeSandboxConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteRuntimeSandboxRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ExecuteCodeRuntimeSandboxRequestParameters_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, ["inputs"]) is not None: - setv(to_object, ["inputs"], [item for item in getv(from_object, ["inputs"])]) - - return to_object - - -def _GetRuntimeSandboxOperationParameters_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 _GetRuntimeSandboxRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListRuntimeSandboxesConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListRuntimeSandboxesRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListRuntimeSandboxesConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _PauseRuntimeSandboxRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ResumeRuntimeSandboxRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - class Sandboxes(_api_module.BaseModule): def _create( @@ -223,7 +67,7 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._CreateRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -303,7 +147,7 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._DeleteRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -379,8 +223,10 @@ def _execute_code( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ExecuteCodeRuntimeSandboxRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ExecuteCodeRuntimeSandboxRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -456,7 +302,7 @@ def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -540,7 +386,7 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSandboxesRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeSandboxesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -610,7 +456,7 @@ def _get_sandbox_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeSandboxOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -689,7 +535,7 @@ def _pause( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PauseRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._PauseRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -769,7 +615,7 @@ def _resume( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ResumeRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._ResumeRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1554,7 +1400,7 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._CreateRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1636,7 +1482,7 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._DeleteRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1714,8 +1560,10 @@ async def _execute_code( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ExecuteCodeRuntimeSandboxRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ExecuteCodeRuntimeSandboxRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -1793,7 +1641,7 @@ async def _get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1879,7 +1727,7 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSandboxesRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeSandboxesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1951,7 +1799,7 @@ async def _get_sandbox_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSandboxOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeSandboxOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2032,7 +1880,7 @@ async def _pause( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _PauseRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._PauseRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -2114,7 +1962,7 @@ async def _resume( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ResumeRuntimeSandboxRequestParameters_to_vertex( + request_dict = _converters._ResumeRuntimeSandboxRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/serving_profiles.py b/agentplatform/_genai/serving_profiles.py index 7417ac1346..3eb291025d 100644 --- a/agentplatform/_genai/serving_profiles.py +++ b/agentplatform/_genai/serving_profiles.py @@ -22,159 +22,15 @@ from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _operations_utils from . import types logger = logging.getLogger("agentplatform_genai.servingprofiles") -def _CreateServingProfileConfig_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, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - return to_object - - -def _CreateServingProfileRequestParameters_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_profile_id"]) is not None: - setv( - to_object, - ["_query", "servingProfileId"], - getv(from_object, ["serving_profile_id"]), - ) - - if getv(from_object, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["scope"]) is not None: - setv(to_object, ["scope"], getv(from_object, ["scope"])) - - if getv(from_object, ["cmek_config"]) is not None: - setv(to_object, ["cmekConfig"], getv(from_object, ["cmek_config"])) - - if getv(from_object, ["config"]) is not None: - _CreateServingProfileConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteServingProfileRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetServingProfileOperationParameters_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 _GetServingProfileRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListServingProfilesConfig_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 _ListServingProfilesRequestParameters_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"], - _ListServingProfilesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - -def _UpdateServingProfileConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - return to_object - - -def _UpdateServingProfileRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateServingProfileConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class ServingProfiles(_api_module.BaseModule): """Class for managing ServingProfiles.""" @@ -196,7 +52,7 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetServingProfileRequestParameters_to_vertex( + request_dict = _converters._GetServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -276,7 +132,7 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateServingProfileRequestParameters_to_vertex( + request_dict = _converters._CreateServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -350,7 +206,7 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateServingProfileRequestParameters_to_vertex( + request_dict = _converters._UpdateServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -416,7 +272,7 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListServingProfilesRequestParameters_to_vertex( + request_dict = _converters._ListServingProfilesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -490,7 +346,7 @@ def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteServingProfileRequestParameters_to_vertex( + request_dict = _converters._DeleteServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -534,7 +390,7 @@ def _get_serving_profile_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetServingProfileOperationParameters_to_vertex( + request_dict = _converters._GetServingProfileOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -731,7 +587,7 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetServingProfileRequestParameters_to_vertex( + request_dict = _converters._GetServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -813,7 +669,7 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateServingProfileRequestParameters_to_vertex( + request_dict = _converters._CreateServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -889,7 +745,7 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateServingProfileRequestParameters_to_vertex( + request_dict = _converters._UpdateServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -957,7 +813,7 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListServingProfilesRequestParameters_to_vertex( + request_dict = _converters._ListServingProfilesRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1033,7 +889,7 @@ async def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteServingProfileRequestParameters_to_vertex( + request_dict = _converters._DeleteServingProfileRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1077,7 +933,7 @@ async def _get_serving_profile_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetServingProfileOperationParameters_to_vertex( + request_dict = _converters._GetServingProfileOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/session_events.py b/agentplatform/_genai/session_events.py index a4f46f5d49..438eca3999 100644 --- a/agentplatform/_genai/session_events.py +++ b/agentplatform/_genai/session_events.py @@ -19,15 +19,14 @@ import functools import json import logging -from typing import Any, Iterator, Optional, Union +from typing import Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import types logger = logging.getLogger("agentplatform_genai.sessionevents") @@ -35,92 +34,6 @@ logger.setLevel(logging.INFO) -def _AppendRuntimeSessionEventConfig_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, ["content"]) is not None: - setv(parent_object, ["content"], getv(from_object, ["content"])) - - if getv(from_object, ["actions"]) is not None: - setv(parent_object, ["actions"], getv(from_object, ["actions"])) - - if getv(from_object, ["error_code"]) is not None: - setv(parent_object, ["errorCode"], getv(from_object, ["error_code"])) - - if getv(from_object, ["error_message"]) is not None: - setv(parent_object, ["errorMessage"], getv(from_object, ["error_message"])) - - if getv(from_object, ["event_metadata"]) is not None: - setv(parent_object, ["eventMetadata"], getv(from_object, ["event_metadata"])) - - if getv(from_object, ["raw_event"]) is not None: - setv(parent_object, ["rawEvent"], getv(from_object, ["raw_event"])) - - return to_object - - -def _AppendRuntimeSessionEventRequestParameters_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, ["author"]) is not None: - setv(to_object, ["author"], getv(from_object, ["author"])) - - if getv(from_object, ["invocation_id"]) is not None: - setv(to_object, ["invocationId"], getv(from_object, ["invocation_id"])) - - if getv(from_object, ["timestamp"]) is not None: - setv(to_object, ["timestamp"], getv(from_object, ["timestamp"])) - - if getv(from_object, ["config"]) is not None: - _AppendRuntimeSessionEventConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - -def _ListRuntimeSessionEventsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListRuntimeSessionEventsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListRuntimeSessionEventsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ) - - return to_object - - class SessionEvents(_api_module.BaseModule): def append( @@ -163,8 +76,10 @@ def append( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AppendRuntimeSessionEventRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._AppendRuntimeSessionEventRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -247,8 +162,10 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSessionEventsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListRuntimeSessionEventsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -372,8 +289,10 @@ async def append( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _AppendRuntimeSessionEventRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._AppendRuntimeSessionEventRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: @@ -458,8 +377,10 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSessionEventsRequestParameters_to_vertex( - parameter_model + request_dict = ( + _converters._ListRuntimeSessionEventsRequestParameters_to_vertex( + parameter_model + ) ) request_url_dict = request_dict.get("_url") if request_url_dict: diff --git a/agentplatform/_genai/sessions.py b/agentplatform/_genai/sessions.py index 72877210f8..fd1eecf21b 100644 --- a/agentplatform/_genai/sessions.py +++ b/agentplatform/_genai/sessions.py @@ -20,15 +20,14 @@ import json import logging import typing -from typing import Any, Iterator, Optional, Union +from typing import Iterator, Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _runtimes_utils from . import types @@ -43,166 +42,6 @@ logger.setLevel(logging.INFO) -def _CreateRuntimeSessionConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["session_state"]) is not None: - setv(parent_object, ["sessionState"], getv(from_object, ["session_state"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - if getv(from_object, ["expire_time"]) is not None: - setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) - - if getv(from_object, ["labels"]) is not None: - setv(parent_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["session_id"]) is not None: - setv(parent_object, ["_query", "sessionId"], getv(from_object, ["session_id"])) - - return to_object - - -def _CreateRuntimeSessionRequestParameters_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, ["user_id"]) is not None: - setv(to_object, ["userId"], getv(from_object, ["user_id"])) - - if getv(from_object, ["config"]) is not None: - _CreateRuntimeSessionConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteRuntimeSessionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetRuntimeSessionOperationParameters_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 _GetRuntimeSessionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _ListRuntimeSessionsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListRuntimeSessionsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _ListRuntimeSessionsConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _UpdateRuntimeSessionConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["session_state"]) is not None: - setv(parent_object, ["sessionState"], getv(from_object, ["session_state"])) - - if getv(from_object, ["ttl"]) is not None: - setv(parent_object, ["ttl"], getv(from_object, ["ttl"])) - - if getv(from_object, ["expire_time"]) is not None: - setv(parent_object, ["expireTime"], getv(from_object, ["expire_time"])) - - if getv(from_object, ["labels"]) is not None: - setv(parent_object, ["labels"], getv(from_object, ["labels"])) - - if getv(from_object, ["session_id"]) is not None: - setv(parent_object, ["_query", "sessionId"], getv(from_object, ["session_id"])) - - if getv(from_object, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - if getv(from_object, ["user_id"]) is not None: - setv(parent_object, ["userId"], getv(from_object, ["user_id"])) - - return to_object - - -def _UpdateRuntimeSessionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateRuntimeSessionConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class Sessions(_api_module.BaseModule): def _create( @@ -239,7 +78,7 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._CreateRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -323,7 +162,7 @@ def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._DeleteRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -404,7 +243,7 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -488,7 +327,7 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSessionsRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeSessionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -558,7 +397,7 @@ def _get_session_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSessionOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeSessionOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -642,7 +481,7 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._UpdateRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -850,7 +689,7 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._CreateRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -936,7 +775,7 @@ async def delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._DeleteRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1019,7 +858,7 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._GetRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1105,7 +944,7 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListRuntimeSessionsRequestParameters_to_vertex( + request_dict = _converters._ListRuntimeSessionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1177,7 +1016,7 @@ async def _get_session_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetRuntimeSessionOperationParameters_to_vertex( + request_dict = _converters._GetRuntimeSessionOperationParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -1263,7 +1102,7 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateRuntimeSessionRequestParameters_to_vertex( + request_dict = _converters._UpdateRuntimeSessionRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/skill_revisions.py b/agentplatform/_genai/skill_revisions.py index 11484e428b..730e9341aa 100644 --- a/agentplatform/_genai/skill_revisions.py +++ b/agentplatform/_genai/skill_revisions.py @@ -17,68 +17,18 @@ import json import logging -from typing import Any, Optional, Union +from typing import Optional from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv +from . import _converters from . import types logger = logging.getLogger("agentplatform_genai.skillrevisions") -def _GetSkillRevisionRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListSkillRevisionsConfig_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 _ListSkillRevisionsRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListSkillRevisionsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) - - return to_object - - class SkillRevisions(_api_module.BaseModule): """Class for managing Skill Revisions in the Skill Registry.""" @@ -100,7 +50,9 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillRevisionRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillRevisionRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -172,7 +124,7 @@ def list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSkillRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListSkillRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") @@ -247,7 +199,9 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillRevisionRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillRevisionRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -321,7 +275,7 @@ async def list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSkillRevisionsRequestParameters_to_vertex( + request_dict = _converters._ListSkillRevisionsRequestParameters_to_vertex( parameter_model ) request_url_dict = request_dict.get("_url") diff --git a/agentplatform/_genai/skills.py b/agentplatform/_genai/skills.py index 536c0b1dff..98d392676d 100644 --- a/agentplatform/_genai/skills.py +++ b/agentplatform/_genai/skills.py @@ -21,15 +21,14 @@ import json import logging import typing -from typing import Any, Optional, Union +from typing import Optional, Union from urllib.parse import urlencode from google.genai import _api_module from google.genai import _common -from google.genai._common import get_value_by_path as getv -from google.genai._common import set_value_by_path as setv from google.genai.pagers import AsyncPager, Pager +from . import _converters from . import _operations_utils from . import _skills_utils from . import types @@ -43,184 +42,6 @@ logger = logging.getLogger("agentplatform_genai.skills") -def _CreateSkillConfig_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, ["zipped_filesystem"]) is not None: - setv( - parent_object, - ["zippedFilesystem"], - getv(from_object, ["zipped_filesystem"]), - ) - - return to_object - - -def _CreateSkillRequestParameters_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, ["display_name"]) is not None: - setv(to_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(to_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["skill_id"]) is not None: - setv(to_object, ["_query", "skillId"], getv(from_object, ["skill_id"])) - - if getv(from_object, ["config"]) is not None: - _CreateSkillConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - -def _DeleteSkillRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - return to_object - - -def _GetSkillOperationParameters_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 _GetSkillRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - - return to_object - - -def _ListSkillsConfig_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - - if getv(from_object, ["page_size"]) is not None: - setv(parent_object, ["_query", "pageSize"], getv(from_object, ["page_size"])) - - if getv(from_object, ["page_token"]) is not None: - setv(parent_object, ["_query", "pageToken"], getv(from_object, ["page_token"])) - - if getv(from_object, ["filter"]) is not None: - setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"])) - - return to_object - - -def _ListSkillsRequestParameters_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"], - _ListSkillsConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - -def _RetrieveSkillsConfig_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, ["top_k"]) is not None: - setv(parent_object, ["_query", "topK"], getv(from_object, ["top_k"])) - - return to_object - - -def _RetrieveSkillsRequestParameters_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", "query"], getv(from_object, ["query"])) - - if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _RetrieveSkillsConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) - - return to_object - - -def _UpdateSkillConfig_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, ["display_name"]) is not None: - setv(parent_object, ["displayName"], getv(from_object, ["display_name"])) - - if getv(from_object, ["description"]) is not None: - setv(parent_object, ["description"], getv(from_object, ["description"])) - - if getv(from_object, ["zipped_filesystem"]) is not None: - setv( - parent_object, - ["zippedFilesystem"], - getv(from_object, ["zipped_filesystem"]), - ) - - if getv(from_object, ["update_mask"]) is not None: - setv( - parent_object, ["_query", "updateMask"], getv(from_object, ["update_mask"]) - ) - - return to_object - - -def _UpdateSkillRequestParameters_to_vertex( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ["name"]) is not None: - setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - - if getv(from_object, ["config"]) is not None: - _UpdateSkillConfig_to_vertex(getv(from_object, ["config"]), to_object) - - return to_object - - class Skills(_api_module.BaseModule): """Class for managing Skills in the Skill Registry.""" @@ -242,7 +63,9 @@ def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -311,7 +134,9 @@ def retrieve( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveSkillsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RetrieveSkillsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills:retrieve".format_map(request_url_dict) @@ -387,7 +212,9 @@ def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills".format_map(request_url_dict) @@ -456,7 +283,9 @@ def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -520,7 +349,9 @@ def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSkillsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListSkillsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills".format_map(request_url_dict) @@ -589,7 +420,9 @@ def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -657,7 +490,9 @@ def _get_skill_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict) @@ -982,7 +817,9 @@ async def get( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1053,7 +890,9 @@ async def retrieve( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _RetrieveSkillsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._RetrieveSkillsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills:retrieve".format_map(request_url_dict) @@ -1131,7 +970,9 @@ async def _create( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _CreateSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._CreateSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills".format_map(request_url_dict) @@ -1202,7 +1043,9 @@ async def _update( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _UpdateSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._UpdateSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1268,7 +1111,9 @@ async def _list( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _ListSkillsRequestParameters_to_vertex(parameter_model) + request_dict = _converters._ListSkillsRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "skills".format_map(request_url_dict) @@ -1339,7 +1184,9 @@ async def _delete( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _DeleteSkillRequestParameters_to_vertex(parameter_model) + request_dict = _converters._DeleteSkillRequestParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{name}".format_map(request_url_dict) @@ -1409,7 +1256,9 @@ async def _get_skill_operation( "This method is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode." ) else: - request_dict = _GetSkillOperationParameters_to_vertex(parameter_model) + request_dict = _converters._GetSkillOperationParameters_to_vertex( + parameter_model + ) request_url_dict = request_dict.get("_url") if request_url_dict: path = "{operationName}".format_map(request_url_dict)