|
| 1 | +// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models |
1 | 2 | import { createVertex } from '@ai-sdk/google-vertex'; |
2 | 3 | import { createVertexAnthropic } from '@ai-sdk/google-vertex/anthropic'; |
3 | 4 | import { LanguageModel } from 'ai'; |
4 | 5 | import { AiProvider } from '../providers'; |
5 | 6 |
|
6 | | -const googleVertexGeminiModels = [ |
7 | | - 'gemini-3-pro-preview', |
8 | | - 'gemini-2.5-pro', |
9 | | - 'gemini-2.5-flash', |
10 | | - 'gemini-2.5-flash-lite', |
11 | | - 'gemini-2.0-flash-001', |
12 | | - 'gemini-2.0-flash-lite-001', |
13 | | -]; |
14 | | - |
15 | | -const googleVertexClaudeModels = [ |
16 | | - 'claude-opus-4-6', |
17 | | - 'claude-sonnet-4-6', |
18 | | - 'claude-opus-4-5@20251101', |
19 | | - 'claude-sonnet-4-5@20250929', |
20 | | - 'claude-haiku-4-5@20251001', |
21 | | - 'claude-3-haiku@20240307', |
| 7 | +const googleVertexModels = [ |
22 | 8 | 'claude-3-5-haiku@20241022', |
23 | 9 | 'claude-3-7-sonnet@20250219', |
24 | | - 'claude-sonnet-4@20250514', |
25 | | - 'claude-opus-4@20250514', |
| 10 | + 'claude-3-haiku@20240307', |
| 11 | + 'claude-haiku-4-5@20251001', |
26 | 12 | 'claude-opus-4-1@20250805', |
| 13 | + 'claude-opus-4-5@20251101', |
| 14 | + 'claude-opus-4-6', |
| 15 | + 'claude-opus-4@20250514', |
| 16 | + 'claude-sonnet-4-5@20250929', |
| 17 | + 'claude-sonnet-4-6', |
| 18 | + 'claude-sonnet-4@20250514', |
| 19 | + 'gemini-2.0-flash-001', |
| 20 | + 'gemini-2.0-flash-lite-001', |
| 21 | + 'gemini-2.5-flash', |
| 22 | + 'gemini-2.5-flash-lite', |
| 23 | + 'gemini-2.5-pro', |
| 24 | + 'gemini-3-pro-preview', |
27 | 25 | ]; |
28 | 26 |
|
29 | 27 | function safeParseJson(value: string): Record<string, unknown> | null { |
@@ -80,6 +78,6 @@ function createLanguageModel(params: { |
80 | 78 | } |
81 | 79 |
|
82 | 80 | export const googleVertexProvider: AiProvider = { |
83 | | - models: [...googleVertexGeminiModels, ...googleVertexClaudeModels], |
| 81 | + models: googleVertexModels, |
84 | 82 | createLanguageModel, |
85 | 83 | }; |
0 commit comments