From 2f1e14d0d7bf3e6c071d84dea20ccdef268f0dd3 Mon Sep 17 00:00:00 2001 From: Kwaroran Date: Thu, 12 Dec 2024 07:53:25 +0900 Subject: [PATCH] Add hasVideoInput flag and new Gemini model with audio and video support --- src/ts/model/modellist.ts | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/src/ts/model/modellist.ts b/src/ts/model/modellist.ts index baa1b626..49711994 100644 --- a/src/ts/model/modellist.ts +++ b/src/ts/model/modellist.ts @@ -13,7 +13,8 @@ export enum LLMFlags{ hasStreaming, requiresAlternateRole, mustStartWithUserInput, - poolSupported + poolSupported, + hasVideoInput } export enum LLMProvider{ @@ -777,10 +778,18 @@ export const LLMModels: LLMModel[] = [ provider: LLMProvider.GoogleCloud, format: LLMFormat.GoogleCloud, flags: [LLMFlags.hasImageInput, LLMFlags.hasFirstSystemPrompt, LLMFlags.poolSupported], - recommended: true, parameters: ['temperature', 'top_k', 'top_p'], tokenizer: LLMTokenizer.GoogleCloud }, + { + name: "gemini-2.0-flash-exp", + id: 'gemini-2.0-flash-exp', + provider: LLMProvider.GoogleCloud, + format: LLMFormat.GoogleCloud, + flags: [LLMFlags.hasImageInput, LLMFlags.hasFirstSystemPrompt, LLMFlags.poolSupported, LLMFlags.hasAudioInput, LLMFlags.hasVideoInput], + parameters: ['temperature', 'top_k', 'top_p', 'presence_penalty', 'frequency_penalty'], + tokenizer: LLMTokenizer.GoogleCloud + }, { name: "Gemini Pro 1.5", id: 'gemini-1.5-pro-latest',