diff --git a/js/core/src/logging.ts b/js/core/src/logging.ts index dd15e05ed..c1d49f7e1 100644 --- a/js/core/src/logging.ts +++ b/js/core/src/logging.ts @@ -81,4 +81,4 @@ class Logger { } } -export const logger = new Logger(); \ No newline at end of file +export const logger = new Logger(); diff --git a/js/plugins/googleai/src/gemini.ts b/js/plugins/googleai/src/gemini.ts index 81a6809e5..a734acac3 100644 --- a/js/plugins/googleai/src/gemini.ts +++ b/js/plugins/googleai/src/gemini.ts @@ -157,6 +157,21 @@ export const gemini15Flash8b = modelRef({ configSchema: GeminiConfigSchema, }); +export const gemini20FlashExp = modelRef({ + name: 'googleai/gemini-2.0-flash-exp', + info: { + label: 'Google AI - Gemini 2.0 Flash (Experimental)', + versions: [], + supports: { + multiturn: true, + media: true, + tools: true, + systemRole: true, + }, + }, + configSchema: GeminiConfigSchema, +}); + export const SUPPORTED_V1_MODELS = { 'gemini-1.0-pro': gemini10Pro, }; @@ -165,6 +180,7 @@ export const SUPPORTED_V15_MODELS = { 'gemini-1.5-pro': gemini15Pro, 'gemini-1.5-flash': gemini15Flash, 'gemini-1.5-flash-8b': gemini15Flash8b, + 'gemini-2.0-flash-exp': gemini20FlashExp, }; export const SUPPORTED_GEMINI_MODELS: Record< diff --git a/js/plugins/googleai/src/index.ts b/js/plugins/googleai/src/index.ts index abb731426..be1de65f1 100644 --- a/js/plugins/googleai/src/index.ts +++ b/js/plugins/googleai/src/index.ts @@ -27,9 +27,18 @@ import { defineGoogleAIModel, gemini10Pro, gemini15Flash, + gemini15Flash8b, gemini15Pro, + gemini20FlashExp, } from './gemini.js'; -export { gemini10Pro, gemini15Flash, gemini15Pro, textEmbeddingGecko001 }; +export { + gemini10Pro, + gemini15Flash, + gemini15Flash8b, + gemini15Pro, + gemini20FlashExp, + textEmbeddingGecko001, +}; export interface PluginOptions { apiKey?: string; diff --git a/js/plugins/vertexai/src/gemini.ts b/js/plugins/vertexai/src/gemini.ts index 574698763..12caa1bbc 100644 --- a/js/plugins/vertexai/src/gemini.ts +++ b/js/plugins/vertexai/src/gemini.ts @@ -126,6 +126,21 @@ export const gemini15Flash = modelRef({ configSchema: GeminiConfigSchema, }); +export const gemini20FlashExp = modelRef({ + name: 'vertexai/gemini-2.0-flash-exp', + info: { + label: 'Vertex AI - Gemini 2.0 Flash (Experimental)', + versions: [], + supports: { + multiturn: true, + media: true, + tools: true, + systemRole: true, + }, + }, + configSchema: GeminiConfigSchema, +}); + export const SUPPORTED_V1_MODELS = { 'gemini-1.0-pro': gemini10Pro, }; @@ -133,6 +148,7 @@ export const SUPPORTED_V1_MODELS = { export const SUPPORTED_V15_MODELS = { 'gemini-1.5-pro': gemini15Pro, 'gemini-1.5-flash': gemini15Flash, + 'gemini-2.0-flash-exp': gemini20FlashExp, }; export const SUPPORTED_GEMINI_MODELS = { diff --git a/js/plugins/vertexai/src/index.ts b/js/plugins/vertexai/src/index.ts index 688be3f82..6648ca686 100644 --- a/js/plugins/vertexai/src/index.ts +++ b/js/plugins/vertexai/src/index.ts @@ -32,6 +32,7 @@ import { gemini10Pro, gemini15Flash, gemini15Pro, + gemini20FlashExp, } from './gemini.js'; import { SUPPORTED_IMAGEN_MODELS, @@ -45,6 +46,7 @@ export { gemini10Pro, gemini15Flash, gemini15Pro, + gemini20FlashExp, imagen2, imagen3, imagen3Fast,