feat: support claude 3.5 v2

rxliuli · Nov 21, 2024 · 5a4d2cb · 5a4d2cb
1 parent d344af3
commit 5a4d2cb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/llm/__tests__/anthropic.test.ts b/src/llm/__tests__/anthropic.test.ts
@@ -1,11 +1,9 @@
 import { beforeAll, expect, it } from 'vitest'
 import {
-  anthropic,
   anthropicVertex,
   getImageAsBase64,
   IAnthropicVertex,
 } from '../anthropic'
-import { AnthropicVertex } from '@anthropic-ai/vertex-sdk'
 
 let client: IAnthropicVertex
 beforeAll(() => {

diff --git a/src/llm/anthropic.ts b/src/llm/anthropic.ts
@@ -294,6 +294,8 @@ export function anthropicVertex(env: Record<string, string>): IAnthropicVertex {
     'VERTEX_ANTROPIC_PROJECTID',
   ]
   r.supportModels = [
+    'claude-3-5-sonnet-v2@20241022',
+    'claude-3-5-haiku@20241022',
     'claude-3-5-sonnet@20240620',
     'claude-3-haiku@20240307',
     'claude-3-opus@20240229',
@@ -308,6 +310,8 @@ export function anthropic(env: Record<string, string>): IAnthropicVertex {
   r.name = 'anthropic'
   r.requiredEnv = ['ANTROPIC_API_KEY']
   r.supportModels = [
+    'claude-3-5-sonnet-20241022',
+    'claude-3-5-haiku-20241022',
     'claude-3-5-sonnet-20240620',
     'claude-3-opus-20240229',
     'claude-3-sonnet-20240229',