From 430c0a1fe92dc322ed23c2eea2703e7f9c8f5493 Mon Sep 17 00:00:00 2001 From: msukkari Date: Fri, 25 Jul 2025 20:08:24 -0700 Subject: [PATCH] add env vars for google thinking config --- packages/web/src/app/api/(server)/chat/route.ts | 8 ++++++++ packages/web/src/env.mjs | 2 ++ 2 files changed, 10 insertions(+) diff --git a/packages/web/src/app/api/(server)/chat/route.ts b/packages/web/src/app/api/(server)/chat/route.ts index 4248c449..05a6857d 100644 --- a/packages/web/src/app/api/(server)/chat/route.ts +++ b/packages/web/src/app/api/(server)/chat/route.ts @@ -364,6 +364,14 @@ const getAISDKLanguageModelAndOptions = async (config: LanguageModel, orgId: num return { model: vertex(modelId), + providerOptions: { + google: { + thinkingConfig: { + thinkingBudget: env.GOOGLE_VERTEX_THINKING_BUDGET_TOKENS, + includeThoughts: env.GOOGLE_VERTEX_INCLUDE_THOUGHTS === 'true', + } + } + }, }; } case 'google-vertex-anthropic': { diff --git a/packages/web/src/env.mjs b/packages/web/src/env.mjs index 0991b4bb..0059b791 100644 --- a/packages/web/src/env.mjs +++ b/packages/web/src/env.mjs @@ -116,6 +116,8 @@ export const env = createEnv({ GOOGLE_VERTEX_PROJECT: z.string().optional(), GOOGLE_VERTEX_REGION: z.string().default('us-central1'), GOOGLE_APPLICATION_CREDENTIALS: z.string().optional(), + GOOGLE_VERTEX_THINKING_BUDGET_TOKENS: numberSchema.default(-1), + GOOGLE_VERTEX_INCLUDE_THOUGHTS: booleanSchema.default('true'), AWS_ACCESS_KEY_ID: z.string().optional(), AWS_SECRET_ACCESS_KEY: z.string().optional(),