Skip to content

Commit 088fe48

Browse files
authored
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.
1 parent ec43251 commit 088fe48

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

llm_gemini.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def register_models(register):
4141
"gemini-exp-1121",
4242
"gemini-exp-1206",
4343
"gemini-2.0-flash-exp",
44+
"learnlm-1.5-pro-experimental",
4445
"gemini-2.0-flash-thinking-exp-1219",
4546
"gemini-2.0-flash-thinking-exp-01-21",
4647
]:
@@ -336,4 +337,4 @@ def embed_batch(self, items):
336337
)
337338

338339
response.raise_for_status()
339-
return [item["values"] for item in response.json()["embeddings"]]
340+
return [item["values"] for item in response.json()["embeddings"]]

0 commit comments

Comments
 (0)