From 088fe4826332a9e2f1b5dd382d5e0ccb23b88459 Mon Sep 17 00:00:00 2001 From: Leonard Tulipan Date: Wed, 22 Jan 2025 03:39:46 +0100 Subject: [PATCH] Add model "learnlm-1.5-pro-experimental" (#36) See https://ai.google.dev/gemini-api/docs/models/experimental-models for all available experimental models. learnlm was missing from the list in the plugin. https://ai.google.dev/gemini-api/docs/learnlm for more details. --- llm_gemini.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llm_gemini.py b/llm_gemini.py index a38a4d5..ede38bb 100644 --- a/llm_gemini.py +++ b/llm_gemini.py @@ -41,6 +41,7 @@ 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", ]: @@ -336,4 +337,4 @@ def embed_batch(self, items): ) response.raise_for_status() - return [item["values"] for item in response.json()["embeddings"]] \ No newline at end of file + return [item["values"] for item in response.json()["embeddings"]]