Add webp to mimetypes (#440)

* Add webp to mimetypes Change-Id: Ib68aecf16156b09e3261158dd5915a0c9bea642a * format Change-Id: I43edff4274d09ce636ceffdf39e4cbc0bbe1da49 * add-test Change-Id: I829e3ce10e6f6a9d406ebf79662a4d8c83b48f9d
google-gemini · Jul 8, 2024 · d9a17ea · d9a17ea
1 parent 13d2f52
commit d9a17ea
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/google/generativeai/files.py b/google/generativeai/files.py
@@ -28,6 +28,8 @@
 
 __all__ = ["upload_file", "get_file", "list_files", "delete_file"]
 
+mimetypes.add_type("image/webp", ".webp")
+
 
 def upload_file(
  path: str | pathlib.Path | os.PathLike,

diff --git a/tests/test_files.py b/tests/test_files.py
@@ -139,3 +139,10 @@ def test_video_metadata(self):
  def test_to_file_data(self, file_data):
  file_data = file_types.to_file_data(file_data)
  self.assertEqual(protos.FileData(file_uri="https://test_uri"), file_data)
+
+ def test_webp_mimetype(self):
+ self.responses["create_file"].append(protos.File())
+
+ response = genai.upload_file("test.webp")
+
+ self.assertEqual("image/webp", self.observed_requests[0]["mime_type"])