Add Gemini 1.5 Pro Flash

Author: MichaelDoyle

js/plugins/googleai/src/gemini.ts

@@ -117,6 +117,20 @@ export const gemini15Pro = modelRef({
   configSchema: GeminiConfigSchema,
 });
 
+export const gemini15Flash = modelRef({
+  name: 'googleai/gemini-1.5-flash-latest',
+  info: {
+    label: 'Google AI - Gemini 1.5 Flash',
+    supports: {
+      multiturn: true,
+      media: true,
+      tools: true,
+      systemRole: true,
+    },
+  },
+  configSchema: GeminiConfigSchema,
+});
+
 export const geminiUltra = modelRef({
   name: 'googleai/gemini-ultra',
   info: {
@@ -146,6 +160,7 @@ export const SUPPORTED_V15_MODELS: Record<
   ModelReference<z.ZodTypeAny>
 > = {
   'gemini-1.5-pro-latest': gemini15Pro,
+  'gemini-1.5-flash-latest': gemini15Flash,
 };
 
 const SUPPORTED_MODELS = {

js/plugins/googleai/src/index.ts

@@ -20,14 +20,15 @@ import {
   textEmbeddingGeckoEmbedder,
 } from './embedder.js';
 import {
+  gemini15Flash,
   gemini15Pro,
   geminiPro,
   geminiProVision,
   googleAIModel,
   SUPPORTED_V15_MODELS,
   SUPPORTED_V1_MODELS,
 } from './gemini.js';
-export { gemini15Pro, geminiPro, geminiProVision };
+export { gemini15Flash, gemini15Pro, geminiPro, geminiProVision };
 
 export interface PluginOptions {
   apiKey?: string;

js/plugins/vertexai/src/gemini.ts

@@ -103,6 +103,22 @@ export const gemini15ProPreview = modelRef({
   version: 'gemini-1.5-pro-preview-0409',
 });
 
+export const gemini15FlashPreview = modelRef({
+  name: 'vertexai/gemini-1.5-flash-preview',
+  info: {
+    label: 'Vertex AI - Gemini 1.5 Flash',
+    versions: ['gemini-1.5-flash-preview-0514'],
+    supports: {
+      multiturn: true,
+      media: true,
+      tools: true,
+      systemRole: true,
+    },
+  },
+  configSchema: GeminiConfigSchema,
+  version: 'gemini-1.5-flash-preview-0514',
+});
+
 export const SUPPORTED_V1_MODELS = {
   'gemini-1.0-pro': geminiPro,
   'gemini-1.0-pro-vision': geminiProVision,
@@ -111,6 +127,7 @@ export const SUPPORTED_V1_MODELS = {
 
 export const SUPPORTED_V15_MODELS = {
   'gemini-1.5-pro-preview': gemini15ProPreview,
+  'gemini-1.5-flash-preview': gemini15FlashPreview,
 };
 
 export const SUPPORTED_GEMINI_MODELS = {

js/plugins/vertexai/src/index.ts

@@ -36,6 +36,7 @@ import {
   vertexEvaluators,
 } from './evaluation.js';
 import {
+  gemini15FlashPreview,
   gemini15ProPreview,
   geminiModel,
   geminiPro,
@@ -48,6 +49,7 @@ export {
   claude3Haiku,
   claude3Opus,
   claude3Sonnet,
+  gemini15FlashPreview,
   gemini15ProPreview,
   geminiPro,
   geminiProVision,