gemini-webapi 1.14.4__tar.gz → 1.15.0__tar.gz

{gemini_webapi-1.14.4 → gemini_webapi-1.15.0}/.github/workflows/github-release.yml

@@ -11,7 +11,7 @@ jobs:
     permissions:
       contents: write
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: ncipollo/release-action@v1
         with:
           body: ${{ github.event.head_commit.message }}

{gemini_webapi-1.14.4 → gemini_webapi-1.15.0}/.github/workflows/pypi-publish.yml

@@ -24,7 +24,7 @@ jobs:
     name: Build package
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Set up Python
         uses: actions/setup-python@v5
         with:

{gemini_webapi-1.14.4 → gemini_webapi-1.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.4
+Version: 1.15.0
 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 → gemini_webapi-1.15.0}/README.md

@@ -50,6 +50,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)
@@ -279,6 +283,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 → gemini_webapi-1.15.0}/src/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-1.14.4 → gemini_webapi-1.15.0}/src/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-1.14.4 → gemini_webapi-1.15.0}/src/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"
 
 

{gemini_webapi-1.14.4 → gemini_webapi-1.15.0}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.14.4
+Version: 1.15.0
 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.15.0/tests/test_gem_mixin.py

@@ -0,0 +1,88 @@
+import os
+import random
+import unittest
+import logging
+
+from gemini_webapi import GeminiClient, set_log_level, logger
+from gemini_webapi.exceptions import AuthError
+
+logging.getLogger("asyncio").setLevel(logging.ERROR)
+set_log_level("DEBUG")
+
+
+class TestGemMixin(unittest.IsolatedAsyncioTestCase):
+    async def asyncSetUp(self):
+        self.geminiclient = GeminiClient(
+            os.getenv("SECURE_1PSID"), os.getenv("SECURE_1PSIDTS"), verify=False
+        )
+
+        try:
+            await self.geminiclient.init(timeout=60, auto_refresh=False)
+        except AuthError as e:
+            self.skipTest(e)
+
+    @logger.catch(reraise=True)
+    async def test_fetch_gems(self):
+        await self.geminiclient.fetch_gems(include_hidden=True)
+        gems = self.geminiclient.gems
+        self.assertTrue(len(gems.filter(predefined=True)) > 0)
+        for gem in gems:
+            logger.debug(gem.name)
+
+        custom_gems = gems.filter(predefined=False)
+        if custom_gems:
+            logger.debug(f"Found {len(custom_gems)} custom gems:")
+            for gem in custom_gems:
+                logger.debug(gem)
+
+    @logger.catch(reraise=True)
+    async def test_create_gem(self):
+        gem = await self.geminiclient.create_gem(
+            name="Test Gem",
+            prompt="Gemini API has launched creating gem functionality on Aug 1st, 2025",
+            description="This gem is used for testing the functionality of Gemini API",
+        )
+        logger.debug(f"Gem created: {gem}")
+
+    @logger.catch(reraise=True)
+    async def test_update_gem(self):
+        await self.geminiclient.fetch_gems()
+        custom_gems = self.geminiclient.gems.filter(predefined=False)
+        if not custom_gems:
+            self.skipTest("No custom gems available to update.")
+
+        last_created_gem = next(iter(custom_gems.values()))
+        randint = random.randint(0, 100)
+        updated_gem = await self.geminiclient.update_gem(
+            last_created_gem.id,
+            name="Updated Test Gem",
+            prompt="Updated prompt for the gem.",
+            description=f"{randint}",
+        )
+        logger.debug(f"Gem updated: {updated_gem}")
+
+        await self.geminiclient.fetch_gems()
+        custom_gems = self.geminiclient.gems.filter(predefined=False)
+        last_created_gem = next(iter(custom_gems.values()))
+        self.assertEqual(last_created_gem.description, updated_gem.description)
+
+    @logger.catch(reraise=True)
+    async def test_delete_gem(self):
+        await self.geminiclient.fetch_gems()
+        custom_gems = self.geminiclient.gems.filter(predefined=False)
+        total_before_deletion = len(custom_gems)
+        if total_before_deletion == 0:
+            self.skipTest("No custom gems available to delete.")
+
+        last_created_gem = next(iter(custom_gems.values()))
+        await self.geminiclient.delete_gem(last_created_gem.id)
+        logger.debug(f"Gem deleted: {last_created_gem}")
+
+        await self.geminiclient.fetch_gems()
+        custom_gems = self.geminiclient.gems.filter(predefined=False)
+        total_after_deletion = len(custom_gems)
+        self.assertEqual(total_after_deletion, total_before_deletion - 1)
+
+
+if __name__ == "__main__":
+    unittest.main()

gemini_webapi-1.14.4/tests/test_gem_mixin.py

@@ -1,33 +0,0 @@
-import os
-import unittest
-import logging
-
-from gemini_webapi import GeminiClient, set_log_level, logger
-from gemini_webapi.exceptions import AuthError
-
-logging.getLogger("asyncio").setLevel(logging.ERROR)
-set_log_level("DEBUG")
-
-
-class TestGemMixin(unittest.IsolatedAsyncioTestCase):
-    async def asyncSetUp(self):
-        self.geminiclient = GeminiClient(
-            os.getenv("SECURE_1PSID"), os.getenv("SECURE_1PSIDTS"), verify=False
-        )
-
-        try:
-            await self.geminiclient.init(timeout=60, auto_refresh=False)
-        except AuthError as e:
-            self.skipTest(e)
-
-    @logger.catch(reraise=True)
-    async def test_fetch_gems(self):
-        await self.geminiclient.fetch_gems(include_hidden=True)
-        gems = self.geminiclient.gems
-        self.assertTrue(len(gems.filter(predefined=True)) > 0)
-        for gem in gems:
-            logger.debug(gem.name)
-
-
-if __name__ == "__main__":
-    unittest.main()