llm-gemini 0.10__py3-none-any.whl → 0.12__py3-none-any.whl

{llm_gemini-0.10.dist-info → llm_gemini-0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: llm-gemini
-Version: 0.10
+Version: 0.12
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -11,12 +11,13 @@ Project-URL: CI, https://github.com/simonw/llm-gemini/actions
 Classifier: License :: OSI Approved :: Apache Software License
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: llm>=0.19
+Requires-Dist: llm>=0.22
 Requires-Dist: httpx
 Requires-Dist: ijson
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-recording; extra == "test"
+Requires-Dist: nest-asyncio; extra == "test"
 
 # llm-gemini
 
@@ -68,7 +69,7 @@ Other models are:
 - `gemini-2.0-flash-thinking-exp-1219` - experimental "thinking" model from December 2024
 - `gemini-2.0-flash-thinking-exp-01-21` - experimental "thinking" model from January 2025
 - `gemini-2.0-flash` - Gemini 2.0 Flash
-- `gemini-2.0-flash-lite-preview-02-05` - Gemini 2.0 Flash-Lite
+- `gemini-2.0-flash-lite` - Gemini 2.0 Flash-Lite
 - `gemini-2.0-pro-exp-02-05` - experimental release of Gemini 2.0 Pro
 
 ### Images, audio and video

llm_gemini-0.12.dist-info/RECORD

@@ -0,0 +1,7 @@
+llm_gemini.py,sha256=er-6W-CJEOZdC0O6iBYkWsAstlYhtYASkQUXfqhE7Wk,12916
+llm_gemini-0.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+llm_gemini-0.12.dist-info/METADATA,sha256=CKh9OfBqO_Ep7NU2eyGMnkDyuqwnOaBA-geYJpCugac,7014
+llm_gemini-0.12.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
+llm_gemini-0.12.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
+llm_gemini-0.12.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
+llm_gemini-0.12.dist-info/RECORD,,

{llm_gemini-0.10.dist-info → llm_gemini-0.12.dist-info}/WHEEL

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

llm_gemini.py

@@ -32,6 +32,7 @@ GOOGLE_SEARCH_MODELS = {
     "gemini-1.5-pro-002",
     "gemini-1.5-flash-002",
     "gemini-2.0-flash-exp",
+    "gemini-2.0-flash",
 }
 
 
@@ -57,8 +58,9 @@ def register_models(register):
         "gemini-2.0-flash-thinking-exp-01-21",
         # Released 5th Feb 2025:
         "gemini-2.0-flash",
-        "gemini-2.0-flash-lite-preview-02-05",
         "gemini-2.0-pro-exp-02-05",
+        # Released 25th Feb 2025:
+        "gemini-2.0-flash-lite",
     ]:
         can_google_search = model_id in GOOGLE_SEARCH_MODELS
         register(
@@ -220,7 +222,7 @@ class _SharedGemini:
         if prompt.options and prompt.options.code_execution:
             body["tools"] = [{"codeExecution": {}}]
         if prompt.options and self.can_google_search and prompt.options.google_search:
-            body["tools"] = [{"google_search_retrieval": {}}]
+            body["tools"] = [{"google_search": {}}]
         if prompt.system:
             body["systemInstruction"] = {"parts": [{"text": prompt.system}]}
 
@@ -268,9 +270,8 @@ class _SharedGemini:
             pass
 
 
-class GeminiPro(_SharedGemini, llm.Model):
-    def execute(self, prompt, stream, response, conversation):
-        key = self.get_key()
+class GeminiPro(_SharedGemini, llm.KeyModel):
+    def execute(self, prompt, stream, response, conversation, key):
         url = f"https://generativelanguage.googleapis.com/v1beta/models/{self.model_id}:streamGenerateContent"
         gathered = []
         body = self.build_request_body(prompt, conversation)
@@ -279,7 +280,7 @@ class GeminiPro(_SharedGemini, llm.Model):
             "POST",
             url,
             timeout=None,
-            headers={"x-goog-api-key": key},
+            headers={"x-goog-api-key": self.get_key(key)},
             json=body,
         ) as http_response:
             events = ijson.sendable_list()
@@ -301,9 +302,8 @@ class GeminiPro(_SharedGemini, llm.Model):
         self.set_usage(response)
 
 
-class AsyncGeminiPro(_SharedGemini, llm.AsyncModel):
-    async def execute(self, prompt, stream, response, conversation):
-        key = self.get_key()
+class AsyncGeminiPro(_SharedGemini, llm.AsyncKeyModel):
+    async def execute(self, prompt, stream, response, conversation, key):
         url = f"https://generativelanguage.googleapis.com/v1beta/models/{self.model_id}:streamGenerateContent"
         gathered = []
         body = self.build_request_body(prompt, conversation)
@@ -313,7 +313,7 @@ class AsyncGeminiPro(_SharedGemini, llm.AsyncModel):
                 "POST",
                 url,
                 timeout=None,
-                headers={"x-goog-api-key": key},
+                headers={"x-goog-api-key": self.get_key(key)},
                 json=body,
             ) as http_response:
                 events = ijson.sendable_list()

llm_gemini-0.10.dist-info/RECORD

@@ -1,7 +0,0 @@
-llm_gemini.py,sha256=JC7QoAvcH-JDCZDuFRtivqJ7aCXOLvWxST6zv6tG79E,12896
-llm_gemini-0.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-llm_gemini-0.10.dist-info/METADATA,sha256=pQ3PVDQtW23dMmfUe24V0NZ7cPS9oDgmS21Ug9ncfuo,6983
-llm_gemini-0.10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-llm_gemini-0.10.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
-llm_gemini-0.10.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
-llm_gemini-0.10.dist-info/RECORD,,