llm-gemini 0.7__tar.gz → 0.8__tar.gz

{llm_gemini-0.7 → llm_gemini-0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-gemini
-Version: 0.7
+Version: 0.8
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0

{llm_gemini-0.7 → llm_gemini-0.8}/llm_gemini.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-gemini
-Version: 0.7
+Version: 0.8
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0

{llm_gemini-0.7 → llm_gemini-0.8}/llm_gemini.py

@@ -41,6 +41,7 @@ def register_models(register):
         "gemini-exp-1121",
         "gemini-exp-1206",
         "gemini-2.0-flash-exp",
+        "gemini-2.0-flash-thinking-exp-1219",
     ]:
         register(GeminiPro(model_id), AsyncGeminiPro(model_id))
 
@@ -154,7 +155,7 @@ class _SharedGemini:
                 if response.prompt.prompt:
                     parts.append({"text": response.prompt.prompt})
                 messages.append({"role": "user", "parts": parts})
-                messages.append({"role": "model", "parts": [{"text": response.text()}]})
+                messages.append({"role": "model", "parts": [{"text": response.text_or_raise()}]})
 
         parts = []
         if prompt.prompt:
@@ -334,4 +335,4 @@ class GeminiEmbeddingModel(llm.EmbeddingModel):
             )
 
         response.raise_for_status()
-        return [item["values"] for item in response.json()["embeddings"]]
+        return [item["values"] for item in response.json()["embeddings"]]

{llm_gemini-0.7 → llm_gemini-0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llm-gemini"
-version = "0.7"
+version = "0.8"
 description = "LLM plugin to access Google's Gemini family of models"
 readme = "README.md"
 authors = [{name = "Simon Willison"}]