wave-agent-sdk 0.17.1 → 0.17.2

package/dist/managers/workflowManager.js

@@ -0,0 +1,380 @@
+import { randomUUID } from "crypto";
+import * as os from "os";
+import * as fs from "fs";
+import * as path from "path";
+import { taskNotificationToXml } from "../utils/notificationXml.js";
+import { ConcurrencyLimiter } from "../workflow/concurrencyLimiter.js";
+import { BudgetTracker } from "../workflow/budgetTracker.js";
+import { ProgressReporter } from "../workflow/progressReporter.js";
+import { Journal } from "../workflow/journal.js";
+import { createWorkflowApis } from "../workflow/workflowApis.js";
+import { validateScript, parseScript, executeScript, } from "../workflow/scriptRuntime.js";
+import { RunStateStore } from "../workflow/runState.js";
+import { logger } from "../utils/globalLogger.js";
+const DEFAULT_CONCURRENCY = () => Math.max(1, Math.min(16, os.cpus().length - 2));
+export class WorkflowManager {
+    constructor(container) {
+        this.runs = new Map();
+        this.abortControllers = new Map();
+        this.agentControllers = new Map();
+        this.runStateStore = null;
+        this.container = container;
+        // Initialize RunStateStore lazily (sessionDir depends on MessageManager)
+    }
+    get stateStore() {
+        if (!this.runStateStore) {
+            this.runStateStore = new RunStateStore(path.join(this.sessionDir, "workflows"));
+        }
+        return this.runStateStore;
+    }
+    get backgroundTaskManager() {
+        return this.container.get("BackgroundTaskManager");
+    }
+    get notificationQueue() {
+        return this.container.get("NotificationQueue");
+    }
+    get subagentManager() {
+        return this.container.get("SubagentManager");
+    }
+    get workdir() {
+        return this.container.get("workdir") || process.cwd();
+    }
+    get sessionDir() {
+        const messageManager = this.container.get("MessageManager");
+        return (messageManager?.getSessionDir() ||
+            path.join(os.homedir(), ".wave", "sessions"));
+    }
+    /**
+     * Create a new workflow run from a script string or file path.
+     * Persists the script to the session directory.
+     */
+    async createRun(script, args, opts) {
+        // Validate script
+        const validation = validateScript(script);
+        if (!validation.valid) {
+            throw new Error(`Script validation failed:\n${validation.errors.join("\n")}`);
+        }
+        // Parse meta for the run object
+        const { meta } = parseScript(script);
+        // Validate resumeFromRunId if provided
+        if (opts?.resumeFromRunId) {
+            const prevRun = this.runs.get(opts.resumeFromRunId);
+            if (!prevRun) {
+                throw new Error(`Cannot resume: run ${opts.resumeFromRunId} not found`);
+            }
+        }
+        // Generate run ID and persist script
+        const runId = `wf_${randomUUID().slice(0, 8)}`;
+        const runDir = path.join(this.sessionDir, "workflows", runId);
+        await fs.promises.mkdir(path.join(runDir, "agents"), { recursive: true });
+        const scriptPath = path.join(runDir, "script.js");
+        await fs.promises.writeFile(scriptPath, script, "utf-8");
+        const run = {
+            runId,
+            meta,
+            status: "running",
+            scriptPath,
+            args,
+            startTime: Date.now(),
+            phases: [],
+            totalAgents: 0,
+            totalTokens: 0,
+            resumeFromRunId: opts?.resumeFromRunId,
+        };
+        this.runs.set(runId, run);
+        // Persist run state
+        this.stateStore.save(run).catch((err) => {
+            logger.warn(`[Workflow] Failed to persist run state: ${err instanceof Error ? err.message : String(err)}`);
+        });
+        return run;
+    }
+    /**
+     * Start executing a workflow run in the background.
+     * Returns immediately; the workflow runs asynchronously.
+     */
+    async startRun(runId, opts) {
+        const run = this.runs.get(runId);
+        if (!run)
+            throw new Error(`Workflow run ${runId} not found`);
+        if (run.status !== "running")
+            throw new Error(`Workflow run ${runId} is not in running state`);
+        const abortController = new AbortController();
+        this.abortControllers.set(runId, abortController);
+        const runAgentControllers = new Map();
+        this.agentControllers.set(runId, runAgentControllers);
+        // Read script from file
+        const script = await fs.promises.readFile(run.scriptPath, "utf-8");
+        // Set up infrastructure
+        const concurrencyLimiter = new ConcurrencyLimiter(DEFAULT_CONCURRENCY());
+        const budgetTracker = new BudgetTracker(run.args &&
+            typeof run.args === "object" &&
+            "budget" in run.args
+            ? run.args.budget
+            : null);
+        const progressReporter = new ProgressReporter(run.meta, runId);
+        // Forward progress events via onProgress callback
+        const onProgress = (event) => {
+            logger.debug(`[Workflow:${runId}] progress: ${event.type} phase=${event.phaseIndex} agent=${event.agentIndex}`);
+        };
+        progressReporter.onEvent(onProgress);
+        // Journal — load previous journal if resuming
+        let journal;
+        let initialAgentCount = 0;
+        if (run.resumeFromRunId || opts?.retryAgentIndex !== undefined) {
+            // Try new-format path first, fall back to old format
+            const sourceRunId = run.resumeFromRunId || runId;
+            const newJournalPath = path.join(this.sessionDir, "workflows", sourceRunId, "journal.jsonl");
+            const oldJournalPath = path.join(this.sessionDir, "workflows", `journal-${sourceRunId}.jsonl`);
+            const journalPath = (await fs.promises
+                .access(newJournalPath)
+                .then(() => true)
+                .catch(() => false))
+                ? newJournalPath
+                : oldJournalPath;
+            journal = await Journal.load(journalPath);
+            // When retrying a specific agent, remove its failed entry
+            if (opts?.retryAgentIndex !== undefined) {
+                journal.removeFailedEntry(opts.retryAgentIndex);
+            }
+            // Count existing agent entries to offset the counter
+            initialAgentCount = journal.agentEntryCount;
+            // Re-open for appending (load() doesn't open the write stream)
+            await journal.init();
+            logger.info(`[Workflow] Resuming from ${sourceRunId} with ${initialAgentCount} cached agent results`);
+        }
+        else {
+            const journalPath = path.join(this.sessionDir, "workflows", runId, "journal.jsonl");
+            journal = new Journal(journalPath);
+            await journal.init();
+        }
+        // Register as background task
+        const taskId = this.backgroundTaskManager.generateId();
+        this.backgroundTaskManager.addTask({
+            id: taskId,
+            type: "workflow",
+            status: "running",
+            startTime: Date.now(),
+            stdout: "",
+            stderr: "",
+            description: `Workflow: ${run.meta.name}`,
+            runId,
+            onStop: () => {
+                abortController.abort();
+            },
+        });
+        // Create workflow APIs
+        const apis = createWorkflowApis({
+            subagentManager: this.subagentManager,
+            concurrencyLimiter,
+            budgetTracker,
+            progressReporter,
+            journal,
+            abortSignal: abortController.signal,
+            args: run.args,
+            initialAgentCount,
+            sessionDir: this.sessionDir,
+            runDir: path.join(this.sessionDir, "workflows", runId),
+            agentControllers: runAgentControllers,
+            onProgress: (event) => {
+                logger.debug(`[Workflow:${runId}] progress: ${event.type} phase=${event.phaseIndex} agent=${event.agentIndex}`);
+            },
+            onLog: (message) => {
+                logger.info(`[Workflow:${runId}] ${message}`);
+            },
+        });
+        // Execute in background
+        run.completionPromise = (async () => {
+            try {
+                const { result } = await executeScript(script, apis, abortController.signal);
+                // Update run state
+                run.status = "completed";
+                run.endTime = Date.now();
+                run.result = result;
+                run.phases = progressReporter.getPhaseStates();
+                run.totalAgents = progressReporter.totalAgents;
+                run.totalTokens = progressReporter.totalTokens;
+                // Persist final state
+                this.stateStore.save(run).catch(() => { });
+                // Close journal
+                await journal.close();
+                // Update background task status
+                const task = this.backgroundTaskManager.getTask(taskId);
+                if (task) {
+                    task.status = "completed";
+                    task.endTime = Date.now();
+                }
+                // Enqueue completion notification
+                const journalPath = journal.filePath;
+                this.notificationQueue.enqueue(taskNotificationToXml({
+                    type: "task_notification",
+                    taskId: runId,
+                    taskType: "workflow",
+                    status: "completed",
+                    summary: `Workflow "${run.meta.name}" completed — ${run.totalAgents} agents, ${(run.totalTokens / 1000).toFixed(1)}k tokens`,
+                    ...(journalPath && { outputFile: journalPath }),
+                }));
+                logger.info(`[Workflow] Run ${runId} completed: ${run.totalAgents} agents, ${run.totalTokens} tokens`);
+            }
+            catch (error) {
+                // Only update if stopRun hasn't already set the status
+                if (run.status === "running") {
+                    if (abortController.signal.aborted) {
+                        run.status = "aborted";
+                    }
+                    else {
+                        run.status = "failed";
+                        run.error = error instanceof Error ? error.message : String(error);
+                    }
+                    run.endTime = Date.now();
+                }
+                run.phases = progressReporter.getPhaseStates();
+                run.totalAgents = progressReporter.totalAgents;
+                run.totalTokens = progressReporter.totalTokens;
+                // Persist failure/abort state
+                this.stateStore.save(run).catch(() => { });
+                await journal.close();
+                // Update background task status
+                const task = this.backgroundTaskManager.getTask(taskId);
+                if (task) {
+                    task.status = abortController.signal.aborted ? "killed" : "failed";
+                    task.stderr = run.error || "";
+                    task.endTime = Date.now();
+                }
+                this.notificationQueue.enqueue(taskNotificationToXml({
+                    type: "task_notification",
+                    taskId: runId,
+                    taskType: "workflow",
+                    status: run.status === "aborted" ? "aborted" : "failed",
+                    summary: `Workflow "${run.meta.name}" ${run.status}${run.error ? `: ${run.error}` : ""}`,
+                }));
+                logger.warn(`[Workflow] Run ${runId} ${run.status}: ${run.error || "aborted"}`);
+            }
+            finally {
+                this.abortControllers.delete(runId);
+                this.agentControllers.delete(runId);
+            }
+        })();
+        // Don't await — let it run in background
+        run.completionPromise.catch(() => { }); // Prevent unhandled rejection
+    }
+    /**
+     * Resume a workflow run from its journal.
+     */
+    async resumeRun(runId) {
+        const run = this.runs.get(runId);
+        if (!run)
+            throw new Error(`Workflow run ${runId} not found`);
+        run.status = "running";
+        await this.startRun(runId);
+    }
+    /**
+     * Stop a running workflow.
+     */
+    stopRun(runId) {
+        const controller = this.abortControllers.get(runId);
+        if (controller) {
+            controller.abort();
+        }
+        const run = this.runs.get(runId);
+        if (run && run.status === "running") {
+            run.status = "aborted";
+            run.endTime = Date.now();
+        }
+    }
+    /**
+     * Skip a specific agent in a running workflow.
+     * Aborts the agent's controller and lets the workflow continue.
+     */
+    skipAgent(runId, agentIndex) {
+        const run = this.runs.get(runId);
+        if (!run || run.status !== "running")
+            return;
+        const agentController = this.agentControllers.get(runId)?.get(agentIndex);
+        if (agentController) {
+            agentController.abort();
+        }
+    }
+    /**
+     * Retry a specific agent by removing its journal entry and resuming.
+     */
+    async retryAgent(runId, agentIndex) {
+        const run = this.runs.get(runId);
+        if (!run)
+            throw new Error(`Workflow run ${runId} not found`);
+        // Stop the current execution
+        this.stopRun(runId);
+        // The journal will have cached results; when we resume,
+        // the agent_failed entry for this index will cause getCachedResult
+        // to return undefined, forcing re-execution
+        run.status = "running";
+        run.error = undefined;
+        run.failedAgentIndex = undefined;
+        run.failedAgentError = undefined;
+        await this.startRun(runId, { retryAgentIndex: agentIndex });
+    }
+    /**
+     * Kill a running workflow — aborts all agent controllers plus the run controller.
+     */
+    killRun(runId) {
+        // Abort all per-agent controllers
+        const agentControllers = this.agentControllers.get(runId);
+        if (agentControllers) {
+            for (const controller of agentControllers.values()) {
+                controller.abort();
+            }
+        }
+        // Abort the run controller
+        const controller = this.abortControllers.get(runId);
+        if (controller) {
+            controller.abort();
+        }
+        const run = this.runs.get(runId);
+        if (run && run.status === "running") {
+            run.status = "aborted";
+            run.endTime = Date.now();
+        }
+    }
+    /**
+     * List all workflow runs (includes in-memory and persisted).
+     */
+    async listRuns() {
+        // Load persisted runs that aren't already in memory
+        const persistedIds = await this.stateStore.listRuns();
+        for (const runId of persistedIds) {
+            if (!this.runs.has(runId)) {
+                const run = await this.stateStore.load(runId);
+                if (run) {
+                    this.runs.set(runId, run);
+                }
+            }
+        }
+        return Array.from(this.runs.values());
+    }
+    /**
+     * Get a specific workflow run.
+     */
+    getRun(runId) {
+        return this.runs.get(runId);
+    }
+    /**
+     * Clean up all running workflows.
+     */
+    cleanup() {
+        for (const controller of this.abortControllers.values()) {
+            controller.abort();
+        }
+        this.abortControllers.clear();
+        for (const agentMap of this.agentControllers.values()) {
+            for (const controller of agentMap.values()) {
+                controller.abort();
+            }
+        }
+        this.agentControllers.clear();
+        for (const run of this.runs.values()) {
+            if (run.status === "running") {
+                run.status = "aborted";
+                run.endTime = Date.now();
+            }
+        }
+    }
+}

package/dist/prompts/index.d.ts

@@ -16,6 +16,7 @@ export declare const WEB_CONTENT_SYSTEM_PROMPT = "You are a helpful assistant th
 export declare const BTW_SYSTEM_PROMPT = "You are a helpful assistant. Answer the user's side question based on the conversation history. \nDo NOT say things like \"Let me try...\", \"I'll now...\", \"Let me check...\", or promise to take any action. \nIf you don't know the answer, say so - do not offer to look it up or investigate. \nSimply answer the question with the information you have.";
 export declare function buildSystemPrompt(basePrompt: string | undefined, tools: ToolPlugin[], options?: {
     workdir?: string;
+    originalWorkdir?: string;
     memory?: string;
     language?: string;
     isSubagent?: boolean;
@@ -25,5 +26,5 @@ export declare function buildSystemPrompt(basePrompt: string | undefined, tools:
     };
     permissionMode?: PermissionMode;
 }): string;
-export declare function enhanceSystemPromptWithEnvDetails(existingSystemPrompt: string, workdir: string): string;
+export declare function enhanceSystemPromptWithEnvDetails(existingSystemPrompt: string, workdir: string, originalWorkdir?: string): string;
 //# sourceMappingURL=index.d.ts.map

package/dist/prompts/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/prompts/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAI/C,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAiBzD,eAAO,MAAM,kBAAkB,oKAAoK,CAAC;AAEpM,eAAO,MAAM,kBAAkB,opIAcqQ,CAAC;AAErS,eAAO,MAAM,wBAAwB,25DASqmB,CAAC;AAE3oB,eAAO,MAAM,WAAW,ihDAWqH,CAAC;AAE9I;;GAEG;AACH,eAAO,MAAM,wBAAwB,+uBAWiH,CAAC;AAEvJ,eAAO,MAAM,qBAAqB,6sBAMuL,CAAC;AAE1N,wBAAgB,mBAAmB,CACjC,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,OAAO,EACnB,UAAU,GAAE,OAAe,GAC1B,MAAM,CAmFR;AAED,eAAO,MAAM,qBAAqB,oKAAqB,CAAC;AAExD,eAAO,MAAM,8BAA8B,44DA8CI,CAAC;AAEhD,eAAO,MAAM,yBAAyB,wHAAwH,CAAC;AAC/J,eAAO,MAAM,iBAAiB,qWAG4B,CAAC;AAE3D,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,KAAK,EAAE,UAAU,EAAE,EACnB,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,UAAU,CAAC,EAAE;QACX,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,cAAc,CAAC,EAAE,cAAc,CAAC;CAC5B,GACL,MAAM,CA4DR;AAED,wBAAgB,iCAAiC,CAC/C,oBAAoB,EAAE,MAAM,EAC5B,OAAO,EAAE,MAAM,GACd,MAAM,CAkCR"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/prompts/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAI/C,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAiBzD,eAAO,MAAM,kBAAkB,oKAAoK,CAAC;AAEpM,eAAO,MAAM,kBAAkB,opIAcqQ,CAAC;AAErS,eAAO,MAAM,wBAAwB,25DASqmB,CAAC;AAE3oB,eAAO,MAAM,WAAW,ihDAWqH,CAAC;AAE9I;;GAEG;AACH,eAAO,MAAM,wBAAwB,+uBAWiH,CAAC;AAEvJ,eAAO,MAAM,qBAAqB,6sBAMuL,CAAC;AAE1N,wBAAgB,mBAAmB,CACjC,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,OAAO,EACnB,UAAU,GAAE,OAAe,GAC1B,MAAM,CAmFR;AAED,eAAO,MAAM,qBAAqB,oKAAqB,CAAC;AAExD,eAAO,MAAM,8BAA8B,44DA8CI,CAAC;AAEhD,eAAO,MAAM,yBAAyB,wHAAwH,CAAC;AAC/J,eAAO,MAAM,iBAAiB,qWAG4B,CAAC;AAE3D,wBAAgB,iBAAiB,CAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,KAAK,EAAE,UAAU,EAAE,EACnB,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,UAAU,CAAC,EAAE;QACX,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,cAAc,CAAC,EAAE,cAAc,CAAC;CAC5B,GACL,MAAM,CA4DR;AAED,wBAAgB,iCAAiC,CAC/C,oBAAoB,EAAE,MAAM,EAC5B,OAAO,EAAE,MAAM,EACf,eAAe,CAAC,EAAE,MAAM,GACvB,MAAM,CAkCR"}

package/dist/prompts/index.js

@@ -230,7 +230,7 @@ export function buildSystemPrompt(basePrompt, tools, options = {}) {
 
 Here is useful information about the environment you are running in:
 <env>
-Working directory: ${options.workdir}${worktreeSession ? `\nThis is a git worktree — an isolated copy of the repository. Run all commands from this directory. Do NOT \`cd\` to the original repository root at ${worktreeSession.originalCwd}.` : ""}
+Primary working directory: ${options.originalWorkdir ?? options.workdir}${worktreeSession ? `\nThis is a git worktree — an isolated copy of the repository. Run all commands from this directory. Do NOT \`cd\` to the original repository root at ${worktreeSession.originalCwd}.` : ""}
 Is directory a git repo: ${isGitRepo}
 Platform: ${platform}
 Shell: ${shellName}
@@ -250,7 +250,7 @@ Today's date: ${today}
     }
     return prompt;
 }
-export function enhanceSystemPromptWithEnvDetails(existingSystemPrompt, workdir) {
+export function enhanceSystemPromptWithEnvDetails(existingSystemPrompt, workdir, originalWorkdir) {
     const isGitRepo = isGitRepository(workdir);
     const platform = os.platform();
     const osVersion = `${os.type()} ${os.release()}`;
@@ -273,7 +273,7 @@ ${notes}
 
 Here is useful information about the environment you are running in:
 <env>
-Working directory: ${workdir}${worktreeSession ? `\nThis is a git worktree — an isolated copy of the repository. Run all commands from this directory. Do NOT \`cd\` to the original repository root at ${worktreeSession.originalCwd}.` : ""}
+Primary working directory: ${originalWorkdir ?? workdir}${worktreeSession ? `\nThis is a git worktree — an isolated copy of the repository. Run all commands from this directory. Do NOT \`cd\` to the original repository root at ${worktreeSession.originalCwd}.` : ""}
 Is directory a git repo: ${isGitRepo}
 Platform: ${platform}
 Shell: ${shellName}

package/dist/services/aiService.d.ts

@@ -17,6 +17,12 @@ export interface CallAgentOptions {
     model?: string;
     systemPrompt?: string;
     maxTokens?: number;
+    toolChoice?: "auto" | "none" | "required" | {
+        type: "function";
+        function: {
+            name: string;
+        };
+    };
     onContentUpdate?: (content: string) => void;
     onToolUpdate?: (toolCall: {
         id: string;
@@ -88,4 +94,21 @@ export interface BtwResult {
     };
 }
 export declare function btw(options: BtwOptions): Promise<BtwResult>;
+export interface EvaluateGoalOptions {
+    gatewayConfig: GatewayConfig;
+    modelConfig: ModelConfig;
+    model: string;
+    goalCondition: string;
+    messages: ChatCompletionMessageParam[];
+    abortSignal?: AbortSignal;
+}
+export interface EvaluateGoalResult {
+    content: string;
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+}
+export declare function evaluateGoal(options: EvaluateGoalOptions): Promise<EvaluateGoalResult>;
 //# sourceMappingURL=aiService.d.ts.map

package/dist/services/aiService.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiService.d.ts","sourceRoot":"","sources":["../../src/services/aiService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,6BAA6B,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAGL,0BAA0B,EAC1B,0BAA0B,EAE3B,MAAM,qBAAqB,CAAC;AAI7B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAKL,KAAK,WAAW,EACjB,MAAM,+BAA+B,CAAC;AAgEvC;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,IAAI,CAEvC;AA6DD,MAAM,WAAW,gBAAgB;IAE/B,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE;QACxB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;KACnD,KAAK,IAAI,CAAC;IACX,iBAAiB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,6BAA6B,EAAE,CAAC;IAC7C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,aAAa,CAAC,EACV,MAAM,GACN,QAAQ,GACR,YAAY,GACZ,gBAAgB,GAChB,eAAe,GACf,IAAI,CAAC;IACT,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC5C;AAED,wBAAsB,SAAS,CAC7B,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CAiU1B;AA6OD,MAAM,WAAW,sBAAsB;IAErC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,eAAe,CACnC,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,qBAAqB,CAAC,CAqGhC;AAED,MAAM,WAAW,wBAAwB;IAEvC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,wBAAwB,GAChC,OAAO,CAAC,uBAAuB,CAAC,CAmFlC;AAED,MAAM,WAAW,UAAU;IAEzB,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,CAuFjE"}
+{"version":3,"file":"aiService.d.ts","sourceRoot":"","sources":["../../src/services/aiService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,6BAA6B,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAGL,0BAA0B,EAC1B,0BAA0B,EAE3B,MAAM,qBAAqB,CAAC;AAI7B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEpE,OAAO,EAKL,KAAK,WAAW,EACjB,MAAM,+BAA+B,CAAC;AAiEvC;;GAEG;AACH,wBAAgB,gBAAgB,IAAI,IAAI,CAEvC;AA6DD,MAAM,WAAW,gBAAgB;IAE/B,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EACP,MAAM,GACN,MAAM,GACN,UAAU,GACV;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;IAGrD,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE;QACxB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;KACnD,KAAK,IAAI,CAAC;IACX,iBAAiB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,6BAA6B,EAAE,CAAC;IAC7C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,aAAa,CAAC,EACV,MAAM,GACN,QAAQ,GACR,YAAY,GACZ,gBAAgB,GAChB,eAAe,GACf,IAAI,CAAC;IACT,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC5C;AAuBD,wBAAsB,SAAS,CAC7B,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CAwU1B;AAyOD,MAAM,WAAW,sBAAsB;IAErC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,eAAe,CACnC,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,qBAAqB,CAAC,CAwGhC;AAED,MAAM,WAAW,wBAAwB;IAEvC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,wBAAwB,GAChC,OAAO,CAAC,uBAAuB,CAAC,CAsFlC;AAED,MAAM,WAAW,UAAU;IAEzB,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,CA0FjE;AAED,MAAM,WAAW,mBAAmB;IAClC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,kBAAkB,CAAC,CA4F7B"}

package/dist/services/aiService.js

@@ -1,11 +1,13 @@
 import { OpenAIClient } from "../utils/openaiClient.js";
 import { logger } from "../utils/globalLogger.js";
 import { addOnceAbortListener } from "../utils/abortUtils.js";
+import { ConfigurationError, CONFIG_ERRORS } from "../types/index.js";
 import { transformMessagesForClaudeCache, addCacheControlToLastTool, supportsPromptCaching, extendUsageWithCacheMetrics, } from "../utils/cacheControlUtils.js";
 import * as os from "os";
 import * as fs from "fs";
 import * as path from "path";
 import { COMPACT_MESSAGES_SYSTEM_PROMPT, WEB_CONTENT_SYSTEM_PROMPT, BTW_SYSTEM_PROMPT, } from "../prompts/index.js";
+import { GOAL_EVALUATION_SYSTEM_PROMPT } from "../constants/goalPrompts.js";
 // Global rate limiter state for 1 QPS
 let nextAllowedTime = 0;
 const MIN_INTERVAL = 1000; // 1 second for 1 QPS
@@ -63,8 +65,24 @@ function getModelConfig(modelName, baseConfig = {}) {
     }
     return config;
 }
+function validateModelConfig(modelConfig) {
+    if (!modelConfig.model) {
+        throw new ConfigurationError(CONFIG_ERRORS.MISSING_MODEL, "model", {
+            constructor: undefined,
+            environment: process.env.WAVE_MODEL,
+        });
+    }
+    if (!modelConfig.fastModel) {
+        throw new ConfigurationError(CONFIG_ERRORS.MISSING_FAST_MODEL, "fastModel", {
+            constructor: undefined,
+            environment: process.env.WAVE_FAST_MODEL,
+        });
+    }
+}
 export async function callAgent(options) {
     const { gatewayConfig, modelConfig, messages, abortSignal, workdir, tools, model, systemPrompt, onContentUpdate, onToolUpdate, onReasoningUpdate, } = options;
+    // Validate model config at call time
+    validateModelConfig(modelConfig);
     // Apply global 1 QPS rate limit
     if (process.env.NODE_ENV !== "test" ||
         modelConfig.model === "rate-limit-test") {
@@ -127,6 +145,10 @@ export async function callAgent(options) {
         if (processedTools && processedTools.length > 0) {
             createParams.tools = processedTools;
         }
+        // Add tool_choice if specified
+        if (options.toolChoice) {
+            createParams.tool_choice = options.toolChoice;
+        }
         if (isStreaming) {
             // Handle streaming response
             const { data: stream, response } = await openai.chat.completions
@@ -139,7 +161,7 @@ export async function callAgent(options) {
             response.headers.forEach((value, key) => {
                 responseHeaders[key] = value;
             });
-            return await processStreamingResponse(stream, onContentUpdate, onToolUpdate, onReasoningUpdate, abortSignal, responseHeaders, currentModel);
+            return await processStreamingResponse(stream, onContentUpdate, onToolUpdate, onReasoningUpdate, abortSignal, responseHeaders);
         }
         else {
             // Handle non-streaming response
@@ -162,8 +184,8 @@ export async function callAgent(options) {
                     total_tokens: response.usage.total_tokens,
                 }
                 : undefined;
-            // Extend usage with cache metrics for Claude models
-            if (totalUsage && supportsPromptCaching(currentModel) && response.usage) {
+            // Extend usage with cache metrics (Claude top-level + OpenAI prompt_tokens_details)
+            if (totalUsage && response.usage) {
                 totalUsage = extendUsageWithCacheMetrics(totalUsage, response.usage);
             }
             const result = {};
@@ -288,10 +310,9 @@ export async function callAgent(options) {
  * @param onToolUpdate Callback for tool updates
  * @param abortSignal Optional abort signal
  * @param responseHeaders Response headers from the initial request
- * @param modelName Model name for cache control processing
  * @returns Final result with accumulated content and tool calls
  */
-async function processStreamingResponse(stream, onContentUpdate, onToolUpdate, onReasoningUpdate, abortSignal, responseHeaders, modelName) {
+async function processStreamingResponse(stream, onContentUpdate, onToolUpdate, onReasoningUpdate, abortSignal, responseHeaders) {
     let accumulatedContent = "";
     let accumulatedReasoningContent = "";
     let hasReasoningContent = false;
@@ -312,10 +333,8 @@ async function processStreamingResponse(stream, onContentUpdate, onToolUpdate, o
                     completion_tokens: chunk.usage.completion_tokens,
                     total_tokens: chunk.usage.total_tokens,
                 };
-                // Extend usage with cache metrics for Claude models
-                if (modelName && supportsPromptCaching(modelName)) {
-                    chunkUsage = extendUsageWithCacheMetrics(chunkUsage, chunk.usage);
-                }
+                // Extend usage with cache metrics (Claude top-level + OpenAI prompt_tokens_details)
+                chunkUsage = extendUsageWithCacheMetrics(chunkUsage, chunk.usage);
                 usage = chunkUsage;
             }
             // Check for finish_reason in the choice
@@ -454,6 +473,8 @@ async function processStreamingResponse(stream, onContentUpdate, onToolUpdate, o
 }
 export async function compactMessages(options) {
     const { gatewayConfig, modelConfig, messages, abortSignal } = options;
+    // Validate model config at call time
+    validateModelConfig(modelConfig);
     // Apply global 1 QPS rate limit
     if (process.env.NODE_ENV !== "test" ||
         modelConfig.model === "rate-limit-test") {
@@ -534,6 +555,8 @@ export async function compactMessages(options) {
 }
 export async function processWebContent(options) {
     const { gatewayConfig, modelConfig, content, prompt, abortSignal } = options;
+    // Validate model config at call time
+    validateModelConfig(modelConfig);
     // Apply global 1 QPS rate limit
     if (process.env.NODE_ENV !== "test" ||
         modelConfig.model === "rate-limit-test") {
@@ -601,6 +624,8 @@ export async function processWebContent(options) {
 }
 export async function btw(options) {
     const { gatewayConfig, modelConfig, messages, question, abortSignal } = options;
+    // Validate model config at call time
+    validateModelConfig(modelConfig);
     // Apply global 1 QPS rate limit
     if (process.env.NODE_ENV !== "test" ||
         modelConfig.model === "rate-limit-test") {
@@ -667,3 +692,71 @@ export async function btw(options) {
         throw error;
     }
 }
+export async function evaluateGoal(options) {
+    const { gatewayConfig, modelConfig, model, goalCondition, messages, abortSignal, } = options;
+    // Create OpenAI client with injected configuration (no rate limiter — bypasses 1 QPS)
+    const openai = new OpenAIClient({
+        apiKey: gatewayConfig.apiKey,
+        baseURL: gatewayConfig.baseURL,
+        defaultHeaders: gatewayConfig.defaultHeaders,
+        fetchOptions: gatewayConfig.fetchOptions,
+        fetch: gatewayConfig.fetch,
+    });
+    const { model: _model, fastModel: _fastModel, maxTokens: _maxTokens, permissionMode: _permissionMode, ...extraParams } = modelConfig;
+    void _model;
+    void _fastModel;
+    void _maxTokens;
+    void _permissionMode;
+    const openaiModelConfig = getModelConfig(model, {
+        temperature: 0,
+        max_tokens: 200,
+        ...extraParams,
+    });
+    // Strip images from messages to reduce token usage (same as compact)
+    const cleanedMessages = messages.map((msg) => {
+        if (Array.isArray(msg.content)) {
+            const textParts = msg.content.filter((part) => part.type === "text");
+            const text = textParts.map((p) => p.text).join("\n");
+            return { ...msg, content: text || "(empty message)" };
+        }
+        return msg;
+    });
+    try {
+        const response = await openai.chat.completions.create({
+            ...openaiModelConfig,
+            messages: [
+                {
+                    role: "system",
+                    content: GOAL_EVALUATION_SYSTEM_PROMPT,
+                },
+                ...cleanedMessages,
+                {
+                    role: "user",
+                    content: `Goal condition: ${goalCondition}\n\nHas this goal been achieved based on the conversation above?`,
+                },
+            ],
+        }, {
+            signal: abortSignal,
+        });
+        const result = response.choices[0]?.message?.content?.trim();
+        if (!result) {
+            throw new Error("Goal evaluation returned empty response");
+        }
+        const usage = response.usage
+            ? {
+                prompt_tokens: response.usage.prompt_tokens,
+                completion_tokens: response.usage.completion_tokens,
+                total_tokens: response.usage.total_tokens,
+            }
+            : undefined;
+        return { content: result, usage };
+    }
+    catch (error) {
+        if (error.name === "AbortError") {
+            logger.info("Goal evaluation was aborted");
+            throw new Error("Goal evaluation was aborted");
+        }
+        logger.error("Goal evaluation failed:", error);
+        throw error;
+    }
+}