gemini-webapi 1.12.2__tar.gz → 1.13.0__tar.gz

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.12.2
+Version: 1.13.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -736,7 +736,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [Generate images with Imagen3](#generate-images-with-imagen3)
   - [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)
+  - [Logging Configuration](#logging-configuration)
 - [References](#references)
 - [Stargazers](#stargazers)
 
@@ -1036,9 +1036,9 @@ async def main():
 asyncio.run(main())
 ```
 
-### Control log level
+### Logging Configuration
 
-You can set the log level of the package to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
+This package uses [loguru](https://loguru.readthedocs.io/en/stable/) for logging, and exposes a function `set_log_level` to control log level. You can set log level to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
 
 ```python
 from gemini_webapi import set_log_level
@@ -1046,6 +1046,10 @@ from gemini_webapi import set_log_level
 set_log_level("DEBUG")
 ```
 
+> [!NOTE]
+>
+> Calling `set_log_level` for the first time will **globally** remove all existing loguru handlers. You may want to configure logging directly with loguru to avoid this issue and have more advanced control over logging behaviors.
+
 ## References
 
 [Google AI Studio](https://ai.google.dev/tutorials/ai-studio_quickstart)

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/README.md

@@ -53,7 +53,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [Generate images with Imagen3](#generate-images-with-imagen3)
   - [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)
+  - [Logging Configuration](#logging-configuration)
 - [References](#references)
 - [Stargazers](#stargazers)
 
@@ -353,9 +353,9 @@ async def main():
 asyncio.run(main())
 ```
 
-### Control log level
+### Logging Configuration
 
-You can set the log level of the package to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
+This package uses [loguru](https://loguru.readthedocs.io/en/stable/) for logging, and exposes a function `set_log_level` to control log level. You can set log level to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
 
 ```python
 from gemini_webapi import set_log_level
@@ -363,6 +363,10 @@ from gemini_webapi import set_log_level
 set_log_level("DEBUG")
 ```
 
+> [!NOTE]
+>
+> Calling `set_log_level` for the first time will **globally** remove all existing loguru handlers. You may want to configure logging directly with loguru to avoid this issue and have more advanced control over logging behaviors.
+
 ## References
 
 [Google AI Studio](https://ai.google.dev/tutorials/ai-studio_quickstart)

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/src/gemini_webapi/__init__.py

@@ -1,4 +1,4 @@
 from .client import GeminiClient, ChatSession  # noqa: F401
 from .exceptions import *  # noqa: F401, F403
 from .types import *  # noqa: F401, F403
-from .utils import set_log_level  # noqa: F401
+from .utils import set_log_level, logger  # noqa: F401

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/src/gemini_webapi/client.py

@@ -422,7 +422,7 @@ class GeminiClient:
                 for candidate_index, candidate in enumerate(body[4]):
                     text = candidate[1][0]
                     if re.match(
-                        r"^http://googleusercontent\.com/card_content/\d+$", text
+                        r"^http://googleusercontent\.com/card_content/\d+", text
                     ):
                         text = candidate[22] and candidate[22][0] or text
 
@@ -470,7 +470,7 @@ class GeminiClient:
                         img_candidate = img_body[4][candidate_index]
 
                         text = re.sub(
-                            r"http://googleusercontent\.com/image_generation_content/\d+$",
+                            r"http://googleusercontent\.com/image_generation_content/\d+",
                             "",
                             img_candidate[1][0],
                         ).rstrip()

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/src/gemini_webapi/constants.py

@@ -27,7 +27,7 @@ class Model(Enum):
     UNSPECIFIED = ("unspecified", {}, False)
     G_2_5_FLASH = (
         "gemini-2.5-flash",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"9ec249fc9ad08861"]'},
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"71c2d248d3b102ff"]'},
         False,
     )
     G_2_5_PRO = (
@@ -49,12 +49,12 @@ class Model(Enum):
         "gemini-2.0-exp-advanced",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
         True,
-    )
+    )  # Deprecated
     G_2_5_EXP_ADVANCED = (
         "gemini-2.5-exp-advanced",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"203e6bb81620bcfe"]'},
         True,
-    )
+    )  # Deprecated
 
     def __init__(self, name, header, advanced_only):
         self.model_name = name

gemini_webapi-1.13.0/src/gemini_webapi/utils/logger.py

@@ -0,0 +1,37 @@
+import sys
+from loguru import logger as _logger
+
+_handler_id = None
+
+
+def set_log_level(level: str | int) -> None:
+    """
+    Set the log level for gemini_webapi. The default log level is "INFO".
+
+    Note: calling this function for the first time will globally remove all existing loguru
+    handlers. To avoid this, you may want to set logging behaviors directly with loguru.
+
+    Parameters
+    ----------
+    level : `str | int`
+        Log level: "TRACE", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"
+
+    Examples
+    --------
+    >>> from gemini_webapi import set_log_level
+    >>> set_log_level("DEBUG")  # Show debug messages
+    >>> set_log_level("ERROR")  # Only show errors
+    """
+
+    global _handler_id
+
+    _logger.remove(_handler_id)
+
+    _handler_id = _logger.add(
+        sys.stderr,
+        level=level,
+        filter=lambda record: record["extra"].get("name") == "gemini_webapi",
+    )
+
+
+logger = _logger.bind(name="gemini_webapi")

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/src/gemini_webapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.12.2
+Version: 1.13.0
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -736,7 +736,7 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
   - [Generate images with Imagen3](#generate-images-with-imagen3)
   - [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)
+  - [Logging Configuration](#logging-configuration)
 - [References](#references)
 - [Stargazers](#stargazers)
 
@@ -1036,9 +1036,9 @@ async def main():
 asyncio.run(main())
 ```
 
-### Control log level
+### Logging Configuration
 
-You can set the log level of the package to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
+This package uses [loguru](https://loguru.readthedocs.io/en/stable/) for logging, and exposes a function `set_log_level` to control log level. You can set log level to one of the following values: `DEBUG`, `INFO`, `WARNING`, `ERROR` and `CRITICAL`. The default value is `INFO`.
 
 ```python
 from gemini_webapi import set_log_level
@@ -1046,6 +1046,10 @@ from gemini_webapi import set_log_level
 set_log_level("DEBUG")
 ```
 
+> [!NOTE]
+>
+> Calling `set_log_level` for the first time will **globally** remove all existing loguru handlers. You may want to configure logging directly with loguru to avoid this issue and have more advanced control over logging behaviors.
+
 ## References
 
 [Google AI Studio](https://ai.google.dev/tutorials/ai-studio_quickstart)

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/src/gemini_webapi.egg-info/SOURCES.txt

@@ -31,6 +31,5 @@ src/gemini_webapi/utils/logger.py
 src/gemini_webapi/utils/rotate_1psidts.py
 src/gemini_webapi/utils/upload_file.py
 tests/test_client_features.py
-tests/test_html_entity_decode.py
 tests/test_rotate_cookies.py
 tests/test_save_image.py

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/tests/test_client_features.py

@@ -3,9 +3,7 @@ import unittest
 import logging
 from pathlib import Path
 
-from loguru import logger
-
-from gemini_webapi import GeminiClient, AuthError, set_log_level
+from gemini_webapi import GeminiClient, AuthError, set_log_level, logger
 from gemini_webapi.constants import Model
 from gemini_webapi.exceptions import UsageLimitExceeded, ModelInvalid
 
@@ -20,7 +18,7 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
         )
 
         try:
-            await self.geminiclient.init(timeout=60)
+            await self.geminiclient.init(timeout=60, auto_refresh=False)
         except AuthError as e:
             self.skipTest(e)
 

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/tests/test_rotate_cookies.py

@@ -1,9 +1,7 @@
 import os
 import asyncio
 
-from loguru import logger
-
-from gemini_webapi import GeminiClient, set_log_level
+from gemini_webapi import GeminiClient, set_log_level, logger
 
 set_log_level("DEBUG")
 

{gemini_webapi-1.12.2 → gemini_webapi-1.13.0}/tests/test_save_image.py

@@ -3,9 +3,8 @@ import unittest
 import logging
 
 from httpx import HTTPError
-from loguru import logger
 
-from gemini_webapi import GeminiClient, AuthError, set_log_level
+from gemini_webapi import GeminiClient, AuthError, set_log_level, logger
 
 logging.getLogger("asyncio").setLevel(logging.ERROR)
 set_log_level("DEBUG")
@@ -18,7 +17,7 @@ class TestGeminiClient(unittest.IsolatedAsyncioTestCase):
         )
 
         try:
-            await self.geminiclient.init()
+            await self.geminiclient.init(auto_refresh=False)
         except AuthError:
             self.skipTest("Test was skipped due to invalid cookies")
 

gemini_webapi-1.12.2/src/gemini_webapi/utils/logger.py

@@ -1,39 +0,0 @@
-import atexit
-from sys import stderr
-
-from loguru._logger import Core as _Core
-from loguru._logger import Logger as _Logger
-
-logger = _Logger(
-    core=_Core(),
-    exception=None,
-    depth=0,
-    record=False,
-    lazy=False,
-    colors=False,
-    raw=False,
-    capture=True,
-    patchers=[],
-    extra={},
-)
-
-if stderr:
-    logger.add(stderr, level="INFO")
-
-atexit.register(logger.remove)
-
-
-def set_log_level(level: str):
-    """
-    Set the log level for the whole module. Default is "INFO". Set to "DEBUG" to see more detailed logs.
-
-    Parameters
-    ----------
-    level : str
-        The log level to set. Must be one of "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL".
-    """
-
-    assert level in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
-
-    logger.remove()
-    logger.add(stderr, level=level)

gemini_webapi-1.12.2/tests/test_html_entity_decode.py

@@ -1,48 +0,0 @@
-import unittest
-from gemini_webapi.types.candidate import Candidate
-
-
-class TestHtmlEntityDecode(unittest.TestCase):
-    def test_html_entity_decoding(self):
-        # Test HTML entity decoding functionality
-        html_encoded_text = (
-            "This is a code snippet: <code>print('Hello, World!')</code>"
-        )
-        expected_decoded_text = (
-            "This is a code snippet: <code>print('Hello, World!')</code>"
-        )
-
-        # Create Candidate instance which should automatically decode HTML entities
-        candidate = Candidate(
-            rcid="test_rcid",
-            text=html_encoded_text,
-            thoughts="Testing &lt;b&gt;HTML&lt;/b&gt; entity decoding",
-        )
-
-        # Verify that text property is correctly decoded
-        self.assertEqual(candidate.text, expected_decoded_text)
-
-        # Verify that thoughts property is correctly decoded
-        self.assertEqual(candidate.thoughts, "Testing <b>HTML</b> entity decoding")
-
-    def test_non_html_text(self):
-        # Test plain text without any HTML entities
-        plain_text = "This is regular text with no HTML entities"
-
-        candidate = Candidate(rcid="test_rcid", text=plain_text)
-
-        # Verify the text remains unchanged
-        self.assertEqual(candidate.text, plain_text)
-
-    def test_complex_html_entities(self):
-        # Test more complex combinations of HTML entities
-        complex_html = "&lt;div&gt;This has &amp;amp; character\n and &quot;quotes&quot;&lt;/div&gt;"
-        expected_decoded = '<div>This has &amp; character\n and "quotes"</div>'
-
-        candidate = Candidate(rcid="test_rcid", text=complex_html)
-
-        self.assertEqual(candidate.text, expected_decoded)
-
-
-if __name__ == "__main__":
-    unittest.main()