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

Add more expiermental models from Gemini ()

Browse source
This commit is contained in:
Timothy Carambat 2024-11-20 09:52:33 -08:00 • committed by GitHub
parent 907a03bb67
commit 26e2d8cc3b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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

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

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

@ -20,6 +20,8 @@ const PROVIDER_DEFAULT_MODELS = {
"gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
],
anthropic: [
"claude-instant-1.2",

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

@ -26,6 +26,8 @@ class GeminiLLM {
"gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
].includes(this.model)
? "v1beta"
: "v1",
@ -111,6 +113,8 @@ class GeminiLLM {
"gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
];
return validModels.includes(modelName);
}

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

@ -34,6 +34,8 @@ const MODEL_MAP = {
"gemini-1.5-pro-exp-0827": 2_097_152,
"gemini-1.5-flash-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: {
"gemma2-9b-it": 8192,

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

@ -701,6 +701,8 @@ function validGeminiModel(input = "") {
"gemini-1.5-pro-exp-0827",
"gemini-1.5-flash-exp-0827",
"gemini-1.5-flash-8b-exp-0827",
"gemini-exp-1114",
"learnlm-1.5-pro-experimental",
];
return validModels.includes(input)
? null