fix

nbonamy · May 16, 2024 · 76ad2bf · 76ad2bf
1 parent eb77b6c
commit 76ad2bf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git 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
@@ -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)