Skip to content

Commit 686c9d2

Browse files
committed
feat: adds support for custom endpoints in vertex
1 parent 5ef3893 commit 686c9d2

File tree

5 files changed

+276
-237
lines changed

5 files changed

+276
-237
lines changed

apps/web/src/app/(private)/settings/_components/ProviderApiKeys/New/_components/Configuration/Vertex/index.tsx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,15 +38,15 @@ export function VertexConfiguration() {
3838
required
3939
type='text'
4040
label='Private Key ID'
41-
info='The private key ID from the service account JSON file.'
41+
info='The private key ID from the service account JSON file. The credentials file can be obtained from the Google Cloud Console: https://console.cloud.google.com/apis/credentials'
4242
name={buildConfigFieldName({
4343
fieldNamespace: '[googleCredentials][privateKeyId]',
4444
})}
4545
/>
4646
<TextArea
4747
required
4848
label='Private Key'
49-
info='The private key from the service account JSON file.'
49+
info='The private key from the service account JSON file. The credentials file can be obtained from the Google Cloud Console: https://console.cloud.google.com/apis/credentials'
5050
name={buildConfigFieldName({
5151
fieldNamespace: '[googleCredentials][privateKey]',
5252
})}

packages/core/package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@
134134
"@ai-sdk/azure": "1.1.0",
135135
"@ai-sdk/deepseek": "0.2.5",
136136
"@ai-sdk/google": "1.2.18",
137-
"@ai-sdk/google-vertex": "2.2.27",
137+
"@ai-sdk/google-vertex": "3.0.25",
138138
"@ai-sdk/mistral": "1.1.0",
139139
"@ai-sdk/openai": "1.3.22",
140140
"@ai-sdk/perplexity": "1.1.3",

packages/core/src/services/ai/helpers.ts

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ import {
99
import {
1010
createVertexAnthropic,
1111
GoogleVertexAnthropicProvider,
12-
} from '@ai-sdk/google-vertex/anthropic/edge'
13-
import { createVertex, GoogleVertexProvider } from '@ai-sdk/google-vertex/edge'
12+
} from '@ai-sdk/google-vertex/anthropic'
13+
import { createVertex, GoogleVertexProvider } from '@ai-sdk/google-vertex'
1414
import { createMistral, MistralProvider } from '@ai-sdk/mistral'
1515
import { createOpenAI, type OpenAIProvider } from '@ai-sdk/openai'
1616
import { createPerplexity, PerplexityProvider } from '@ai-sdk/perplexity'
@@ -86,16 +86,20 @@ function validateVertexConfig({
8686
name: string
8787
maybeConfig: unknown
8888
}) {
89+
console.log('maybeConfig', maybeConfig)
8990
const result = vertexConfigurationSchema.safeParse(maybeConfig)
91+
console.log('result', result)
9092
if (result.success) {
9193
const config = result.data
9294
const privateKey = config.googleCredentials.privateKey.replace(/\\n/g, '\n')
9395

9496
return Result.ok({
9597
...config,
96-
googleCredentials: {
97-
...config.googleCredentials,
98-
privateKey,
98+
googleAuthOptions: {
99+
credentials: {
100+
client_email: config.googleCredentials.clientEmail,
101+
private_key: privateKey,
102+
},
99103
},
100104
})
101105
}

0 commit comments

Comments
 (0)