sij/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2025-04-23 13:08:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions1

add gemini 1206 and gemini-2.0-flash exp models

Browse source 
connect
This commit is contained in:
timothycarambat 2024-12-11 09:04:29 -08:00
parent a69997a715
commit 69b672b625
5 changed files with 12 additions and 0 deletions
frontend/src
components/LLMSelection/GeminiLLMOptions
index.jsx
hooks
useGetProvidersModels.js
server/utils
AiProviders
gemini
index.js
modelMap.js
helpers
updateENV.js

2
frontend/src/components/LLMSelection/GeminiLLMOptions/index.jsx
View file

@ -52,7 +52,9 @@ export default function GeminiLLMOptions({ settings }) {
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114", "gemini-exp-1114",
"gemini-exp-1121", "gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental", "learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
].map((model) => { ].map((model) => {
return ( return (
<option key={model} value={model}> <option key={model} value={model}>

2
frontend/src/hooks/useGetProvidersModels.js
View file

@ -22,7 +22,9 @@ const PROVIDER_DEFAULT_MODELS = {
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114", "gemini-exp-1114",
"gemini-exp-1121", "gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental", "learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
], ],
anthropic: [ anthropic: [
"claude-instant-1.2", "claude-instant-1.2",

4
server/utils/AiProviders/gemini/index.js
View file

@ -28,7 +28,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114", "gemini-exp-1114",
"gemini-exp-1121", "gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental", "learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
].includes(this.model) ].includes(this.model)
? "v1beta" ? "v1beta"
: "v1", : "v1",
@ -116,7 +118,9 @@ class GeminiLLM {
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114", "gemini-exp-1114",
"gemini-exp-1121", "gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental", "learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
]; ];
return validModels.includes(modelName); return validModels.includes(modelName);
} }

2
server/utils/AiProviders/modelMap.js
View file

@ -36,7 +36,9 @@ const MODEL_MAP = {
"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, "gemini-exp-1114": 32_767,
"gemini-exp-1121": 32_767, "gemini-exp-1121": 32_767,
"gemini-exp-1206": 32_767,
"learnlm-1.5-pro-experimental": 32_767, "learnlm-1.5-pro-experimental": 32_767,
"gemini-2.0-flash-exp-1206": 1_048_576,
}, },
groq: { groq: {
"gemma2-9b-it": 8192, "gemma2-9b-it": 8192,

2
server/utils/helpers/updateENV.js
View file

@ -731,7 +731,9 @@ function validGeminiModel(input = "") {
"gemini-1.5-flash-8b-exp-0827", "gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114", "gemini-exp-1114",
"gemini-exp-1121", "gemini-exp-1121",
"gemini-exp-1206",
"learnlm-1.5-pro-experimental", "learnlm-1.5-pro-experimental",
"gemini-2.0-flash-exp-1206",
]; ];
return validModels.includes(input) return validModels.includes(input)
? null ? null