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

google/genai/files.py

@@ -13,8 +13,11 @@
 # limitations under the License.
 #
 
+# Code generated by the Google Gen AI SDK generator DO NOT EDIT.
+
 import mimetypes
 import os
+import pathlib
 from typing import Optional, Union
 from urllib.parse import urlencode
 from . import _common
@@ -98,7 +101,7 @@ def _ListFilesParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -128,13 +131,13 @@ def _FileStatus_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['details']):
+  if getv(from_object, ['details']) is not None:
     raise ValueError('details parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['message']):
+  if getv(from_object, ['message']) is not None:
     raise ValueError('message parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['code']):
+  if getv(from_object, ['code']) is not None:
     raise ValueError('code parameter is not supported in Vertex AI.')
 
   return to_object
@@ -197,40 +200,40 @@ def _File_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['display_name']):
+  if getv(from_object, ['display_name']) is not None:
     raise ValueError('display_name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['mime_type']):
+  if getv(from_object, ['mime_type']) is not None:
     raise ValueError('mime_type parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['size_bytes']):
+  if getv(from_object, ['size_bytes']) is not None:
     raise ValueError('size_bytes parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['create_time']):
+  if getv(from_object, ['create_time']) is not None:
     raise ValueError('create_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['expiration_time']):
+  if getv(from_object, ['expiration_time']) is not None:
     raise ValueError('expiration_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['update_time']):
+  if getv(from_object, ['update_time']) is not None:
     raise ValueError('update_time parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['sha256_hash']):
+  if getv(from_object, ['sha256_hash']) is not None:
     raise ValueError('sha256_hash parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['uri']):
+  if getv(from_object, ['uri']) is not None:
     raise ValueError('uri parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['state']):
+  if getv(from_object, ['state']) is not None:
     raise ValueError('state parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['video_metadata']):
+  if getv(from_object, ['video_metadata']) is not None:
     raise ValueError('video_metadata parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['error']):
+  if getv(from_object, ['error']) is not None:
     raise ValueError('error parameter is not supported in Vertex AI.')
 
   return to_object
@@ -291,10 +294,10 @@ def _CreateFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['file']):
+  if getv(from_object, ['file']) is not None:
     raise ValueError('file parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -355,10 +358,10 @@ def _GetFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -419,10 +422,10 @@ def _DeleteFileParameters_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['name']):
+  if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['config']):
+  if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Vertex AI.')
 
   return to_object
@@ -615,11 +618,11 @@ class Files(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _ListFilesParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files'.format_map(request_dict.get('_url'))
 
@@ -632,23 +635,23 @@ class Files(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListFilesResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListFilesResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListFilesResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def _create(
@@ -662,11 +665,11 @@ class Files(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _CreateFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'upload/v1beta/files'.format_map(request_dict.get('_url'))
 
@@ -679,23 +682,23 @@ class Files(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _CreateFileResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _CreateFileResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.CreateFileResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def get(
@@ -723,11 +726,11 @@ class Files(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _GetFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files/{file}'.format_map(request_dict.get('_url'))
 
@@ -740,17 +743,17 @@ class Files(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _File_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _File_from_vertex(self._api_client, response_dict)
     else:
-      response_dict = _File_from_mldev(self.api_client, response_dict)
+      response_dict = _File_from_mldev(self._api_client, response_dict)
 
     return_value = types.File._from_response(response_dict, parameter_model)
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def delete(
@@ -777,11 +780,11 @@ class Files(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _DeleteFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files/{file}'.format_map(request_dict.get('_url'))
 
@@ -794,39 +797,38 @@ class Files(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'delete', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _DeleteFileResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _DeleteFileResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.DeleteFileResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def upload(
       self,
       *,
-      path: str,
+      path: str | pathlib.Path | os.PathLike,
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file using a supported file service.
 
     Args:
-      path: The path to the file or a file-like object (e.g. `BytesIO`) to be
-        uploaded.
+      path: The path or a path-like object points to the file to to be uploaded.
       config: Optional parameters to set `diplay_name`, `mime_type`, and `name`.
     """
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError(
           'Vertex AI does not support creating files. You can upload files to'
           ' GCS files instead.'
@@ -884,12 +886,12 @@ class Files(_common.BaseModule):
       )
     upload_url = response['headers']['X-Goog-Upload-URL']
 
-    return_file = self.api_client.upload_file(
+    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
+        _File_from_mldev(self._api_client, return_file['file']), None
     )
 
   def list(
@@ -929,11 +931,11 @@ class AsyncFiles(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _ListFilesParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files'.format_map(request_dict.get('_url'))
 
@@ -946,23 +948,23 @@ class AsyncFiles(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListFilesResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListFilesResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListFilesResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def _create(
@@ -976,11 +978,11 @@ class AsyncFiles(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _CreateFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'upload/v1beta/files'.format_map(request_dict.get('_url'))
 
@@ -993,23 +995,23 @@ class AsyncFiles(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _CreateFileResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _CreateFileResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.CreateFileResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def get(
@@ -1037,11 +1039,11 @@ class AsyncFiles(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _GetFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files/{file}'.format_map(request_dict.get('_url'))
 
@@ -1054,17 +1056,17 @@ class AsyncFiles(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _File_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _File_from_vertex(self._api_client, response_dict)
     else:
-      response_dict = _File_from_mldev(self.api_client, response_dict)
+      response_dict = _File_from_mldev(self._api_client, response_dict)
 
     return_value = types.File._from_response(response_dict, parameter_model)
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def delete(
@@ -1091,11 +1093,11 @@ class AsyncFiles(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError('This method is only supported in the default client.')
     else:
       request_dict = _DeleteFileParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'files/{file}'.format_map(request_dict.get('_url'))
 
@@ -1108,39 +1110,38 @@ class AsyncFiles(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'delete', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _DeleteFileResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _DeleteFileResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.DeleteFileResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def upload(
       self,
       *,
-      path: str,
+      path: str | pathlib.Path | os.PathLike,
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file asynchronously using a supported file service.
 
     Args:
-      path: The path to the file or a file-like object (e.g. `BytesIO`) to be
-        uploaded.
+      path:  The path or a path-like object points to the file to be uploaded.
       config: Optional parameters to set `diplay_name`, `mime_type`, and `name`.
     """
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       raise ValueError(
           'Vertex AI does not support creating files. You can upload files to'
           ' GCS files instead.'
@@ -1198,12 +1199,12 @@ class AsyncFiles(_common.BaseModule):
       )
     upload_url = response['headers']['X-Goog-Upload-URL']
 
-    return_file = await self.api_client.async_upload_file(
+    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
+        _File_from_mldev(self._api_client, return_file['file']), None
     )
 
   async def list(