llm-gemini 0.1a0__tar.gz → 0.1a1__tar.gz

{llm-gemini-0.1a0 → llm-gemini-0.1a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-gemini
-Version: 0.1a0
+Version: 0.1a1
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -34,7 +34,7 @@ llm install llm-gemini
 ```
 ## Usage
 
-Configure the model by setting a key called "gemini" to your API key:
+Configure the model by setting a key called "gemini" to your [API key](https://ai.google.dev/):
 
 ```bash
 llm keys set gemini
@@ -59,6 +59,8 @@ To chat interactively with the model, run `llm chat`:
 llm chat -m gemini-pro
 ```
 
+If you have access to the Gemini 1.5 Pro preview you can use `-m gemini-1.5-pro-latest` to work with that model.
+
 ## Development
 
 To set up this plugin locally, first checkout the code. Then create a new virtual environment:

{llm-gemini-0.1a0 → llm-gemini-0.1a1}/README.md

@@ -15,7 +15,7 @@ llm install llm-gemini
 ```
 ## Usage
 
-Configure the model by setting a key called "gemini" to your API key:
+Configure the model by setting a key called "gemini" to your [API key](https://ai.google.dev/):
 
 ```bash
 llm keys set gemini
@@ -40,6 +40,8 @@ To chat interactively with the model, run `llm chat`:
 llm chat -m gemini-pro
 ```
 
+If you have access to the Gemini 1.5 Pro preview you can use `-m gemini-1.5-pro-latest` to work with that model.
+
 ## Development
 
 To set up this plugin locally, first checkout the code. Then create a new virtual environment:

{llm-gemini-0.1a0 → llm-gemini-0.1a1}/llm_gemini.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-gemini
-Version: 0.1a0
+Version: 0.1a1
 Summary: LLM plugin to access Google's Gemini family of models
 Author: Simon Willison
 License: Apache-2.0
@@ -34,7 +34,7 @@ llm install llm-gemini
 ```
 ## Usage
 
-Configure the model by setting a key called "gemini" to your API key:
+Configure the model by setting a key called "gemini" to your [API key](https://ai.google.dev/):
 
 ```bash
 llm keys set gemini
@@ -59,6 +59,8 @@ To chat interactively with the model, run `llm chat`:
 llm chat -m gemini-pro
 ```
 
+If you have access to the Gemini 1.5 Pro preview you can use `-m gemini-1.5-pro-latest` to work with that model.
+
 ## Development
 
 To set up this plugin locally, first checkout the code. Then create a new virtual environment:

{llm-gemini-0.1a0 → llm-gemini-0.1a1}/llm_gemini.py

@@ -26,28 +26,34 @@ SAFETY_SETTINGS = [
 
 @llm.hookimpl
 def register_models(register):
-    register(GeminiPro())
+    register(GeminiPro("gemini-pro"))
+    register(GeminiPro("gemini-1.5-pro-latest"))
 
 
 class GeminiPro(llm.Model):
-    model_id = "gemini-pro"
     can_stream = True
 
+    def __init__(self, model_id):
+        self.model_id = model_id
+
     def build_messages(self, prompt, conversation):
         if not conversation:
             return [{"role": "user", "parts": [{"text": prompt.prompt}]}]
         messages = []
         for response in conversation.responses:
-            messages.append({"role": "user", "parts": [{"text": response.prompt.prompt}]})
+            messages.append(
+                {"role": "user", "parts": [{"text": response.prompt.prompt}]}
+            )
             messages.append({"role": "model", "parts": [{"text": response.text()}]})
         messages.append({"role": "user", "parts": [{"text": prompt.prompt}]})
         return messages
 
     def execute(self, prompt, stream, response, conversation):
         key = llm.get_key("", "gemini", "LLM_GEMINI_KEY")
-        url = (
-            "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:streamGenerateContent?"
-            + urllib.parse.urlencode({"key": key})
+        url = "https://generativelanguage.googleapis.com/v1beta/models/{}:streamGenerateContent?".format(
+            self.model_id
+        ) + urllib.parse.urlencode(
+            {"key": key}
         )
         gathered = []
         with httpx.stream(

{llm-gemini-0.1a0 → llm-gemini-0.1a1}/pyproject.toml

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