llm-gemini 0.14__py3-none-any.whl → 0.15__py3-none-any.whl

{llm_gemini-0.14.dist-info → llm_gemini-0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: llm-gemini
-Version: 0.14
+Version: 0.15
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -72,6 +72,7 @@ Other models are:
 - `gemini-2.0-flash` - Gemini 2.0 Flash
 - `gemini-2.0-flash-lite` - Gemini 2.0 Flash-Lite
 - `gemini-2.0-pro-exp-02-05` - experimental release of Gemini 2.0 Pro
+- `gemma-3-27b-it` - [Gemma 3](https://blog.google/technology/developers/gemma-3/) 27B
 
 ### Images, audio and video
 

llm_gemini-0.15.dist-info/RECORD

@@ -0,0 +1,7 @@
+llm_gemini.py,sha256=DdgyLfTuDS5RNt2iaFX9UZnDGUHezObGj-TkyWLJirw,15938
+llm_gemini-0.15.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+llm_gemini-0.15.dist-info/METADATA,sha256=8qhtAEFniW5fn-6Qz1zFFrjaZwqqD_kNXJNheU8SG_M,7643
+llm_gemini-0.15.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+llm_gemini-0.15.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
+llm_gemini-0.15.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
+llm_gemini-0.15.dist-info/RECORD,,

{llm_gemini-0.14.dist-info → llm_gemini-0.15.dist-info}/WHEEL

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

llm_gemini.py

@@ -1,6 +1,8 @@
+import click
 import copy
 import httpx
 import ijson
+import json
 import llm
 from pydantic import Field
 from typing import Optional
@@ -62,6 +64,8 @@ def register_models(register):
         "gemini-2.0-pro-exp-02-05",
         # Released 25th Feb 2025:
         "gemini-2.0-flash-lite",
+        # Released 12th March 2025:
+        "gemma-3-27b-it",
     ]:
         can_google_search = model_id in GOOGLE_SEARCH_MODELS
         register(
@@ -88,18 +92,24 @@ def resolve_type(attachment):
     return mime_type
 
 
-def cleanup_schema(schema):
+def cleanup_schema(schema, in_properties=False):
     "Gemini supports only a subset of JSON schema"
     keys_to_remove = ("$schema", "additionalProperties", "title")
-    # Recursively remove them
+
     if isinstance(schema, dict):
-        for key in keys_to_remove:
-            schema.pop(key, None)
-        for value in schema.values():
-            cleanup_schema(value)
+        # Only remove keys if we're not inside a 'properties' block.
+        if not in_properties:
+            for key in keys_to_remove:
+                schema.pop(key, None)
+        for key, value in list(schema.items()):
+            # If the key is 'properties', set the flag for its value.
+            if key == "properties" and isinstance(value, dict):
+                cleanup_schema(value, in_properties=True)
+            else:
+                cleanup_schema(value, in_properties=in_properties)
     elif isinstance(schema, list):
-        for value in schema:
-            cleanup_schema(value)
+        for item in schema:
+            cleanup_schema(item, in_properties=in_properties)
     return schema
 
 
@@ -431,3 +441,21 @@ class GeminiEmbeddingModel(llm.EmbeddingModel):
         if self.truncate:
             values = [value[: self.truncate] for value in values]
         return values
+
+
+@llm.hookimpl
+def register_commands(cli):
+    @cli.group()
+    def gemini():
+        "Commands relating to the llm-gemini plugin"
+
+    @gemini.command()
+    @click.option("--key", help="API key to use")
+    def models(key):
+        "List of Gemini models pulled from their API"
+        key = llm.get_key(key, "gemini", "LLM_GEMINI_KEY")
+        response = httpx.get(
+            f"https://generativelanguage.googleapis.com/v1beta/models?key={key}",
+        )
+        response.raise_for_status()
+        click.echo(json.dumps(response.json(), indent=2))

llm_gemini-0.14.dist-info/RECORD

@@ -1,7 +0,0 @@
-llm_gemini.py,sha256=KgiKMPY6q-GkAMzH0gfQWp0cjBBKizNhlnZ3nX5pXWY,14917
-llm_gemini-0.14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-llm_gemini-0.14.dist-info/METADATA,sha256=J8oA7hZmNrFsRyH3_s4oesSjWNEaK8pWcT2p-_quPTA,7556
-llm_gemini-0.14.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-llm_gemini-0.14.dist-info/entry_points.txt,sha256=n544bpgUPIBc5l_cnwsTxPc3gMGJHPtAyqBNp-CkMWk,26
-llm_gemini-0.14.dist-info/top_level.txt,sha256=WUQmG6_2QKbT_8W4HH93qyKl_0SUteL4Ra6_PhyNGKU,11
-llm_gemini-0.14.dist-info/RECORD,,