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

How does this all work?

- Web and node need to have __identical__ interfaces.
- We overload the public registerFiles method in NodeFiles
- The generator generates registerFilesInternal as private. the generator doesn't have a protected option. so we use an additional `protected async _registerFiles` wrapper so that registerFiles can call through to internalRegisterFiles.
- The public RegisterFilesParameters type uses `any` for the type of auth, because web can't access the GoogleAuth types.
- Renamed the old RegisterFilesParameters to InternalRegisterFilesParameters to abvoid conflict. this is backwards compatible - it wasn't publicly used.

PiperOrigin-RevId: 861969178

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/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'

google/genai/types.py

@@ -14026,8 +14026,8 @@ class RegisterFilesConfigDict(TypedDict, total=False):
 RegisterFilesConfigOrDict = Union[RegisterFilesConfig, RegisterFilesConfigDict]
 
 
-class _RegisterFilesParameters(_common.BaseModel):
-  """Generates the parameters for the private _Register method."""
+class _InternalRegisterFilesParameters(_common.BaseModel):
+  """Parameters for the private _Register method."""
 
   uris: Optional[list[str]] = Field(
       default=None,
@@ -14039,8 +14039,8 @@ class _RegisterFilesParameters(_common.BaseModel):
   )
 
 
-class _RegisterFilesParametersDict(TypedDict, total=False):
-  """Generates the parameters for the private _Register method."""
+class _InternalRegisterFilesParametersDict(TypedDict, total=False):
+  """Parameters for the private _Register method."""
 
   uris: Optional[list[str]]
   """The Google Cloud Storage URIs to register. Example: `gs://bucket/object`."""
@@ -14049,8 +14049,8 @@ class _RegisterFilesParametersDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
 
-_RegisterFilesParametersOrDict = Union[
-    _RegisterFilesParameters, _RegisterFilesParametersDict
+_InternalRegisterFilesParametersOrDict = Union[
+    _InternalRegisterFilesParameters, _InternalRegisterFilesParametersDict
 ]