mirror of
https://github.com/Mintplex-Labs/anything-llm.git
synced 2025-04-24 21:48:12 +00:00
[FEAT] Add support for gemini-1.5-flash-latest model (#1502)
* add support for gemini-1.5-flash-latest * update comment in gemini LLM provider
This commit is contained in:
parent
961630f7d5
commit
cc7e7fb3ac
4 changed files with 22 additions and 6 deletions
frontend/src
server/utils
|
@ -30,7 +30,11 @@ export default function GeminiLLMOptions({ settings }) {
|
|||
required={true}
|
||||
className="bg-zinc-900 border-gray-500 text-white text-sm rounded-lg block w-full p-2.5"
|
||||
>
|
||||
{["gemini-pro", "gemini-1.5-pro-latest"].map((model) => {
|
||||
{[
|
||||
"gemini-pro",
|
||||
"gemini-1.5-pro-latest",
|
||||
"gemini-1.5-flash-latest",
|
||||
].map((model) => {
|
||||
return (
|
||||
<option key={model} value={model}>
|
||||
{model}
|
||||
|
|
|
@ -10,7 +10,7 @@ export const DISABLED_PROVIDERS = [
|
|||
];
|
||||
const PROVIDER_DEFAULT_MODELS = {
|
||||
openai: [],
|
||||
gemini: ["gemini-pro", "gemini-1.5-pro-latest"],
|
||||
gemini: ["gemini-pro", "gemini-1.5-pro-latest", "gemini-1.5-flash-latest"],
|
||||
anthropic: [
|
||||
"claude-instant-1.2",
|
||||
"claude-2.0",
|
||||
|
|
|
@ -17,8 +17,12 @@ class GeminiLLM {
|
|||
this.gemini = genAI.getGenerativeModel(
|
||||
{ model: this.model },
|
||||
{
|
||||
// Gemini-1.5-pro is only available on the v1beta API.
|
||||
apiVersion: this.model === "gemini-1.5-pro-latest" ? "v1beta" : "v1",
|
||||
// Gemini-1.5-pro and Gemini-1.5-flash are only available on the v1beta API.
|
||||
apiVersion:
|
||||
this.model === "gemini-1.5-pro-latest" ||
|
||||
this.model === "gemini-1.5-flash-latest"
|
||||
? "v1beta"
|
||||
: "v1",
|
||||
}
|
||||
);
|
||||
this.limits = {
|
||||
|
@ -95,7 +99,11 @@ class GeminiLLM {
|
|||
}
|
||||
|
||||
isValidChatCompletionModel(modelName = "") {
|
||||
const validModels = ["gemini-pro", "gemini-1.5-pro-latest"];
|
||||
const validModels = [
|
||||
"gemini-pro",
|
||||
"gemini-1.5-pro-latest",
|
||||
"gemini-1.5-flash-latest",
|
||||
];
|
||||
return validModels.includes(modelName);
|
||||
}
|
||||
|
||||
|
|
|
@ -526,7 +526,11 @@ function supportedTranscriptionProvider(input = "") {
|
|||
}
|
||||
|
||||
function validGeminiModel(input = "") {
|
||||
const validModels = ["gemini-pro", "gemini-1.5-pro-latest"];
|
||||
const validModels = [
|
||||
"gemini-pro",
|
||||
"gemini-1.5-pro-latest",
|
||||
"gemini-1.5-flash-latest",
|
||||
];
|
||||
return validModels.includes(input)
|
||||
? null
|
||||
: `Invalid Model type. Must be one of ${validModels.join(", ")}.`;
|
||||
|
|
Loading…
Add table
Reference in a new issue