Skip to content
Snippets Groups Projects
Commit 69b672b6 authored by timothycarambat's avatar timothycarambat
Browse files

add gemini 1206 and gemini-2.0-flash exp models

connect #2788
parent a69997a7
No related branches found
No related tags found
No related merge requests found
......@@ -52,7 +52,9 @@ export default function GeminiLLMOptions({ settings }) {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
].map((model) => {
return (
<option key={model} value={model}>
......
......@@ -22,7 +22,9 @@ const PROVIDER_DEFAULT_MODELS = {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
],
anthropic: [
"claude-instant-1.2",
......
......@@ -28,7 +28,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
].includes(this.model)
? "v1beta"
: "v1",
......@@ -116,7 +118,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
];
return validModels.includes(modelName);
}
......
......@@ -36,7 +36,9 @@ const MODEL_MAP = {
"gemini-1.5-flash-8b-exp-0827": 1_048_576,
"gemini-exp-1114": 32_767,
"gemini-exp-1121": 32_767,
"gemini-exp-1206": 32_767,
"learnlm-1.5-pro-experimental": 32_767,
"gemini-2.0-flash-exp-1206": 1_048_576,
},
groq: {
"gemma2-9b-it": 8192,
......
......@@ -731,7 +731,9 @@ function validGeminiModel(input = "") {
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
];
return validModels.includes(input)
? null
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment