From 23dbbd4cd8b8f5d8e5ac117e567b8540862577c4 Mon Sep 17 00:00:00 2001
From: Michael Doyle <michaeldoyle@google.com>
Date: Thu, 2 May 2024 15:05:35 -0400
Subject: [PATCH] Update dev-ui-gallery to show both v1 and v1beta models for
 GoogleAI

---
 js/samples/dev-ui-gallery/src/genkit.config.ts | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/js/samples/dev-ui-gallery/src/genkit.config.ts b/js/samples/dev-ui-gallery/src/genkit.config.ts
index aa78601e0f..7d868b3435 100644
--- a/js/samples/dev-ui-gallery/src/genkit.config.ts
+++ b/js/samples/dev-ui-gallery/src/genkit.config.ts
@@ -44,9 +44,7 @@ export default configureGenkit({
     firebase(),
 
     // model providers
-    googleAI({
-      apiVersion: 'v1beta', // enables Gemini 1.5
-    }),
+    googleAI({ apiVersion: ['v1', 'v1beta'] }),
     ollama({
       models: [
         { name: 'llama2' },