gemini-webapi 1.14.4__py3-none-any.whl → 1.15.1__py3-none-any.whl

gemini_webapi/client.py

@@ -292,7 +292,9 @@ class GeminiClient(GemMixin):
             model = Model.from_name(model)
 
         if isinstance(gem, Gem):
-            gem = gem.id
+            gem_id = gem.id
+        else:
+            gem_id = gem
 
         if self.auto_close:
             await self.reset_close_task()
@@ -325,7 +327,7 @@ class GeminiClient(GemMixin):
                                     None,
                                     chat and chat.metadata,
                                 ]
-                                + (gem and [None] * 16 + [gem] or [])
+                                + (gem_id and [None] * 16 + [gem_id] or [])
                             ).decode(),
                         ]
                     ).decode(),
@@ -551,12 +553,10 @@ class GeminiClient(GemMixin):
                 "consider setting a higher `timeout` value when initializing GeminiClient."
             )
 
+        # ? Seems like batch execution will immediately invalidate the current access token,
+        # ? causing the next request to fail with 401 Unauthorized.
         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}"
-            )
+            await self.close()
             raise APIError(
                 f"Batch execution failed with status code {response.status_code}"
             )

gemini_webapi/components/gem_mixin.py

@@ -131,10 +131,145 @@ class GemMixin:
 
         return self._gems
 
+    @running(retry=2)
+    async def create_gem(self, name: str, prompt: str, description: str = "") -> Gem:
+        """
+        Create a new custom gem.
+
+        Parameters
+        ----------
+        name: `str`
+            Name of the custom gem.
+        prompt: `str`
+            System instructions for the custom gem.
+        description: `str`, optional
+            Description of the custom gem (has no effect on the model's behavior).
+
+        Returns
+        -------
+        :class:`Gem`
+            The created gem.
+        """
+
+        response = await self._batch_execute(
+            [
+                RPCData(
+                    rpcid=GRPC.CREATE_GEM,
+                    payload=json.dumps(
+                        [
+                            [
+                                name,
+                                description,
+                                prompt,
+                                None,
+                                None,
+                                None,
+                                None,
+                                None,
+                                0,
+                                None,
+                                1,
+                                None,
+                                None,
+                                None,
+                                [],
+                            ]
+                        ]
+                    ).decode(),
+                )
+            ]
+        )
+
+        try:
+            response_json = json.loads(response.text.split("\n")[2])
+            gem_id = json.loads(response_json[0][2])[0]
+        except Exception:
+            await self.close()
+            logger.debug(f"Invalid response: {response.text}")
+            raise APIError(
+                "Failed to create gem. Invalid response data received. Client will try to re-initialize on next request."
+            )
+
+        return Gem(
+            id=gem_id,
+            name=name,
+            description=description,
+            prompt=prompt,
+            predefined=False,
+        )
+
+    @running(retry=2)
+    async def update_gem(
+        self, gem: Gem | str, name: str, prompt: str, description: str = ""
+    ) -> Gem:
+        """
+        Update an existing custom gem.
+
+        Parameters
+        ----------
+        gem: `Gem | str`
+            Gem to update, can be either a `gemini_webapi.types.Gem` object or a gem id string.
+        name: `str`
+            New name for the custom gem.
+        prompt: `str`
+            New system instructions for the custom gem.
+        description: `str`, optional
+            New description of the custom gem (has no effect on the model's behavior).
+
+        Returns
+        -------
+        :class:`Gem`
+            The updated gem.
+        """
+
+        if isinstance(gem, Gem):
+            gem_id = gem.id
+        else:
+            gem_id = gem
+
+        await self._batch_execute(
+            [
+                RPCData(
+                    rpcid=GRPC.UPDATE_GEM,
+                    payload=json.dumps(
+                        [
+                            gem_id,
+                            [
+                                name,
+                                description,
+                                prompt,
+                                None,
+                                None,
+                                None,
+                                None,
+                                None,
+                                0,
+                                None,
+                                1,
+                                None,
+                                None,
+                                None,
+                                [],
+                                0,
+                            ],
+                        ]
+                    ).decode(),
+                )
+            ]
+        )
+
+        return Gem(
+            id=gem_id,
+            name=name,
+            description=description,
+            prompt=prompt,
+            predefined=False,
+        )
+
     @running(retry=2)
     async def delete_gem(self, gem: Gem | str, **kwargs) -> None:
         """
-        Delete a custom gem from gemini.google.com.
+        Delete a custom gem.
 
         Parameters
         ----------
@@ -143,9 +278,11 @@ class GemMixin:
         """
 
         if isinstance(gem, Gem):
-            gem = gem.id
+            gem_id = gem.id
+        else:
+            gem_id = gem
 
         await self._batch_execute(
-            [RPCData(rpcid=GRPC.DELETE_GEM, payload=[gem])],
+            [RPCData(rpcid=GRPC.DELETE_GEM, payload=json.dumps([gem_id]).decode())],
             **kwargs,
         )

gemini_webapi/constants.py

@@ -20,6 +20,8 @@ class GRPC(StrEnum):
 
     # Gem methods
     LIST_GEMS = "CNgdBe"
+    CREATE_GEM = "oMH3Zd"
+    UPDATE_GEM = "kHv0Vd"
     DELETE_GEM = "UXcSJb"
 
 
@@ -42,12 +44,12 @@ class Model(Enum):
     UNSPECIFIED = ("unspecified", {}, False)
     G_2_5_FLASH = (
         "gemini-2.5-flash",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"71c2d248d3b102ff"]'},
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"71c2d248d3b102ff",null,null,0,[4]]'},
         False,
     )
     G_2_5_PRO = (
         "gemini-2.5-pro",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"2525e3954d185b3c"]'},
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"4af6c7f5da75d65d",null,null,0,[4]]'},
         False,
     )
     G_2_0_FLASH = (

{gemini_webapi-1.14.4.dist-info → gemini_webapi-1.15.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.4
+Version: 1.15.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -734,6 +734,10 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [Continue previous conversations](#continue-previous-conversations)
   - [Select language model](#select-language-model)
   - [Apply system prompt with Gemini Gems](#apply-system-prompt-with-gemini-gems)
+  - [Manage Custom Gems](#manage-custom-gems)
+    - [Create a custom gem](#create-a-custom-gem)
+    - [Update an existing gem](#update-an-existing-gem)
+    - [Delete a custom gem](#delete-a-custom-gem)
   - [Retrieve model's thought process](#retrieve-models-thought-process)
   - [Retrieve images in response](#retrieve-images-in-response)
   - [Generate images with Imagen4](#generate-images-with-imagen4)
@@ -963,6 +967,75 @@ async def main():
     print(response2)
 ```
 
+### Manage Custom Gems
+
+You can create, update, and delete your custom gems programmatically with the API. Note that predefined system gems cannot be modified or deleted.
+
+#### Create a custom gem
+
+Create a new custom gem with a name, system prompt (instructions), and optional description:
+
+```python
+async def main():
+    # Create a new custom gem
+    new_gem = await client.create_gem(
+        name="Python Tutor",
+        prompt="You are a helpful Python programming tutor.",
+        description="A specialized gem for Python programming"
+    )
+
+    print(f"Custom gem created: {new_gem}")
+
+    # Use the newly created gem in a conversation
+    response = await client.generate_content(
+        "Explain how list comprehensions work in Python",
+        gem=new_gem
+    )
+    print(response.text)
+
+asyncio.run(main())
+```
+
+#### Update an existing gem
+
+> [!NOTE]
+>
+> When updating a gem, you must provide all parameters (name, prompt, description) even if you only want to change one of them.
+
+```python
+async def main():
+    # Get a custom gem (assuming you have one named "Python Tutor")
+    await client.fetch_gems()
+    python_tutor = client.gems.get(name="Python Tutor")
+
+    # Update the gem with new instructions
+    updated_gem = await client.update_gem(
+        gem=python_tutor,  # Can also pass gem ID string
+        name="Advanced Python Tutor",
+        prompt="You are an expert Python programming tutor.",
+        description="An advanced Python programming assistant"
+    )
+
+    print(f"Custom gem updated: {updated_gem}")
+
+asyncio.run(main())
+```
+
+#### Delete a custom gem
+
+```python
+async def main():
+    # Get the gem to delete
+    await client.fetch_gems()
+    gem_to_delete = client.gems.get(name="Advanced Python Tutor")
+
+    # Delete the gem
+    await client.delete_gem(gem_to_delete)  # Can also pass gem ID string
+    print(f"Custom gem deleted: {gem_to_delete.name}")
+
+asyncio.run(main())
+```
+
 ### Retrieve model's thought process
 
 When using models with thinking capabilities, the model's thought process will be populated in `ModelOutput.thoughts`.

{gemini_webapi-1.14.4.dist-info → gemini_webapi-1.15.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
 gemini_webapi/__init__.py,sha256=7ELCiUoI10ea3daeJxnv0UwqLVKpM7rxsgOZsPMstO8,150
-gemini_webapi/client.py,sha256=nnInjBRsBLLLcc07f2oFv4HjwMXXCBx5Y_RMwo8NYtM,26891
-gemini_webapi/constants.py,sha256=Ulj38JVv8lPW7GZRTb25sLM2utuY-q7aDTrgm-bXbrU,2545
+gemini_webapi/client.py,sha256=G0nmRCeEmT0BpqgYY4zxb_ozFDmb39JqtddKpIN6JlA,26870
+gemini_webapi/constants.py,sha256=z8Jr6NQk79mgZOgmwx8vbgzyORHQVBjoJ4ZhUdDWWXw,2629
 gemini_webapi/exceptions.py,sha256=qkXrIpr0L7LtGbq3VcTO8D1xZ50pJtt0dDRp5I3uDSg,1038
 gemini_webapi/components/__init__.py,sha256=wolxuAJJ32-jmHOKgpsesexP7hXea1JMo5vI52wysTI,48
-gemini_webapi/components/gem_mixin.py,sha256=HUrTBHQC9Lzc53Le2droi_UoFxTOZMRLMbn6MRdcHfI,4727
+gemini_webapi/components/gem_mixin.py,sha256=WPJkYDS4yQpLMBNQ94LQo5w59RgkllWaSiHsFG1k5GU,8795
 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
@@ -17,8 +17,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.4.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.14.4.dist-info/METADATA,sha256=-d9j4G1z2fh6ohTAOeNW0I2O2HzlG0D48TKH67KCXFE,59156
-gemini_webapi-1.14.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gemini_webapi-1.14.4.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.14.4.dist-info/RECORD,,
+gemini_webapi-1.15.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.15.1.dist-info/METADATA,sha256=baIL2RVMPLqKe1l_lMxi5QDpTEJbNdehoF4uk2-467Q,61279
+gemini_webapi-1.15.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gemini_webapi-1.15.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.15.1.dist-info/RECORD,,