gemini-webapi 1.8.4__py3-none-any.whl → 1.9.1__py3-none-any.whl

gemini_webapi/client.py

@@ -361,14 +361,17 @@ class GeminiClient:
             try:
                 response_json = json.loads(response.text.split("\n")[2])
 
-                # Plain request
-                body = json.loads(response_json[0][2])
-
-                if not body[4]:
-                    # Request with Gemini extensions enabled
-                    body = json.loads(response_json[4][2])
-
-                if not body[4]:
+                body = None
+                for part in response_json:
+                    try:
+                        main_part = json.loads(part[2])
+                        if main_part[4]:
+                            body = main_part
+                            break
+                    except (IndexError, TypeError, ValueError):
+                        continue
+
+                if not body:
                     raise Exception
             except Exception:
                 await self.close()
@@ -386,6 +389,11 @@ class GeminiClient:
                     ):
                         text = candidate[22] and candidate[22][0] or text
 
+                    try:
+                        thoughts = candidate[37][0][0]
+                    except (TypeError, IndexError):
+                        thoughts = None
+
                     web_images = (
                         candidate[12]
                         and candidate[12][1]
@@ -435,6 +443,7 @@ class GeminiClient:
                         Candidate(
                             rcid=candidate[0],
                             text=text,
+                            thoughts=thoughts,
                             web_images=web_images,
                             generated_images=generated_images,
                         )

gemini_webapi/constants.py

@@ -25,6 +25,31 @@ class Headers(Enum):
 
 class Model(Enum):
     UNSPECIFIED = ("unspecified", {}, False)
+    G_2_0_FLASH = (
+        "gemini-2.0-flash",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f299729663a2343f"]'},
+        False,
+    )
+    G_2_0_FLASH_EXP = (
+        "gemini-2.0-flash-exp",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f299729663a2343f"]'},
+        False,
+    )  # Deprecated, should be removed in the future
+    G_2_0_FLASH_THINKING = (
+        "gemini-2.0-flash-thinking",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"9c17b1863f581b8a"]'},
+        False,
+    )
+    G_2_0_FLASH_THINKING_WITH_APPS = (
+        "gemini-2.0-flash-thinking-with-apps",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f8f8f5ea629f5d37"]'},
+        False,
+    )
+    G_2_0_EXP_ADVANCED = (
+        "gemini-2.0-exp-advanced",
+        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
+        True,
+    )
     G_1_5_FLASH = (
         "gemini-1.5-flash",
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"418ab5ea040b5c43"]'},
@@ -40,16 +65,6 @@ class Model(Enum):
         {"x-goog-ext-525001261-jspb": '[null,null,null,null,"e5a44cb1dae2b489"]'},
         True,
     )
-    G_2_0_FLASH_EXP = (
-        "gemini-2.0-flash-exp",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"f299729663a2343f"]'},
-        False,
-    )
-    G_2_0_EXP_ADVANCED = (
-        "gemini-2.0-exp-advanced",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
-        True,
-    )
 
     def __init__(self, name, header, advanced_only):
         self.model_name = name

gemini_webapi/types/candidate.py

@@ -13,6 +13,8 @@ class Candidate(BaseModel):
         Reply candidate ID to build the metadata
     text: `str`
         Text output
+    thoughts: `str`, optional
+        Model's thought process, can be empty. Only populated with `-thinking` models
     web_images: `list[WebImage]`, optional
         List of web images in reply, can be empty.
     generated_images: `list[GeneratedImage]`, optional
@@ -21,6 +23,7 @@ class Candidate(BaseModel):
 
     rcid: str
     text: str
+    thoughts: str | None = None
     web_images: list[WebImage] = []
     generated_images: list[GeneratedImage] = []
 

gemini_webapi/types/modeloutput.py

@@ -32,6 +32,10 @@ class ModelOutput(BaseModel):
     def text(self) -> str:
         return self.candidates[self.chosen].text
 
+    @property
+    def thoughts(self) -> str | None:
+        return self.candidates[self.chosen].thoughts
+
     @property
     def images(self) -> list[Image]:
         return self.candidates[self.chosen].images

{gemini_webapi-1.8.4.dist-info → gemini_webapi-1.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: gemini-webapi
-Version: 1.8.4
+Version: 1.9.1
 Summary: ✨ An elegant async Python wrapper for Google Gemini web app
 Author: UZQueen
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE
@@ -679,6 +679,7 @@ License-File: LICENSE
 Requires-Dist: httpx[http2]~=0.28.1
 Requires-Dist: pydantic~=2.10.5
 Requires-Dist: loguru~=0.7.3
+Dynamic: license-file
 
 <p align="center">
     <img src="https://raw.githubusercontent.com/HanaokaYuzu/Gemini-API/master/assets/banner.png" width="55%" alt="Gemini Banner" align="center">
@@ -725,14 +726,15 @@ A reverse-engineered asynchronous python wrapper for [Google Gemini](https://gem
 - [Authentication](#authentication)
 - [Usage](#usage)
   - [Initialization](#initialization)
+  - [Select language model](#select-language-model)
   - [Generate contents from text](#generate-contents-from-text)
   - [Generate contents from image](#generate-contents-from-image)
   - [Conversations across multiple turns](#conversations-across-multiple-turns)
   - [Continue previous conversations](#continue-previous-conversations)
+  - [Retrieve model's thought process](#retrieve-models-thought-process)
   - [Retrieve images in response](#retrieve-images-in-response)
   - [Generate images with ImageFx](#generate-images-with-imagefx)
   - [Save images to local files](#save-images-to-local-files)
-  - [Specify language model version](#specify-language-model-version)
   - [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)
@@ -815,6 +817,41 @@ asyncio.run(main())
 >
 > `auto_close` and `close_delay` are optional arguments for automatically closing the client after a certain period of inactivity. This feature is disabled by default. In an always-on service like chatbot, it's recommended to set `auto_close` to `True` combined with reasonable seconds of `close_delay` for better resource management.
 
+### Select language model
+
+You can specify which language model to use by passing `model` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. The default value is `unspecified`.
+
+Currently available models (as of Feb 5, 2025):
+
+- `unspecified` - Default model (same as `gemini-2.0-flash` if account does NOT have Gemini Advanced subscription)
+- `gemini-2.0-flash` - Gemini 2.0 Flash
+- `gemini-2.0-flash-thinking` - Gemini 2.0 Flash Thinking Experimental
+- `gemini-2.0-flash-thinking-with-apps` - Gemini 2.0 Flash Thinking Experimental with apps
+- `gemini-1.5-flash` - Gemini 1.5 Flash
+
+Models pending update (may not work as expected):
+
+- `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced **(requires Gemini Advanced account)**
+- `gemini-1.5-pro` - Gemini 1.5 Pro **(requires Gemini Advanced account)**
+- `gemini-1.5-pro-research` - Gemini 1.5 Pro with Deep Research **(requires Gemini Advanced account)**
+
+```python
+from gemini_webapi.constants import Model
+
+async def main():
+    response1 = await client.generate_content(
+        "What's you language model version? Reply version number only.",
+        model=Model.G_2_0_FLASH,
+    )
+    print(f"Model version ({Model.G_2_0_FLASH.model_name}): {response1.text}")
+
+    chat = client.start_chat(model="gemini-2.0-flash-thinking")
+    response2 = await chat.send_message("What's you language model version? Reply version number only.")
+    print(f"Model version (gemini-2.0-flash-thinking): {response2.text}")
+
+asyncio.run(main())
+```
+
 ### Generate contents from text
 
 Ask a one-turn quick question by calling `GeminiClient.generate_content`.
@@ -885,6 +922,21 @@ async def main():
 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`.
+
+```python
+async def main():
+    response = await client.generate_content(
+            "What's 1+1?", model="gemini-2.0-flash-thinking"
+        )
+    print(response.thoughts)
+    print(response.text)
+
+asyncio.run(main())
+```
+
 ### Retrieve images in response
 
 Images in the API's output are stored as a list of `Image` objects. You can access the image title, URL, and description by calling `image.title`, `image.url` and `image.alt` respectively.
@@ -938,36 +990,6 @@ async def main():
 asyncio.run(main())
 ```
 
-### Specify language model version
-
-You can choose a specified language model version by passing `model` argument to `GeminiClient.generate_content` or `GeminiClient.start_chat`. The default value is `unspecified`.
-
-Currently available models (as of Feb 1, 2025):
-
-- `unspecified` - Default model (Gemini 2.0 Flash if account does NOT have Gemini Advanced subscription)
-- `gemini-1.5-flash` - Gemini 1.5 Flash
-- `gemini-1.5-pro` - Gemini 1.5 Pro **(requires Gemini Advanced account)**
-- `gemini-1.5-pro-research` - Gemini 1.5 Pro with Deep Research **(requires Gemini Advanced account)**
-- `gemini-2.0-flash-exp` - Gemini 2.0 Flash Experimental
-- `gemini-2.0-exp-advanced` - Gemini 2.0 Experimental Advanced **(requires Gemini Advanced account)**
-
-```python
-from gemini_webapi.constants import Model
-
-async def main():
-    response1 = await client.generate_content(
-        "What's you language model version? Reply version number only.",
-        model="gemini-1.5-flash",
-    )
-    print(f"Model version (gemini-1.5-flash): {response1.text}")
-
-    chat = client.start_chat(model=Model.G_2_0_FLASH_EXP)
-    response2 = await chat.send_message("What's you language model version? Reply version number only.")
-    print(f"Model version ({Model.G_2_0_FLASH_EXP.model_name}): {response2.text}")
-
-asyncio.run(main())
-```
-
 ### Generate contents with Gemini extensions
 
 > [!IMPORTANT]

{gemini_webapi-1.8.4.dist-info → gemini_webapi-1.9.1.dist-info}/RECORD

@@ -1,19 +1,19 @@
 gemini_webapi/__init__.py,sha256=28uNIywK4vCXxENaSagNWUzhqr1RyNtLzDF6WRRM4KQ,194
-gemini_webapi/client.py,sha256=PPr8hZljMKOxszKFJihvqUZaMV0eSNJ-X2d4Hvz7jFU,22759
-gemini_webapi/constants.py,sha256=_glqlB6Wy_VDi7h6ruorGGdtoyKfVg6shYfA_BAT9Q4,2178
+gemini_webapi/client.py,sha256=4ptFBYy-ipcxox1x7WDOK04YMYU_cu6NwS8CSrmz3kM,23105
+gemini_webapi/constants.py,sha256=_8TdJH3dvzSDo0qfhKCAerWXBbH71u0B0mSGOkkJQ5w,2737
 gemini_webapi/exceptions.py,sha256=6e-EXHGApi4iC0GDw7RKc3YqVK8UvEkHYaJyGQbReLw,548
 gemini_webapi/types/__init__.py,sha256=d2kvXnE004s2E2KDmPPLi5N-BQ59FgDSlrGrO3Wphww,163
-gemini_webapi/types/candidate.py,sha256=Z9bpIK4l8UWbUVMLEoophfhgROo93dxOM9cAwx77CkU,1030
+gemini_webapi/types/candidate.py,sha256=dMoGr53WR7FYDRrXpG9Yd_n9YTHGwGaFHZ8zPt8RMWw,1178
 gemini_webapi/types/image.py,sha256=4BC8hxAWJrYFwzA60CivF1di4RZkzPKjcaSPPFKmRdY,5237
-gemini_webapi/types/modeloutput.py,sha256=sGEnaQtSOJE68ve5R5sFgW4POmsyWZAV5zei3Z4BRVk,1090
+gemini_webapi/types/modeloutput.py,sha256=h07kQOkL5r-oPLvZ59uVtO1eP4FGy5ZpzuYQzAeQdr8,1196
 gemini_webapi/utils/__init__.py,sha256=mcm1kgQ5HHKyZrhHS-rd_GXbKMpIDUsq02XtmlQNN_I,357
 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/rotate_1psidts.py,sha256=NyQ9OYPLBOcvpc8bodvEYDIVFrsYN0kdfc831lPEctM,1680
 gemini_webapi/utils/upload_file.py,sha256=A-ZQ54gy2wOVA_giEEVLQQK5zTJfrgq99Z5YYYOAFYc,1163
-gemini_webapi-1.8.4.dist-info/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-gemini_webapi-1.8.4.dist-info/METADATA,sha256=zKj_zbFMsF1Gkm2VEsIkXX6Zab_x6dBNpPwjsz1efZQ,57135
-gemini_webapi-1.8.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-gemini_webapi-1.8.4.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
-gemini_webapi-1.8.4.dist-info/RECORD,,
+gemini_webapi-1.9.1.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+gemini_webapi-1.9.1.dist-info/METADATA,sha256=LXrhPGqkC1MfBvX868TaWV-xKuVpQmsDbi90KSrKHzg,57797
+gemini_webapi-1.9.1.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
+gemini_webapi-1.9.1.dist-info/top_level.txt,sha256=dtWtug_ZrmnUqCYuu8NmGzTgWglHeNzhHU_hXmqZGWE,14
+gemini_webapi-1.9.1.dist-info/RECORD,,

{gemini_webapi-1.8.4.dist-info → gemini_webapi-1.9.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (77.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any