llm-gemini 0.18.1__py3-none-any.whl → 0.19.1__py3-none-any.whl

{llm_gemini-0.18.1.dist-info → llm_gemini-0.19.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llm-gemini
-Version: 0.18.1
+Version: 0.19.1
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -66,6 +66,7 @@ llm "A joke about a pelican and a walrus"
 
 Other models are:
 
+- `gemini-2.5-pro-preview-05-06` - latest paid Gemini 2.5 Pro preview
 - `gemini-2.5-flash-preview-04-17` - Gemini 2.5 Flash preview
 - `gemini-2.5-pro-exp-03-25` - free experimental release of Gemini 2.5 Pro
 - `gemini-2.5-pro-preview-03-25` - paid preview of Gemini 2.5 Pro

llm_gemini-0.19.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+llm_gemini.py,sha256=GJp1oDwcdLXy2QAYwp7jGN0KZsbmj_pJcqktYRC139Q,18700
+llm_gemini-0.19.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+llm_gemini-0.19.1.dist-info/METADATA,sha256=PsXdnxt56uLdmbYnnCF9PZgQUi2EBx9uZEgEFW0l3IQ,8119
+llm_gemini-0.19.1.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+llm_gemini-0.19.1.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
+llm_gemini-0.19.1.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
+llm_gemini-0.19.1.dist-info/RECORD,,

{llm_gemini-0.18.1.dist-info → llm_gemini-0.19.1.dist-info}/WHEEL

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

llm_gemini.py

@@ -38,6 +38,8 @@ GOOGLE_SEARCH_MODELS = {
     "gemini-2.0-flash",
     "gemini-2.5-pro-preview-03-25",
     "gemini-2.5-pro-exp-03-25",
+    "gemini-2.5-flash-preview-04-17",
+    "gemini-2.5-pro-preview-05-06",
 }
 
 # Older Google models used google_search_retrieval instead of google_search
@@ -89,6 +91,8 @@ def register_models(register):
         "gemini-2.5-pro-preview-03-25",
         # 17th April 2025:
         "gemini-2.5-flash-preview-04-17",
+        # 6th May 2025:
+        "gemini-2.5-pro-preview-05-06",
     ]:
         can_google_search = model_id in GOOGLE_SEARCH_MODELS
         can_thinking_budget = model_id in THINKING_BUDGET_MODELS
@@ -365,7 +369,10 @@ class _SharedGemini:
                 candidate.pop("content", None)
             usage = response.response_json.pop("usageMetadata")
             input_tokens = usage.pop("promptTokenCount", None)
-            output_tokens = usage.pop("candidatesTokenCount", None)
+            # See https://github.com/simonw/llm-gemini/issues/75#issuecomment-2861827509
+            candidates_token_count = usage.get("candidatesTokenCount") or 0
+            thoughts_token_count = usage.get("thoughtsTokenCount") or 0
+            output_tokens = candidates_token_count + thoughts_token_count
             usage.pop("totalTokenCount", None)
             if input_tokens is not None:
                 response.set_usage(

llm_gemini-0.18.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-llm_gemini.py,sha256=mPnnJ5gWU1NS8cxmq2JDiKGq7tXaP1oz7lsMtRv3Ae4,18319
-llm_gemini-0.18.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-llm_gemini-0.18.1.dist-info/METADATA,sha256=Y6sJT4JCMXtnk6zP6EaQ6mEEZYL7vwfwOg04IdTI6UU,8049
-llm_gemini-0.18.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-llm_gemini-0.18.1.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
-llm_gemini-0.18.1.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
-llm_gemini-0.18.1.dist-info/RECORD,,