gemini-webapi 1.12.3__py3-none-any.whl → 1.13.0__py3-none-any.whl

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/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/utils/logger.py

@@ -1,39 +1,37 @@
-import atexit
-from sys import stderr
+import sys
+from loguru import logger as _logger
 
-from loguru._logger import Core as _Core
-from loguru._logger import Logger as _Logger
+_handler_id = None
 
-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):
+def set_log_level(level: str | int) -> None:
     """
-    Set the log level for the whole module. Default is "INFO". Set to "DEBUG" to see more detailed logs.
+    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
-        The log level to set. Must be one of "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL".
+    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
     """
 
-    assert level in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
+    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.remove()
-    logger.add(stderr, level=level)
+logger = _logger.bind(name="gemini_webapi")

{gemini_webapi-1.12.3.dist-info → gemini_webapi-1.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.12.3
+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.3.dist-info → gemini_webapi-1.13.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
-gemini_webapi/__init__.py,sha256=28uNIywK4vCXxENaSagNWUzhqr1RyNtLzDF6WRRM4KQ,194
-gemini_webapi/client.py,sha256=W4t4z9Q8tBiS7URdVKFytyELzQylV-K_82NsSqkvrgU,25398
+gemini_webapi/__init__.py,sha256=UBrwmqT-7nWrEr90xz78KqWqMIdapIdjkUyYEZsIktc,202
+gemini_webapi/client.py,sha256=wyLDWl-CXLg4YN13HqWj8TKX-IzAzArDP8Jwofppj18,25396
 gemini_webapi/constants.py,sha256=2KKPqz0r3DKYuxQhwCQpTBQmp9sqFMrH88n3wG41V7g,2572
 gemini_webapi/exceptions.py,sha256=qkXrIpr0L7LtGbq3VcTO8D1xZ50pJtt0dDRp5I3uDSg,1038
 gemini_webapi/types/__init__.py,sha256=d2kvXnE004s2E2KDmPPLi5N-BQ59FgDSlrGrO3Wphww,163
@@ -9,11 +9,11 @@ gemini_webapi/types/modeloutput.py,sha256=h07kQOkL5r-oPLvZ59uVtO1eP4FGy5ZpzuYQzA
 gemini_webapi/utils/__init__.py,sha256=tO6Sx-3fcPeyITZcvUmcFKBFlR5XW87xUFbNrIh3_mE,374
 gemini_webapi/utils/get_access_token.py,sha256=uyb6tuzPr3mHttCjiM86M29ykrnHqsUClYdf5sVkyEQ,5465
 gemini_webapi/utils/load_browser_cookies.py,sha256=A5n_VsB7Rm8ck5lpy856UNJEhv30l3dvQ3j0g3ln1fE,1535
-gemini_webapi/utils/logger.py,sha256=PF4ROQq7scRRrWzeYdeYiYs2S2Jqr0bgjyrPbXVOCqE,816
+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.12.3.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.12.3.dist-info/METADATA,sha256=A-7kQZNvMeii9afd8XtpLo8pX2oQ8ivqpoj3WOljoJI,57167
-gemini_webapi-1.12.3.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-gemini_webapi-1.12.3.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.12.3.dist-info/RECORD,,
+gemini_webapi-1.13.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.13.0.dist-info/METADATA,sha256=4JhYUbv5-iqYquPepYL_wE46lut6y6dFw5DFdBP89vg,57546
+gemini_webapi-1.13.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+gemini_webapi-1.13.0.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.13.0.dist-info/RECORD,,