gemini-webapi 1.14.1__tar.gz → 1.14.2__tar.gz

{gemini_webapi-1.14.1 → gemini_webapi-1.14.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.1
+Version: 1.14.2
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{gemini_webapi-1.14.1 → gemini_webapi-1.14.2}/src/gemini_webapi/client.py

@@ -357,6 +357,7 @@ class GeminiClient:
                 if not predefined_gems and not custom_gems:
                     raise Exception
             except Exception:
+                await self.close()
                 logger.debug(f"Invalid response: {response.text}")
                 raise APIError(
                     "Failed to fetch gems. Invalid response data received. Client will try to re-initialize on next request."

{gemini_webapi-1.14.1 → gemini_webapi-1.14.2}/src/gemini_webapi/types/image.py

@@ -137,7 +137,7 @@ class GeneratedImage(Image):
         return v
 
     # @override
-    async def save(self, **kwargs) -> str | None:
+    async def save(self, full_size=True, **kwargs) -> str | None:
         """
         Save the image to disk.
 
@@ -146,6 +146,8 @@ class GeneratedImage(Image):
         filename: `str`, optional
             Filename to save the image, generated images are always in .png format, but file extension will not be included in the URL.
             And since the URL ends with a long hash, by default will use timestamp + end of the hash as the filename.
+        full_size: `bool`, optional
+            If True, will modify the default preview (512*512) URL to get the full size image.
         kwargs: `dict`, optional
             Other arguments to pass to `Image.save`.
 
@@ -155,6 +157,9 @@ class GeneratedImage(Image):
             Absolute path of the saved image if successfully saved.
         """
 
+        if full_size:
+            self.url += "=s2048"
+
         return await super().save(
             filename=kwargs.pop("filename", None)
             or f"{datetime.now().strftime('%Y%m%d%H%M%S')}_{self.url[-10:]}.png",

{gemini_webapi-1.14.1 → gemini_webapi-1.14.2}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.1
+Version: 1.14.2
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{gemini_webapi-1.14.1 → gemini_webapi-1.14.2}/tests/test_save_image.py

@@ -35,12 +35,12 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
 
     async def test_save_generated_image(self):
         response = await self.geminiclient.generate_content(
-            "Generate some pictures of random subjects"
+            "Generate a picture of random subjects"
         )
         self.assertTrue(response.images)
         for i, image in enumerate(response.images):
             self.assertTrue(image.url)
-            await image.save(verbose=True)
+            await image.save(verbose=True, full_size=True)
 
 
 if __name__ == "__main__":