gemini-webapi 1.9.1__py3-none-any.whl → 1.10.0__py3-none-any.whl

gemini_webapi/client.py

@@ -13,6 +13,7 @@ from .exceptions import AuthError, APIError, TimeoutError, GeminiError
 from .types import WebImage, GeneratedImage, Candidate, ModelOutput
 from .utils import (
     upload_file,
+    parse_file_name,
     rotate_1psidts,
     get_access_token,
     load_browser_cookies,
@@ -263,7 +264,7 @@ class GeminiClient:
     async def generate_content(
         self,
         prompt: str,
-        images: list[bytes | str | Path] | None = None,
+        files: list[str | Path] | None = None,
         model: Model | str = Model.UNSPECIFIED,
         chat: Optional["ChatSession"] = None,
         **kwargs,
@@ -275,8 +276,8 @@ class GeminiClient:
         ----------
         prompt: `str`
             Prompt provided by user.
-        images: `list[bytes | str | Path]`, optional
-            List of image file paths or file data in bytes.
+        files: `list[str | Path]`, optional
+            List of file paths to be attached.
         model: `Model` | `str`, optional
             Specify the model to use for generation.
             Pass either a `gemini_webapi.constants.Model` enum or a model name string.
@@ -324,17 +325,17 @@ class GeminiClient:
                             None,
                             json.dumps(
                                 [
-                                    images
+                                    files
                                     and [
                                         prompt,
                                         0,
                                         None,
                                         [
                                             [
-                                                [await upload_file(image, self.proxy)],
-                                                "filename.jpg",
+                                                [await upload_file(file, self.proxy)],
+                                                parse_file_name(file),
                                             ]
-                                            for image in images
+                                            for file in files
                                         ],
                                     ]
                                     or [prompt],
@@ -550,7 +551,7 @@ class ChatSession:
     async def send_message(
         self,
         prompt: str,
-        images: list[bytes | str | Path] | None = None,
+        files: list[str | Path] | None = None,
         **kwargs,
     ) -> ModelOutput:
         """
@@ -561,8 +562,8 @@ class ChatSession:
         ----------
         prompt: `str`
             Prompt provided by user.
-        images: `list[bytes | str | Path]`, optional
-            List of image file paths or file data in bytes.
+        files: `list[str | Path]`, optional
+            List of file paths to be attached.
         kwargs: `dict`, optional
             Additional arguments which will be passed to the post request.
             Refer to `httpx.AsyncClient.request` for more information.
@@ -587,7 +588,7 @@ class ChatSession:
         """
 
         return await self.geminiclient.generate_content(
-            prompt=prompt, images=images, model=self.model, chat=self, **kwargs
+            prompt=prompt, files=files, model=self.model, chat=self, **kwargs
         )
 
     def choose_candidate(self, index: int) -> ModelOutput:

gemini_webapi/constants.py

@@ -44,7 +44,7 @@ class Model(Enum):
         "gemini-2.0-flash-thinking-with-apps",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f8f8f5ea629f5d37"]'},
         False,
-    )
+    )  # Deprecated, should be removed in the future
     G_2_0_EXP_ADVANCED = (
         "gemini-2.0-exp-advanced",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
@@ -54,17 +54,17 @@ class Model(Enum):
         "gemini-1.5-flash",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"418ab5ea040b5c43"]'},
         False,
-    )
+    )  # Deprecated, should be removed in the future
     G_1_5_PRO = (
         "gemini-1.5-pro",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"9d60dfae93c9ff1f"]'},
         True,
-    )
+    )  # Deprecated, should be removed in the future
     G_1_5_PRO_RESEARCH = (
         "gemini-1.5-pro-research",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"e5a44cb1dae2b489"]'},
         True,
-    )
+    )  # Deprecated, should be removed in the future
 
     def __init__(self, name, header, advanced_only):
         self.model_name = name

gemini_webapi/utils/__init__.py

@@ -1,6 +1,6 @@
 from asyncio import Task
 
-from .upload_file import upload_file  # noqa: F401
+from .upload_file import upload_file, parse_file_name  # noqa: F401
 from .rotate_1psidts import rotate_1psidts  # noqa: F401
 from .get_access_token import get_access_token  # noqa: F401
 from .load_browser_cookies import load_browser_cookies  # noqa: F401

gemini_webapi/utils/upload_file.py

@@ -7,14 +7,14 @@ from ..constants import Endpoint, Headers
 
 
 @validate_call
-async def upload_file(file: bytes | str | Path, proxy: str | None = None) -> str:
+async def upload_file(file: str | Path, proxy: str | None = None) -> str:
     """
     Upload a file to Google's server and return its identifier.
 
     Parameters
     ----------
-    file : `bytes` | `str` | `Path`
-        File data in bytes, or path to the file to be uploaded.
+    file : `str` | `Path`
+        Path to the file to be uploaded.
     proxy: `str`, optional
         Proxy URL.
 
@@ -30,9 +30,8 @@ async def upload_file(file: bytes | str | Path, proxy: str | None = None) -> str
         If the upload request failed.
     """
 
-    if not isinstance(file, bytes):
-        with open(file, "rb") as f:
-            file = f.read()
+    with open(file, "rb") as f:
+        file = f.read()
 
     async with AsyncClient(http2=True, proxy=proxy) as client:
         response = await client.post(
@@ -43,3 +42,25 @@ async def upload_file(file: bytes | str | Path, proxy: str | None = None) -> str
         )
         response.raise_for_status()
         return response.text
+
+
+def parse_file_name(file: str | Path) -> str:
+    """
+    Parse the file name from the given path.
+
+    Parameters
+    ----------
+    file : `str` | `Path`
+        Path to the file.
+
+    Returns
+    -------
+    `str`
+        File name with extension.
+    """
+
+    file = Path(file)
+    if not file.is_file():
+        raise ValueError(f"{file} is not a valid file.")
+
+    return file.name

{gemini_webapi-1.9.1.dist-info → gemini_webapi-1.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.9.1
+Version: 1.10.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -712,7 +712,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
 ## Features
 
 - **Persistent Cookies** - Automatically refreshes cookies in background. Optimized for always-on services.
-- **ImageFx Support** - Supports retrieving images generated by ImageFx, Google's latest AI image generator.
+- **Image Generation** - Natively supports generating and modifying images with natural language.
 - **Extension Support** - Supports generating contents with [Gemini extensions](https://gemini.google.com/extensions) on, like YouTube and Gmail.
 - **Classified Outputs** - Automatically categorizes texts, web images and AI generated images in the response.
 - **Official Flavor** - Provides a simple and elegant interface inspired by [Google Generative AI](https://ai.google.dev/tutorials/python_quickstart)'s official API.
@@ -728,13 +728,12 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [Initialization](#initialization)
   - [Select language model](#select-language-model)
   - [Generate contents from text](#generate-contents-from-text)
-  - [Generate contents from image](#generate-contents-from-image)
+  - [Generate contents with files](#generate-contents-with-files)
   - [Conversations across multiple turns](#conversations-across-multiple-turns)
   - [Continue previous conversations](#continue-previous-conversations)
   - [Retrieve model's thought process](#retrieve-models-thought-process)
   - [Retrieve images in response](#retrieve-images-in-response)
-  - [Generate images with ImageFx](#generate-images-with-imagefx)
-  - [Save images to local files](#save-images-to-local-files)
+  - [Generate images with Imagen3](#generate-images-with-imagen3)
   - [Generate contents with Gemini extensions](#generate-contents-with-gemini-extensions)
   - [Check and switch to other reply candidates](#check-and-switch-to-other-reply-candidates)
   - [Control log level](#control-log-level)
@@ -868,15 +867,15 @@ asyncio.run(main())
 >
 > Simply use `print(response)` to get the same output if you just want to see the response text
 
-### Generate contents from image
+### Generate contents with files
 
-Gemini supports image recognition and generating contents from images. Optionally, you can pass images in a list of file data in `bytes` or their paths in `str` or `pathlib.Path` to `GeminiClient.generate_content` together with text prompt.
+Gemini supports file input, including images and documents. Optionally, you can pass files as a list of paths in `str` or `pathlib.Path` to `GeminiClient.generate_content` together with text prompt.
 
 ```python
 async def main():
     response = await client.generate_content(
-            "Describe each of these images",
-            images=["assets/banner.png", "assets/favicon.png"],
+            "Introduce the contents of these two files. Is there any connection between them?",
+            files=["assets/sample.pdf", Path("assets/banner.png")],
         )
     print(response.text)
 
@@ -890,9 +889,15 @@ If you want to keep conversation continuous, please use `GeminiClient.start_chat
 ```python
 async def main():
     chat = client.start_chat()
-    response1 = await chat.send_message("Briefly introduce Europe")
-    response2 = await chat.send_message("What's the population there?")
-    print(response1.text, response2.text, sep="\n\n----------------------------------\n\n")
+    response1 = await chat.send_message(
+        "Introduce the contents of these two files. Is there any connection between them?",
+        files=["assets/sample.pdf", Path("assets/banner.png")],
+    )
+    print(response1.text)
+    response2 = await chat.send_message(
+        "Use image generation tool to modify the banner with another font and design."
+    )
+    print(response2.text, response2.images, sep="\n\n----------------------------------\n\n")
 
 asyncio.run(main())
 ```
@@ -950,24 +955,27 @@ async def main():
 asyncio.run(main())
 ```
 
-### Generate images with ImageFx
+### Generate images with Imagen3
 
-In February 2022, Google introduced a new AI image generator called ImageFx and integrated it into Gemini. You can ask Gemini to generate images with ImageFx simply by natural language.
+You can ask Gemini to generate and modify images with Imagen3, Google's latest AI image generator, simply by natural language.
 
 > [!IMPORTANT]
 >
-> Google has some limitations on the image generation feature in Gemini, so its availability could be different per region/account. Here's a summary copied from [official documentation](https://support.google.com/gemini/answer/14286560) (as of February 15th, 2024):
+> Google has some limitations on the image generation feature in Gemini, so its availability could be different per region/account. Here's a summary copied from [official documentation](https://support.google.com/gemini/answer/14286560) (as of March 19th, 2025):
 >
-> > Image generation in Gemini Apps is available in most countries, except in the European Economic Area (EEA), Switzerland, and the UK. It’s only available for **English prompts**.
-> >
 > > This feature’s availability in any specific Gemini app is also limited to the supported languages and countries of that app.
 > >
 > > For now, this feature isn’t available to users under 18.
+> >
+> > To use this feature, you must be signed in to Gemini Apps.
+
+You can save images returned from Gemini to local by calling `Image.save()`. Optionally, you can specify the file path and file name by passing `path` and `filename` arguments to the function and skip images with invalid file names by passing `skip_invalid_filename=True`. Works for both `WebImage` and `GeneratedImage`.
 
 ```python
 async def main():
     response = await client.generate_content("Generate some pictures of cats")
-    for image in response.images:
+    for i, image in enumerate(response.images):
+        await image.save(path="temp/", filename=f"cat_{i}.png", verbose=True)
         print(image, "\n\n----------------------------------\n")
 
 asyncio.run(main())
@@ -977,32 +985,17 @@ asyncio.run(main())
 >
 > by default, when asked to send images (like the previous example), Gemini will send images fetched from web instead of generating images with AI model, unless you specifically require to "generate" images in your prompt. In this package, web images and generated images are treated differently as `WebImage` and `GeneratedImage`, and will be automatically categorized in the output.
 
-### Save images to local files
-
-You can save images returned from Gemini to local files under `/temp` by calling `Image.save()`. Optionally, you can specify the file path and file name by passing `path` and `filename` arguments to the function and skip images with invalid file names by passing `skip_invalid_filename=True`. Works for both `WebImage` and `GeneratedImage`.
-
-```python
-async def main():
-    response = await client.generate_content("Generate some pictures of cats")
-    for i, image in enumerate(response.images):
-        await image.save(path="temp/", filename=f"cat_{i}.png", verbose=True)
-
-asyncio.run(main())
-```
-
 ### Generate contents with Gemini extensions
 
 > [!IMPORTANT]
 >
-> To access Gemini extensions in API, you must activate them on the [Gemini website](https://gemini.google.com/extensions) first. Same as image generation, Google also has limitations on the availability of Gemini extensions. Here's a summary copied from [official documentation](https://support.google.com/gemini/answer/13695044) (as of February 18th, 2024):
+> To access Gemini extensions in API, you must activate them on the [Gemini website](https://gemini.google.com/extensions) first. Same as image generation, Google also has limitations on the availability of Gemini extensions. Here's a summary copied from [official documentation](https://support.google.com/gemini/answer/13695044) (as of March 19th, 2025):
 >
-> > To use extensions in Gemini Apps:
-> >
-> > Sign in with your personal Google Account that you manage on your own. Extensions, including the Google Workspace extension, are currently not available to Google Workspace accounts for school, business, or other organizations.
+> > To connect apps to Gemini, you must have​​​​ Gemini Apps Activity on.
 > >
-> > Have Gemini Apps Activity on. Extensions are only available when Gemini Apps Activity is turned on.
+> > To use this feature, you must be signed in to Gemini Apps.
 > >
-> > Important: For now, extensions are available in **English, Japanese, and Korean** only.
+> > Important: If you’re under 18, Google Workspace and Maps apps currently only work with English prompts in Gemini.
 
 After activating extensions for your account, you can access them in your prompts either by natural language or by starting your prompt with "@" followed by the extension keyword.
 

{gemini_webapi-1.9.1.dist-info → gemini_webapi-1.10.0.dist-info}/RECORD

@@ -1,19 +1,19 @@
 gemini_webapi/__init__.py,sha256=28uNIywK4vCXxENaSagNWUzhqr1RyNtLzDF6WRRM4KQ,194
-gemini_webapi/client.py,sha256=4ptFBYy-ipcxox1x7WDOK04YMYU_cu6NwS8CSrmz3kM,23105
-gemini_webapi/constants.py,sha256=_8TdJH3dvzSDo0qfhKCAerWXBbH71u0B0mSGOkkJQ5w,2737
+gemini_webapi/client.py,sha256=WsH07qepNSQzq4pdYIJW393Jr4gcdp8D64ti6GlDj9E,23065
+gemini_webapi/constants.py,sha256=pg0JiSbUgrsMqVkWwPSPRqKfKIo2qpDAEJ7Kl-XOBc4,2925
 gemini_webapi/exceptions.py,sha256=6e-EXHGApi4iC0GDw7RKc3YqVK8UvEkHYaJyGQbReLw,548
 gemini_webapi/types/__init__.py,sha256=d2kvXnE004s2E2KDmPPLi5N-BQ59FgDSlrGrO3Wphww,163
 gemini_webapi/types/candidate.py,sha256=dMoGr53WR7FYDRrXpG9Yd_n9YTHGwGaFHZ8zPt8RMWw,1178
 gemini_webapi/types/image.py,sha256=4BC8hxAWJrYFwzA60CivF1di4RZkzPKjcaSPPFKmRdY,5237
 gemini_webapi/types/modeloutput.py,sha256=h07kQOkL5r-oPLvZ59uVtO1eP4FGy5ZpzuYQzAeQdr8,1196
-gemini_webapi/utils/__init__.py,sha256=mcm1kgQ5HHKyZrhHS-rd_GXbKMpIDUsq02XtmlQNN_I,357
+gemini_webapi/utils/__init__.py,sha256=tO6Sx-3fcPeyITZcvUmcFKBFlR5XW87xUFbNrIh3_mE,374
 gemini_webapi/utils/get_access_token.py,sha256=uyb6tuzPr3mHttCjiM86M29ykrnHqsUClYdf5sVkyEQ,5465
 gemini_webapi/utils/load_browser_cookies.py,sha256=A5n_VsB7Rm8ck5lpy856UNJEhv30l3dvQ3j0g3ln1fE,1535
 gemini_webapi/utils/logger.py,sha256=PF4ROQq7scRRrWzeYdeYiYs2S2Jqr0bgjyrPbXVOCqE,816
 gemini_webapi/utils/rotate_1psidts.py,sha256=NyQ9OYPLBOcvpc8bodvEYDIVFrsYN0kdfc831lPEctM,1680
-gemini_webapi/utils/upload_file.py,sha256=A-ZQ54gy2wOVA_giEEVLQQK5zTJfrgq99Z5YYYOAFYc,1163
-gemini_webapi-1.9.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.9.1.dist-info/METADATA,sha256=LXrhPGqkC1MfBvX868TaWV-xKuVpQmsDbi90KSrKHzg,57797
-gemini_webapi-1.9.1.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
-gemini_webapi-1.9.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.9.1.dist-info/RECORD,,
+gemini_webapi/utils/upload_file.py,sha256=SJOMr6kryK_ClrKmqI96fqZBNFOMPsyAvFINAGAU3rk,1468
+gemini_webapi-1.10.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.10.0.dist-info/METADATA,sha256=j5G5uhgjXjfAUYwK8UmWVP85ZCXoAxdwVLrsiu7FtKQ,57353
+gemini_webapi-1.10.0.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
+gemini_webapi-1.10.0.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.10.0.dist-info/RECORD,,