From 88c2287ac7c0a500099676fdddae1870c4d2dfbe Mon Sep 17 00:00:00 2001 From: Mehdi Chaabouni Date: Wed, 19 Feb 2025 10:50:48 -0500 Subject: [PATCH] feat: updated gemini supported models --- packages/kurt-vertex-ai/spec/isSupportedModel.spec.ts | 1 - packages/kurt-vertex-ai/src/KurtVertexAI.ts | 1 - 2 files changed, 2 deletions(-) diff --git a/packages/kurt-vertex-ai/spec/isSupportedModel.spec.ts b/packages/kurt-vertex-ai/spec/isSupportedModel.spec.ts index 20d0635..efc2bc3 100644 --- a/packages/kurt-vertex-ai/spec/isSupportedModel.spec.ts +++ b/packages/kurt-vertex-ai/spec/isSupportedModel.spec.ts @@ -6,7 +6,6 @@ test("KurtVertexAI.isSupportedModel", () => { // https://ai.google.dev/gemini-api/docs/models/gemini expect(KurtVertexAI.isSupportedModel("gemini-1.5-pro")).toBe(true) expect(KurtVertexAI.isSupportedModel("gemini-1.5-flash")).toBe(true) - expect(KurtVertexAI.isSupportedModel("gemini-1.5-flash-8b")).toBe(true) expect(KurtVertexAI.isSupportedModel("gemini-2.0-flash")).toBe(true) expect(KurtVertexAI.isSupportedModel("gemini-1.0-pro")).toBe(false) expect(KurtVertexAI.isSupportedModel("gemini-1.0-pro-vision")).toBe(false) diff --git a/packages/kurt-vertex-ai/src/KurtVertexAI.ts b/packages/kurt-vertex-ai/src/KurtVertexAI.ts index 4868f26..8e7f59c 100644 --- a/packages/kurt-vertex-ai/src/KurtVertexAI.ts +++ b/packages/kurt-vertex-ai/src/KurtVertexAI.ts @@ -37,7 +37,6 @@ import { ZodError } from "zod" const COMPATIBLE_MODELS = [ "gemini-1.5-pro", "gemini-1.5-flash", - "gemini-1.5-flash-8b", "gemini-2.0-flash", ] as const