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