Skip to content

Commit 430c0a1

Browse files
committed
add env vars for google thinking config
1 parent 2c7ae9d commit 430c0a1

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

packages/web/src/app/api/(server)/chat/route.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -364,6 +364,14 @@ const getAISDKLanguageModelAndOptions = async (config: LanguageModel, orgId: num
364364

365365
return {
366366
model: vertex(modelId),
367+
providerOptions: {
368+
google: {
369+
thinkingConfig: {
370+
thinkingBudget: env.GOOGLE_VERTEX_THINKING_BUDGET_TOKENS,
371+
includeThoughts: env.GOOGLE_VERTEX_INCLUDE_THOUGHTS === 'true',
372+
}
373+
}
374+
},
367375
};
368376
}
369377
case 'google-vertex-anthropic': {

packages/web/src/env.mjs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,8 @@ export const env = createEnv({
116116
GOOGLE_VERTEX_PROJECT: z.string().optional(),
117117
GOOGLE_VERTEX_REGION: z.string().default('us-central1'),
118118
GOOGLE_APPLICATION_CREDENTIALS: z.string().optional(),
119+
GOOGLE_VERTEX_THINKING_BUDGET_TOKENS: numberSchema.default(-1),
120+
GOOGLE_VERTEX_INCLUDE_THOUGHTS: booleanSchema.default('true'),
119121

120122
AWS_ACCESS_KEY_ID: z.string().optional(),
121123
AWS_SECRET_ACCESS_KEY: z.string().optional(),

0 commit comments

Comments
 (0)