gemini-webapi 1.15.1__py3-none-any.whl → 1.16.0__py3-none-any.whl

gemini_webapi/client.py

@@ -32,7 +32,6 @@ from .utils import (
     parse_file_name,
     rotate_1psidts,
     get_access_token,
-    load_browser_cookies,
     running,
     rotate_tasks,
     logger,
@@ -101,24 +100,14 @@ class GeminiClient(GemMixin):
         self.refresh_interval: float = 540
         self.kwargs = kwargs
 
-        # Validate cookies
         if secure_1psid:
             self.cookies["__Secure-1PSID"] = secure_1psid
             if secure_1psidts:
                 self.cookies["__Secure-1PSIDTS"] = secure_1psidts
-        else:
-            try:
-                cookies = load_browser_cookies(domain_name="google.com")
-                if not (cookies and cookies.get("__Secure-1PSID")):
-                    raise ValueError(
-                        "Failed to load cookies from local browser. Please pass cookie values manually."
-                    )
-            except ImportError:
-                pass
 
     async def init(
         self,
-        timeout: float = 30,
+        timeout: float = 300,
         auto_close: bool = False,
         close_delay: float = 300,
         auto_refresh: bool = True,
@@ -371,6 +360,11 @@ class GeminiClient(GemMixin):
                             raise UsageLimitExceeded(
                                 f"Failed to generate contents. Usage limit of {model.model_name} model has exceeded. Please try switching to another model."
                             )
+                        case ErrorCode.MODEL_INCONSISTENT:
+                            raise ModelInvalid(
+                                "Failed to generate contents. The specified model is inconsistent with the chat history. Please make sure to pass the same "
+                                "`model` parameter when starting a chat session with previous metadata."
+                            )
                         case ErrorCode.MODEL_HEADER_INVALID:
                             raise ModelInvalid(
                                 "Failed to generate contents. The specified model is not available. Please update gemini_webapi to the latest version. "

gemini_webapi/constants.py

@@ -16,6 +16,7 @@ class GRPC(StrEnum):
     """
 
     # Chat methods
+    LIST_CHATS = "MaZiqc"
     READ_CHAT = "hNvQHb"
 
     # Gem methods
@@ -84,5 +85,6 @@ class ErrorCode(IntEnum):
     """
 
     USAGE_LIMIT_EXCEEDED = 1037
+    MODEL_INCONSISTENT = 1050
     MODEL_HEADER_INVALID = 1052
     IP_TEMPORARILY_BLOCKED = 1060

gemini_webapi/utils/get_access_token.py

@@ -1,3 +1,4 @@
+import os
 import re
 import asyncio
 from asyncio import Task
@@ -88,7 +89,11 @@ async def get_access_token(
         )
 
     # Cached cookies in local file
-    cache_dir = Path(__file__).parent / "temp"
+    cache_dir = (
+        (GEMINI_COOKIE_PATH := os.getenv("GEMINI_COOKIE_PATH"))
+        and Path(GEMINI_COOKIE_PATH)
+        or (Path(__file__).parent / "temp")
+    )
     if "__Secure-1PSID" in base_cookies:
         filename = f".cached_1psidts_{base_cookies['__Secure-1PSID']}.txt"
         cache_file = cache_dir / filename
@@ -126,17 +131,35 @@ async def get_access_token(
 
     # Browser cookies (if browser-cookie3 is installed)
     try:
+        valid_browser_cookies = 0
         browser_cookies = load_browser_cookies(
             domain_name="google.com", verbose=verbose
         )
-        if browser_cookies and (secure_1psid := browser_cookies.get("__Secure-1PSID")):
-            local_cookies = {"__Secure-1PSID": secure_1psid}
-            if secure_1psidts := browser_cookies.get("__Secure-1PSIDTS"):
-                local_cookies["__Secure-1PSIDTS"] = secure_1psidts
-            if nid := browser_cookies.get("NID"):
-                local_cookies["NID"] = nid
-            tasks.append(Task(send_request(local_cookies, proxy=proxy)))
-        elif verbose:
+        if browser_cookies:
+            for browser, cookies in browser_cookies.items():
+                if secure_1psid := cookies.get("__Secure-1PSID"):
+                    if (
+                        "__Secure-1PSID" in base_cookies
+                        and base_cookies["__Secure-1PSID"] != secure_1psid
+                    ):
+                        if verbose:
+                            logger.debug(
+                                f"Skipping loading local browser cookies from {browser}. "
+                                f"__Secure-1PSID does not match the one provided."
+                            )
+                        continue
+
+                    local_cookies = {"__Secure-1PSID": secure_1psid}
+                    if secure_1psidts := cookies.get("__Secure-1PSIDTS"):
+                        local_cookies["__Secure-1PSIDTS"] = secure_1psidts
+                    if nid := cookies.get("NID"):
+                        local_cookies["NID"] = nid
+                    tasks.append(Task(send_request(local_cookies, proxy=proxy)))
+                    valid_browser_cookies += 1
+                    if verbose:
+                        logger.debug(f"Loaded local browser cookies from {browser}")
+
+        if valid_browser_cookies == 0 and verbose:
             logger.debug(
                 "Skipping loading local browser cookies. Login to gemini.google.com in your browser first."
             )
@@ -149,6 +172,11 @@ async def get_access_token(
         if verbose:
             logger.warning(f"Skipping loading local browser cookies. {e}")
 
+    if not tasks:
+        raise AuthError(
+            "No valid cookies available for initialization. Please pass __Secure-1PSID and __Secure-1PSIDTS manually."
+        )
+
     for i, future in enumerate(asyncio.as_completed(tasks)):
         try:
             response, request_cookies = await future

gemini_webapi/utils/load_browser_cookies.py

@@ -1,3 +1,5 @@
+from http.cookiejar import CookieJar
+
 from .logger import logger
 
 
@@ -15,8 +17,9 @@ def load_browser_cookies(domain_name: str = "", verbose=True) -> dict:
 
     Returns
     -------
-    `dict`
-        Dictionary with cookie name as key and cookie value as value.
+    `dict[str, dict]`
+        Dictionary with browser as keys and their cookies for the specified domain as values.
+        Only browsers that have cookies for the specified domain will be included.
     """
 
     import browser_cookie3 as bc3
@@ -35,8 +38,11 @@ def load_browser_cookies(domain_name: str = "", verbose=True) -> dict:
         bc3.safari,
     ]:
         try:
-            for cookie in cookie_fn(domain_name=domain_name):
-                cookies[cookie.name] = cookie.value
+            jar: CookieJar = cookie_fn(domain_name=domain_name)
+            if jar:
+                cookies[cookie_fn.__name__] = {
+                    cookie.name: cookie.value for cookie in jar
+                }
         except bc3.BrowserCookieError:
             pass
         except PermissionError as e:

gemini_webapi/utils/rotate_1psidts.py

@@ -32,7 +32,11 @@ async def rotate_1psidts(cookies: dict, proxy: str | None = None) -> str:
         If request failed with other status codes.
     """
 
-    path = Path(__file__).parent / "temp"
+    path = (
+        (GEMINI_COOKIE_PATH := os.getenv("GEMINI_COOKIE_PATH"))
+        and Path(GEMINI_COOKIE_PATH)
+        or (Path(__file__).parent / "temp")
+    )
     path.mkdir(parents=True, exist_ok=True)
     filename = f".cached_1psidts_{cookies['__Secure-1PSID']}.txt"
     path = path / filename

{gemini_webapi-1.15.1.dist-info → gemini_webapi-1.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.15.1
+Version: 1.16.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -679,7 +679,7 @@ License-File: LICENSE
 Requires-Dist: httpx[http2]~=0.28.1
 Requires-Dist: loguru~=0.7.3
 Requires-Dist: orjson~=3.11.1
-Requires-Dist: pydantic~=2.11.5
+Requires-Dist: pydantic~=2.12.2
 Dynamic: license-file
 
 <p align="center">
@@ -713,7 +713,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.
-- **Image Generation** - Natively supports generating and modifying images with natural language.
+- **Image Generation** - Natively supports generating and editing images with natural language.
 - **System Prompt** - Supports customizing model's system prompt with [Gemini Gems](https://gemini.google.com/gems/view).
 - **Extension Support** - Supports generating contents with [Gemini extensions](https://gemini.google.com/extensions) on, like YouTube and Gmail.
 - **Classified Outputs** - Categorizes texts, thoughts, web images and AI generated images in the response.
@@ -740,7 +740,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
     - [Delete a custom gem](#delete-a-custom-gem)
   - [Retrieve model's thought process](#retrieve-models-thought-process)
   - [Retrieve images in response](#retrieve-images-in-response)
-  - [Generate images with Imagen4](#generate-images-with-imagen4)
+  - [Generate and edit images](#generate-and-edit-images)
   - [Generate contents with Gemini extensions](#generate-contents-with-gemini-extensions)
   - [Check and switch to other reply candidates](#check-and-switch-to-other-reply-candidates)
   - [Logging Configuration](#logging-configuration)
@@ -777,15 +777,17 @@ pip install -U browser-cookie3
 
 > [!NOTE]
 >
-> If your application is deployed in a containerized environment (e.g. Docker), you may want to persist the cookies with a volume to avoid re-authentication every time the container rebuilds.
+> If your application is deployed in a containerized environment (e.g. Docker), you may want to persist the cookies with a volume to avoid re-authentication every time the container rebuilds. You can set `GEMINI_COOKIE_PATH` environment variable to specify the path where auto-refreshed cookies are stored. Make sure the path is writable by the application.
 >
 > Here's part of a sample `docker-compose.yml` file:
 
 ```yaml
 services:
-    main:
-        volumes:
-            - ./gemini_cookies:/usr/local/lib/python3.12/site-packages/gemini_webapi/utils/temp
+  main:
+    environment:
+      GEMINI_COOKIE_PATH: /tmp/gemini_webapi
+    volumes:
+      - ./gemini_cookies:/tmp/gemini_webapi
 ```
 
 > [!NOTE]
@@ -1064,13 +1066,13 @@ async def main():
 asyncio.run(main())
 ```
 
-### Generate images with Imagen4
+### Generate and edit images
 
-You can ask Gemini to generate and modify images with Imagen4, Google's latest AI image generator, simply by natural language.
+You can ask Gemini to generate and edit images with Nano Banana, Google's latest image model, 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 March 19th, 2025):
+> 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 Sep 10, 2025):
 >
 > > This feature’s availability in any specific Gemini app is also limited to the supported languages and countries of that app.
 > >

{gemini_webapi-1.15.1.dist-info → gemini_webapi-1.16.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 gemini_webapi/__init__.py,sha256=7ELCiUoI10ea3daeJxnv0UwqLVKpM7rxsgOZsPMstO8,150
-gemini_webapi/client.py,sha256=G0nmRCeEmT0BpqgYY4zxb_ozFDmb39JqtddKpIN6JlA,26870
-gemini_webapi/constants.py,sha256=z8Jr6NQk79mgZOgmwx8vbgzyORHQVBjoJ4ZhUdDWWXw,2629
+gemini_webapi/client.py,sha256=YlgBiIQWkxCjPaqMQ1Zs-MvRz6ww0U3uqdS__azqvwA,26827
+gemini_webapi/constants.py,sha256=ZzyZRqwJsqjvDdj3GYIdk0SpQVbKa1Vd5NwEsvIz6pI,2685
 gemini_webapi/exceptions.py,sha256=qkXrIpr0L7LtGbq3VcTO8D1xZ50pJtt0dDRp5I3uDSg,1038
 gemini_webapi/components/__init__.py,sha256=wolxuAJJ32-jmHOKgpsesexP7hXea1JMo5vI52wysTI,48
 gemini_webapi/components/gem_mixin.py,sha256=WPJkYDS4yQpLMBNQ94LQo5w59RgkllWaSiHsFG1k5GU,8795
@@ -12,13 +12,13 @@ gemini_webapi/types/image.py,sha256=NhW1QY2Agzb6UbrIjZLnqxqukabDSGbmoCsVguzko10,
 gemini_webapi/types/modeloutput.py,sha256=h07kQOkL5r-oPLvZ59uVtO1eP4FGy5ZpzuYQzAeQdr8,1196
 gemini_webapi/utils/__init__.py,sha256=RIU1MBqYNjih0XMMt7wCxAA-X9KVF53nDbAjaxaGTo8,352
 gemini_webapi/utils/decorators.py,sha256=AuY6sU1_6_ZqeL92dTAi3eRPQ7zubB5VuBZEiz16aBM,1741
-gemini_webapi/utils/get_access_token.py,sha256=eNn1omFO41wWXco1eM-KXR2CEi0Tb-chlph7H-PCNjg,6137
-gemini_webapi/utils/load_browser_cookies.py,sha256=A5n_VsB7Rm8ck5lpy856UNJEhv30l3dvQ3j0g3ln1fE,1535
+gemini_webapi/utils/get_access_token.py,sha256=VjrHW8VMN3LPs6zCdXQtraWygurOjTY0SJZhe49aQwc,7265
+gemini_webapi/utils/load_browser_cookies.py,sha256=OHCfe27DpV_rloIDgW9Xpeb0mkfzbYONNiholw0ElXU,1791
 gemini_webapi/utils/logger.py,sha256=0VcxhVLhHBRDQutNCpapP1y_MhPoQ2ud1uIFLqxC3Z8,958
-gemini_webapi/utils/rotate_1psidts.py,sha256=NyQ9OYPLBOcvpc8bodvEYDIVFrsYN0kdfc831lPEctM,1680
+gemini_webapi/utils/rotate_1psidts.py,sha256=XjEeQnZS3ZI6wOl0Zb5CvsbIrg0BVVNas7cE6f3x_XE,1802
 gemini_webapi/utils/upload_file.py,sha256=SJOMr6kryK_ClrKmqI96fqZBNFOMPsyAvFINAGAU3rk,1468
-gemini_webapi-1.15.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.15.1.dist-info/METADATA,sha256=baIL2RVMPLqKe1l_lMxi5QDpTEJbNdehoF4uk2-467Q,61279
-gemini_webapi-1.15.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gemini_webapi-1.15.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.15.1.dist-info/RECORD,,
+gemini_webapi-1.16.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.16.0.dist-info/METADATA,sha256=vRnA79JpF9eMcgjzCqbpok_UGTvw5EPYKRO1bFtjZ5Q,61426
+gemini_webapi-1.16.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gemini_webapi-1.16.0.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.16.0.dist-info/RECORD,,