From 76ad2bf0c3bbd0f2915349798c69c5ace7e1b90b Mon Sep 17 00:00:00 2001 From: Nicolas Bonamy Date: Thu, 16 May 2024 13:05:12 -0500 Subject: [PATCH] fix --- src/services/anthropic.ts | 7 +------ tests/unit/engine_anthropic.test.ts | 1 - 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/src/services/anthropic.ts b/src/services/anthropic.ts index 89ae457..41630f2 100644 --- a/src/services/anthropic.ts +++ b/src/services/anthropic.ts @@ -26,12 +26,7 @@ export default class extends LlmEngine { } getVisionModels(): string[] { - return [] - } - - // eslint-disable-next-line @typescript-eslint/no-unused-vars - isVisionModel(model: string): boolean { - return true + return ['*'] } getRountingModel(): string|null { diff --git a/tests/unit/engine_anthropic.test.ts b/tests/unit/engine_anthropic.test.ts index 479f584..25429dd 100644 --- a/tests/unit/engine_anthropic.test.ts +++ b/tests/unit/engine_anthropic.test.ts @@ -60,7 +60,6 @@ test('Anthropic Load Models', async () => { test('Anthropic Basic', async () => { const anthropic = new Anthropic(store.config) expect(anthropic.getName()).toBe('anthropic') - expect(anthropic.getVisionModels()).toStrictEqual([]) expect(anthropic.isVisionModel('claude-3-haiku-20240307')).toBe(true) expect(anthropic.isVisionModel('claude-3-sonnet-20240229')).toBe(true) expect(anthropic.isVisionModel('claude-3-opus-2024022')).toBe(true)