diff --git a/google/genai/files.py b/google/genai/files.py index 2f13d8bf8..6bbbd65b0 100644 --- a/google/genai/files.py +++ b/google/genai/files.py @@ -101,6 +101,17 @@ def _GetFileParameters_to_mldev( return to_object +def _InternalRegisterFilesParameters_to_mldev( + from_object: Union[dict[str, Any], object], + parent_object: Optional[dict[str, Any]] = None, +) -> dict[str, Any]: + to_object: dict[str, Any] = {} + if getv(from_object, ['uris']) is not None: + setv(to_object, ['uris'], getv(from_object, ['uris'])) + + return to_object + + def _ListFilesConfig_to_mldev( from_object: Union[dict[str, Any], object], parent_object: Optional[dict[str, Any]] = None, @@ -152,17 +163,6 @@ def _ListFilesResponse_from_mldev( return to_object -def _RegisterFilesParameters_to_mldev( - from_object: Union[dict[str, Any], object], - parent_object: Optional[dict[str, Any]] = None, -) -> dict[str, Any]: - to_object: dict[str, Any] = {} - if getv(from_object, ['uris']) is not None: - setv(to_object, ['uris'], getv(from_object, ['uris'])) - - return to_object - - def _RegisterFilesResponse_from_mldev( from_object: Union[dict[str, Any], object], parent_object: Optional[dict[str, Any]] = None, @@ -438,7 +438,7 @@ def _register_files( uris: list[str], config: Optional[types.RegisterFilesConfigOrDict] = None, ) -> types.RegisterFilesResponse: - parameter_model = types._RegisterFilesParameters( + parameter_model = types._InternalRegisterFilesParameters( uris=uris, config=config, ) @@ -449,7 +449,7 @@ def _register_files( 'This method is only supported in the Gemini Developer client.' ) else: - request_dict = _RegisterFilesParameters_to_mldev(parameter_model) + request_dict = _InternalRegisterFilesParameters_to_mldev(parameter_model) request_url_dict = request_dict.get('_url') if request_url_dict: path = 'files:register'.format_map(request_url_dict) @@ -977,7 +977,7 @@ async def _register_files( uris: list[str], config: Optional[types.RegisterFilesConfigOrDict] = None, ) -> types.RegisterFilesResponse: - parameter_model = types._RegisterFilesParameters( + parameter_model = types._InternalRegisterFilesParameters( uris=uris, config=config, ) @@ -988,7 +988,7 @@ async def _register_files( 'This method is only supported in the Gemini Developer client.' ) else: - request_dict = _RegisterFilesParameters_to_mldev(parameter_model) + request_dict = _InternalRegisterFilesParameters_to_mldev(parameter_model) request_url_dict = request_dict.get('_url') if request_url_dict: path = 'files:register'.format_map(request_url_dict) diff --git a/google/genai/tests/files/test_register_table.py b/google/genai/tests/files/test_register_table.py index ff70dc0f2..5a8e43ae4 100644 --- a/google/genai/tests/files/test_register_table.py +++ b/google/genai/tests/files/test_register_table.py @@ -42,7 +42,7 @@ def get_headers(): test_table: list[pytest_helper.TestTableItem] = [ pytest_helper.TestTableItem( name='test_register', - parameters=types._RegisterFilesParameters(uris=['gs://unified-genai-dev/image.jpg']), + parameters=types._InternalRegisterFilesParameters(uris=['gs://unified-genai-dev/image.jpg']), exception_if_vertex='only supported in the Gemini Developer client', skip_in_api_mode=( 'The files have a TTL, they cannot be reliably retrieved for a long' diff --git a/google/genai/types.py b/google/genai/types.py index e932e0373..ade5e3db6 100644 --- a/google/genai/types.py +++ b/google/genai/types.py @@ -14026,7 +14026,7 @@ class RegisterFilesConfigDict(TypedDict, total=False): RegisterFilesConfigOrDict = Union[RegisterFilesConfig, RegisterFilesConfigDict] -class _RegisterFilesParameters(_common.BaseModel): +class _InternalRegisterFilesParameters(_common.BaseModel): """Generates the parameters for the private _Register method.""" uris: Optional[list[str]] = Field( @@ -14039,7 +14039,7 @@ class _RegisterFilesParameters(_common.BaseModel): ) -class _RegisterFilesParametersDict(TypedDict, total=False): +class _InternalRegisterFilesParametersDict(TypedDict, total=False): """Generates the parameters for the private _Register method.""" uris: Optional[list[str]] @@ -14049,8 +14049,8 @@ class _RegisterFilesParametersDict(TypedDict, total=False): """Used to override the default configuration.""" -_RegisterFilesParametersOrDict = Union[ - _RegisterFilesParameters, _RegisterFilesParametersDict +_InternalRegisterFilesParametersOrDict = Union[ + _InternalRegisterFilesParameters, _InternalRegisterFilesParametersDict ]