gemini-webapi 1.17.0__tar.gz → 1.17.1__tar.gz

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/.github/workflows/pypi-publish.yml

@@ -36,7 +36,7 @@ jobs:
       - name: Build package
         run: python -m build
       - name: Archive production artifacts
-        uses: actions/upload-artifact@v4.6.2
+        uses: actions/upload-artifact@v5.0.0
         with:
           name: dist
           path: dist
@@ -52,7 +52,7 @@ jobs:
       id-token: write  # IMPORTANT: this permission is mandatory for trusted publishing
     steps:
       - name: Retrieve built artifacts
-        uses: actions/download-artifact@v5.0.0
+        uses: actions/download-artifact@v6.0.0
         with:
           name: dist
           path: dist

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.17.0
+Version: 1.17.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -783,11 +783,11 @@ pip install -U browser-cookie3
 
 ```yaml
 services:
-  main:
-    environment:
-      GEMINI_COOKIE_PATH: /tmp/gemini_webapi
-    volumes:
-      - ./gemini_cookies:/tmp/gemini_webapi
+    main:
+        environment:
+            GEMINI_COOKIE_PATH: /tmp/gemini_webapi
+        volumes:
+            - ./gemini_cookies:/tmp/gemini_webapi
 ```
 
 > [!NOTE]
@@ -929,6 +929,23 @@ async def main():
 asyncio.run(main())
 ```
 
+You can also pass custom model header strings directly to access models which are not listed above.
+
+```python
+# "model_name" and "model_header" keys must be present
+custom_model = {
+    "model_name": "xxx",
+    "model_header": {
+        "x-goog-ext-525001261-jspb": "[1,null,null,null,'e6fa609c3fa255c0',null,null,null,[4]]"
+    },
+}
+
+response = await client.generate_content(
+    "What's your model version?",
+    model=custom_model
+)
+```
+
 ### Apply system prompt with Gemini Gems
 
 System prompt can be applied to conversations via [Gemini Gems](https://gemini.google.com/gems/view). To use a gem, you can pass `gem` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. `gem` can be either a string of gem id or a `gemini_webapi.Gem` object. Only one gem can be applied to a single conversation.

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/README.md

@@ -99,11 +99,11 @@ pip install -U browser-cookie3
 
 ```yaml
 services:
-  main:
-    environment:
-      GEMINI_COOKIE_PATH: /tmp/gemini_webapi
-    volumes:
-      - ./gemini_cookies:/tmp/gemini_webapi
+    main:
+        environment:
+            GEMINI_COOKIE_PATH: /tmp/gemini_webapi
+        volumes:
+            - ./gemini_cookies:/tmp/gemini_webapi
 ```
 
 > [!NOTE]
@@ -245,6 +245,23 @@ async def main():
 asyncio.run(main())
 ```
 
+You can also pass custom model header strings directly to access models which are not listed above.
+
+```python
+# "model_name" and "model_header" keys must be present
+custom_model = {
+    "model_name": "xxx",
+    "model_header": {
+        "x-goog-ext-525001261-jspb": "[1,null,null,null,'e6fa609c3fa255c0',null,null,null,[4]]"
+    },
+}
+
+response = await client.generate_content(
+    "What's your model version?",
+    model=custom_model
+)
+```
+
 ### Apply system prompt with Gemini Gems
 
 System prompt can be applied to conversations via [Gemini Gems](https://gemini.google.com/gems/view). To use a gem, you can pass `gem` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. `gem` can be either a string of gem id or a `gemini_webapi.Gem` object. Only one gem can be applied to a single conversation.

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/src/gemini_webapi/client.py

@@ -240,7 +240,7 @@ class GeminiClient(GemMixin):
         self,
         prompt: str,
         files: list[str | Path] | None = None,
-        model: Model | str = Model.UNSPECIFIED,
+        model: Model | str | dict = Model.UNSPECIFIED,
         gem: Gem | str | None = None,
         chat: Optional["ChatSession"] = None,
         **kwargs,
@@ -254,9 +254,10 @@ class GeminiClient(GemMixin):
             Prompt provided by user.
         files: `list[str | Path]`, optional
             List of file paths to be attached.
-        model: `Model` | `str`, optional
+        model: `Model | str | dict`, optional
             Specify the model to use for generation.
-            Pass either a `gemini_webapi.constants.Model` enum or a model name string.
+            Pass either a `gemini_webapi.constants.Model` enum or a model name string to use predefined models.
+            Pass a dictionary to use custom model header strings ("model_name" and "model_header" keys must be provided).
         gem: `Gem | str`, optional
             Specify a gem to use as system prompt for the chat session.
             Pass either a `gemini_webapi.types.Gem` object or a gem id string.
@@ -287,8 +288,15 @@ class GeminiClient(GemMixin):
 
         assert prompt, "Prompt cannot be empty."
 
-        if not isinstance(model, Model):
+        if isinstance(model, str):
             model = Model.from_name(model)
+        elif isinstance(model, dict):
+            model = Model.from_dict(model)
+        elif not isinstance(model, Model):
+            raise TypeError(
+                f"'model' must be a `gemini_webapi.constants.Model` instance, "
+                f"string, or dictionary; got `{type(model).__name__}`"
+            )
 
         if isinstance(gem, Gem):
             gem_id = gem.id
@@ -362,7 +370,7 @@ class GeminiClient(GemMixin):
                         if get_nested_value(part_json, [4]):
                             body_index, body = part_index, part_json
                             break
-                    except (TypeError, ValueError):
+                    except json.JSONDecodeError:
                         continue
 
                 if not body:
@@ -443,13 +451,17 @@ class GeminiClient(GemMixin):
                             if img_part_index < body_index:
                                 continue
                             try:
-                                img_part = json.loads(part[2])
+                                img_part_body = get_nested_value(part, [2])
+                                if not img_part_body:
+                                    continue
+
+                                img_part_json = json.loads(img_part_body)
                                 if get_nested_value(
-                                    img_part, [4, candidate_index, 12, 7, 0]
+                                    img_part_json, [4, candidate_index, 12, 7, 0]
                                 ):
-                                    img_body = img_part
+                                    img_body = img_part_json
                                     break
-                            except (IndexError, TypeError, ValueError):
+                            except json.JSONDecodeError:
                                 continue
 
                         if not img_body:
@@ -614,9 +626,10 @@ class ChatSession:
         Reply id, if provided together with metadata, will override the second value in it.
     rcid: `str`, optional
         Reply candidate id, if provided together with metadata, will override the third value in it.
-    model: `Model` | `str`, optional
+    model: `Model | str | dict`, optional
         Specify the model to use for generation.
-        Pass either a `gemini_webapi.constants.Model` enum or a model name string.
+        Pass either a `gemini_webapi.constants.Model` enum or a model name string to use predefined models.
+        Pass a dictionary to use custom model header strings ("model_name" and "model_header" keys must be provided).
     gem: `Gem | str`, optional
         Specify a gem to use as system prompt for the chat session.
         Pass either a `gemini_webapi.types.Gem` object or a gem id string.
@@ -637,13 +650,13 @@ class ChatSession:
         cid: str | None = None,  # chat id
         rid: str | None = None,  # reply id
         rcid: str | None = None,  # reply candidate id
-        model: Model | str = Model.UNSPECIFIED,
+        model: Model | str | dict = Model.UNSPECIFIED,
         gem: Gem | str | None = None,
     ):
         self.__metadata: list[str | None] = [None, None, None]
         self.geminiclient: GeminiClient = geminiclient
         self.last_output: ModelOutput | None = None
-        self.model: Model | str = model
+        self.model: Model | str | dict = model
         self.gem: Gem | str | None = gem
 
         if metadata:

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/src/gemini_webapi/constants.py

@@ -45,17 +45,23 @@ class Model(Enum):
     UNSPECIFIED = ("unspecified", {}, False)
     G_3_0_PRO = (
         "gemini-3.0-pro",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"9d8ca3786ebdfbea",null,null,null,[4]]'},
-        False,
-    )
-    G_2_5_FLASH = (
-        "gemini-2.5-flash",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"9ec249fc9ad08861",null,null,0,[4]]'},
+        {
+            "x-goog-ext-525001261-jspb": '[1,null,null,null,"9d8ca3786ebdfbea",null,null,0,[4]]'
+        },
         False,
     )
     G_2_5_PRO = (
         "gemini-2.5-pro",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"4af6c7f5da75d65d",null,null,0,[4]]'},
+        {
+            "x-goog-ext-525001261-jspb": '[1,null,null,null,"4af6c7f5da75d65d",null,null,0,[4]]'
+        },
+        False,
+    )
+    G_2_5_FLASH = (
+        "gemini-2.5-flash",
+        {
+            "x-goog-ext-525001261-jspb": '[1,null,null,null,"9ec249fc9ad08861",null,null,0,[4]]'
+        },
         False,
     )
 
@@ -69,10 +75,28 @@ class Model(Enum):
         for model in cls:
             if model.model_name == name:
                 return model
+
         raise ValueError(
             f"Unknown model name: {name}. Available models: {', '.join([model.model_name for model in cls])}"
         )
 
+    @classmethod
+    def from_dict(cls, model_dict: dict):
+        if "model_name" not in model_dict or "model_header" not in model_dict:
+            raise ValueError(
+                "When passing a custom model as a dictionary, 'model_name' and 'model_header' keys must be provided."
+            )
+
+        if not isinstance(model_dict["model_header"], dict):
+            raise ValueError(
+                "When passing a custom model as a dictionary, 'model_header' must be a dictionary containing valid header strings."
+            )
+
+        custom_model = cls.UNSPECIFIED
+        custom_model.model_name = model_dict["model_name"]
+        custom_model.model_header = model_dict["model_header"]
+        return custom_model
+
 
 class ErrorCode(IntEnum):
     """

{gemini_webapi-1.17.0 → gemini_webapi-1.17.1}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.17.0
+Version: 1.17.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -783,11 +783,11 @@ pip install -U browser-cookie3
 
 ```yaml
 services:
-  main:
-    environment:
-      GEMINI_COOKIE_PATH: /tmp/gemini_webapi
-    volumes:
-      - ./gemini_cookies:/tmp/gemini_webapi
+    main:
+        environment:
+            GEMINI_COOKIE_PATH: /tmp/gemini_webapi
+        volumes:
+            - ./gemini_cookies:/tmp/gemini_webapi
 ```
 
 > [!NOTE]
@@ -929,6 +929,23 @@ async def main():
 asyncio.run(main())
 ```
 
+You can also pass custom model header strings directly to access models which are not listed above.
+
+```python
+# "model_name" and "model_header" keys must be present
+custom_model = {
+    "model_name": "xxx",
+    "model_header": {
+        "x-goog-ext-525001261-jspb": "[1,null,null,null,'e6fa609c3fa255c0',null,null,null,[4]]"
+    },
+}
+
+response = await client.generate_content(
+    "What's your model version?",
+    model=custom_model
+)
+```
+
 ### Apply system prompt with Gemini Gems
 
 System prompt can be applied to conversations via [Gemini Gems](https://gemini.google.com/gems/view). To use a gem, you can pass `gem` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. `gem` can be either a string of gem id or a `gemini_webapi.Gem` object. Only one gem can be applied to a single conversation.