getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-17 05:00:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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;
}