llm-gemini 0.8__py3-none-any.whl → 0.9__py3-none-any.whl

{llm_gemini-0.8.dist-info → llm_gemini-0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: llm-gemini
-Version: 0.8
+Version: 0.9
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -64,6 +64,9 @@ Other models are:
 - `gemini-exp-1121` - recent experimental #2
 - `gemini-exp-1206` - recent experimental #3
 - `gemini-2.0-flash-exp` - [Gemini 2.0 Flash](https://blog.google/technology/google-deepmind/google-gemini-ai-update-december-2024/#gemini-2-0-flash)
+- `learnlm-1.5-pro-experimental` - "an experimental task-specific model that has been trained to align with learning science principles" - [more details here](https://ai.google.dev/gemini-api/docs/learnlm).
+- `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
 
 ### Images, audio and video
 
@@ -113,6 +116,20 @@ To enable this feature, use `-o code_execution 1`:
 llm -m gemini-1.5-pro-latest -o code_execution 1 \
 'use python to calculate (factorial of 13) * 3'
 ```
+### Google search
+
+Some Gemini models support [Grounding with Google Search](https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/ground-gemini#web-ground-gemini), where the model can run a Google search and use the results as part of answering a prompt.
+
+Using this feature may incur additional requirements in terms of how you use the results. Consult [Google's documentation](https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/ground-gemini#web-ground-gemini) for more details.
+
+To run a prompt with Google search enabled, use `-o google_search 1`:
+
+```bash
+llm -m gemini-1.5-pro-latest -o google_search 1 \
+  'What happened in Ireland today?'
+```
+
+Use `llm logs -c --json` after running a prompt to see the full JSON response, which includes [additional information](https://github.com/simonw/llm-gemini/pull/29#issuecomment-2606201877) about grounded results.
 
 ### Chat
 

llm_gemini-0.9.dist-info/RECORD

@@ -0,0 +1,7 @@
+llm_gemini.py,sha256=sCouoSbzOe4GoTsskAKJZjhDTxRYqSxgNuODwi2O1z0,12752
+llm_gemini-0.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+llm_gemini-0.9.dist-info/METADATA,sha256=UEr_dRMMSev9YY9U34QMQHnhuyRPM7E7sHT1o8uA0qg,6808
+llm_gemini-0.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+llm_gemini-0.9.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
+llm_gemini-0.9.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
+llm_gemini-0.9.dist-info/RECORD,,

{llm_gemini-0.8.dist-info → llm_gemini-0.9.dist-info}/WHEEL

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

llm_gemini.py

@@ -23,6 +23,17 @@ SAFETY_SETTINGS = [
     },
 ]
 
+# https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/ground-gemini#supported_models_2
+GOOGLE_SEARCH_MODELS = {
+    "gemini-1.5-pro-latest",
+    "gemini-1.5-flash-latest",
+    "gemini-1.5-pro-001",
+    "gemini-1.5-flash-001",
+    "gemini-1.5-pro-002",
+    "gemini-1.5-flash-002",
+    "gemini-2.0-flash-exp",
+}
+
 
 @llm.hookimpl
 def register_models(register):
@@ -41,9 +52,15 @@ def register_models(register):
         "gemini-exp-1121",
         "gemini-exp-1206",
         "gemini-2.0-flash-exp",
+        "learnlm-1.5-pro-experimental",
         "gemini-2.0-flash-thinking-exp-1219",
+        "gemini-2.0-flash-thinking-exp-01-21",
     ]:
-        register(GeminiPro(model_id), AsyncGeminiPro(model_id))
+        can_google_search = model_id in GOOGLE_SEARCH_MODELS
+        register(
+            GeminiPro(model_id, can_google_search=can_google_search),
+            AsyncGeminiPro(model_id, can_google_search=can_google_search),
+        )
 
 
 def resolve_type(attachment):
@@ -51,6 +68,8 @@ def resolve_type(attachment):
     # https://github.com/simonw/llm/issues/587#issuecomment-2439785140
     if mime_type == "audio/mpeg":
         mime_type = "audio/mp3"
+    if mime_type == "application/ogg":
+        mime_type = "audio/ogg"
     return mime_type
 
 
@@ -60,6 +79,9 @@ class _SharedGemini:
     can_stream = True
 
     attachment_types = (
+        # Text
+        "text/plain",
+        "text/csv",
         # PDF
         "application/pdf",
         # Images
@@ -74,6 +96,7 @@ class _SharedGemini:
         "audio/aiff",
         "audio/aac",
         "audio/ogg",
+        "application/ogg",
         "audio/flac",
         "audio/mpeg",  # Treated as audio/mp3
         # Video
@@ -134,8 +157,17 @@ class _SharedGemini:
             default=None,
         )
 
-    def __init__(self, model_id):
+    class OptionsWithGoogleSearch(Options):
+        google_search: Optional[bool] = Field(
+            description="Enables the model to use Google Search to improve the accuracy and recency of responses from the model",
+            default=None,
+        )
+
+    def __init__(self, model_id, can_google_search=False):
         self.model_id = model_id
+        self.can_google_search = can_google_search
+        if can_google_search:
+            self.Options = self.OptionsWithGoogleSearch
 
     def build_messages(self, prompt, conversation):
         messages = []
@@ -155,7 +187,9 @@ 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_or_raise()}]})
+                messages.append(
+                    {"role": "model", "parts": [{"text": response.text_or_raise()}]}
+                )
 
         parts = []
         if prompt.prompt:
@@ -181,6 +215,8 @@ 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": {}}]
         if prompt.system:
             body["systemInstruction"] = {"parts": [{"text": prompt.system}]}
 
@@ -335,4 +371,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.8.dist-info/RECORD

@@ -1,7 +0,0 @@
-llm_gemini.py,sha256=6xRF1uP64O-nYAGgKytFh1Wj0N-cRny4bs69GmTwJLU,11408
-llm_gemini-0.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-llm_gemini-0.8.dist-info/METADATA,sha256=cCkBf00ebzzwl2wyUxOTuwTWfG_peiB7mtD-8bzHBkc,5530
-llm_gemini-0.8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-llm_gemini-0.8.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
-llm_gemini-0.8.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
-llm_gemini-0.8.dist-info/RECORD,,