ti.sil/sourcebot
1
0
Fork 0
mirror of https://github.com/sourcebot-dev/sourcebot.git synced 2025-12-11 20:05:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add env vars for google thinking config

Browse source
This commit is contained in:
msukkari 2025-07-25 20:08:24 -07:00
parent 2c7ae9dae5
commit 430c0a1fe9
2 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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': {

2
packages/web/src/env.mjs
View file

@ -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(),