gemini-webapi 1.7.0__py3-none-any.whl → 1.8.1__py3-none-any.whl

gemini_webapi/client.py

@@ -8,7 +8,7 @@ from typing import Any, Optional
 
 from httpx import AsyncClient, ReadTimeout
 
-from .constants import Endpoint, Headers
+from .constants import Endpoint, Headers, Model
 from .exceptions import AuthError, APIError, TimeoutError, GeminiError
 from .types import WebImage, GeneratedImage, Candidate, ModelOutput
 from .utils import (
@@ -79,6 +79,9 @@ class GeminiClient:
         __Secure-1PSIDTS cookie value, some google accounts don't require this value, provide only if it's in the cookie list.
     proxy: `str`, optional
         Proxy URL.
+    kwargs: `dict`, optional
+        Additional arguments which will be passed to the http client.
+        Refer to `httpx.AsyncClient` for more information.
 
     Raises
     ------
@@ -98,6 +101,7 @@ class GeminiClient:
         "close_task",
         "auto_refresh",
         "refresh_interval",
+        "kwargs",
     ]
 
     def __init__(
@@ -105,6 +109,7 @@ class GeminiClient:
         secure_1psid: str | None = None,
         secure_1psidts: str | None = None,
         proxy: str | None = None,
+        **kwargs,
     ):
         self.cookies = {}
         self.proxy = proxy
@@ -117,6 +122,7 @@ class GeminiClient:
         self.close_task: Task | None = None
         self.auto_refresh: bool = True
         self.refresh_interval: float = 540
+        self.kwargs = kwargs
 
         # Validate cookies
         if secure_1psid:
@@ -173,6 +179,7 @@ class GeminiClient:
                 follow_redirects=True,
                 headers=Headers.GEMINI.value,
                 cookies=valid_cookies,
+                **self.kwargs,
             )
             self.access_token = access_token
             self.cookies = valid_cookies
@@ -256,7 +263,9 @@ class GeminiClient:
         self,
         prompt: str,
         images: list[bytes | str | Path] | None = None,
+        model: Model | str = Model.UNSPECIFIED,
         chat: Optional["ChatSession"] = None,
+        **kwargs,
     ) -> ModelOutput:
         """
         Generates contents with prompt.
@@ -267,8 +276,14 @@ class GeminiClient:
             Prompt provided by user.
         images: `list[bytes | str | Path]`, optional
             List of image file paths or file data in bytes.
+        model: `Model` | `str`, optional
+            Specify the model to use for generation.
+            Pass either a `gemini_webapi.constants.Model` enum or a model name string.
         chat: `ChatSession`, optional
             Chat data to retrieve conversation history. If None, will automatically generate a new chat id when sending post request.
+        kwargs: `dict`, optional
+            Additional arguments which will be passed to the post request.
+            Refer to `httpx.AsyncClient.request` for more information.
 
         Returns
         -------
@@ -291,12 +306,16 @@ class GeminiClient:
 
         assert prompt, "Prompt cannot be empty."
 
+        if not isinstance(model, Model):
+            model = Model.from_name(model)
+
         if self.auto_close:
             await self.reset_close_task()
 
         try:
             response = await self.client.post(
                 Endpoint.GENERATE.value,
+                headers=model.model_header,
                 data={
                     "at": self.access_token,
                     "f.req": json.dumps(
@@ -311,12 +330,8 @@ class GeminiClient:
                                         None,
                                         [
                                             [
-                                                [
-                                                    await upload_file(
-                                                        image, self.proxy
-                                                    ),
-                                                    1,
-                                                ]
+                                                [await upload_file(image, self.proxy)],
+                                                "filename.jpg",
                                             ]
                                             for image in images
                                         ],
@@ -329,6 +344,7 @@ class GeminiClient:
                         ]
                     ),
                 },
+                **kwargs,
             )
         except ReadTimeout:
             raise TimeoutError(
@@ -435,12 +451,13 @@ class GeminiClient:
         Parameters
         ----------
         kwargs: `dict`, optional
-            Other arguments to pass to `ChatSession.__init__`.
+            Additional arguments which will be passed to the chat session.
+            Refer to `gemini_webapi.ChatSession` for more information.
 
         Returns
         -------
         :class:`ChatSession`
-            Empty chat object for retrieving conversation history.
+            Empty chat session object for retrieving conversation history.
         """
 
         return ChatSession(geminiclient=self, **kwargs)
@@ -462,9 +479,17 @@ 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
+        Specify the model to use for generation.
+        Pass either a `gemini_webapi.constants.Model` enum or a model name string.
     """
 
-    __slots__ = ["__metadata", "geminiclient", "last_output"]
+    __slots__ = [
+        "__metadata",
+        "geminiclient",
+        "last_output",
+        "model",
+    ]
 
     def __init__(
         self,
@@ -473,10 +498,12 @@ 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,
     ):
         self.__metadata: list[str | None] = [None, None, None]
         self.geminiclient: GeminiClient = geminiclient
         self.last_output: ModelOutput | None = None
+        self.model = model
 
         if metadata:
             self.metadata = metadata
@@ -503,6 +530,7 @@ class ChatSession:
         self,
         prompt: str,
         images: list[bytes | str | Path] | None = None,
+        **kwargs,
     ) -> ModelOutput:
         """
         Generates contents with prompt.
@@ -514,6 +542,9 @@ class ChatSession:
             Prompt provided by user.
         images: `list[bytes | str | Path]`, optional
             List of image file paths or file data in bytes.
+        kwargs: `dict`, optional
+            Additional arguments which will be passed to the post request.
+            Refer to `httpx.AsyncClient.request` for more information.
 
         Returns
         -------
@@ -535,7 +566,7 @@ class ChatSession:
         """
 
         return await self.geminiclient.generate_content(
-            prompt=prompt, images=images, chat=self
+            prompt=prompt, images=images, model=self.model, chat=self, **kwargs
         )
 
     def choose_candidate(self, index: int) -> ModelOutput:

gemini_webapi/constants.py

@@ -21,3 +21,36 @@ class Headers(Enum):
         "Content-Type": "application/json",
     }
     UPLOAD = {"Push-ID": "feeds/mcudyrk2a4khkz"}
+
+
+class Model(Enum):
+    UNSPECIFIED = ("unspecified", {}, False)
+    G_1_5_FLASH = (
+        "gemini-1.5-flash",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"7daceb7ef88130f5"]'},
+        False,
+    )
+    G_2_0_FLASH_EXP = (
+        "gemini-2.0-flash-exp",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"948b866104ccf484"]'},
+        False,
+    )
+    G_2_0_EXP_ADVANCED = (
+        "gemini-2.0-exp-advanced",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
+        True,
+    )
+
+    def __init__(self, name, header, advanced_only):
+        self.model_name = name
+        self.model_header = header
+        self.advanced_only = advanced_only
+
+    @classmethod
+    def from_name(cls, name: str):
+        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])}"
+        )

{gemini_webapi-1.7.0.dist-info → gemini_webapi-1.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: gemini-webapi
-Version: 1.7.0
+Version: 1.8.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -732,6 +732,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [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)
+  - [Specify language model version](#specify-language-model-version)
   - [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)
@@ -774,9 +775,9 @@ pip install -U browser-cookie3
 
 ```yaml
 services:
-  main:
-    volumes:
-      - ./gemini_cookies:/usr/local/lib/python3.12/site-packages/gemini_webapi/utils/temp
+    main:
+        volumes:
+            - ./gemini_cookies:/usr/local/lib/python3.12/site-packages/gemini_webapi/utils/temp
 ```
 
 > [!NOTE]
@@ -937,6 +938,34 @@ async def main():
 asyncio.run(main())
 ```
 
+### Specify language model version
+
+You can choose a specified language model version by passing `model` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. The default value is `unspecified`.
+
+Currently available models (as of Dec 21, 2024):
+
+- `unspecified` - Default model (Gemini 1.5 Flash)
+- `gemini-1.5-flash` - Gemini 1.5 Flash
+- `gemini-2.0-flash-exp` - Gemini 2.0 Flash Experimental
+- `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced (requires Gemini Advanced account)
+
+```python
+from gemini_webapi.constants import Model
+
+async def main():
+    response1 = await client.generate_content(
+        "What's you language model version? Reply version number only.",
+        model="gemini-1.5-flash",
+    )
+    print(f"Model version (gemini-1.5-flash): {response1.text}")
+
+    chat = client.start_chat(model=Model.G_2_0_FLASH_EXP)
+    response2 = await chat.send_message("What's you language model version? Reply version number only.")
+    print(f"Model version ({Model.G_2_0_FLASH_EXP.model_name}): {response2.text}")
+
+asyncio.run(main())
+```
+
 ### Generate contents with Gemini extensions
 
 > [!IMPORTANT]

{gemini_webapi-1.7.0.dist-info → gemini_webapi-1.8.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 gemini_webapi/__init__.py,sha256=28uNIywK4vCXxENaSagNWUzhqr1RyNtLzDF6WRRM4KQ,194
-gemini_webapi/client.py,sha256=A_unTPUCSSLch0zojvTG4PrS7Hvpxgr7mnIXG-uW_kM,20922
-gemini_webapi/constants.py,sha256=ezJ0Squ5FPTspAp5IGRe5gg9CYaPHnzx5N95W5gGm5s,865
+gemini_webapi/client.py,sha256=SZEihRytBcBNa8ZJllEU3xFRB1cX8ZrEY_3TRjDdN4g,22169
+gemini_webapi/constants.py,sha256=43tV363_lv6J-idZ48n0_G-A4GZSwIW0gJ60GibNMa0,1866
 gemini_webapi/exceptions.py,sha256=6e-EXHGApi4iC0GDw7RKc3YqVK8UvEkHYaJyGQbReLw,548
 gemini_webapi/types/__init__.py,sha256=d2kvXnE004s2E2KDmPPLi5N-BQ59FgDSlrGrO3Wphww,163
 gemini_webapi/types/candidate.py,sha256=Z9bpIK4l8UWbUVMLEoophfhgROo93dxOM9cAwx77CkU,1030
@@ -12,8 +12,8 @@ gemini_webapi/utils/load_browser_cookies.py,sha256=A5n_VsB7Rm8ck5lpy856UNJEhv30l
 gemini_webapi/utils/logger.py,sha256=PF4ROQq7scRRrWzeYdeYiYs2S2Jqr0bgjyrPbXVOCqE,816
 gemini_webapi/utils/rotate_1psidts.py,sha256=cwCXsE_mX2knaKCuquZT2jsRXHQo-GK2Ldom-cO9Gio,1668
 gemini_webapi/utils/upload_file.py,sha256=X0k-3jW7usheFo5z0nXL9mUt-XF-2L2OgwpJnPSIcKw,1151
-gemini_webapi-1.7.0.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.7.0.dist-info/METADATA,sha256=XhfeQVH9p_5SYZ1t12mVhrhbIJMCw9Ucqhmzq0PvKI0,55729
-gemini_webapi-1.7.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-gemini_webapi-1.7.0.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.7.0.dist-info/RECORD,,
+gemini_webapi-1.8.1.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.8.1.dist-info/METADATA,sha256=cV33fx9JrgMZJQjiIe5sLUZYkobSIyzW5pmK8mCRBuw,56893
+gemini_webapi-1.8.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+gemini_webapi-1.8.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.8.1.dist-info/RECORD,,

{gemini_webapi-1.7.0.dist-info → gemini_webapi-1.8.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any