shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix google-vertex build

Browse Source
This commit is contained in:
Anton Kuzmenko
2025-12-31 16:54:32 -08:00
committed by Mario Zechner
parent 6467e70995
commit 3b61d83d29
3 changed files with 205 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/ai/src/stream.ts
View File

@@ -259,23 +259,14 @@ function mapOptionsForApi<TApi extends Api>(
const vertexModel = model as Model<"google-vertex">;
const effort = clampReasoning(options.reasoning)!;
const geminiModel = vertexModel as unknown as Model<"google-generative-ai">;
if (isGemini3ProModel(vertexModel as unknown as Model<"google-generative-ai">)) {
if (isGemini3ProModel(geminiModel) || isGemini3FlashModel(geminiModel)) {
return {
...base,
thinking: {
enabled: true,
level: getGemini3ThinkingLevel(effort, vertexModel as unknown as Model<"google-generative-ai">),
},
} satisfies GoogleVertexOptions;
}
if (isGemini3FlashModel(vertexModel as unknown as Model<"google-generative-ai">)) {
return {
...base,
thinking: {
enabled: true,
level: getGemini3ThinkingLevel(effort, vertexModel as unknown as Model<"google-generative-ai">),
level: getGemini3ThinkingLevel(effort, geminiModel),
},
} satisfies GoogleVertexOptions;
}
@@ -284,7 +275,7 @@ function mapOptionsForApi<TApi extends Api>(
...base,
thinking: {
enabled: true,
budgetTokens: getGoogleBudget(vertexModel as unknown as Model<"google-generative-ai">, effort),
budgetTokens: getGoogleBudget(geminiModel, effort),
},
} satisfies GoogleVertexOptions;
}