gemini-webapi 1.10.1__tar.gz → 1.11.0__tar.gz

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.10.1
+Version: 1.11.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -677,8 +677,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: httpx[http2]~=0.28.1
-Requires-Dist: pydantic~=2.10.5
 Requires-Dist: loguru~=0.7.3
+Requires-Dist: pydantic~=2.11.3
 Dynamic: license-file
 
 <p align="center">
@@ -822,18 +822,16 @@ You can specify which language model to use by passing `model` argument to `Gemi
 
 Currently available models (as of Feb 5, 2025):
 
-- `unspecified` - Default model (same as `gemini-2.0-flash` if account does NOT have Gemini Advanced subscription)
+- `unspecified` - Default model
 - `gemini-2.0-flash` - Gemini 2.0 Flash
 - `gemini-2.0-flash-thinking` - Gemini 2.0 Flash Thinking Experimental
-- `gemini-2.0-flash-thinking-with-apps` - Gemini 2.0 Flash Thinking Experimental with apps
-- `gemini-1.5-flash` - Gemini 1.5 Flash
+- `gemini-2.5-flash` - Gemini 2.5 Flash
+- `gemini-2.5-pro` - Gemini 2.5 Pro (daily usage limit imposed)
 
 Models pending update (may not work as expected):
 
 - `gemini-2.5-exp-advanced` - Gemini 2.5 Experimental Advanced **(requires Gemini Advanced account)**
 - `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced **(requires Gemini Advanced account)**
-- `gemini-1.5-pro` - Gemini 1.5 Pro **(requires Gemini Advanced account)**
-- `gemini-1.5-pro-research` - Gemini 1.5 Pro with Deep Research **(requires Gemini Advanced account)**
 
 ```python
 from gemini_webapi.constants import Model

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/README.md

@@ -139,18 +139,16 @@ You can specify which language model to use by passing `model` argument to `Gemi
 
 Currently available models (as of Feb 5, 2025):
 
-- `unspecified` - Default model (same as `gemini-2.0-flash` if account does NOT have Gemini Advanced subscription)
+- `unspecified` - Default model
 - `gemini-2.0-flash` - Gemini 2.0 Flash
 - `gemini-2.0-flash-thinking` - Gemini 2.0 Flash Thinking Experimental
-- `gemini-2.0-flash-thinking-with-apps` - Gemini 2.0 Flash Thinking Experimental with apps
-- `gemini-1.5-flash` - Gemini 1.5 Flash
+- `gemini-2.5-flash` - Gemini 2.5 Flash
+- `gemini-2.5-pro` - Gemini 2.5 Pro (daily usage limit imposed)
 
 Models pending update (may not work as expected):
 
 - `gemini-2.5-exp-advanced` - Gemini 2.5 Experimental Advanced **(requires Gemini Advanced account)**
 - `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced **(requires Gemini Advanced account)**
-- `gemini-1.5-pro` - Gemini 1.5 Pro **(requires Gemini Advanced account)**
-- `gemini-1.5-pro-research` - Gemini 1.5 Pro with Deep Research **(requires Gemini Advanced account)**
 
 ```python
 from gemini_webapi.constants import Model

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/pyproject.toml

@@ -21,8 +21,8 @@ classifiers = [
 requires-python = ">=3.10"
 dependencies = [
     "httpx[http2]~=0.28.1",
-    "pydantic~=2.10.5",
     "loguru~=0.7.3",
+    "pydantic~=2.11.3",
 ]
 dynamic = ["version"]
 

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/src/gemini_webapi/client.py

@@ -8,8 +8,15 @@ from typing import Any, Optional
 
 from httpx import AsyncClient, ReadTimeout
 
-from .constants import Endpoint, Headers, Model
-from .exceptions import AuthError, APIError, TimeoutError, GeminiError
+from .constants import Endpoint, ErrorCode, Headers, Model
+from .exceptions import (
+    AuthError,
+    APIError,
+    TimeoutError,
+    GeminiError,
+    UsageLimitExceeded,
+    ModelInvalid,
+)
 from .types import WebImage, GeneratedImage, Candidate, ModelOutput
 from .utils import (
     upload_file,
@@ -376,10 +383,27 @@ class GeminiClient:
                     raise Exception
             except Exception:
                 await self.close()
-                logger.debug(f"Invalid response: {response.text}")
-                raise APIError(
-                    "Failed to generate contents. Invalid response data received. Client will try to re-initialize on next request."
-                )
+
+                try:
+                    match ErrorCode(response_json[0][5][2][0][1][0]):
+                        case ErrorCode.USAGE_LIMIT_EXCEEDED:
+                            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_HEADER_INVALID:
+                            raise ModelInvalid(
+                                "Failed to generate contents. The specified model is not available. Please update gemini_webapi to the latest version. "
+                                "If the error persists and is caused by the package, please report it on GitHub."
+                            )
+                        case _:
+                            raise Exception
+                except GeminiError:
+                    raise
+                except Exception:
+                    logger.debug(f"Invalid response: {response.text}")
+                    raise APIError(
+                        "Failed to generate contents. Invalid response data received. Client will try to re-initialize on next request."
+                    )
 
             try:
                 candidates = []

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/src/gemini_webapi/constants.py

@@ -27,24 +27,24 @@ class Model(Enum):
     UNSPECIFIED = ("unspecified", {}, False)
     G_2_0_FLASH = (
         "gemini-2.0-flash",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f299729663a2343f"]'},
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"f299729663a2343f"]'},
         False,
     )
-    G_2_0_FLASH_EXP = (
-        "gemini-2.0-flash-exp",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f299729663a2343f"]'},
-        False,
-    )  # Deprecated, should be removed in the future
     G_2_0_FLASH_THINKING = (
         "gemini-2.0-flash-thinking",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"9c17b1863f581b8a"]'},
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"7ca48d02d802f20a"]'},
+        False,
+    )  # Deprecated
+    G_2_5_FLASH = (
+        "gemini-2.5-flash",
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"35609594dbe934d8"]'},
         False,
     )
-    G_2_0_FLASH_THINKING_WITH_APPS = (
-        "gemini-2.0-flash-thinking-with-apps",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f8f8f5ea629f5d37"]'},
+    G_2_5_PRO = (
+        "gemini-2.5-pro",
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"2525e3954d185b3c"]'},
         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"]'},
@@ -55,21 +55,6 @@ class Model(Enum):
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"203e6bb81620bcfe"]'},
         True,
     )
-    G_1_5_FLASH = (
-        "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
@@ -84,3 +69,12 @@ class Model(Enum):
         raise ValueError(
             f"Unknown model name: {name}. Available models: {', '.join([model.model_name for model in cls])}"
         )
+
+
+class ErrorCode(Enum):
+    """
+    Known error codes returned from server.
+    """
+
+    USAGE_LIMIT_EXCEEDED = 1037
+    MODEL_HEADER_INVALID = 1052

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/src/gemini_webapi/exceptions.py

@@ -28,3 +28,19 @@ class TimeoutError(GeminiError):
     """
 
     pass
+
+
+class UsageLimitExceeded(GeminiError):
+    """
+    Exception for model usage limit exceeded errors.
+    """
+
+    pass
+
+
+class ModelInvalid(GeminiError):
+    """
+    Exception for invalid model header string errors.
+    """
+
+    pass

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.10.1
+Version: 1.11.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -677,8 +677,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: httpx[http2]~=0.28.1
-Requires-Dist: pydantic~=2.10.5
 Requires-Dist: loguru~=0.7.3
+Requires-Dist: pydantic~=2.11.3
 Dynamic: license-file
 
 <p align="center">
@@ -822,18 +822,16 @@ You can specify which language model to use by passing `model` argument to `Gemi
 
 Currently available models (as of Feb 5, 2025):
 
-- `unspecified` - Default model (same as `gemini-2.0-flash` if account does NOT have Gemini Advanced subscription)
+- `unspecified` - Default model
 - `gemini-2.0-flash` - Gemini 2.0 Flash
 - `gemini-2.0-flash-thinking` - Gemini 2.0 Flash Thinking Experimental
-- `gemini-2.0-flash-thinking-with-apps` - Gemini 2.0 Flash Thinking Experimental with apps
-- `gemini-1.5-flash` - Gemini 1.5 Flash
+- `gemini-2.5-flash` - Gemini 2.5 Flash
+- `gemini-2.5-pro` - Gemini 2.5 Pro (daily usage limit imposed)
 
 Models pending update (may not work as expected):
 
 - `gemini-2.5-exp-advanced` - Gemini 2.5 Experimental Advanced **(requires Gemini Advanced account)**
 - `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced **(requires Gemini Advanced account)**
-- `gemini-1.5-pro` - Gemini 1.5 Pro **(requires Gemini Advanced account)**
-- `gemini-1.5-pro-research` - Gemini 1.5 Pro with Deep Research **(requires Gemini Advanced account)**
 
 ```python
 from gemini_webapi.constants import Model

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/src/gemini_webapi.egg-info/requires.txt

@@ -1,3 +1,3 @@
 httpx[http2]~=0.28.1
-pydantic~=2.10.5
 loguru~=0.7.3
+pydantic~=2.11.3

{gemini_webapi-1.10.1 → gemini_webapi-1.11.0}/tests/test_client_features.py

@@ -7,6 +7,7 @@ from loguru import logger
 
 from gemini_webapi import GeminiClient, AuthError, set_log_level
 from gemini_webapi.constants import Model
+from gemini_webapi.exceptions import UsageLimitExceeded, ModelInvalid
 
 logging.getLogger("asyncio").setLevel(logging.ERROR)
 set_log_level("DEBUG")
@@ -19,22 +20,23 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
         )
 
         try:
-            await self.geminiclient.init()
+            await self.geminiclient.init(timeout=60)
         except AuthError as e:
             self.skipTest(e)
 
     @logger.catch(reraise=True)
     async def test_successful_request(self):
         response = await self.geminiclient.generate_content(
-            "Hello World!", model=Model.G_2_0_FLASH
+            "Tell me a fact about today in history and illustrate it with a youtube video",
+            model=Model.G_2_5_FLASH,
         )
         logger.debug(response.text)
 
     @logger.catch(reraise=True)
     async def test_thinking_model(self):
         response = await self.geminiclient.generate_content(
-            "Tell me a fact about today in history and illustrate it with a youtube video",
-            model=Model.G_2_0_FLASH_THINKING,
+            "1+1=?",
+            model=Model.G_2_5_FLASH,
         )
         logger.debug(response.thoughts)
         logger.debug(response.text)
@@ -46,11 +48,16 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
                 logger.debug(f"Model {model.model_name} requires an advanced account")
                 continue
 
-            response = await self.geminiclient.generate_content(
-                "What's you language model version? Reply version number only.",
-                model=model,
-            )
-            logger.debug(f"Model version ({model.model_name}): {response.text}")
+            try:
+                response = await self.geminiclient.generate_content(
+                    "What's you language model version? Reply version number only.",
+                    model=model,
+                )
+                logger.debug(f"Model version ({model.model_name}): {response.text}")
+            except UsageLimitExceeded:
+                logger.debug(f"Model {model.model_name} usage limit exceeded")
+            except ModelInvalid:
+                logger.debug(f"Model {model.model_name} is not available anymore")
 
     @logger.catch(reraise=True)
     async def test_upload_files(self):