feat: Add registerFiles for you can use gcs files with mldev.

PiperOrigin-RevId: 852951219

Author: MarkDaoust

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)

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
 ]