Add webp to mimetypes (#440)

* Add webp to mimetypes

Change-Id: Ib68aecf16156b09e3261158dd5915a0c9bea642a

* format

Change-Id: I43edff4274d09ce636ceffdf39e4cbc0bbe1da49

* add-test

Change-Id: I829e3ce10e6f6a9d406ebf79662a4d8c83b48f9d

Author: MarkDaoust

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,

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"])