Add Vertex AI routing support (#753)

* add Vertex AI routing support

* include Vertex smokes in action CI
This commit is contained in:
Colin McDonnell
2026-05-20 16:49:08 +00:00
committed by pullfrog[bot]
parent 09344a9ec9
commit c43ed65c3b
15 changed files with 532 additions and 47 deletions
+10
View File
@@ -49,6 +49,11 @@ import { resolveTimeoutMs, TIMEOUT_DISABLED } from "./utils/time.ts";
import { Timer } from "./utils/timer.ts";
import { createTodoTracker } from "./utils/todoTracking.ts";
import { getJobToken, resolveTokens } from "./utils/token.ts";
import {
cleanupVertexCredentials,
materializeVertexCredentials,
type VertexCredentials,
} from "./utils/vertex.ts";
import { resolveRun } from "./utils/workflow.ts";
export { Inputs } from "./utils/payload.ts";
@@ -177,6 +182,7 @@ export async function main(): Promise<MainResult> {
let toolContext: ToolContext | undefined;
let progressCallbackDisabled = false;
let todoTracker: ReturnType<typeof createTodoTracker> | undefined;
let vertexCredentials: VertexCredentials | undefined;
try {
if (payload.cwd && process.cwd() !== payload.cwd) {
@@ -233,6 +239,8 @@ export async function main(): Promise<MainResult> {
toolState.modelFallback = { from: fallback.from };
}
vertexCredentials = materializeVertexCredentials({ model: resolvedModel });
const agent = resolveAgent({ model: resolvedModel });
// surface the effective model in comment/review footers. payload.model is
@@ -475,6 +483,7 @@ export async function main(): Promise<MainResult> {
resolvedModel,
mcpServerUrl: mcpHttpServer.url,
tmpdir,
secretDenyPaths: vertexCredentials ? [vertexCredentials.secretDir] : [],
instructions,
todoTracker,
stopScript: runContext.repoSettings.stopScript,
@@ -619,5 +628,6 @@ export async function main(): Promise<MainResult> {
await patchWorkflowRunFields(toolContext, patch);
}
}
cleanupVertexCredentials(vertexCredentials);
}
}