mirror of
https://github.com/sourcebot-dev/sourcebot.git
synced 2025-12-12 12:25:22 +00:00
add env vars for google thinking config
This commit is contained in:
parent
2c7ae9dae5
commit
430c0a1fe9
2 changed files with 10 additions and 0 deletions
|
|
@ -364,6 +364,14 @@ const getAISDKLanguageModelAndOptions = async (config: LanguageModel, orgId: num
|
||||||
|
|
||||||
return {
|
return {
|
||||||
model: vertex(modelId),
|
model: vertex(modelId),
|
||||||
|
providerOptions: {
|
||||||
|
google: {
|
||||||
|
thinkingConfig: {
|
||||||
|
thinkingBudget: env.GOOGLE_VERTEX_THINKING_BUDGET_TOKENS,
|
||||||
|
includeThoughts: env.GOOGLE_VERTEX_INCLUDE_THOUGHTS === 'true',
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
case 'google-vertex-anthropic': {
|
case 'google-vertex-anthropic': {
|
||||||
|
|
|
||||||
|
|
@ -116,6 +116,8 @@ export const env = createEnv({
|
||||||
GOOGLE_VERTEX_PROJECT: z.string().optional(),
|
GOOGLE_VERTEX_PROJECT: z.string().optional(),
|
||||||
GOOGLE_VERTEX_REGION: z.string().default('us-central1'),
|
GOOGLE_VERTEX_REGION: z.string().default('us-central1'),
|
||||||
GOOGLE_APPLICATION_CREDENTIALS: z.string().optional(),
|
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_ACCESS_KEY_ID: z.string().optional(),
|
||||||
AWS_SECRET_ACCESS_KEY: z.string().optional(),
|
AWS_SECRET_ACCESS_KEY: z.string().optional(),
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue