getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-20 19:02:10 +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

15
packages/ai/src/providers/google-vertex.ts
View file

@ -3,7 +3,7 @@ import {
type GenerateContentParameters,
GoogleGenAI,
type ThinkingConfig,
type ThinkingLevel,
ThinkingLevel,
} from "@google/genai";
import { calculateCost } from "../models.js";
import type {
@ -19,6 +19,7 @@ import type {
} from "../types.js";
import { AssistantMessageEventStream } from "../utils/event-stream.js";
import { sanitizeSurrogates } from "../utils/sanitize-unicode.js";
import type { GoogleThinkingLevel } from "./google-gemini-cli.js";
import { convertMessages, convertTools, mapStopReason, mapToolChoice } from "./google-shared.js";
export interface GoogleVertexOptions extends StreamOptions {
@ -26,7 +27,7 @@ export interface GoogleVertexOptions extends StreamOptions {
thinking?: {
enabled: boolean;
budgetTokens?: number; // -1 for dynamic, 0 to disable
level?: ThinkingLevel;
level?: GoogleThinkingLevel;
};
project?: string;
location?: string;
@ -34,6 +35,14 @@ export interface GoogleVertexOptions extends StreamOptions {
const API_VERSION = "v1";
const THINKING_LEVEL_MAP: Record<GoogleThinkingLevel, ThinkingLevel> = {
THINKING_LEVEL_UNSPECIFIED: ThinkingLevel.THINKING_LEVEL_UNSPECIFIED,
MINIMAL: ThinkingLevel.MINIMAL,
LOW: ThinkingLevel.LOW,
MEDIUM: ThinkingLevel.MEDIUM,
HIGH: ThinkingLevel.HIGH,
};
// Counter for generating unique tool call IDs
let toolCallCounter = 0;
@ -322,7 +331,7 @@ function buildParams(
if (options.thinking?.enabled && model.reasoning) {
const thinkingConfig: ThinkingConfig = { includeThoughts: true };
if (options.thinking.level !== undefined) {
thinkingConfig.thinkingLevel = options.thinking.level;
thinkingConfig.thinkingLevel = THINKING_LEVEL_MAP[options.thinking.level];
} else if (options.thinking.budgetTokens !== undefined) {
thinkingConfig.thinkingBudget = options.thinking.budgetTokens;
}