2488 lines
71 KiB
Python
2488 lines
71 KiB
Python
# 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 . import _transformers as t
|
|
from ._api_client import BaseApiClient
|
|
from ._common import get_value_by_path as getv
|
|
from ._common import set_value_by_path as setv
|
|
|
|
|
|
def _Part_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['video_metadata']) is not None:
|
|
raise ValueError('video_metadata parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['thought']) is not None:
|
|
setv(to_object, ['thought'], getv(from_object, ['thought']))
|
|
|
|
if getv(from_object, ['code_execution_result']) is not None:
|
|
setv(
|
|
to_object,
|
|
['codeExecutionResult'],
|
|
getv(from_object, ['code_execution_result']),
|
|
)
|
|
|
|
if getv(from_object, ['executable_code']) is not None:
|
|
setv(to_object, ['executableCode'], getv(from_object, ['executable_code']))
|
|
|
|
if getv(from_object, ['file_data']) is not None:
|
|
setv(to_object, ['fileData'], getv(from_object, ['file_data']))
|
|
|
|
if getv(from_object, ['function_call']) is not None:
|
|
setv(to_object, ['functionCall'], getv(from_object, ['function_call']))
|
|
|
|
if getv(from_object, ['function_response']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionResponse'],
|
|
getv(from_object, ['function_response']),
|
|
)
|
|
|
|
if getv(from_object, ['inline_data']) is not None:
|
|
setv(to_object, ['inlineData'], getv(from_object, ['inline_data']))
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Part_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['video_metadata']) is not None:
|
|
setv(to_object, ['videoMetadata'], getv(from_object, ['video_metadata']))
|
|
|
|
if getv(from_object, ['thought']) is not None:
|
|
setv(to_object, ['thought'], getv(from_object, ['thought']))
|
|
|
|
if getv(from_object, ['code_execution_result']) is not None:
|
|
setv(
|
|
to_object,
|
|
['codeExecutionResult'],
|
|
getv(from_object, ['code_execution_result']),
|
|
)
|
|
|
|
if getv(from_object, ['executable_code']) is not None:
|
|
setv(to_object, ['executableCode'], getv(from_object, ['executable_code']))
|
|
|
|
if getv(from_object, ['file_data']) is not None:
|
|
setv(to_object, ['fileData'], getv(from_object, ['file_data']))
|
|
|
|
if getv(from_object, ['function_call']) is not None:
|
|
setv(to_object, ['functionCall'], getv(from_object, ['function_call']))
|
|
|
|
if getv(from_object, ['function_response']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionResponse'],
|
|
getv(from_object, ['function_response']),
|
|
)
|
|
|
|
if getv(from_object, ['inline_data']) is not None:
|
|
setv(to_object, ['inlineData'], getv(from_object, ['inline_data']))
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Content_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['parts']) is not None:
|
|
setv(
|
|
to_object,
|
|
['parts'],
|
|
[
|
|
_Part_to_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['parts'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['role']) is not None:
|
|
setv(to_object, ['role'], getv(from_object, ['role']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Content_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['parts']) is not None:
|
|
setv(
|
|
to_object,
|
|
['parts'],
|
|
[
|
|
_Part_to_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['parts'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['role']) is not None:
|
|
setv(to_object, ['role'], getv(from_object, ['role']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Schema_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: 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']) is not None:
|
|
raise ValueError('example parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['pattern']) is not None:
|
|
raise ValueError('pattern parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['default']) is not None:
|
|
raise ValueError('default parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['max_length']) is not None:
|
|
raise ValueError('max_length parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['min_length']) is not None:
|
|
raise ValueError('min_length parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['min_properties']) is not None:
|
|
raise ValueError('min_properties parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['max_properties']) is not None:
|
|
raise ValueError('max_properties parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['any_of']) is not None:
|
|
setv(to_object, ['anyOf'], getv(from_object, ['any_of']))
|
|
|
|
if getv(from_object, ['description']) is not None:
|
|
setv(to_object, ['description'], getv(from_object, ['description']))
|
|
|
|
if getv(from_object, ['enum']) is not None:
|
|
setv(to_object, ['enum'], getv(from_object, ['enum']))
|
|
|
|
if getv(from_object, ['format']) is not None:
|
|
setv(to_object, ['format'], getv(from_object, ['format']))
|
|
|
|
if getv(from_object, ['items']) is not None:
|
|
setv(to_object, ['items'], getv(from_object, ['items']))
|
|
|
|
if getv(from_object, ['max_items']) is not None:
|
|
setv(to_object, ['maxItems'], getv(from_object, ['max_items']))
|
|
|
|
if getv(from_object, ['maximum']) is not None:
|
|
setv(to_object, ['maximum'], getv(from_object, ['maximum']))
|
|
|
|
if getv(from_object, ['min_items']) is not None:
|
|
setv(to_object, ['minItems'], getv(from_object, ['min_items']))
|
|
|
|
if getv(from_object, ['minimum']) is not None:
|
|
setv(to_object, ['minimum'], getv(from_object, ['minimum']))
|
|
|
|
if getv(from_object, ['nullable']) is not None:
|
|
setv(to_object, ['nullable'], getv(from_object, ['nullable']))
|
|
|
|
if getv(from_object, ['properties']) is not None:
|
|
setv(to_object, ['properties'], getv(from_object, ['properties']))
|
|
|
|
if getv(from_object, ['property_ordering']) is not None:
|
|
setv(
|
|
to_object,
|
|
['propertyOrdering'],
|
|
getv(from_object, ['property_ordering']),
|
|
)
|
|
|
|
if getv(from_object, ['required']) is not None:
|
|
setv(to_object, ['required'], getv(from_object, ['required']))
|
|
|
|
if getv(from_object, ['title']) is not None:
|
|
setv(to_object, ['title'], getv(from_object, ['title']))
|
|
|
|
if getv(from_object, ['type']) is not None:
|
|
setv(to_object, ['type'], getv(from_object, ['type']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Schema_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: 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']) is not None:
|
|
setv(to_object, ['example'], getv(from_object, ['example']))
|
|
|
|
if getv(from_object, ['pattern']) is not None:
|
|
setv(to_object, ['pattern'], getv(from_object, ['pattern']))
|
|
|
|
if getv(from_object, ['default']) is not None:
|
|
setv(to_object, ['default'], getv(from_object, ['default']))
|
|
|
|
if getv(from_object, ['max_length']) is not None:
|
|
setv(to_object, ['maxLength'], getv(from_object, ['max_length']))
|
|
|
|
if getv(from_object, ['min_length']) is not None:
|
|
setv(to_object, ['minLength'], getv(from_object, ['min_length']))
|
|
|
|
if getv(from_object, ['min_properties']) is not None:
|
|
setv(to_object, ['minProperties'], getv(from_object, ['min_properties']))
|
|
|
|
if getv(from_object, ['max_properties']) is not None:
|
|
setv(to_object, ['maxProperties'], getv(from_object, ['max_properties']))
|
|
|
|
if getv(from_object, ['any_of']) is not None:
|
|
setv(to_object, ['anyOf'], getv(from_object, ['any_of']))
|
|
|
|
if getv(from_object, ['description']) is not None:
|
|
setv(to_object, ['description'], getv(from_object, ['description']))
|
|
|
|
if getv(from_object, ['enum']) is not None:
|
|
setv(to_object, ['enum'], getv(from_object, ['enum']))
|
|
|
|
if getv(from_object, ['format']) is not None:
|
|
setv(to_object, ['format'], getv(from_object, ['format']))
|
|
|
|
if getv(from_object, ['items']) is not None:
|
|
setv(to_object, ['items'], getv(from_object, ['items']))
|
|
|
|
if getv(from_object, ['max_items']) is not None:
|
|
setv(to_object, ['maxItems'], getv(from_object, ['max_items']))
|
|
|
|
if getv(from_object, ['maximum']) is not None:
|
|
setv(to_object, ['maximum'], getv(from_object, ['maximum']))
|
|
|
|
if getv(from_object, ['min_items']) is not None:
|
|
setv(to_object, ['minItems'], getv(from_object, ['min_items']))
|
|
|
|
if getv(from_object, ['minimum']) is not None:
|
|
setv(to_object, ['minimum'], getv(from_object, ['minimum']))
|
|
|
|
if getv(from_object, ['nullable']) is not None:
|
|
setv(to_object, ['nullable'], getv(from_object, ['nullable']))
|
|
|
|
if getv(from_object, ['properties']) is not None:
|
|
setv(to_object, ['properties'], getv(from_object, ['properties']))
|
|
|
|
if getv(from_object, ['property_ordering']) is not None:
|
|
setv(
|
|
to_object,
|
|
['propertyOrdering'],
|
|
getv(from_object, ['property_ordering']),
|
|
)
|
|
|
|
if getv(from_object, ['required']) is not None:
|
|
setv(to_object, ['required'], getv(from_object, ['required']))
|
|
|
|
if getv(from_object, ['title']) is not None:
|
|
setv(to_object, ['title'], getv(from_object, ['title']))
|
|
|
|
if getv(from_object, ['type']) is not None:
|
|
setv(to_object, ['type'], getv(from_object, ['type']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionDeclaration_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['response']) is not None:
|
|
raise ValueError('response parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['description']) is not None:
|
|
setv(to_object, ['description'], getv(from_object, ['description']))
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
if getv(from_object, ['parameters']) is not None:
|
|
setv(to_object, ['parameters'], getv(from_object, ['parameters']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionDeclaration_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['response']) is not None:
|
|
setv(
|
|
to_object,
|
|
['response'],
|
|
_Schema_to_vertex(
|
|
api_client, getv(from_object, ['response']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['description']) is not None:
|
|
setv(to_object, ['description'], getv(from_object, ['description']))
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
if getv(from_object, ['parameters']) is not None:
|
|
setv(to_object, ['parameters'], getv(from_object, ['parameters']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _GoogleSearch_to_mldev(
|
|
api_client: BaseApiClient,
|
|
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 _GoogleSearch_to_vertex(
|
|
api_client: BaseApiClient,
|
|
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 _DynamicRetrievalConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['mode']) is not None:
|
|
setv(to_object, ['mode'], getv(from_object, ['mode']))
|
|
|
|
if getv(from_object, ['dynamic_threshold']) is not None:
|
|
setv(
|
|
to_object,
|
|
['dynamicThreshold'],
|
|
getv(from_object, ['dynamic_threshold']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _DynamicRetrievalConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['mode']) is not None:
|
|
setv(to_object, ['mode'], getv(from_object, ['mode']))
|
|
|
|
if getv(from_object, ['dynamic_threshold']) is not None:
|
|
setv(
|
|
to_object,
|
|
['dynamicThreshold'],
|
|
getv(from_object, ['dynamic_threshold']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _GoogleSearchRetrieval_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['dynamic_retrieval_config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['dynamicRetrievalConfig'],
|
|
_DynamicRetrievalConfig_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['dynamic_retrieval_config']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _GoogleSearchRetrieval_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['dynamic_retrieval_config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['dynamicRetrievalConfig'],
|
|
_DynamicRetrievalConfig_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['dynamic_retrieval_config']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _Tool_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['function_declarations']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionDeclarations'],
|
|
[
|
|
_FunctionDeclaration_to_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['function_declarations'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['retrieval']) is not None:
|
|
raise ValueError('retrieval parameter is not supported in Gemini API.')
|
|
|
|
if getv(from_object, ['google_search']) is not None:
|
|
setv(
|
|
to_object,
|
|
['googleSearch'],
|
|
_GoogleSearch_to_mldev(
|
|
api_client, getv(from_object, ['google_search']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['google_search_retrieval']) is not None:
|
|
setv(
|
|
to_object,
|
|
['googleSearchRetrieval'],
|
|
_GoogleSearchRetrieval_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['google_search_retrieval']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['code_execution']) is not None:
|
|
setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Tool_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['function_declarations']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionDeclarations'],
|
|
[
|
|
_FunctionDeclaration_to_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['function_declarations'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['retrieval']) is not None:
|
|
setv(to_object, ['retrieval'], getv(from_object, ['retrieval']))
|
|
|
|
if getv(from_object, ['google_search']) is not None:
|
|
setv(
|
|
to_object,
|
|
['googleSearch'],
|
|
_GoogleSearch_to_vertex(
|
|
api_client, getv(from_object, ['google_search']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['google_search_retrieval']) is not None:
|
|
setv(
|
|
to_object,
|
|
['googleSearchRetrieval'],
|
|
_GoogleSearchRetrieval_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['google_search_retrieval']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['code_execution']) is not None:
|
|
setv(to_object, ['codeExecution'], getv(from_object, ['code_execution']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _SessionResumptionConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['handle']) is not None:
|
|
setv(to_object, ['handle'], getv(from_object, ['handle']))
|
|
|
|
if getv(from_object, ['transparent']) is not None:
|
|
raise ValueError('transparent parameter is not supported in Gemini API.')
|
|
|
|
return to_object
|
|
|
|
|
|
def _SessionResumptionConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['handle']) is not None:
|
|
setv(to_object, ['handle'], getv(from_object, ['handle']))
|
|
|
|
if getv(from_object, ['transparent']) is not None:
|
|
setv(to_object, ['transparent'], getv(from_object, ['transparent']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _AudioTranscriptionConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
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 _AudioTranscriptionConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
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 _AutomaticActivityDetection_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['disabled']) is not None:
|
|
setv(to_object, ['disabled'], getv(from_object, ['disabled']))
|
|
|
|
if getv(from_object, ['start_of_speech_sensitivity']) is not None:
|
|
setv(
|
|
to_object,
|
|
['startOfSpeechSensitivity'],
|
|
getv(from_object, ['start_of_speech_sensitivity']),
|
|
)
|
|
|
|
if getv(from_object, ['end_of_speech_sensitivity']) is not None:
|
|
setv(
|
|
to_object,
|
|
['endOfSpeechSensitivity'],
|
|
getv(from_object, ['end_of_speech_sensitivity']),
|
|
)
|
|
|
|
if getv(from_object, ['prefix_padding_ms']) is not None:
|
|
setv(
|
|
to_object, ['prefixPaddingMs'], getv(from_object, ['prefix_padding_ms'])
|
|
)
|
|
|
|
if getv(from_object, ['silence_duration_ms']) is not None:
|
|
setv(
|
|
to_object,
|
|
['silenceDurationMs'],
|
|
getv(from_object, ['silence_duration_ms']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _AutomaticActivityDetection_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['disabled']) is not None:
|
|
setv(to_object, ['disabled'], getv(from_object, ['disabled']))
|
|
|
|
if getv(from_object, ['start_of_speech_sensitivity']) is not None:
|
|
setv(
|
|
to_object,
|
|
['startOfSpeechSensitivity'],
|
|
getv(from_object, ['start_of_speech_sensitivity']),
|
|
)
|
|
|
|
if getv(from_object, ['end_of_speech_sensitivity']) is not None:
|
|
setv(
|
|
to_object,
|
|
['endOfSpeechSensitivity'],
|
|
getv(from_object, ['end_of_speech_sensitivity']),
|
|
)
|
|
|
|
if getv(from_object, ['prefix_padding_ms']) is not None:
|
|
setv(
|
|
to_object, ['prefixPaddingMs'], getv(from_object, ['prefix_padding_ms'])
|
|
)
|
|
|
|
if getv(from_object, ['silence_duration_ms']) is not None:
|
|
setv(
|
|
to_object,
|
|
['silenceDurationMs'],
|
|
getv(from_object, ['silence_duration_ms']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _RealtimeInputConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['automatic_activity_detection']) is not None:
|
|
setv(
|
|
to_object,
|
|
['automaticActivityDetection'],
|
|
_AutomaticActivityDetection_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['automatic_activity_detection']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_handling']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityHandling'],
|
|
getv(from_object, ['activity_handling']),
|
|
)
|
|
|
|
if getv(from_object, ['turn_coverage']) is not None:
|
|
setv(to_object, ['turnCoverage'], getv(from_object, ['turn_coverage']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _RealtimeInputConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['automatic_activity_detection']) is not None:
|
|
setv(
|
|
to_object,
|
|
['automaticActivityDetection'],
|
|
_AutomaticActivityDetection_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['automatic_activity_detection']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_handling']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityHandling'],
|
|
getv(from_object, ['activity_handling']),
|
|
)
|
|
|
|
if getv(from_object, ['turn_coverage']) is not None:
|
|
setv(to_object, ['turnCoverage'], getv(from_object, ['turn_coverage']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _SlidingWindow_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['target_tokens']) is not None:
|
|
setv(to_object, ['targetTokens'], getv(from_object, ['target_tokens']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _SlidingWindow_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['target_tokens']) is not None:
|
|
setv(to_object, ['targetTokens'], getv(from_object, ['target_tokens']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _ContextWindowCompressionConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['trigger_tokens']) is not None:
|
|
setv(to_object, ['triggerTokens'], getv(from_object, ['trigger_tokens']))
|
|
|
|
if getv(from_object, ['sliding_window']) is not None:
|
|
setv(
|
|
to_object,
|
|
['slidingWindow'],
|
|
_SlidingWindow_to_mldev(
|
|
api_client, getv(from_object, ['sliding_window']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _ContextWindowCompressionConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['trigger_tokens']) is not None:
|
|
setv(to_object, ['triggerTokens'], getv(from_object, ['trigger_tokens']))
|
|
|
|
if getv(from_object, ['sliding_window']) is not None:
|
|
setv(
|
|
to_object,
|
|
['slidingWindow'],
|
|
_SlidingWindow_to_vertex(
|
|
api_client, getv(from_object, ['sliding_window']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveConnectConfig_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['generation_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig'],
|
|
getv(from_object, ['generation_config']),
|
|
)
|
|
|
|
if getv(from_object, ['response_modalities']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'responseModalities'],
|
|
getv(from_object, ['response_modalities']),
|
|
)
|
|
|
|
if getv(from_object, ['temperature']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'temperature'],
|
|
getv(from_object, ['temperature']),
|
|
)
|
|
|
|
if getv(from_object, ['top_p']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'topP'],
|
|
getv(from_object, ['top_p']),
|
|
)
|
|
|
|
if getv(from_object, ['top_k']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'topK'],
|
|
getv(from_object, ['top_k']),
|
|
)
|
|
|
|
if getv(from_object, ['max_output_tokens']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'maxOutputTokens'],
|
|
getv(from_object, ['max_output_tokens']),
|
|
)
|
|
|
|
if getv(from_object, ['media_resolution']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'mediaResolution'],
|
|
getv(from_object, ['media_resolution']),
|
|
)
|
|
|
|
if getv(from_object, ['seed']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'seed'],
|
|
getv(from_object, ['seed']),
|
|
)
|
|
|
|
if getv(from_object, ['speech_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'speechConfig'],
|
|
getv(from_object, ['speech_config']),
|
|
)
|
|
|
|
if getv(from_object, ['system_instruction']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'systemInstruction'],
|
|
_Content_to_mldev(
|
|
api_client,
|
|
t.t_content(api_client, getv(from_object, ['system_instruction'])),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tools']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'tools'],
|
|
[
|
|
_Tool_to_mldev(api_client, t.t_tool(api_client, item), to_object)
|
|
for item in t.t_tools(api_client, getv(from_object, ['tools']))
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['session_resumption']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'sessionResumption'],
|
|
_SessionResumptionConfig_to_mldev(
|
|
api_client, getv(from_object, ['session_resumption']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['input_audio_transcription']) is not None:
|
|
raise ValueError(
|
|
'input_audio_transcription parameter is not supported in Gemini API.'
|
|
)
|
|
|
|
if getv(from_object, ['output_audio_transcription']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'outputAudioTranscription'],
|
|
_AudioTranscriptionConfig_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['output_audio_transcription']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['realtime_input_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'realtimeInputConfig'],
|
|
_RealtimeInputConfig_to_mldev(
|
|
api_client, getv(from_object, ['realtime_input_config']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['context_window_compression']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'contextWindowCompression'],
|
|
_ContextWindowCompressionConfig_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['context_window_compression']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveConnectConfig_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['generation_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig'],
|
|
getv(from_object, ['generation_config']),
|
|
)
|
|
|
|
if getv(from_object, ['response_modalities']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'responseModalities'],
|
|
getv(from_object, ['response_modalities']),
|
|
)
|
|
|
|
if getv(from_object, ['temperature']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'temperature'],
|
|
getv(from_object, ['temperature']),
|
|
)
|
|
|
|
if getv(from_object, ['top_p']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'topP'],
|
|
getv(from_object, ['top_p']),
|
|
)
|
|
|
|
if getv(from_object, ['top_k']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'topK'],
|
|
getv(from_object, ['top_k']),
|
|
)
|
|
|
|
if getv(from_object, ['max_output_tokens']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'maxOutputTokens'],
|
|
getv(from_object, ['max_output_tokens']),
|
|
)
|
|
|
|
if getv(from_object, ['media_resolution']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'mediaResolution'],
|
|
getv(from_object, ['media_resolution']),
|
|
)
|
|
|
|
if getv(from_object, ['seed']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'seed'],
|
|
getv(from_object, ['seed']),
|
|
)
|
|
|
|
if getv(from_object, ['speech_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'generationConfig', 'speechConfig'],
|
|
getv(from_object, ['speech_config']),
|
|
)
|
|
|
|
if getv(from_object, ['system_instruction']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'systemInstruction'],
|
|
_Content_to_vertex(
|
|
api_client,
|
|
t.t_content(api_client, getv(from_object, ['system_instruction'])),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tools']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'tools'],
|
|
[
|
|
_Tool_to_vertex(api_client, t.t_tool(api_client, item), to_object)
|
|
for item in t.t_tools(api_client, getv(from_object, ['tools']))
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['session_resumption']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'sessionResumption'],
|
|
_SessionResumptionConfig_to_vertex(
|
|
api_client, getv(from_object, ['session_resumption']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['input_audio_transcription']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'inputAudioTranscription'],
|
|
_AudioTranscriptionConfig_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['input_audio_transcription']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['output_audio_transcription']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'outputAudioTranscription'],
|
|
_AudioTranscriptionConfig_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['output_audio_transcription']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['realtime_input_config']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'realtimeInputConfig'],
|
|
_RealtimeInputConfig_to_vertex(
|
|
api_client, getv(from_object, ['realtime_input_config']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['context_window_compression']) is not None:
|
|
setv(
|
|
parent_object,
|
|
['setup', 'contextWindowCompression'],
|
|
_ContextWindowCompressionConfig_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['context_window_compression']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveConnectParameters_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['model']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup', 'model'],
|
|
t.t_model(api_client, getv(from_object, ['model'])),
|
|
)
|
|
|
|
if getv(from_object, ['config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['config'],
|
|
_LiveConnectConfig_to_mldev(
|
|
api_client, getv(from_object, ['config']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveConnectParameters_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['model']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup', 'model'],
|
|
t.t_model(api_client, getv(from_object, ['model'])),
|
|
)
|
|
|
|
if getv(from_object, ['config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['config'],
|
|
_LiveConnectConfig_to_vertex(
|
|
api_client, getv(from_object, ['config']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _ActivityStart_to_mldev(
|
|
api_client: BaseApiClient,
|
|
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 _ActivityStart_to_vertex(
|
|
api_client: BaseApiClient,
|
|
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 _ActivityEnd_to_mldev(
|
|
api_client: BaseApiClient,
|
|
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 _ActivityEnd_to_vertex(
|
|
api_client: BaseApiClient,
|
|
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 _LiveSendRealtimeInputParameters_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['media']) is not None:
|
|
setv(
|
|
to_object,
|
|
['mediaChunks'],
|
|
t.t_blobs(api_client, getv(from_object, ['media'])),
|
|
)
|
|
|
|
if getv(from_object, ['audio']) is not None:
|
|
setv(
|
|
to_object,
|
|
['audio'],
|
|
t.t_audio_blob(api_client, getv(from_object, ['audio'])),
|
|
)
|
|
|
|
if getv(from_object, ['audio_stream_end']) is not None:
|
|
setv(to_object, ['audioStreamEnd'], getv(from_object, ['audio_stream_end']))
|
|
|
|
if getv(from_object, ['video']) is not None:
|
|
setv(
|
|
to_object,
|
|
['video'],
|
|
t.t_image_blob(api_client, getv(from_object, ['video'])),
|
|
)
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
if getv(from_object, ['activity_start']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityStart'],
|
|
_ActivityStart_to_mldev(
|
|
api_client, getv(from_object, ['activity_start']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_end']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityEnd'],
|
|
_ActivityEnd_to_mldev(
|
|
api_client, getv(from_object, ['activity_end']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveSendRealtimeInputParameters_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['media']) is not None:
|
|
setv(
|
|
to_object,
|
|
['mediaChunks'],
|
|
t.t_blobs(api_client, getv(from_object, ['media'])),
|
|
)
|
|
|
|
if getv(from_object, ['audio']) is not None:
|
|
raise ValueError('audio parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['audio_stream_end']) is not None:
|
|
setv(to_object, ['audioStreamEnd'], getv(from_object, ['audio_stream_end']))
|
|
|
|
if getv(from_object, ['video']) is not None:
|
|
raise ValueError('video parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
raise ValueError('text parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['activity_start']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityStart'],
|
|
_ActivityStart_to_vertex(
|
|
api_client, getv(from_object, ['activity_start']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_end']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityEnd'],
|
|
_ActivityEnd_to_vertex(
|
|
api_client, getv(from_object, ['activity_end']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientSetup_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['model']) is not None:
|
|
setv(to_object, ['model'], getv(from_object, ['model']))
|
|
|
|
if getv(from_object, ['generation_config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['generationConfig'],
|
|
getv(from_object, ['generation_config']),
|
|
)
|
|
|
|
if getv(from_object, ['system_instruction']) is not None:
|
|
setv(
|
|
to_object,
|
|
['systemInstruction'],
|
|
_Content_to_mldev(
|
|
api_client,
|
|
t.t_content(api_client, getv(from_object, ['system_instruction'])),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tools']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tools'],
|
|
[
|
|
_Tool_to_mldev(api_client, t.t_tool(api_client, item), to_object)
|
|
for item in t.t_tools(api_client, getv(from_object, ['tools']))
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['session_resumption']) is not None:
|
|
setv(
|
|
to_object,
|
|
['sessionResumption'],
|
|
_SessionResumptionConfig_to_mldev(
|
|
api_client, getv(from_object, ['session_resumption']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['context_window_compression']) is not None:
|
|
setv(
|
|
to_object,
|
|
['contextWindowCompression'],
|
|
_ContextWindowCompressionConfig_to_mldev(
|
|
api_client,
|
|
getv(from_object, ['context_window_compression']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientSetup_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['model']) is not None:
|
|
setv(to_object, ['model'], getv(from_object, ['model']))
|
|
|
|
if getv(from_object, ['generation_config']) is not None:
|
|
setv(
|
|
to_object,
|
|
['generationConfig'],
|
|
getv(from_object, ['generation_config']),
|
|
)
|
|
|
|
if getv(from_object, ['system_instruction']) is not None:
|
|
setv(
|
|
to_object,
|
|
['systemInstruction'],
|
|
_Content_to_vertex(
|
|
api_client,
|
|
t.t_content(api_client, getv(from_object, ['system_instruction'])),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tools']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tools'],
|
|
[
|
|
_Tool_to_vertex(api_client, t.t_tool(api_client, item), to_object)
|
|
for item in t.t_tools(api_client, getv(from_object, ['tools']))
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['session_resumption']) is not None:
|
|
setv(
|
|
to_object,
|
|
['sessionResumption'],
|
|
_SessionResumptionConfig_to_vertex(
|
|
api_client, getv(from_object, ['session_resumption']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['context_window_compression']) is not None:
|
|
setv(
|
|
to_object,
|
|
['contextWindowCompression'],
|
|
_ContextWindowCompressionConfig_to_vertex(
|
|
api_client,
|
|
getv(from_object, ['context_window_compression']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientContent_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['turns']) is not None:
|
|
setv(
|
|
to_object,
|
|
['turns'],
|
|
[
|
|
_Content_to_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['turns'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['turn_complete']) is not None:
|
|
setv(to_object, ['turnComplete'], getv(from_object, ['turn_complete']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientContent_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['turns']) is not None:
|
|
setv(
|
|
to_object,
|
|
['turns'],
|
|
[
|
|
_Content_to_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['turns'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['turn_complete']) is not None:
|
|
setv(to_object, ['turnComplete'], getv(from_object, ['turn_complete']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientRealtimeInput_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['media_chunks']) is not None:
|
|
setv(to_object, ['mediaChunks'], getv(from_object, ['media_chunks']))
|
|
|
|
if getv(from_object, ['audio']) is not None:
|
|
setv(to_object, ['audio'], getv(from_object, ['audio']))
|
|
|
|
if getv(from_object, ['audio_stream_end']) is not None:
|
|
setv(to_object, ['audioStreamEnd'], getv(from_object, ['audio_stream_end']))
|
|
|
|
if getv(from_object, ['video']) is not None:
|
|
setv(to_object, ['video'], getv(from_object, ['video']))
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
if getv(from_object, ['activity_start']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityStart'],
|
|
_ActivityStart_to_mldev(
|
|
api_client, getv(from_object, ['activity_start']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_end']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityEnd'],
|
|
_ActivityEnd_to_mldev(
|
|
api_client, getv(from_object, ['activity_end']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientRealtimeInput_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['media_chunks']) is not None:
|
|
setv(to_object, ['mediaChunks'], getv(from_object, ['media_chunks']))
|
|
|
|
if getv(from_object, ['audio']) is not None:
|
|
raise ValueError('audio parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['audio_stream_end']) is not None:
|
|
raise ValueError(
|
|
'audio_stream_end parameter is not supported in Vertex AI.'
|
|
)
|
|
|
|
if getv(from_object, ['video']) is not None:
|
|
raise ValueError('video parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
raise ValueError('text parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['activity_start']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityStart'],
|
|
_ActivityStart_to_vertex(
|
|
api_client, getv(from_object, ['activity_start']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['activity_end']) is not None:
|
|
setv(
|
|
to_object,
|
|
['activityEnd'],
|
|
_ActivityEnd_to_vertex(
|
|
api_client, getv(from_object, ['activity_end']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionResponse_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['id']) is not None:
|
|
setv(to_object, ['id'], getv(from_object, ['id']))
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
if getv(from_object, ['response']) is not None:
|
|
setv(to_object, ['response'], getv(from_object, ['response']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionResponse_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['id']) is not None:
|
|
raise ValueError('id parameter is not supported in Vertex AI.')
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
if getv(from_object, ['response']) is not None:
|
|
setv(to_object, ['response'], getv(from_object, ['response']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientToolResponse_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['function_responses']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionResponses'],
|
|
[
|
|
_FunctionResponse_to_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['function_responses'])
|
|
],
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientToolResponse_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['function_responses']) is not None:
|
|
setv(
|
|
to_object,
|
|
['functionResponses'],
|
|
[
|
|
_FunctionResponse_to_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['function_responses'])
|
|
],
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientMessage_to_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['setup']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup'],
|
|
_LiveClientSetup_to_mldev(
|
|
api_client, getv(from_object, ['setup']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['client_content']) is not None:
|
|
setv(
|
|
to_object,
|
|
['clientContent'],
|
|
_LiveClientContent_to_mldev(
|
|
api_client, getv(from_object, ['client_content']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['realtime_input']) is not None:
|
|
setv(
|
|
to_object,
|
|
['realtimeInput'],
|
|
_LiveClientRealtimeInput_to_mldev(
|
|
api_client, getv(from_object, ['realtime_input']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tool_response']) is not None:
|
|
setv(
|
|
to_object,
|
|
['toolResponse'],
|
|
_LiveClientToolResponse_to_mldev(
|
|
api_client, getv(from_object, ['tool_response']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveClientMessage_to_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['setup']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup'],
|
|
_LiveClientSetup_to_vertex(
|
|
api_client, getv(from_object, ['setup']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['client_content']) is not None:
|
|
setv(
|
|
to_object,
|
|
['clientContent'],
|
|
_LiveClientContent_to_vertex(
|
|
api_client, getv(from_object, ['client_content']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['realtime_input']) is not None:
|
|
setv(
|
|
to_object,
|
|
['realtimeInput'],
|
|
_LiveClientRealtimeInput_to_vertex(
|
|
api_client, getv(from_object, ['realtime_input']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['tool_response']) is not None:
|
|
setv(
|
|
to_object,
|
|
['toolResponse'],
|
|
_LiveClientToolResponse_to_vertex(
|
|
api_client, getv(from_object, ['tool_response']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerSetupComplete_from_mldev(
|
|
api_client: BaseApiClient,
|
|
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 _LiveServerSetupComplete_from_vertex(
|
|
api_client: BaseApiClient,
|
|
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 _Part_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
|
|
if getv(from_object, ['thought']) is not None:
|
|
setv(to_object, ['thought'], getv(from_object, ['thought']))
|
|
|
|
if getv(from_object, ['codeExecutionResult']) is not None:
|
|
setv(
|
|
to_object,
|
|
['code_execution_result'],
|
|
getv(from_object, ['codeExecutionResult']),
|
|
)
|
|
|
|
if getv(from_object, ['executableCode']) is not None:
|
|
setv(to_object, ['executable_code'], getv(from_object, ['executableCode']))
|
|
|
|
if getv(from_object, ['fileData']) is not None:
|
|
setv(to_object, ['file_data'], getv(from_object, ['fileData']))
|
|
|
|
if getv(from_object, ['functionCall']) is not None:
|
|
setv(to_object, ['function_call'], getv(from_object, ['functionCall']))
|
|
|
|
if getv(from_object, ['functionResponse']) is not None:
|
|
setv(
|
|
to_object,
|
|
['function_response'],
|
|
getv(from_object, ['functionResponse']),
|
|
)
|
|
|
|
if getv(from_object, ['inlineData']) is not None:
|
|
setv(to_object, ['inline_data'], getv(from_object, ['inlineData']))
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Part_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['videoMetadata']) is not None:
|
|
setv(to_object, ['video_metadata'], getv(from_object, ['videoMetadata']))
|
|
|
|
if getv(from_object, ['thought']) is not None:
|
|
setv(to_object, ['thought'], getv(from_object, ['thought']))
|
|
|
|
if getv(from_object, ['codeExecutionResult']) is not None:
|
|
setv(
|
|
to_object,
|
|
['code_execution_result'],
|
|
getv(from_object, ['codeExecutionResult']),
|
|
)
|
|
|
|
if getv(from_object, ['executableCode']) is not None:
|
|
setv(to_object, ['executable_code'], getv(from_object, ['executableCode']))
|
|
|
|
if getv(from_object, ['fileData']) is not None:
|
|
setv(to_object, ['file_data'], getv(from_object, ['fileData']))
|
|
|
|
if getv(from_object, ['functionCall']) is not None:
|
|
setv(to_object, ['function_call'], getv(from_object, ['functionCall']))
|
|
|
|
if getv(from_object, ['functionResponse']) is not None:
|
|
setv(
|
|
to_object,
|
|
['function_response'],
|
|
getv(from_object, ['functionResponse']),
|
|
)
|
|
|
|
if getv(from_object, ['inlineData']) is not None:
|
|
setv(to_object, ['inline_data'], getv(from_object, ['inlineData']))
|
|
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Content_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['parts']) is not None:
|
|
setv(
|
|
to_object,
|
|
['parts'],
|
|
[
|
|
_Part_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['parts'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['role']) is not None:
|
|
setv(to_object, ['role'], getv(from_object, ['role']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Content_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['parts']) is not None:
|
|
setv(
|
|
to_object,
|
|
['parts'],
|
|
[
|
|
_Part_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['parts'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['role']) is not None:
|
|
setv(to_object, ['role'], getv(from_object, ['role']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Transcription_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
if getv(from_object, ['finished']) is not None:
|
|
setv(to_object, ['finished'], getv(from_object, ['finished']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _Transcription_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['text']) is not None:
|
|
setv(to_object, ['text'], getv(from_object, ['text']))
|
|
|
|
if getv(from_object, ['finished']) is not None:
|
|
setv(to_object, ['finished'], getv(from_object, ['finished']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerContent_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['modelTurn']) is not None:
|
|
setv(
|
|
to_object,
|
|
['model_turn'],
|
|
_Content_from_mldev(
|
|
api_client, getv(from_object, ['modelTurn']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['turnComplete']) is not None:
|
|
setv(to_object, ['turn_complete'], getv(from_object, ['turnComplete']))
|
|
|
|
if getv(from_object, ['interrupted']) is not None:
|
|
setv(to_object, ['interrupted'], getv(from_object, ['interrupted']))
|
|
|
|
if getv(from_object, ['generationComplete']) is not None:
|
|
setv(
|
|
to_object,
|
|
['generation_complete'],
|
|
getv(from_object, ['generationComplete']),
|
|
)
|
|
|
|
if getv(from_object, ['inputTranscription']) is not None:
|
|
setv(
|
|
to_object,
|
|
['input_transcription'],
|
|
_Transcription_from_mldev(
|
|
api_client, getv(from_object, ['inputTranscription']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['outputTranscription']) is not None:
|
|
setv(
|
|
to_object,
|
|
['output_transcription'],
|
|
_Transcription_from_mldev(
|
|
api_client, getv(from_object, ['outputTranscription']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerContent_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['modelTurn']) is not None:
|
|
setv(
|
|
to_object,
|
|
['model_turn'],
|
|
_Content_from_vertex(
|
|
api_client, getv(from_object, ['modelTurn']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['turnComplete']) is not None:
|
|
setv(to_object, ['turn_complete'], getv(from_object, ['turnComplete']))
|
|
|
|
if getv(from_object, ['interrupted']) is not None:
|
|
setv(to_object, ['interrupted'], getv(from_object, ['interrupted']))
|
|
|
|
if getv(from_object, ['generationComplete']) is not None:
|
|
setv(
|
|
to_object,
|
|
['generation_complete'],
|
|
getv(from_object, ['generationComplete']),
|
|
)
|
|
|
|
if getv(from_object, ['inputTranscription']) is not None:
|
|
setv(
|
|
to_object,
|
|
['input_transcription'],
|
|
_Transcription_from_vertex(
|
|
api_client, getv(from_object, ['inputTranscription']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['outputTranscription']) is not None:
|
|
setv(
|
|
to_object,
|
|
['output_transcription'],
|
|
_Transcription_from_vertex(
|
|
api_client, getv(from_object, ['outputTranscription']), to_object
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionCall_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['id']) is not None:
|
|
setv(to_object, ['id'], getv(from_object, ['id']))
|
|
|
|
if getv(from_object, ['args']) is not None:
|
|
setv(to_object, ['args'], getv(from_object, ['args']))
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _FunctionCall_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
|
|
if getv(from_object, ['args']) is not None:
|
|
setv(to_object, ['args'], getv(from_object, ['args']))
|
|
|
|
if getv(from_object, ['name']) is not None:
|
|
setv(to_object, ['name'], getv(from_object, ['name']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerToolCall_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['functionCalls']) is not None:
|
|
setv(
|
|
to_object,
|
|
['function_calls'],
|
|
[
|
|
_FunctionCall_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['functionCalls'])
|
|
],
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerToolCall_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['functionCalls']) is not None:
|
|
setv(
|
|
to_object,
|
|
['function_calls'],
|
|
[
|
|
_FunctionCall_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['functionCalls'])
|
|
],
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerToolCallCancellation_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['ids']) is not None:
|
|
setv(to_object, ['ids'], getv(from_object, ['ids']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerToolCallCancellation_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['ids']) is not None:
|
|
setv(to_object, ['ids'], getv(from_object, ['ids']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _ModalityTokenCount_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['modality']) is not None:
|
|
setv(to_object, ['modality'], getv(from_object, ['modality']))
|
|
|
|
if getv(from_object, ['tokenCount']) is not None:
|
|
setv(to_object, ['token_count'], getv(from_object, ['tokenCount']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _ModalityTokenCount_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['modality']) is not None:
|
|
setv(to_object, ['modality'], getv(from_object, ['modality']))
|
|
|
|
if getv(from_object, ['tokenCount']) is not None:
|
|
setv(to_object, ['token_count'], getv(from_object, ['tokenCount']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _UsageMetadata_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['promptTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['prompt_token_count'],
|
|
getv(from_object, ['promptTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['cachedContentTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['cached_content_token_count'],
|
|
getv(from_object, ['cachedContentTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['responseTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['response_token_count'],
|
|
getv(from_object, ['responseTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['toolUsePromptTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_use_prompt_token_count'],
|
|
getv(from_object, ['toolUsePromptTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['thoughtsTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['thoughts_token_count'],
|
|
getv(from_object, ['thoughtsTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['totalTokenCount']) is not None:
|
|
setv(
|
|
to_object, ['total_token_count'], getv(from_object, ['totalTokenCount'])
|
|
)
|
|
|
|
if getv(from_object, ['promptTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['prompt_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['promptTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['cacheTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['cache_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['cacheTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['responseTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['response_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['responseTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['toolUsePromptTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_use_prompt_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_mldev(api_client, item, to_object)
|
|
for item in getv(from_object, ['toolUsePromptTokensDetails'])
|
|
],
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _UsageMetadata_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['promptTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['prompt_token_count'],
|
|
getv(from_object, ['promptTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['cachedContentTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['cached_content_token_count'],
|
|
getv(from_object, ['cachedContentTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['candidatesTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['response_token_count'],
|
|
getv(from_object, ['candidatesTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['toolUsePromptTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_use_prompt_token_count'],
|
|
getv(from_object, ['toolUsePromptTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['thoughtsTokenCount']) is not None:
|
|
setv(
|
|
to_object,
|
|
['thoughts_token_count'],
|
|
getv(from_object, ['thoughtsTokenCount']),
|
|
)
|
|
|
|
if getv(from_object, ['totalTokenCount']) is not None:
|
|
setv(
|
|
to_object, ['total_token_count'], getv(from_object, ['totalTokenCount'])
|
|
)
|
|
|
|
if getv(from_object, ['promptTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['prompt_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['promptTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['cacheTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['cache_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['cacheTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['candidatesTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['response_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['candidatesTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['toolUsePromptTokensDetails']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_use_prompt_tokens_details'],
|
|
[
|
|
_ModalityTokenCount_from_vertex(api_client, item, to_object)
|
|
for item in getv(from_object, ['toolUsePromptTokensDetails'])
|
|
],
|
|
)
|
|
|
|
if getv(from_object, ['trafficType']) is not None:
|
|
setv(to_object, ['traffic_type'], getv(from_object, ['trafficType']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerGoAway_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['timeLeft']) is not None:
|
|
setv(to_object, ['time_left'], getv(from_object, ['timeLeft']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerGoAway_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['timeLeft']) is not None:
|
|
setv(to_object, ['time_left'], getv(from_object, ['timeLeft']))
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerSessionResumptionUpdate_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['newHandle']) is not None:
|
|
setv(to_object, ['new_handle'], getv(from_object, ['newHandle']))
|
|
|
|
if getv(from_object, ['resumable']) is not None:
|
|
setv(to_object, ['resumable'], getv(from_object, ['resumable']))
|
|
|
|
if getv(from_object, ['lastConsumedClientMessageIndex']) is not None:
|
|
setv(
|
|
to_object,
|
|
['last_consumed_client_message_index'],
|
|
getv(from_object, ['lastConsumedClientMessageIndex']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerSessionResumptionUpdate_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['newHandle']) is not None:
|
|
setv(to_object, ['new_handle'], getv(from_object, ['newHandle']))
|
|
|
|
if getv(from_object, ['resumable']) is not None:
|
|
setv(to_object, ['resumable'], getv(from_object, ['resumable']))
|
|
|
|
if getv(from_object, ['lastConsumedClientMessageIndex']) is not None:
|
|
setv(
|
|
to_object,
|
|
['last_consumed_client_message_index'],
|
|
getv(from_object, ['lastConsumedClientMessageIndex']),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerMessage_from_mldev(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['setupComplete']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup_complete'],
|
|
_LiveServerSetupComplete_from_mldev(
|
|
api_client, getv(from_object, ['setupComplete']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['serverContent']) is not None:
|
|
setv(
|
|
to_object,
|
|
['server_content'],
|
|
_LiveServerContent_from_mldev(
|
|
api_client, getv(from_object, ['serverContent']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['toolCall']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_call'],
|
|
_LiveServerToolCall_from_mldev(
|
|
api_client, getv(from_object, ['toolCall']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['toolCallCancellation']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_call_cancellation'],
|
|
_LiveServerToolCallCancellation_from_mldev(
|
|
api_client, getv(from_object, ['toolCallCancellation']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['usageMetadata']) is not None:
|
|
setv(
|
|
to_object,
|
|
['usage_metadata'],
|
|
_UsageMetadata_from_mldev(
|
|
api_client, getv(from_object, ['usageMetadata']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['goAway']) is not None:
|
|
setv(
|
|
to_object,
|
|
['go_away'],
|
|
_LiveServerGoAway_from_mldev(
|
|
api_client, getv(from_object, ['goAway']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['sessionResumptionUpdate']) is not None:
|
|
setv(
|
|
to_object,
|
|
['session_resumption_update'],
|
|
_LiveServerSessionResumptionUpdate_from_mldev(
|
|
api_client,
|
|
getv(from_object, ['sessionResumptionUpdate']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|
|
|
|
|
|
def _LiveServerMessage_from_vertex(
|
|
api_client: BaseApiClient,
|
|
from_object: Union[dict[str, Any], object],
|
|
parent_object: Optional[dict[str, Any]] = None,
|
|
) -> dict[str, Any]:
|
|
to_object: dict[str, Any] = {}
|
|
if getv(from_object, ['setupComplete']) is not None:
|
|
setv(
|
|
to_object,
|
|
['setup_complete'],
|
|
_LiveServerSetupComplete_from_vertex(
|
|
api_client, getv(from_object, ['setupComplete']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['serverContent']) is not None:
|
|
setv(
|
|
to_object,
|
|
['server_content'],
|
|
_LiveServerContent_from_vertex(
|
|
api_client, getv(from_object, ['serverContent']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['toolCall']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_call'],
|
|
_LiveServerToolCall_from_vertex(
|
|
api_client, getv(from_object, ['toolCall']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['toolCallCancellation']) is not None:
|
|
setv(
|
|
to_object,
|
|
['tool_call_cancellation'],
|
|
_LiveServerToolCallCancellation_from_vertex(
|
|
api_client, getv(from_object, ['toolCallCancellation']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['usageMetadata']) is not None:
|
|
setv(
|
|
to_object,
|
|
['usage_metadata'],
|
|
_UsageMetadata_from_vertex(
|
|
api_client, getv(from_object, ['usageMetadata']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['goAway']) is not None:
|
|
setv(
|
|
to_object,
|
|
['go_away'],
|
|
_LiveServerGoAway_from_vertex(
|
|
api_client, getv(from_object, ['goAway']), to_object
|
|
),
|
|
)
|
|
|
|
if getv(from_object, ['sessionResumptionUpdate']) is not None:
|
|
setv(
|
|
to_object,
|
|
['session_resumption_update'],
|
|
_LiveServerSessionResumptionUpdate_from_vertex(
|
|
api_client,
|
|
getv(from_object, ['sessionResumptionUpdate']),
|
|
to_object,
|
|
),
|
|
)
|
|
|
|
return to_object
|