gemini-webapi 1.14.1__py3-none-any.whl → 1.14.3__py3-none-any.whl

gemini_webapi/client.py

@@ -7,9 +7,9 @@ from pathlib import Path
 from typing import Any, Optional
 
 import orjson as json
-from httpx import AsyncClient, ReadTimeout
+from httpx import AsyncClient, ReadTimeout, Response
 
-from .constants import Endpoint, ErrorCode, Headers, Model
+from .constants import Endpoint, ErrorCode, Headers, Model, GRPC
 from .exceptions import (
     AuthError,
     APIError,
@@ -20,7 +20,15 @@ from .exceptions import (
     ModelInvalid,
     TemporarilyBlocked,
 )
-from .types import WebImage, GeneratedImage, Candidate, ModelOutput, Gem, GemJar
+from .types import (
+    WebImage,
+    GeneratedImage,
+    Candidate,
+    ModelOutput,
+    Gem,
+    GemJar,
+    RPCData,
+)
 from .utils import (
     upload_file,
     parse_file_name,
@@ -301,66 +309,61 @@ class GeminiClient:
 
         return self._gems
 
-    @running(retry=2)
-    async def fetch_gems(self, **kwargs) -> GemJar:
+    async def fetch_gems(self, include_hidden: bool = False, **kwargs) -> GemJar:
         """
         Get a list of available gems from gemini, including system predefined gems and user-created custom gems.
 
         Note that network request will be sent every time this method is called.
         Once the gems are fetched, they will be cached and accessible via `GeminiClient.gems` property.
 
+        Parameters
+        ----------
+        include_hidden: `bool`, optional
+            There are some predefined gems that by default are not shown to users (and therefore may not work properly).
+            Set this parameter to `True` to include them in the fetched gem list.
+
         Returns
         -------
         :class:`GemJar`
             Refer to `gemini_webapi.types.GemJar`.
         """
 
-        try:
-            response = await self.client.post(
-                Endpoint.BATCH_EXEC,
-                data={
-                    "at": self.access_token,
-                    "f.req": json.dumps(
-                        [
-                            [
-                                ["CNgdBe", '[2,["en"],0]', None, "custom"],
-                                ["CNgdBe", '[3,["en"],0]', None, "system"],
-                            ]
-                        ]
-                    ).decode(),
-                },
-                **kwargs,
-            )
-        except ReadTimeout:
-            raise TimeoutError(
-                "Fetch gems request timed out, please try again. If the problem persists, "
-                "consider setting a higher `timeout` value when initializing GeminiClient."
-            )
+        response = await self._batch_execute(
+            [
+                RPCData(
+                    rpcid=GRPC.LIST_GEMS,
+                    payload="[4]" if include_hidden else "[3]",
+                    identifier="system",
+                ),
+                RPCData(
+                    rpcid=GRPC.LIST_GEMS,
+                    payload="[2]",
+                    identifier="custom",
+                ),
+            ],
+            **kwargs,
+        )
 
-        if response.status_code != 200:
-            raise APIError(
-                f"Failed to fetch gems. Request failed with status code {response.status_code}"
-            )
-        else:
-            try:
-                response_json = json.loads(response.text.split("\n")[2])
+        try:
+            response_json = json.loads(response.text.split("\n")[2])
 
-                predefined_gems, custom_gems = [], []
+            predefined_gems, custom_gems = [], []
 
-                for part in response_json:
-                    if part[-1] == "system":
-                        predefined_gems = json.loads(part[2])[2]
-                    elif part[-1] == "custom":
-                        if custom_gems_container := json.loads(part[2]):
-                            custom_gems = custom_gems_container[2]
+            for part in response_json:
+                if part[-1] == "system":
+                    predefined_gems = json.loads(part[2])[2]
+                elif part[-1] == "custom":
+                    if custom_gems_container := json.loads(part[2]):
+                        custom_gems = custom_gems_container[2]
 
-                if not predefined_gems and not custom_gems:
-                    raise Exception
-            except Exception:
-                logger.debug(f"Invalid response: {response.text}")
-                raise APIError(
-                    "Failed to fetch gems. Invalid response data received. Client will try to re-initialize on next request."
-                )
+            if not predefined_gems and not custom_gems:
+                raise Exception
+        except Exception:
+            await self.close()
+            logger.debug(f"Invalid response: {response.text}")
+            raise APIError(
+                "Failed to fetch gems. Invalid response data received. Client will try to re-initialize on next request."
+            )
 
         self._gems = GemJar(
             itertools.chain(
@@ -664,6 +667,54 @@ class GeminiClient:
 
         return ChatSession(geminiclient=self, **kwargs)
 
+    @running(retry=2)
+    async def _batch_execute(self, payloads: list[RPCData], **kwargs) -> Response:
+        """
+        Execute a batch of requests to Gemini API.
+
+        Parameters
+        ----------
+        payloads: `list[GRPC]`
+            List of `gemini_webapi.types.GRPC` objects to be executed.
+        kwargs: `dict`, optional
+            Additional arguments which will be passed to the post request.
+            Refer to `httpx.AsyncClient.request` for more information.
+
+        Returns
+        -------
+        :class:`httpx.Response`
+            Response object containing the result of the batch execution.
+        """
+
+        try:
+            response = await self.client.post(
+                Endpoint.BATCH_EXEC,
+                data={
+                    "at": self.access_token,
+                    "f.req": json.dumps(
+                        [[payload.serialize() for payload in payloads]]
+                    ).decode(),
+                },
+                **kwargs,
+            )
+        except ReadTimeout:
+            raise TimeoutError(
+                "Batch execute request timed out, please try again. If the problem persists, "
+                "consider setting a higher `timeout` value when initializing GeminiClient."
+            )
+
+        if response.status_code != 200:
+            logger.debug(
+                f"Batch execution failed with status code {response.status_code}. "
+                f"RPC: {', '.join({payload.rpcid.name for payload in payloads})}; "
+                f"Invalid response: {response.text}"
+            )
+            raise APIError(
+                f"Batch execution failed with status code {response.status_code}"
+            )
+
+        return response
+
 
 class ChatSession:
     """

gemini_webapi/constants.py

@@ -10,6 +10,18 @@ class Endpoint(StrEnum):
     BATCH_EXEC = "https://gemini.google.com/_/BardChatUi/data/batchexecute"
 
 
+class GRPC(StrEnum):
+    """
+    Google RPC ids used in Gemini API.
+    """
+
+    # Chat methods
+    READ_CHAT = "hNvQHb"
+
+    # Gem methods
+    LIST_GEMS = "CNgdBe"
+
+
 class Headers(Enum):
     GEMINI = {
         "Content-Type": "application/x-www-form-urlencoded;charset=utf-8",

gemini_webapi/types/__init__.py

@@ -2,5 +2,6 @@
 
 from .candidate import Candidate
 from .gem import Gem, GemJar
+from .grpc import RPCData
 from .image import Image, WebImage, GeneratedImage
 from .modeloutput import ModelOutput

gemini_webapi/types/grpc.py

@@ -0,0 +1,34 @@
+from pydantic import BaseModel
+
+from ..constants import GRPC
+
+
+class RPCData(BaseModel):
+    """
+    Helper class containing necessary data for Google RPC calls.
+
+    Parameters
+    ----------
+    rpcid : GRPC
+        Google RPC ID.
+    payload : str
+        Payload for the RPC call.
+    identifier : str, optional
+        Identifier/order for the RPC call, defaults to "generic".
+        Makes sense if there are multiple RPC calls in a batch, where this identifier
+        can be used to distinguish between responses.
+    """
+
+    rpcid: GRPC
+    payload: str
+    identifier: str = "generic"
+
+    def __repr__(self):
+        return f"GRPC(rpcid='{self.rpcid}', payload='{self.payload}', identifier='{self.identifier}')"
+
+    def serialize(self) -> list:
+        """
+        Serializes object into formatted payload ready for RPC call.
+        """
+
+        return [self.rpcid, self.payload, None, self.identifier]

gemini_webapi/types/image.py

@@ -137,7 +137,7 @@ class GeneratedImage(Image):
         return v
 
     # @override
-    async def save(self, **kwargs) -> str | None:
+    async def save(self, full_size=True, **kwargs) -> str | None:
         """
         Save the image to disk.
 
@@ -146,6 +146,8 @@ class GeneratedImage(Image):
         filename: `str`, optional
             Filename to save the image, generated images are always in .png format, but file extension will not be included in the URL.
             And since the URL ends with a long hash, by default will use timestamp + end of the hash as the filename.
+        full_size: `bool`, optional
+            If True, will modify the default preview (512*512) URL to get the full size image.
         kwargs: `dict`, optional
             Other arguments to pass to `Image.save`.
 
@@ -155,6 +157,9 @@ class GeneratedImage(Image):
             Absolute path of the saved image if successfully saved.
         """
 
+        if full_size:
+            self.url += "=s2048"
+
         return await super().save(
             filename=kwargs.pop("filename", None)
             or f"{datetime.now().strftime('%Y%m%d%H%M%S')}_{self.url[-10:]}.png",

{gemini_webapi-1.14.1.dist-info → gemini_webapi-1.14.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.1
+Version: 1.14.3
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -931,10 +931,14 @@ asyncio.run(main())
 
 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.
 
+> [!TIP]
+>
+> There are some system predefined gems that by default are not shown to users (and therefore may not work properly). Use `client.fetch_gems(include_hidden=True)` to include them in the fetch result.
+
 ```python
 async def main():
     # Fetch all gems for the current account, including both predefined and user-created ones
-    await client.fetch_gems()
+    await client.fetch_gems(include_hidden=False)
 
     # Once fetched, gems will be cached in `GeminiClient.gems`
     gems = client.gems

{gemini_webapi-1.14.1.dist-info → gemini_webapi-1.14.3.dist-info}/RECORD

@@ -1,11 +1,12 @@
 gemini_webapi/__init__.py,sha256=7ELCiUoI10ea3daeJxnv0UwqLVKpM7rxsgOZsPMstO8,150
-gemini_webapi/client.py,sha256=9t_ytH4ClkjJWUwvLJ6KyArSRbNeGxd2MJVyPUmBIUA,30333
-gemini_webapi/constants.py,sha256=rkugkck9qXW_f-9dc0XXEjVIzJ9Ja4gHBVYDP71BESg,2352
+gemini_webapi/client.py,sha256=XrPG5LtJDYpEOMyjL8zxRO9ExFwp9EO5UpsNxhouSvk,31820
+gemini_webapi/constants.py,sha256=M2fbQSSVhF0ROv_9CTR4n8sDOujJmLRxwA6ARQ0Uf7k,2519
 gemini_webapi/exceptions.py,sha256=qkXrIpr0L7LtGbq3VcTO8D1xZ50pJtt0dDRp5I3uDSg,1038
-gemini_webapi/types/__init__.py,sha256=Xap_FGOKOOC9mU3bmp_VT_1pQgFMuqbys_fOcyvTnuE,166
+gemini_webapi/types/__init__.py,sha256=1DU4JEw2KHQJbtOgOuvoEXtzQCMwAkyo0koSFVdT9JY,192
 gemini_webapi/types/candidate.py,sha256=67BhY75toE5mVuB21cmHcTFtw332V_KmCjr3-9VTbJo,1477
 gemini_webapi/types/gem.py,sha256=3Ppjq9V22Zp4Lb9a9ZnDviDKQpfSQf8UZxqOEjeEWd4,4070
-gemini_webapi/types/image.py,sha256=MflOex2tAxBF5zQkYTGTR78CuiqH3Wa6KxMKKXr5Yvo,5233
+gemini_webapi/types/grpc.py,sha256=S64h1oeC7ZJC50kmS_C2CQ7WVTanhJ4kqTFx5ZYayXI,917
+gemini_webapi/types/image.py,sha256=xCPaMzstIM5C09ZkJ0545vVPAsrF3y89XnRJrEWGz5k,5436
 gemini_webapi/types/modeloutput.py,sha256=h07kQOkL5r-oPLvZ59uVtO1eP4FGy5ZpzuYQzAeQdr8,1196
 gemini_webapi/utils/__init__.py,sha256=cJ9HQYxr8l0CsY61TFlho-5DdPxCaEOpRfjxPX-eils,320
 gemini_webapi/utils/get_access_token.py,sha256=eNn1omFO41wWXco1eM-KXR2CEi0Tb-chlph7H-PCNjg,6137
@@ -13,8 +14,8 @@ gemini_webapi/utils/load_browser_cookies.py,sha256=A5n_VsB7Rm8ck5lpy856UNJEhv30l
 gemini_webapi/utils/logger.py,sha256=0VcxhVLhHBRDQutNCpapP1y_MhPoQ2ud1uIFLqxC3Z8,958
 gemini_webapi/utils/rotate_1psidts.py,sha256=NyQ9OYPLBOcvpc8bodvEYDIVFrsYN0kdfc831lPEctM,1680
 gemini_webapi/utils/upload_file.py,sha256=SJOMr6kryK_ClrKmqI96fqZBNFOMPsyAvFINAGAU3rk,1468
-gemini_webapi-1.14.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.14.1.dist-info/METADATA,sha256=Qlym2uwFz03bgKIed0UtBmLMGXfpJwJ8j-2qMraLgJw,58925
-gemini_webapi-1.14.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gemini_webapi-1.14.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.14.1.dist-info/RECORD,,
+gemini_webapi-1.14.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.14.3.dist-info/METADATA,sha256=GNbHaHKIyQD0t3_MhkaxjHufYV7B5S1X5YUDG9-8e5o,59157
+gemini_webapi-1.14.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gemini_webapi-1.14.3.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.14.3.dist-info/RECORD,,