Skip to content
Snippets Groups Projects
Unverified Commit 26e2d8cc authored by Timothy Carambat's avatar Timothy Carambat Committed by GitHub
Browse files

Add more expiermental models from Gemini (#2663)

parent 907a03bb
No related branches found
No related tags found
No related merge requests found
...@@ -50,6 +50,8 @@ export default function GeminiLLMOptions({ settings }) { ...@@ -50,6 +50,8 @@ export default function GeminiLLMOptions({ settings }) {
"gemini-1.5-pro-exp-0827", "gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827", "gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
].map((model) => { ].map((model) => {
return ( return (
<option key={model} value={model}> <option key={model} value={model}>
......
...@@ -20,6 +20,8 @@ const PROVIDER_DEFAULT_MODELS = { ...@@ -20,6 +20,8 @@ const PROVIDER_DEFAULT_MODELS = {
"gemini-1.5-pro-exp-0827", "gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827", "gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
], ],
anthropic: [ anthropic: [
"claude-instant-1.2", "claude-instant-1.2",
......
...@@ -26,6 +26,8 @@ class GeminiLLM { ...@@ -26,6 +26,8 @@ class GeminiLLM {
"gemini-1.5-pro-exp-0827", "gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827", "gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
].includes(this.model) ].includes(this.model)
? "v1beta" ? "v1beta"
: "v1", : "v1",
...@@ -111,6 +113,8 @@ class GeminiLLM { ...@@ -111,6 +113,8 @@ class GeminiLLM {
"gemini-1.5-pro-exp-0827", "gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827", "gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
]; ];
return validModels.includes(modelName); return validModels.includes(modelName);
} }
......
...@@ -34,6 +34,8 @@ const MODEL_MAP = { ...@@ -34,6 +34,8 @@ const MODEL_MAP = {
"gemini-1.5-pro-exp-0827": 2_097_152, "gemini-1.5-pro-exp-0827": 2_097_152,
"gemini-1.5-flash-exp-0827": 1_048_576, "gemini-1.5-flash-exp-0827": 1_048_576,
"gemini-1.5-flash-8b-exp-0827": 1_048_576, "gemini-1.5-flash-8b-exp-0827": 1_048_576,
"gemini-exp-1114": 32_767,
"learnlm-1.5-pro-experimental": 32_767,
}, },
groq: { groq: {
"gemma2-9b-it": 8192, "gemma2-9b-it": 8192,
......
...@@ -701,6 +701,8 @@ function validGeminiModel(input = "") { ...@@ -701,6 +701,8 @@ function validGeminiModel(input = "") {
"gemini-1.5-pro-exp-0827", "gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827", "gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
]; ];
return validModels.includes(input) return validModels.includes(input)
? null ? null
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment