Added support for Gemini 1.5 Flash 8B and updated available versions …

…for Google AI and Vertex AI Gemini models (#1005) * Added support for Gemini 1.5 Flash 8B and updated other models * Reverted default Google AI model to absolute latest * formatted * vertex changes --------- Co-authored-by: “chrisraygill” <“[email protected]”>
firebase · Oct 4, 2024 · 96ea3e8 · 96ea3e8
1 parent 8f28a89
commit 96ea3e8
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 7 deletions.
diff --git a/js/plugins/googleai/src/gemini.ts b/js/plugins/googleai/src/gemini.ts
@@ -81,19 +81,19 @@ export const geminiPro = modelRef({
   name: 'googleai/gemini-pro',
   info: {
     label: 'Google AI - Gemini Pro',
-    versions: ['gemini-1.0-pro', 'gemini-1.0-pro-latest', 'gemini-1.0-pro-001'],
     supports: {
       multiturn: true,
       media: false,
       tools: true,
       systemRole: true,
     },
+    versions: ['gemini-1.0-pro', 'gemini-1.0-pro-latest', 'gemini-1.0-pro-001'],
   },
   configSchema: GeminiConfigSchema,
 });
 
 /**
- * @deprecated Use `gemini15Pro` or `gemini15Flash` instead.
+ * @deprecated Use `gemini15Pro`, `gemini15Flash`, or `gemini15flash8B` instead.
  */
 export const geminiProVision = modelRef({
   name: 'googleai/gemini-pro-vision',
@@ -123,7 +123,12 @@ export const gemini15Pro = modelRef({
       systemRole: true,
       output: ['text', 'json'],
     },
-    versions: ['gemini-1.5-pro-001'],
+    versions: [
+      'gemini-1.5-pro',
+      'gemini-1.5-pro-001',
+      'gemini-1.5-pro-002',
+      'gemini-1.5-pro-exp-0827',
+    ],
   },
   configSchema: GeminiConfigSchema,
 });
@@ -139,7 +144,30 @@ export const gemini15Flash = modelRef({
       systemRole: true,
       output: ['text', 'json'],
     },
-    versions: ['gemini-1.5-flash-001'],
+    versions: [
+      'gemini-1.5-flash',
+      'gemini-1.5-flash-001',
+      'gemini-1.5-flash-002',
+      'gemini-1.5-flash-8b-exp-0924',
+      'gemini-1.5-flash-8b-exp-0827',
+      'gemini-1.5-flash-exp-0827',
+    ],
+  },
+  configSchema: GeminiConfigSchema,
+});
+
+export const gemini15Flash8B = modelRef({
+  name: 'googleai/gemini-1.5-flash-8b-latest',
+  info: {
+    label: 'Google AI - Gemini 1.5 Flash-8B',
+    supports: {
+      multiturn: true,
+      media: true,
+      tools: true,
+      systemRole: true,
+      output: ['text', 'json'],
+    },
+    versions: ['gemini-1.5-flash-8b', 'gemini-1.5-flash-8b-001'],
   },
   configSchema: GeminiConfigSchema,
 });
@@ -174,6 +202,7 @@ export const SUPPORTED_V15_MODELS: Record<
 > = {
   'gemini-1.5-pro-latest': gemini15Pro,
   'gemini-1.5-flash-latest': gemini15Flash,
+  'gemini-1.5-flash-8b-latest': gemini15Flash8B,
 };
 
 const SUPPORTED_MODELS = {

diff --git a/js/plugins/googleai/src/index.ts b/js/plugins/googleai/src/index.ts
@@ -22,6 +22,7 @@ import {
 } from './embedder.js';
 import {
   gemini15Flash,
+  gemini15Flash8B,
   gemini15Pro,
   geminiPro,
   geminiProVision,
@@ -31,6 +32,7 @@ import {
 } from './gemini.js';
 export {
   gemini15Flash,
+  gemini15Flash8B,
   gemini15Pro,
   geminiPro,
   geminiProVision,

diff --git a/js/plugins/vertexai/src/gemini.ts b/js/plugins/vertexai/src/gemini.ts
@@ -80,7 +80,7 @@ export const geminiPro = modelRef({
   name: 'vertexai/gemini-1.0-pro',
   info: {
     label: 'Vertex AI - Gemini Pro',
-    versions: ['gemini-1.0-pro', 'gemini-1.0-pro-001'],
+    versions: ['gemini-1.0-pro', 'gemini-1.0-pro-001', 'gemini-1.0-pro-002'],
     supports: {
       multiturn: true,
       media: false,
@@ -113,7 +113,11 @@ export const gemini15Pro = modelRef({
   name: 'vertexai/gemini-1.5-pro',
   info: {
     label: 'Vertex AI - Gemini 1.5 Pro',
-    versions: ['gemini-1.5-pro-001'],
+    versions: [
+      'gemini-1.5-pro-001',
+      'gemini-1.5-pro-002',
+      'gemini-pro-experimental',
+    ],
     supports: {
       multiturn: true,
       media: true,
@@ -160,7 +164,11 @@ export const gemini15Flash = modelRef({
   name: 'vertexai/gemini-1.5-flash',
   info: {
     label: 'Vertex AI - Gemini 1.5 Flash',
-    versions: ['gemini-1.5-flash-001'],
+    versions: [
+      'gemini-1.5-flash-001',
+      'gemini-1.5-flash-002',
+      'gemini-flash-experimental',
+    ],
     supports: {
       multiturn: true,
       media: true,