gemini-webapi 1.6.1__tar.gz → 1.8.0__tar.gz

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/.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.4.0
+        uses: actions/upload-artifact@v4.4.3
         with:
           name: dist
           path: dist
@@ -57,4 +57,4 @@ jobs:
           name: dist
           path: dist
       - name: Publish package distributions to PyPI
-        uses: pypa/gh-action-pypi-publish@v1.10.1
+        uses: pypa/gh-action-pypi-publish@v1.12.3

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gemini-webapi
-Version: 1.6.1
+Version: 1.8.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -676,9 +676,9 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: httpx>=0.25.2
-Requires-Dist: pydantic>=2.5.3
-Requires-Dist: loguru>=0.7.2
+Requires-Dist: httpx~=0.28.0
+Requires-Dist: pydantic~=2.10.2
+Requires-Dist: loguru~=0.7.2
 
 <p align="center">
     <img src="https://raw.githubusercontent.com/HanaokaYuzu/Gemini-API/master/assets/banner.png" width="55%" alt="Gemini Banner" align="center">
@@ -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]
@@ -804,7 +805,7 @@ Secure_1PSIDTS = "COOKIE VALUE HERE"
 
 async def main():
     # If browser-cookie3 is installed, simply use `client = GeminiClient()`
-    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxies=None)
+    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxy=None)
     await client.init(timeout=30, auto_close=False, close_delay=300, auto_refresh=True)
 
 asyncio.run(main())
@@ -937,6 +938,33 @@ 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
+
+```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.6.1 → gemini_webapi-1.8.0}/README.md

@@ -50,6 +50,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)
@@ -92,9 +93,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]
@@ -122,7 +123,7 @@ Secure_1PSIDTS = "COOKIE VALUE HERE"
 
 async def main():
     # If browser-cookie3 is installed, simply use `client = GeminiClient()`
-    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxies=None)
+    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxy=None)
     await client.init(timeout=30, auto_close=False, close_delay=300, auto_refresh=True)
 
 asyncio.run(main())
@@ -255,6 +256,33 @@ 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
+
+```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.6.1 → gemini_webapi-1.8.0}/pyproject.toml

@@ -20,9 +20,9 @@ classifiers = [
 ]
 requires-python = ">=3.10"
 dependencies = [
-    "httpx>=0.25.2",
-    "pydantic>=2.5.3",
-    "loguru>=0.7.2",
+    "httpx~=0.28.0",
+    "pydantic~=2.10.2",
+    "loguru~=0.7.2",
 ]
 dynamic = ["version"]
 

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/src/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 (
@@ -77,8 +77,11 @@ class GeminiClient:
         __Secure-1PSID cookie value.
     secure_1psidts: `str`, optional
         __Secure-1PSIDTS cookie value, some google accounts don't require this value, provide only if it's in the cookie list.
-    proxies: `dict`, optional
-        Dict of proxies.
+    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
     ------
@@ -88,7 +91,7 @@ class GeminiClient:
 
     __slots__ = [
         "cookies",
-        "proxies",
+        "proxy",
         "running",
         "client",
         "access_token",
@@ -98,16 +101,18 @@ class GeminiClient:
         "close_task",
         "auto_refresh",
         "refresh_interval",
+        "kwargs",
     ]
 
     def __init__(
         self,
         secure_1psid: str | None = None,
         secure_1psidts: str | None = None,
-        proxies: dict | None = None,
+        proxy: str | None = None,
+        **kwargs,
     ):
         self.cookies = {}
-        self.proxies = proxies
+        self.proxy = proxy
         self.running: bool = False
         self.client: AsyncClient | None = None
         self.access_token: str | None = None
@@ -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:
@@ -164,15 +170,16 @@ class GeminiClient:
 
         try:
             access_token, valid_cookies = await get_access_token(
-                base_cookies=self.cookies, proxies=self.proxies, verbose=verbose
+                base_cookies=self.cookies, proxy=self.proxy, verbose=verbose
             )
 
             self.client = AsyncClient(
                 timeout=timeout,
-                proxies=self.proxies,
+                proxy=self.proxy,
                 follow_redirects=True,
                 headers=Headers.GEMINI.value,
                 cookies=valid_cookies,
+                **self.kwargs,
             )
             self.access_token = access_token
             self.cookies = valid_cookies
@@ -238,7 +245,7 @@ class GeminiClient:
 
         while True:
             try:
-                new_1psidts = await rotate_1psidts(self.cookies, self.proxies)
+                new_1psidts = await rotate_1psidts(self.cookies, self.proxy)
             except AuthError:
                 if task := rotate_tasks.get(self.cookies["__Secure-1PSID"]):
                     task.cancel()
@@ -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.proxies
-                                                    ),
-                                                    1,
-                                                ]
+                                                [await upload_file(image, self.proxy)],
+                                                "filename.jpg",
                                             ]
                                             for image in images
                                         ],
@@ -329,6 +344,7 @@ class GeminiClient:
                         ]
                     ),
                 },
+                **kwargs,
             )
         except ReadTimeout:
             raise TimeoutError(
@@ -377,7 +393,7 @@ class GeminiClient:
                                 url=image[0][0][0],
                                 title=image[7][0],
                                 alt=image[0][4],
-                                proxies=self.proxies,
+                                proxy=self.proxy,
                             )
                             for image in candidate[12][1]
                         ]
@@ -395,7 +411,7 @@ class GeminiClient:
                                 alt=len(image[3][5]) > i
                                 and image[3][5][i]
                                 or image[3][5][0],
-                                proxies=self.proxies,
+                                proxy=self.proxy,
                                 cookies=self.cookies,
                             )
                             for i, image in enumerate(
@@ -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-1.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi/constants.py

@@ -21,3 +21,28 @@ class Headers(Enum):
         "Content-Type": "application/json",
     }
     UPLOAD = {"Push-ID": "feeds/mcudyrk2a4khkz"}
+
+
+class Model(Enum):
+    UNSPECIFIED = ("unspecified", {})
+    G_1_5_FLASH = (
+        "gemini-1.5-flash",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"7daceb7ef88130f5"]'},
+    )
+    G_2_0_FLASH_EXP = (
+        "gemini-2.0-flash-exp",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"948b866104ccf484"]'},
+    )
+
+    def __init__(self, name, header):
+        self.model_name = name
+        self.model_header = header
+
+    @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.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi/types/image.py

@@ -20,14 +20,14 @@ class Image(BaseModel):
         Title of the image, by default is "[Image]".
     alt: `str`, optional
         Optional description of the image.
-    proxies: `dict`, optional
-        Dict of proxies used when saving image.
+    proxy: `str`, optional
+        Proxy used when saving image.
     """
 
     url: str
     title: str = "[Image]"
     alt: str = ""
-    proxies: dict | None = None
+    proxy: str | None = None
 
     def __str__(self):
         return f"{self.title}({self.url}) - {self.alt}"
@@ -80,7 +80,7 @@ class Image(BaseModel):
                 return None
 
         async with AsyncClient(
-            follow_redirects=True, cookies=cookies, proxies=self.proxies
+            follow_redirects=True, cookies=cookies, proxy=self.proxy
         ) as client:
             response = await client.get(self.url)
             if response.status_code == 200:

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi/utils/get_access_token.py

@@ -12,7 +12,7 @@ from .logger import logger
 
 
 async def get_access_token(
-    base_cookies: dict, proxies: dict | None = None, verbose: bool = False
+    base_cookies: dict, proxy: str | None = None, verbose: bool = False
 ) -> tuple[str, dict]:
     """
     Send a get request to gemini.google.com for each group of available cookies and return
@@ -27,8 +27,8 @@ async def get_access_token(
     ----------
     base_cookies : `dict`
         Base cookies to be used in the request.
-    proxies: `dict`, optional
-        Dict of proxies.
+    proxy: `str`, optional
+        Proxy URL.
     verbose: `bool`, optional
         If `True`, will print more infomation in logs.
 
@@ -47,7 +47,7 @@ async def get_access_token(
 
     async def send_request(cookies: dict) -> tuple[Response | None, dict]:
         async with AsyncClient(
-            proxies=proxies,
+            proxy=proxy,
             headers=Headers.GEMINI.value,
             cookies=cookies,
             follow_redirects=True,

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi/utils/rotate_1psidts.py

@@ -8,7 +8,7 @@ from ..constants import Endpoint, Headers
 from ..exceptions import AuthError
 
 
-async def rotate_1psidts(cookies: dict, proxies: dict | None = None) -> str:
+async def rotate_1psidts(cookies: dict, proxy: str | None = None) -> str:
     """
     Refresh the __Secure-1PSIDTS cookie and store the refreshed cookie value in cache file.
 
@@ -16,8 +16,8 @@ async def rotate_1psidts(cookies: dict, proxies: dict | None = None) -> str:
     ----------
     cookies : `dict`
         Cookies to be used in the request.
-    proxies: `dict`, optional
-        Dict of proxies.
+    proxy: `str`, optional
+        Proxy URL.
 
     Returns
     -------
@@ -39,7 +39,7 @@ async def rotate_1psidts(cookies: dict, proxies: dict | None = None) -> str:
 
     # Check if the cache file was modified in the last minute to avoid 429 Too Many Requests
     if not (path.is_file() and time.time() - os.path.getmtime(path) <= 60):
-        async with AsyncClient(proxies=proxies) as client:
+        async with AsyncClient(proxy=proxy) as client:
             response = await client.post(
                 url=Endpoint.ROTATE_COOKIES.value,
                 headers=Headers.ROTATE_COOKIES.value,

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi/utils/upload_file.py

@@ -7,7 +7,7 @@ from ..constants import Endpoint, Headers
 
 
 @validate_call
-async def upload_file(file: bytes | str | Path, proxies: dict | None = None) -> str:
+async def upload_file(file: bytes | str | Path, proxy: str | None = None) -> str:
     """
     Upload a file to Google's server and return its identifier.
 
@@ -15,8 +15,8 @@ async def upload_file(file: bytes | str | Path, proxies: dict | None = None) ->
     ----------
     file : `bytes` | `str` | `Path`
         File data in bytes, or path to the file to be uploaded.
-    proxies: `dict`, optional
-        Dict of proxies.
+    proxy: `str`, optional
+        Proxy URL.
 
     Returns
     -------
@@ -34,7 +34,7 @@ async def upload_file(file: bytes | str | Path, proxies: dict | None = None) ->
         with open(file, "rb") as f:
             file = f.read()
 
-    async with AsyncClient(proxies=proxies) as client:
+    async with AsyncClient(proxy=proxy) as client:
         response = await client.post(
             url=Endpoint.UPLOAD.value,
             headers=Headers.UPLOAD.value,

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gemini-webapi
-Version: 1.6.1
+Version: 1.8.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -676,9 +676,9 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: httpx>=0.25.2
-Requires-Dist: pydantic>=2.5.3
-Requires-Dist: loguru>=0.7.2
+Requires-Dist: httpx~=0.28.0
+Requires-Dist: pydantic~=2.10.2
+Requires-Dist: loguru~=0.7.2
 
 <p align="center">
     <img src="https://raw.githubusercontent.com/HanaokaYuzu/Gemini-API/master/assets/banner.png" width="55%" alt="Gemini Banner" align="center">
@@ -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]
@@ -804,7 +805,7 @@ Secure_1PSIDTS = "COOKIE VALUE HERE"
 
 async def main():
     # If browser-cookie3 is installed, simply use `client = GeminiClient()`
-    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxies=None)
+    client = GeminiClient(Secure_1PSID, Secure_1PSIDTS, proxy=None)
     await client.init(timeout=30, auto_close=False, close_delay=300, auto_refresh=True)
 
 asyncio.run(main())
@@ -937,6 +938,33 @@ 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
+
+```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.8.0/src/gemini_webapi.egg-info/requires.txt

@@ -0,0 +1,3 @@
+httpx~=0.28.0
+pydantic~=2.10.2
+loguru~=0.7.2

{gemini_webapi-1.6.1 → gemini_webapi-1.8.0}/tests/test_client_features.py

@@ -6,6 +6,7 @@ from pathlib import Path
 from loguru import logger
 
 from gemini_webapi import GeminiClient, AuthError, set_log_level
+from gemini_webapi.constants import Model
 
 logging.getLogger("asyncio").setLevel(logging.ERROR)
 set_log_level("DEBUG")
@@ -27,10 +28,20 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
         response = await self.geminiclient.generate_content("Hello World!")
         self.assertTrue(response.text)
 
+    @logger.catch(reraise=True)
+    async def test_switch_model(self):
+        for model in Model:
+            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}")
+
     @logger.catch(reraise=True)
     async def test_upload_image(self):
         response = await self.geminiclient.generate_content(
-            "Describe these images", images=[Path("assets/banner.png"), "assets/favicon.png"]
+            "Describe these images",
+            images=[Path("assets/banner.png"), "assets/favicon.png"],
         )
         logger.debug(response.text)
 
@@ -86,9 +97,7 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
 
     @logger.catch(reraise=True)
     async def test_card_content(self):
-        response = await self.geminiclient.generate_content(
-            "How is today's weather?"
-        )
+        response = await self.geminiclient.generate_content("How is today's weather?")
         logger.debug(response.text)
 
     @logger.catch(reraise=True)

gemini_webapi-1.6.1/src/gemini_webapi.egg-info/requires.txt

@@ -1,3 +0,0 @@
-httpx>=0.25.2
-pydantic>=2.5.3
-loguru>=0.7.2