google-genai 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

google/genai/files.py

@@ -15,11 +15,13 @@
 
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
+import io
 import mimetypes
 import os
 import pathlib
 from typing import Optional, Union
 from urllib.parse import urlencode
+from . import _api_module
 from . import _common
 from . import _transformers as t
 from . import types
@@ -35,8 +37,6 @@ def _ListFilesConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
   if getv(from_object, ['page_size']) is not None:
     setv(
@@ -59,8 +59,6 @@ def _ListFilesConfig_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
   if getv(from_object, ['page_size']) is not None:
     setv(
@@ -176,9 +174,15 @@ def _File_to_mldev(
   if getv(from_object, ['uri']) is not None:
     setv(to_object, ['uri'], getv(from_object, ['uri']))
 
+  if getv(from_object, ['download_uri']) is not None:
+    setv(to_object, ['downloadUri'], getv(from_object, ['download_uri']))
+
   if getv(from_object, ['state']) is not None:
     setv(to_object, ['state'], getv(from_object, ['state']))
 
+  if getv(from_object, ['source']) is not None:
+    setv(to_object, ['source'], getv(from_object, ['source']))
+
   if getv(from_object, ['video_metadata']) is not None:
     setv(to_object, ['videoMetadata'], getv(from_object, ['video_metadata']))
 
@@ -227,9 +231,15 @@ def _File_to_vertex(
   if getv(from_object, ['uri']) is not None:
     raise ValueError('uri parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['download_uri']) is not None:
+    raise ValueError('download_uri parameter is not supported in Vertex AI.')
+
   if getv(from_object, ['state']) is not None:
     raise ValueError('state parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['source']) is not None:
+    raise ValueError('source parameter is not supported in Vertex AI.')
+
   if getv(from_object, ['video_metadata']) is not None:
     raise ValueError('video_metadata parameter is not supported in Vertex AI.')
 
@@ -239,30 +249,6 @@ def _File_to_vertex(
   return to_object
 
 
-def _CreateFileConfig_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
-def _CreateFileConfig_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
 def _CreateFileParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -277,13 +263,7 @@ def _CreateFileParameters_to_mldev(
     )
 
   if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _CreateFileConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
+    setv(to_object, ['config'], getv(from_object, ['config']))
 
   return to_object
 
@@ -303,30 +283,6 @@ def _CreateFileParameters_to_vertex(
   return to_object
 
 
-def _GetFileConfig_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
-def _GetFileConfig_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
 def _GetFileParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -341,13 +297,7 @@ def _GetFileParameters_to_mldev(
     )
 
   if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _GetFileConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
+    setv(to_object, ['config'], getv(from_object, ['config']))
 
   return to_object
 
@@ -367,30 +317,6 @@ def _GetFileParameters_to_vertex(
   return to_object
 
 
-def _DeleteFileConfig_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
-def _DeleteFileConfig_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
 def _DeleteFileParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -405,13 +331,7 @@ def _DeleteFileParameters_to_mldev(
     )
 
   if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _DeleteFileConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
+    setv(to_object, ['config'], getv(from_object, ['config']))
 
   return to_object
 
@@ -431,6 +351,16 @@ def _DeleteFileParameters_to_vertex(
   return to_object
 
 
+def _FileState_to_vertex_enum_validate(enum_value: any):
+  if enum_value in set(['STATE_UNSPECIFIED', 'PROCESSING', 'ACTIVE', 'FAILED']):
+    raise ValueError(f'{enum_value} enum value is not supported in Vertex AI.')
+
+
+def _FileSource_to_vertex_enum_validate(enum_value: any):
+  if enum_value in set(['SOURCE_UNSPECIFIED', 'UPLOADED', 'GENERATED']):
+    raise ValueError(f'{enum_value} enum value is not supported in Vertex AI.')
+
+
 def _FileStatus_from_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -492,9 +422,15 @@ def _File_from_mldev(
   if getv(from_object, ['uri']) is not None:
     setv(to_object, ['uri'], getv(from_object, ['uri']))
 
+  if getv(from_object, ['downloadUri']) is not None:
+    setv(to_object, ['download_uri'], getv(from_object, ['downloadUri']))
+
   if getv(from_object, ['state']) is not None:
     setv(to_object, ['state'], getv(from_object, ['state']))
 
+  if getv(from_object, ['source']) is not None:
+    setv(to_object, ['source'], getv(from_object, ['source']))
+
   if getv(from_object, ['videoMetadata']) is not None:
     setv(to_object, ['video_metadata'], getv(from_object, ['videoMetadata']))
 
@@ -592,7 +528,7 @@ def _DeleteFileResponse_from_vertex(
   return to_object
 
 
-class Files(_common.BaseModule):
+class Files(_api_module.BaseModule):
 
   def _list(
       self, *, config: Optional[types.ListFilesConfigOrDict] = None
@@ -630,10 +566,16 @@ class Files(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = self._api_client.request(
         'get', path, request_dict, http_options
@@ -649,7 +591,7 @@ class Files(_common.BaseModule):
       )
 
     return_value = types.ListFilesResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -677,10 +619,16 @@ class Files(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = self._api_client.request(
         'post', path, request_dict, http_options
@@ -696,7 +644,7 @@ class Files(_common.BaseModule):
       )
 
     return_value = types.CreateFileResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -738,10 +686,16 @@ class Files(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = self._api_client.request(
         'get', path, request_dict, http_options
@@ -752,7 +706,9 @@ class Files(_common.BaseModule):
     else:
       response_dict = _File_from_mldev(self._api_client, response_dict)
 
-    return_value = types.File._from_response(response_dict, parameter_model)
+    return_value = types.File._from_response(
+        response=response_dict, kwargs=parameter_model
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -792,10 +748,16 @@ class Files(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = self._api_client.request(
         'delete', path, request_dict, http_options
@@ -811,7 +773,7 @@ class Files(_common.BaseModule):
       )
 
     return_value = types.DeleteFileResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -819,13 +781,16 @@ class Files(_common.BaseModule):
   def upload(
       self,
       *,
-      path: str | pathlib.Path | os.PathLike,
+      path: Union[str, pathlib.Path, os.PathLike, io.IOBase],
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file using a supported file service.
 
     Args:
-      path: The path or a path-like object points to the file to to be uploaded.
+      path: The path to the file or an `IOBase` object to be uploaded. If it's
+        an IOBase object, it must be opened in blocking mode and binary mode. In
+        other words, do not use non-blocking mode or text mode. The given stream
+        must be seekable, that is, it must be able to call seek() on 'path'.
       config: Optional parameters to set `diplay_name`, `mime_type`, and `name`.
     """
     if self._api_client.vertexai:
@@ -849,17 +814,31 @@ class Files(_common.BaseModule):
     if file.name is not None and not file.name.startswith('files/'):
       file.name = f'files/{file.name}'
 
-    fs_path = os.fspath(path)
-    if not fs_path or not os.path.isfile(fs_path):
-      raise FileNotFoundError(f'{path} is not a valid file path.')
-    file.size_bytes = os.path.getsize(fs_path)
-    if file.mime_type is None:
-      file.mime_type, _ = mimetypes.guess_type(fs_path)
-    if file.mime_type is None:
-      raise ValueError(
-          'Unknown mime type: Could not determine the mimetype for your file\n'
-          '    please set the `mime_type` argument'
-      )
+    if isinstance(path, io.IOBase):
+      if file.mime_type is None:
+        raise ValueError(
+            'Unknown mime type: Could not determine the mimetype for your'
+            ' file\n please set the `mime_type` argument'
+        )
+      if hasattr(path, 'mode'):
+        if 'b' not in path.mode:
+          raise ValueError('The file must be opened in binary mode.')
+      offset = path.tell()
+      path.seek(0, os.SEEK_END)
+      file.size_bytes = path.tell() - offset
+      path.seek(offset, os.SEEK_SET)
+    else:
+      fs_path = os.fspath(path)
+      if not fs_path or not os.path.isfile(fs_path):
+        raise FileNotFoundError(f'{path} is not a valid file path.')
+      file.size_bytes = os.path.getsize(fs_path)
+      if file.mime_type is None:
+        file.mime_type, _ = mimetypes.guess_type(fs_path)
+      if file.mime_type is None:
+        raise ValueError(
+            'Unknown mime type: Could not determine the mimetype for your'
+            ' file\n    please set the `mime_type` argument'
+        )
     response = {}
     if config_model and config_model.http_options:
       http_options = config_model.http_options
@@ -873,9 +852,10 @@ class Files(_common.BaseModule):
               'X-Goog-Upload-Header-Content-Length': f'{file.size_bytes}',
               'X-Goog-Upload-Header-Content-Type': f'{file.mime_type}',
           },
-          'response_payload': response,
+          'deprecated_response_payload': response,
       }
     self._create(file=file, config={'http_options': http_options})
+
     if (
         'headers' not in response
         or 'X-Goog-Upload-URL' not in response['headers']
@@ -886,9 +866,14 @@ class Files(_common.BaseModule):
       )
     upload_url = response['headers']['X-Goog-Upload-URL']
 
-    return_file = self._api_client.upload_file(
-        fs_path, upload_url, file.size_bytes
-    )
+    if isinstance(path, io.IOBase):
+      return_file = self._api_client.upload_file(
+          path, upload_url, file.size_bytes
+      )
+    else:
+      return_file = self._api_client.upload_file(
+          fs_path, upload_url, file.size_bytes
+      )
 
     return types.File._from_response(
         _File_from_mldev(self._api_client, return_file['file']), None
@@ -904,8 +889,76 @@ class Files(_common.BaseModule):
         config,
     )
 
+  def download(
+      self,
+      *,
+      file: Union[str, types.File],
+      config: Optional[types.DownloadFileConfigOrDict] = None,
+  ) -> bytes:
+    """Downloads a file's data from storage.
+
+    Files created by `upload` can't be downloaded. You can tell which files are
+    downloadable by checking the `source` or `download_uri` property.
+
+    Args:
+      file (str): A file name, uri, or file object. Identifying which file to
+        download.
+      config (DownloadFileConfigOrDict): Optional, configuration for the get
+        method.
+
+    Returns:
+      File: The file data as bytes.
+
+    Usage:
+
+    .. code-block:: python
+
+      for file client.files.list():
+        if file.download_uri is not None:
+          break
+      else:
+        raise ValueError('No files found with a `download_uri`.')
+      data = client.files.download(file=file)
+      # data = client.files.download(file=file.name)
+      # data = client.files.download(file=file.download_uri)
+    """
+    if self._api_client.vertexai:
+      raise ValueError(
+          'Vertex AI does not support the Files API. Use GCS files instead.'
+      )
+
+    config_model = None
+    if config:
+      if isinstance(config, dict):
+        config_model = types.DownloadFileConfig(**config)
+      else:
+        config_model = config
+
+    if isinstance(file, types.File) and file.download_uri is None:
+      raise ValueError(
+          "Only generated files can be downloaded, uploaded files can't be "
+          'downloaded. You can tell which files are downloadable by checking '
+          'the `source` or `download_uri` property.'
+      )
+    name = t.t_file_name(self, file)
+
+    path = f'files/{name}:download'
+
+    query_params = {'alt': 'media'}
+    path = f'{path}?{urlencode(query_params)}'
+    http_options = None
+    if getv(config_model, ['http_options']) is not None:
+      http_options = getv(config_model, ['http_options'])
+
+    data = self._api_client.download_file(
+        path,
+        http_options,
+    )
+
+    return data
+
 
-class AsyncFiles(_common.BaseModule):
+class AsyncFiles(_api_module.BaseModule):
 
   async def _list(
       self, *, config: Optional[types.ListFilesConfigOrDict] = None
@@ -943,10 +996,16 @@ class AsyncFiles(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
@@ -962,7 +1021,7 @@ class AsyncFiles(_common.BaseModule):
       )
 
     return_value = types.ListFilesResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -990,10 +1049,16 @@ class AsyncFiles(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = await self._api_client.async_request(
         'post', path, request_dict, http_options
@@ -1009,7 +1074,7 @@ class AsyncFiles(_common.BaseModule):
       )
 
     return_value = types.CreateFileResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1051,10 +1116,16 @@ class AsyncFiles(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
@@ -1065,7 +1136,9 @@ class AsyncFiles(_common.BaseModule):
     else:
       response_dict = _File_from_mldev(self._api_client, response_dict)
 
-    return_value = types.File._from_response(response_dict, parameter_model)
+    return_value = types.File._from_response(
+        response=response_dict, kwargs=parameter_model
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -1105,10 +1178,16 @@ class AsyncFiles(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
     response_dict = await self._api_client.async_request(
         'delete', path, request_dict, http_options
@@ -1124,7 +1203,7 @@ class AsyncFiles(_common.BaseModule):
       )
 
     return_value = types.DeleteFileResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1132,13 +1211,16 @@ class AsyncFiles(_common.BaseModule):
   async def upload(
       self,
       *,
-      path: str | pathlib.Path | os.PathLike,
+      path: Union[str, pathlib.Path, os.PathLike, io.IOBase],
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file asynchronously using a supported file service.
 
     Args:
-      path:  The path or a path-like object points to the file to be uploaded.
+      path:  The path to the file or an `IOBase` object to be uploaded. If it's
+        an IOBase object, it must be opened in blocking mode and binary mode. In
+        other words, do not use non-blocking mode or text mode. The given stream
+        must be seekable, that is, it must be able to call seek() on 'path'.
       config: Optional parameters to set `diplay_name`, `mime_type`, and `name`.
     """
     if self._api_client.vertexai:
@@ -1162,17 +1244,32 @@ class AsyncFiles(_common.BaseModule):
     if file.name is not None and not file.name.startswith('files/'):
       file.name = f'files/{file.name}'
 
-    fs_path = os.fspath(path)
-    if not fs_path or not os.path.isfile(fs_path):
-      raise FileNotFoundError(f'{path} is not a valid file path.')
-    file.size_bytes = os.path.getsize(fs_path)
-    if file.mime_type is None:
-      file.mime_type, _ = mimetypes.guess_type(fs_path)
-    if file.mime_type is None:
-      raise ValueError(
-          'Unknown mime type: Could not determine the mimetype for your file\n'
-          '    please set the `mime_type` argument'
-      )
+    if isinstance(path, io.IOBase):
+      if file.mime_type is None:
+        raise ValueError(
+            'Unknown mime type: Could not determine the mimetype for your'
+            ' file\n    please set the `mime_type` argument'
+        )
+      if hasattr(path, 'mode'):
+        if 'b' not in path.mode:
+          raise ValueError('The file must be opened in binary mode.')
+      offset = path.tell()
+      path.seek(0, os.SEEK_END)
+      file.size_bytes = path.tell() - offset
+      path.seek(offset, os.SEEK_SET)
+    else:
+      fs_path = os.fspath(path)
+      if not fs_path or not os.path.isfile(fs_path):
+        raise FileNotFoundError(f'{path} is not a valid file path.')
+      file.size_bytes = os.path.getsize(fs_path)
+      if file.mime_type is None:
+        file.mime_type, _ = mimetypes.guess_type(fs_path)
+      if file.mime_type is None:
+        raise ValueError(
+            'Unknown mime type: Could not determine the mimetype for your'
+            ' file\n    please set the `mime_type` argument'
+        )
+
     response = {}
     if config_model and config_model.http_options:
       http_options = config_model.http_options
@@ -1186,7 +1283,7 @@ class AsyncFiles(_common.BaseModule):
               'X-Goog-Upload-Header-Content-Length': f'{file.size_bytes}',
               'X-Goog-Upload-Header-Content-Type': f'{file.mime_type}',
           },
-          'response_payload': response,
+          'deprecated_response_payload': response,
       }
     await self._create(file=file, config={'http_options': http_options})
     if (
@@ -1199,9 +1296,14 @@ class AsyncFiles(_common.BaseModule):
       )
     upload_url = response['headers']['X-Goog-Upload-URL']
 
-    return_file = await self._api_client.async_upload_file(
-        fs_path, upload_url, file.size_bytes
-    )
+    if isinstance(path, io.IOBase):
+      return_file = await self._api_client.async_upload_file(
+          path, upload_url, file.size_bytes
+      )
+    else:
+      return_file = await self._api_client.async_upload_file(
+          fs_path, upload_url, file.size_bytes
+      )
 
     return types.File._from_response(
         _File_from_mldev(self._api_client, return_file['file']), None
@@ -1216,3 +1318,69 @@ class AsyncFiles(_common.BaseModule):
         await self._list(config=config),
         config,
     )
+
+  async def download(
+      self,
+      *,
+      file: Union[str, types.File],
+      config: Optional[types.DownloadFileConfigOrDict] = None,
+  ) -> bytes:
+    """Downloads a file's data from the file service.
+
+    The Vertex-AI implementation of the API foes not include the file service.
+
+    Files created by `upload` can't be downloaded. You can tell which files are
+    downloadable by checking the `download_uri` property.
+
+    Args:
+      File (str): A file name, uri, or file object. Identifying which file to
+        download.
+      config (DownloadFileConfigOrDict): Optional, configuration for the get
+        method.
+
+    Returns:
+      File: The file data as bytes.
+
+    Usage:
+
+    .. code-block:: python
+
+      for file client.files.list():
+        if file.download_uri is not None:
+          break
+      else:
+        raise ValueError('No files found with a `download_uri`.')
+      data = client.files.download(file=file)
+      # data = client.files.download(file=file.name)
+      # data = client.files.download(file=file.uri)
+    """
+    if self._api_client.vertexai:
+      raise ValueError(
+          'Vertex AI does not support the Files API. Use GCS files instead.'
+      )
+
+    config_model = None
+    if config:
+      if isinstance(config, dict):
+        config_model = types.DownloadFileConfig(**config)
+      else:
+        config_model = config
+
+    name = t.t_file_name(self, file)
+
+    path = f'files/{name}:download'
+
+    http_options = None
+    if getv(config_model, ['http_options']) is not None:
+      http_options = getv(config_model, ['http_options'])
+
+    query_params = {'alt': 'media'}
+    if query_params:
+      path = f'{path}?{urlencode(query_params)}'
+
+    data = await self._api_client.async_download_file(
+        path,
+        http_options,
+    )
+
+    return data