laohuang 0.4.2 → 0.5.0

package/dist/runtime/agent-step-runner.js

@@ -0,0 +1,361 @@
+import { randomUUID } from "node:crypto";
+import { ModelStreamCancelled, ModelStreamError, } from "../model-stream.js";
+import { modelErrorKind } from "../model-adapter.js";
+import { touchedPathOf } from "../tools.js";
+export const FORCED_FINAL_PROMPT = `Tool use has been stopped by the runtime safety guard.
+Do not call any tools. Give the user the best concise answer possible from the
+information already available. Clearly state any limitation caused by stopping
+tool use, but do not mention internal implementation details unless useful.`;
+/** Runs the model/tool steps inside one user turn. */
+export class AgentStepRunner {
+    options;
+    constructor(options) {
+        this.options = options;
+    }
+    async run() {
+        const { cancelToken, context, guardPolicy, history, userInput, } = this.options;
+        let toolRounds = 0;
+        let modelRound = 0;
+        let modelRequests = 0;
+        let totalTokens = 0;
+        const startedAt = performance.now();
+        let guardReason = null;
+        let guardEmitted = false;
+        if (!history.commitInput(userInput)) {
+            throw this.options.createCancelled("cancelled before user input commit");
+        }
+        this.emit("user_message", { content: userInput });
+        this.options.injectBaselineInstructions(cancelToken);
+        for (;;) {
+            this.raiseIfCancelled();
+            guardReason = guardReason ?? guardPolicy.budgetReason(totalTokens, (performance.now() - startedAt) / 1000);
+            const forceFinal = guardReason !== null;
+            if (guardReason !== null && !guardEmitted) {
+                guardEmitted = true;
+                this.emit("agent_guard_triggered", this.guardPayload({
+                    reason: guardReason,
+                    toolRounds,
+                    modelRequests,
+                    totalTokens,
+                    startedAt,
+                }));
+            }
+            if (context?.modelStarted?.() === false) {
+                throw this.options.createCancelled("cancelled before model request");
+            }
+            modelRound += 1;
+            modelRequests += 1;
+            const requestId = modelRound === 1 && this.options.requestId !== null
+                ? this.options.requestId
+                : randomUUID();
+            this.options.onRequestId(requestId);
+            this.emit("model_request", {
+                round: modelRound,
+                request_id: requestId,
+                message_count: history.snapshot().length,
+                tool_rounds: toolRounds,
+                model_requests: modelRequests,
+                total_tokens: totalTokens,
+                force_final: forceFinal,
+                guard_reason: guardReason,
+            });
+            const requestMessages = history.snapshot();
+            if (forceFinal) {
+                const systemMessage = {
+                    ...(requestMessages[0] ?? {}),
+                };
+                systemMessage["content"] =
+                    `${String(systemMessage["content"] ?? "")}\n\n${FORCED_FINAL_PROMPT}`;
+                requestMessages[0] = systemMessage;
+            }
+            let result;
+            try {
+                const modelRequestOpened = context?.modelRequestOpened;
+                result = await this.options.modelRuntime.complete({
+                    client: this.options.client,
+                    model: this.options.model,
+                    messages: requestMessages,
+                    tools: this.options.toolDefinitions,
+                    toolChoice: forceFinal ? "none" : "auto",
+                    requestId,
+                    cancelToken,
+                    isRequestActive: this.options.isRequestActive,
+                    onDelta: (kind, payload) => {
+                        this.emit(kind, { round: modelRound, ...payload });
+                    },
+                    onRequestOpened: modelRequestOpened
+                        ? () => modelRequestOpened.call(context)
+                        : null,
+                });
+            }
+            catch (error) {
+                if (error instanceof ModelStreamCancelled) {
+                    this.emit("model_response_aborted", {
+                        round: modelRound,
+                        request_id: requestId,
+                        reason: errorMessage(error),
+                    });
+                    throw this.options.createCancelled(errorMessage(error), error);
+                }
+                const errorPayload = {
+                    round: modelRound,
+                    request_id: requestId,
+                    error: errorMessage(error),
+                };
+                if (error instanceof ModelStreamError && error.hadDelta) {
+                    this.emit("model_response_aborted", errorPayload);
+                    this.options.emitLegacy("model_error", errorPayload);
+                }
+                else {
+                    this.emit("model_error", errorPayload);
+                }
+                if (forceFinal) {
+                    const payload = this.guardPayload({
+                        reason: guardReason ?? "runtime safety guard",
+                        toolRounds,
+                        modelRequests,
+                        totalTokens,
+                        startedAt,
+                        finalError: errorMessage(error),
+                    });
+                    this.emit("agent_guard_failed", payload);
+                    let message = guardErrorMessage(payload);
+                    if (this.options.provider && modelErrorKind(error) === "authentication") {
+                        message += ` Authentication failed for ${this.options.provider}. ` +
+                            `Run /login ${this.options.provider} to update your API key.`;
+                    }
+                    throw this.options.createError(message, error);
+                }
+                let message = `Model request failed: ${errorMessage(error)}`;
+                if (this.options.provider && modelErrorKind(error) === "authentication") {
+                    message += `\nAuthentication failed for ${this.options.provider}. ` +
+                        `Run /login ${this.options.provider} to update your API key.`;
+                }
+                throw this.options.createError(message, error);
+            }
+            const toolCalls = [...result.toolCalls];
+            let requestTokens = usageTotalTokens(result.usage);
+            const tokensEstimated = requestTokens === 0;
+            if (tokensEstimated) {
+                requestTokens = estimateRequestTokens(requestMessages, result.messageDict());
+            }
+            totalTokens += requestTokens;
+            this.emit("model_response", {
+                round: modelRound,
+                request_id: requestId,
+                finish_reason: result.finishReason,
+                tool_call_count: toolCalls.length,
+                tool_names: toolCalls.map((call) => call.function.name),
+                tool_call_ids: toolCalls.map((call) => call.id),
+                usage: result.usage,
+                request_tokens: requestTokens,
+                tokens_estimated: tokensEstimated,
+                total_tokens: totalTokens,
+                tool_rounds: toolRounds,
+                model_requests: modelRequests,
+                force_final: forceFinal,
+            });
+            if (forceFinal && toolCalls.length > 0) {
+                this.emit("model_response_aborted", {
+                    round: modelRound,
+                    request_id: requestId,
+                    reason: "tool call returned while tools were disabled",
+                });
+                const payload = this.guardPayload({
+                    reason: guardReason ?? "runtime safety guard",
+                    toolRounds,
+                    modelRequests,
+                    totalTokens,
+                    startedAt,
+                });
+                this.emit("agent_guard_failed", payload);
+                throw this.options.createError(guardErrorMessage(payload));
+            }
+            const postResponseGuard = guardPolicy.budgetReason(totalTokens, (performance.now() - startedAt) / 1000);
+            if (toolCalls.length > 0 && postResponseGuard !== null) {
+                this.emit("model_response_aborted", {
+                    round: modelRound,
+                    request_id: requestId,
+                    reason: postResponseGuard,
+                });
+                guardReason = postResponseGuard;
+                continue;
+            }
+            const assistantMessage = result.messageDict();
+            if (!history.commitAssistant(assistantMessage)) {
+                this.emit("model_response_aborted", {
+                    round: modelRound,
+                    request_id: requestId,
+                    reason: "cancelled before history commit",
+                });
+                throw this.options.createCancelled("cancelled before history commit");
+            }
+            this.emit("model_response_committed", { round: modelRound, request_id: requestId });
+            if (toolCalls.length === 0) {
+                if (result.content === null) {
+                    throw this.options.createError("Model response had no content");
+                }
+                this.emit("assistant_response", {
+                    round: modelRound,
+                    content: truncateForEvent(result.content),
+                });
+                return result.content;
+            }
+            toolRounds += 1;
+            context?.toolsStarted?.();
+            const toolResults = (await this.options.toolRuntime.execute({
+                toolCalls,
+                executionMode: this.options.toolExecution,
+                cancelToken,
+                onToolStart: (event) => this.startToolEvent(event, modelRound),
+                onToolResult: (event) => this.completeToolEvent(event, modelRound),
+            })).results;
+            const repeated = guardPolicy.recordRepeatedToolCalls(toolCalls, toolResults);
+            history.commitToolResults(toolCalls, toolResults);
+            this.raiseIfCancelled();
+            const touchedPaths = [];
+            for (const toolResult of toolResults) {
+                const touched = touchedPathOf(toolResult);
+                if (typeof touched === "string") {
+                    touchedPaths.push(touched);
+                }
+            }
+            this.options.discoverForTouchedPaths(touchedPaths);
+            if (repeated !== null) {
+                guardReason = `repeated tool call detected (${repeated.name} repeated ` +
+                    `${repeated.count} times with the same arguments and result)`;
+            }
+            const pendingBatch = context?.safePoint?.();
+            if (pendingBatch !== null && pendingBatch !== undefined && (pendingBatch.content ?? "") !== "") {
+                if (!history.commitPending(pendingBatch)) {
+                    throw this.options.createCancelled("cancelled before pending input commit");
+                }
+                this.emit("user_message", {
+                    content: pendingBatch.content,
+                    pending_event_ids: [...(pendingBatch.eventIds ?? [])],
+                });
+            }
+        }
+    }
+    startToolEvent(event, modelRound) {
+        this.emit("tool_start", {
+            ...this.toolEventContext(event, modelRound),
+            arguments: safeArguments(event.args),
+        });
+    }
+    completeToolEvent(event, modelRound) {
+        this.options.onToolEvent(event.toolCall.function.name, event.args, event.result);
+        const status = event.result["status"];
+        this.emit("tool_result", {
+            ...this.toolEventContext(event, modelRound),
+            status: (typeof status === "string" && status) ||
+                (event.result["ok"] ? "completed" : "failed"),
+            result: safeResult(event.result),
+        });
+    }
+    toolEventContext(event, modelRound) {
+        return {
+            round: modelRound,
+            index: event.index,
+            batch_size: event.batchSize,
+            tool_call_id: event.toolCall.id,
+            name: event.toolCall.function.name,
+        };
+    }
+    guardPayload(options) {
+        const payload = {
+            reason: options.reason,
+            tool_rounds: options.toolRounds,
+            model_requests: options.modelRequests,
+            total_tokens: options.totalTokens,
+            elapsed_ms: Math.round(performance.now() - options.startedAt),
+        };
+        if (options.finalError) {
+            payload["final_error"] = options.finalError;
+        }
+        return payload;
+    }
+    emit(eventType, payload) {
+        this.options.emit(eventType, payload);
+    }
+    isCancelled() {
+        return this.options.cancelToken !== null && this.options.cancelToken.isCancelled();
+    }
+    raiseIfCancelled() {
+        if (this.isCancelled()) {
+            throw this.options.createCancelled(this.options.cancelToken?.reason || "cancelled");
+        }
+    }
+}
+function usageTotalTokens(usage) {
+    if (usage === null || usage === undefined || typeof usage !== "object") {
+        return 0;
+    }
+    const record = usage;
+    const total = record["total_tokens"];
+    if (isJsonInteger(total)) {
+        return Math.max(0, total);
+    }
+    const input = record["prompt_tokens"] ?? record["input_tokens"] ?? 0;
+    const output = record["completion_tokens"] ?? record["output_tokens"] ?? 0;
+    let sum = 0;
+    for (const value of [input, output]) {
+        if (isJsonInteger(value) && value > 0) {
+            sum += value;
+        }
+    }
+    return sum;
+}
+function isJsonInteger(value) {
+    return typeof value === "number" && Number.isInteger(value);
+}
+function estimateRequestTokens(messages, response) {
+    const serialized = JSON.stringify([...messages, response]);
+    return Math.max(1, Math.floor((Buffer.byteLength(serialized, "utf8") + 3) / 4));
+}
+function guardErrorMessage(payload) {
+    let message = "Agent safety guard stopped tool use but could not produce a final " +
+        `answer: ${String(payload["reason"])}. ` +
+        `Tool rounds: ${String(payload["tool_rounds"])}; ` +
+        `model requests: ${String(payload["model_requests"])}; ` +
+        `tokens counted: ${String(payload["total_tokens"])}; ` +
+        `elapsed: ${String(payload["elapsed_ms"])}ms.`;
+    if (payload["final_error"]) {
+        message += ` Final request failed: ${String(payload["final_error"])}`;
+    }
+    return message;
+}
+function safeArguments(args) {
+    const safe = { ...args };
+    for (const key of ["content", "old_text", "new_text"]) {
+        const value = safe[key];
+        if (typeof value === "string") {
+            safe[key] = `<${value.length} chars>`;
+        }
+    }
+    const edits = safe["edits"];
+    if (Array.isArray(edits)) {
+        safe["edits"] = `<${edits.length} edits>`;
+    }
+    const result = {};
+    for (const [key, value] of Object.entries(safe)) {
+        result[key] = typeof value === "string" ? truncateForEvent(value) : value;
+    }
+    return result;
+}
+function safeResult(result) {
+    const safe = {};
+    for (const [key, value] of Object.entries(result)) {
+        safe[key] = typeof value === "string" ? truncateForEvent(value) : value;
+    }
+    return safe;
+}
+function truncateForEvent(value, limit = 4_000) {
+    return value.length <= limit
+        ? value
+        : `${value.slice(0, limit)}\n...[truncated ${value.length - limit} chars]`;
+}
+function errorMessage(error) {
+    return error instanceof Error ? error.message : String(error);
+}
+//# sourceMappingURL=agent-step-runner.js.map

package/dist/runtime/agent-step-runner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-step-runner.js","sourceRoot":"","sources":["../../src/runtime/agent-step-runner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EACL,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,cAAc,EAAuB,MAAM,qBAAqB,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAe5C,MAAM,CAAC,MAAM,mBAAmB,GAAG;;;4EAGyC,CAAC;AAkC7E,sDAAsD;AACtD,MAAM,OAAO,eAAe;IACT,OAAO,CAAyB;IAEjD,YAAY,OAA+B;QACzC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,GAAG;QACP,MAAM,EACJ,WAAW,EACX,OAAO,EACP,WAAW,EACX,OAAO,EACP,SAAS,GACV,GAAG,IAAI,CAAC,OAAO,CAAC;QACjB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,WAAW,GAAkB,IAAI,CAAC;QACtC,IAAI,YAAY,GAAG,KAAK,CAAC;QAEzB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,oCAAoC,CAAC,CAAC;QAC3E,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,0BAA0B,CAAC,WAAW,CAAC,CAAC;QAErD,SAAS,CAAC;YACR,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACxB,WAAW,GAAG,WAAW,IAAI,WAAW,CAAC,YAAY,CACnD,WAAW,EACX,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CACvC,CAAC;YACF,MAAM,UAAU,GAAG,WAAW,KAAK,IAAI,CAAC;YACxC,IAAI,WAAW,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBAC1C,YAAY,GAAG,IAAI,CAAC;gBACpB,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,IAAI,CAAC,YAAY,CAAC;oBACnD,MAAM,EAAE,WAAW;oBACnB,UAAU;oBACV,aAAa;oBACb,WAAW;oBACX,SAAS;iBACV,CAAC,CAAC,CAAC;YACN,CAAC;YACD,IAAI,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,KAAK,EAAE,CAAC;gBACxC,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,gCAAgC,CAAC,CAAC;YACvE,CAAC;YACD,UAAU,IAAI,CAAC,CAAC;YAChB,aAAa,IAAI,CAAC,CAAC;YACnB,MAAM,SAAS,GAAG,UAAU,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,IAAI;gBACnE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS;gBACxB,CAAC,CAAC,UAAU,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YACpC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;gBACzB,KAAK,EAAE,UAAU;gBACjB,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM;gBACxC,WAAW,EAAE,UAAU;gBACvB,cAAc,EAAE,aAAa;gBAC7B,YAAY,EAAE,WAAW;gBACzB,WAAW,EAAE,UAAU;gBACvB,YAAY,EAAE,WAAW;aAC1B,CAAC,CAAC;YACH,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,aAAa,GAA4B;oBAC7C,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;iBAC9B,CAAC;gBACF,aAAa,CAAC,SAAS,CAAC;oBACtB,GAAG,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,OAAO,mBAAmB,EAAE,CAAC;gBACxE,eAAe,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC;YACrC,CAAC;YAED,IAAI,MAAM,CAAC;YACX,IAAI,CAAC;gBACH,MAAM,kBAAkB,GAAG,OAAO,EAAE,kBAAkB,CAAC;gBACvD,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC;oBAChD,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;oBAC3B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;oBACzB,QAAQ,EAAE,eAAe;oBACzB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe;oBACnC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM;oBACxC,SAAS;oBACT,WAAW;oBACX,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe;oBAC7C,OAAO,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE;wBACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;oBACrD,CAAC;oBACD,eAAe,EAAE,kBAAkB;wBACjC,CAAC,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC;wBACxC,CAAC,CAAC,IAAI;iBACT,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,oBAAoB,EAAE,CAAC;oBAC1C,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE;wBAClC,KAAK,EAAE,UAAU;wBACjB,UAAU,EAAE,SAAS;wBACrB,MAAM,EAAE,YAAY,CAAC,KAAK,CAAC;qBAC5B,CAAC,CAAC;oBACH,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;gBACjE,CAAC;gBACD,MAAM,YAAY,GAAG;oBACnB,KAAK,EAAE,UAAU;oBACjB,UAAU,EAAE,SAAS;oBACrB,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC;iBAC3B,CAAC;gBACF,IAAI,KAAK,YAAY,gBAAgB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBACxD,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;gBACvD,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;gBACzC,CAAC;gBACD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;wBAChC,MAAM,EAAE,WAAW,IAAI,sBAAsB;wBAC7C,UAAU;wBACV,aAAa;wBACb,WAAW;wBACX,SAAS;wBACT,UAAU,EAAE,YAAY,CAAC,KAAK,CAAC;qBAChC,CAAC,CAAC;oBACH,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;oBACzC,IAAI,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;oBACzC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,gBAAgB,EAAE,CAAC;wBACxE,OAAO,IAAI,8BAA8B,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI;4BAChE,cAAc,IAAI,CAAC,OAAO,CAAC,QAAQ,0BAA0B,CAAC;oBAClE,CAAC;oBACD,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBACjD,CAAC;gBACD,IAAI,OAAO,GAAG,yBAAyB,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7D,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,gBAAgB,EAAE,CAAC;oBACxE,OAAO,IAAI,+BAA+B,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI;wBACjE,cAAc,IAAI,CAAC,OAAO,CAAC,QAAQ,0BAA0B,CAAC;gBAClE,CAAC;gBACD,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YACxC,IAAI,aAAa,GAAG,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnD,MAAM,eAAe,GAAG,aAAa,KAAK,CAAC,CAAC;YAC5C,IAAI,eAAe,EAAE,CAAC;gBACpB,aAAa,GAAG,qBAAqB,CAAC,eAAe,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/E,CAAC;YACD,WAAW,IAAI,aAAa,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAC1B,KAAK,EAAE,UAAU;gBACjB,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,MAAM,CAAC,YAAY;gBAClC,eAAe,EAAE,SAAS,CAAC,MAAM;gBACjC,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACvD,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/C,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,cAAc,EAAE,aAAa;gBAC7B,gBAAgB,EAAE,eAAe;gBACjC,YAAY,EAAE,WAAW;gBACzB,WAAW,EAAE,UAAU;gBACvB,cAAc,EAAE,aAAa;gBAC7B,WAAW,EAAE,UAAU;aACxB,CAAC,CAAC;YACH,IAAI,UAAU,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE;oBAClC,KAAK,EAAE,UAAU;oBACjB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,8CAA8C;iBACvD,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;oBAChC,MAAM,EAAE,WAAW,IAAI,sBAAsB;oBAC7C,UAAU;oBACV,aAAa;oBACb,WAAW;oBACX,SAAS;iBACV,CAAC,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;gBACzC,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7D,CAAC;YAED,MAAM,iBAAiB,GAAG,WAAW,CAAC,YAAY,CAChD,WAAW,EACX,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CACvC,CAAC;YACF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,iBAAiB,KAAK,IAAI,EAAE,CAAC;gBACvD,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE;oBAClC,KAAK,EAAE,UAAU;oBACjB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,iBAAiB;iBAC1B,CAAC,CAAC;gBACH,WAAW,GAAG,iBAAiB,CAAC;gBAChC,SAAS;YACX,CAAC;YAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;YAC9C,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC/C,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE;oBAClC,KAAK,EAAE,UAAU;oBACjB,UAAU,EAAE,SAAS;oBACrB,MAAM,EAAE,iCAAiC;iBAC1C,CAAC,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,iCAAiC,CAAC,CAAC;YACxE,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YACpF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3B,IAAI,MAAM,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;oBAC5B,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,+BAA+B,CAAC,CAAC;gBAClE,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE;oBAC9B,KAAK,EAAE,UAAU;oBACjB,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;iBAC1C,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC,OAAO,CAAC;YACxB,CAAC;YAED,UAAU,IAAI,CAAC,CAAC;YAChB,OAAO,EAAE,YAAY,EAAE,EAAE,CAAC;YAC1B,MAAM,WAAW,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC;gBAC1D,SAAS;gBACT,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;gBACzC,WAAW;gBACX,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,UAAU,CAAC;gBAC9D,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC;aACnE,CAAC,CAAC,CAAC,OAAO,CAAC;YACZ,MAAM,QAAQ,GAAG,WAAW,CAAC,uBAAuB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAC7E,OAAO,CAAC,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAClD,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACxB,MAAM,YAAY,GAAa,EAAE,CAAC;YAClC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;gBACrC,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBAC1C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAChC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,uBAAuB,CAAC,YAAY,CAAC,CAAC;YACnD,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;gBACtB,WAAW,GAAG,gCAAgC,QAAQ,CAAC,IAAI,YAAY;oBACrE,GAAG,QAAQ,CAAC,KAAK,4CAA4C,CAAC;YAClE,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC;YAC5C,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,KAAK,SAAS,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/F,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,YAAY,CAAC,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,uCAAuC,CAAC,CAAC;gBAC9E,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;oBACxB,OAAO,EAAE,YAAY,CAAC,OAAO;oBAC7B,iBAAiB,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;iBACtD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,KAA2B,EAAE,UAAkB;QACpE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACtB,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,UAAU,CAAC;YAC3C,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC;SACrC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CACvB,KAAiC,EACjC,UAAkB;QAElB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QACjF,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;YACvB,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,UAAU,CAAC;YAC3C,MAAM,EAAE,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC;gBAC5C,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;YAC/C,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC;SACjC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CACtB,KAA2B,EAC3B,UAAkB;QAElB,OAAO;YACL,KAAK,EAAE,UAAU;YACjB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,UAAU,EAAE,KAAK,CAAC,SAAS;YAC3B,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE;YAC/B,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI;SACnC,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,OAOpB;QACC,MAAM,OAAO,GAA4B;YACvC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,WAAW,EAAE,OAAO,CAAC,UAAU;YAC/B,cAAc,EAAE,OAAO,CAAC,aAAa;YACrC,YAAY,EAAE,OAAO,CAAC,WAAW;YACjC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC;SAC9D,CAAC;QACF,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,OAAO,CAAC,aAAa,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;QAC9C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,IAAI,CAAC,SAAiB,EAAE,OAAgC;QAC9D,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC;IAEO,WAAW;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,KAAK,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;IACrF,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,IAAI,WAAW,CAAC,CAAC;QACtF,CAAC;IACH,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvE,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,MAAM,KAAK,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;IACrC,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,CAAC,IAAI,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACrE,MAAM,MAAM,GAAG,MAAM,CAAC,mBAAmB,CAAC,IAAI,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAC3E,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,MAAM,KAAK,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;QACpC,IAAI,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACtC,GAAG,IAAI,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAwC,EACxC,QAAiC;IAEjC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC3D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AAClF,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAgC;IACzD,IAAI,OAAO,GAAG,oEAAoE;QAChF,WAAW,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI;QACxC,gBAAgB,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,IAAI;QAClD,mBAAmB,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI;QACxD,mBAAmB,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,IAAI;QACtD,YAAY,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC;IACjD,IAAI,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,0BAA0B,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC;IACxE,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,aAAa,CAAC,IAA6B;IAClD,MAAM,IAAI,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;IAClD,KAAK,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;QACtD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,KAAK,CAAC,MAAM,SAAS,CAAC;QACxC,CAAC;IACH,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,KAAK,CAAC,MAAM,SAAS,CAAC;IAC5C,CAAC;IACD,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAC5E,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,UAAU,CAAC,MAAkB;IACpC,MAAM,IAAI,GAA4B,EAAE,CAAC;IACzC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAC1E,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa,EAAE,KAAK,GAAG,KAAK;IACpD,OAAO,KAAK,CAAC,MAAM,IAAI,KAAK;QAC1B,CAAC,CAAC,KAAK;QACP,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,mBAAmB,KAAK,CAAC,MAAM,GAAG,KAAK,SAAS,CAAC;AAC/E,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC"}

package/dist/runtime/agent-turn-loop.d.ts

@@ -0,0 +1,36 @@
+import { type CancelToken } from "../cancellation.ts";
+import { TaskLifecycle } from "./task-lifecycle.ts";
+import type { QueueBridge, QueueInputBatch } from "./queue-bridge.ts";
+export interface AgentTurnContext {
+    readonly cancelToken: CancelToken;
+    setClaimedInput(batch: QueueInputBatch | null): void;
+}
+export type AgentTurnRunnerResult = string | null | Promise<string | null>;
+export interface AgentTurnRunnerLike<Context extends AgentTurnContext> {
+    run(userInput: string, context: Context): AgentTurnRunnerResult;
+}
+export type AgentTurnRunner<Context extends AgentTurnContext> = ((content: string, context: Context) => AgentTurnRunnerResult) | AgentTurnRunnerLike<Context>;
+export type TurnLoopResult = {
+    readonly status: "completed";
+    readonly result: string | null;
+} | {
+    readonly status: "cancelled";
+} | {
+    readonly status: "failed";
+    readonly error: string;
+};
+export interface AgentTurnLoopOptions<Context extends AgentTurnContext> {
+    readonly lifecycle: TaskLifecycle;
+    readonly bridge: QueueBridge;
+    readonly runner: AgentTurnRunner<Context>;
+}
+/** Runs consecutive model turns until the task reaches a terminal state. */
+export declare class AgentTurnLoop<Context extends AgentTurnContext> {
+    private readonly lifecycle;
+    private readonly bridge;
+    private readonly runner;
+    constructor(options: AgentTurnLoopOptions<Context>);
+    run(taskId: string, content: string, context: Context): Promise<TurnLoopResult>;
+    private invokeRunner;
+    private finishCancelled;
+}

package/dist/runtime/agent-turn-loop.js

@@ -0,0 +1,71 @@
+import { CancellationError } from "../cancellation.js";
+import { TaskState } from "./task-lifecycle.js";
+/** Runs consecutive model turns until the task reaches a terminal state. */
+export class AgentTurnLoop {
+    lifecycle;
+    bridge;
+    runner;
+    constructor(options) {
+        this.lifecycle = options.lifecycle;
+        this.bridge = options.bridge;
+        this.runner = options.runner;
+    }
+    async run(taskId, content, context) {
+        let currentInput = content;
+        let currentBatch = null;
+        try {
+            for (;;) {
+                context.cancelToken.throwIfCancelled();
+                this.lifecycle.markModelRunning(taskId);
+                context.setClaimedInput(currentBatch);
+                const result = await this.invokeRunner(currentInput, context);
+                if (currentBatch !== null) {
+                    this.bridge.acknowledgeClaimedInput(taskId, currentBatch.eventIds);
+                }
+                context.cancelToken.throwIfCancelled();
+                // Drain or complete synchronously so a boundary cancellation wins.
+                const batch = this.bridge.drainPending(taskId);
+                if (batch !== null) {
+                    currentInput = batch.content;
+                    currentBatch = batch;
+                    continue;
+                }
+                const record = this.lifecycle.taskRegistry.get(taskId);
+                if (record === null ||
+                    record.state === TaskState.Cancelling ||
+                    record.cancelToken.isCancelled()) {
+                    return this.finishCancelled(taskId);
+                }
+                this.lifecycle.completeTask(taskId, result);
+                return { status: "completed", result };
+            }
+        }
+        catch (error) {
+            if (error instanceof CancellationError) {
+                return this.finishCancelled(taskId);
+            }
+            const record = this.lifecycle.taskRegistry.get(taskId);
+            if (record !== null && record.cancelToken.isCancelled()) {
+                return this.finishCancelled(taskId);
+            }
+            const message = error instanceof Error ? error.message : String(error);
+            this.bridge.preserveTaskInputs(taskId, "held because its task failed");
+            this.lifecycle.failTask(taskId, message);
+            return { status: "failed", error: message };
+        }
+    }
+    invokeRunner(content, context) {
+        const runner = this.runner;
+        const fn = typeof runner === "function" ? runner : runner.run.bind(runner);
+        return Promise.resolve(fn(content, context)).then((result) => result ?? null);
+    }
+    finishCancelled(taskId) {
+        const record = this.lifecycle.taskRegistry.get(taskId);
+        if (record !== null && record.state === TaskState.Cancelling) {
+            this.bridge.preserveTaskInputs(taskId, "held because its task was cancelled");
+            this.lifecycle.finishCancelled(taskId);
+        }
+        return { status: "cancelled" };
+    }
+}
+//# sourceMappingURL=agent-turn-loop.js.map

package/dist/runtime/agent-turn-loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-turn-loop.js","sourceRoot":"","sources":["../../src/runtime/agent-turn-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAoB,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAiB,SAAS,EAAE,MAAM,qBAAqB,CAAC;AA6B/D,4EAA4E;AAC5E,MAAM,OAAO,aAAa;IACP,SAAS,CAAgB;IACzB,MAAM,CAAc;IACpB,MAAM,CAA2B;IAElD,YAAY,OAAsC;QAChD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,GAAG,CACP,MAAc,EACd,OAAe,EACf,OAAgB;QAEhB,IAAI,YAAY,GAAG,OAAO,CAAC;QAC3B,IAAI,YAAY,GAA2B,IAAI,CAAC;QAChD,IAAI,CAAC;YACH,SAAS,CAAC;gBACR,OAAO,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC;gBACvC,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;gBACxC,OAAO,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;gBACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;gBAC9D,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;oBAC1B,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,MAAM,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACrE,CAAC;gBACD,OAAO,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC;gBAEvC,mEAAmE;gBACnE,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;gBAC/C,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;oBACnB,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;oBAC7B,YAAY,GAAG,KAAK,CAAC;oBACrB,SAAS;gBACX,CAAC;gBACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACvD,IACE,MAAM,KAAK,IAAI;oBACf,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,UAAU;oBACrC,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE,EAChC,CAAC;oBACD,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;gBACtC,CAAC;gBACD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC5C,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;YACzC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACvD,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,8BAA8B,CAAC,CAAC;YACvE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACzC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,OAAe,EAAE,OAAgB;QACpD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,MAAM,EAAE,GAAG,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3E,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IAChF,CAAC;IAEO,eAAe,CAAC,MAAc;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,UAAU,EAAE,CAAC;YAC7D,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAC5B,MAAM,EACN,qCAAqC,CACtC,CAAC;YACF,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;IACjC,CAAC;CACF"}

package/dist/runtime/guard-policy.d.ts

@@ -0,0 +1,21 @@
+import type { AssembledToolCall } from "../model-stream.ts";
+import type { ToolResult } from "../tools.ts";
+export interface GuardPolicyOptions {
+    maxTotalTokens: number;
+    maxElapsedSeconds: number;
+    repeatedToolCallLimit: number;
+}
+export interface RepeatedToolCall {
+    name: string;
+    count: number;
+}
+/** Owns token, elapsed-time, and repeated-tool-call safety decisions. */
+export declare class GuardPolicy {
+    private readonly maxTotalTokens;
+    private readonly maxElapsedSeconds;
+    private readonly repeatedToolCallLimit;
+    private readonly repeatedCalls;
+    constructor(options: GuardPolicyOptions);
+    budgetReason(totalTokens: number, elapsedSeconds: number): string | null;
+    recordRepeatedToolCalls(toolCalls: readonly AssembledToolCall[], toolResults: readonly ToolResult[]): RepeatedToolCall | null;
+}

package/dist/runtime/guard-policy.js

@@ -0,0 +1,93 @@
+/** Owns token, elapsed-time, and repeated-tool-call safety decisions. */
+export class GuardPolicy {
+    maxTotalTokens;
+    maxElapsedSeconds;
+    repeatedToolCallLimit;
+    repeatedCalls = new Map();
+    constructor(options) {
+        this.maxTotalTokens = options.maxTotalTokens;
+        this.maxElapsedSeconds = options.maxElapsedSeconds;
+        this.repeatedToolCallLimit = options.repeatedToolCallLimit;
+    }
+    budgetReason(totalTokens, elapsedSeconds) {
+        if (totalTokens >= this.maxTotalTokens) {
+            return `token budget reached (${this.maxTotalTokens})`;
+        }
+        if (elapsedSeconds >= this.maxElapsedSeconds) {
+            return `elapsed time budget reached (${String(this.maxElapsedSeconds)} seconds)`;
+        }
+        return null;
+    }
+    recordRepeatedToolCalls(toolCalls, toolResults) {
+        let repeated = null;
+        const seen = new Set();
+        for (let index = 0; index < toolCalls.length; index += 1) {
+            const toolCall = toolCalls[index];
+            if (toolCall === undefined) {
+                continue;
+            }
+            let parsedArguments;
+            try {
+                parsedArguments = JSON.parse(toolCall.function.arguments);
+            }
+            catch {
+                parsedArguments = toolCall.function.arguments;
+            }
+            const fingerprint = stableStringify({
+                name: toolCall.function.name,
+                arguments: parsedArguments,
+                result: stableToolResult(toolResults[index]),
+            });
+            const count = (this.repeatedCalls.get(fingerprint) ?? 0) + 1;
+            this.repeatedCalls.set(fingerprint, count);
+            seen.add(fingerprint);
+            if (repeated === null || count > repeated.count) {
+                repeated = { name: toolCall.function.name, count };
+            }
+        }
+        for (const fingerprint of [...this.repeatedCalls.keys()]) {
+            if (!seen.has(fingerprint)) {
+                this.repeatedCalls.delete(fingerprint);
+            }
+        }
+        return repeated !== null && repeated.count >= this.repeatedToolCallLimit
+            ? repeated
+            : null;
+    }
+}
+function stableToolResult(value) {
+    if (Array.isArray(value)) {
+        return value.map((item) => stableToolResult(item));
+    }
+    if (value !== null && typeof value === "object") {
+        const result = {};
+        for (const [key, item] of Object.entries(value)) {
+            if (key !== "duration_ms") {
+                result[key] = stableToolResult(item);
+            }
+        }
+        return result;
+    }
+    return value;
+}
+function stableStringify(value) {
+    if (value === null) {
+        return "null";
+    }
+    if (Array.isArray(value)) {
+        return `[${value.map((item) => stableStringify(item)).join(",")}]`;
+    }
+    if (typeof value === "object") {
+        const record = value;
+        const parts = Object.keys(record)
+            .filter((key) => record[key] !== undefined)
+            .sort()
+            .map((key) => `${JSON.stringify(key)}:${stableStringify(record[key])}`);
+        return `{${parts.join(",")}}`;
+    }
+    if (typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
+        return JSON.stringify(value);
+    }
+    return JSON.stringify(String(value));
+}
+//# sourceMappingURL=guard-policy.js.map

package/dist/runtime/guard-policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"guard-policy.js","sourceRoot":"","sources":["../../src/runtime/guard-policy.ts"],"names":[],"mappings":"AAcA,yEAAyE;AACzE,MAAM,OAAO,WAAW;IACL,cAAc,CAAS;IACvB,iBAAiB,CAAS;IAC1B,qBAAqB,CAAS;IAC9B,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE3D,YAAY,OAA2B;QACrC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;QACnD,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAC7D,CAAC;IAED,YAAY,CAAC,WAAmB,EAAE,cAAsB;QACtD,IAAI,WAAW,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACvC,OAAO,yBAAyB,IAAI,CAAC,cAAc,GAAG,CAAC;QACzD,CAAC;QACD,IAAI,cAAc,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC7C,OAAO,gCAAgC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC;QACnF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,uBAAuB,CACrB,SAAuC,EACvC,WAAkC;QAElC,IAAI,QAAQ,GAA4B,IAAI,CAAC;QAC7C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,SAAS,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACzD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC3B,SAAS;YACX,CAAC;YACD,IAAI,eAAwB,CAAC;YAC7B,IAAI,CAAC;gBACH,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC5D,CAAC;YAAC,MAAM,CAAC;gBACP,eAAe,GAAG,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;YAChD,CAAC;YACD,MAAM,WAAW,GAAG,eAAe,CAAC;gBAClC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;gBAC5B,SAAS,EAAE,eAAe;gBAC1B,MAAM,EAAE,gBAAgB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;aAC7C,CAAC,CAAC;YACH,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAC7D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC3C,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YACtB,IAAI,QAAQ,KAAK,IAAI,IAAI,KAAK,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAChD,QAAQ,GAAG,EAAE,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;YACrD,CAAC;QACH,CAAC;QACD,KAAK,MAAM,WAAW,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACzD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,KAAK,IAAI,IAAI,QAAQ,CAAC,KAAK,IAAI,IAAI,CAAC,qBAAqB;YACtE,CAAC,CAAC,QAAQ;YACV,CAAC,CAAC,IAAI,CAAC;IACX,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAChD,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC3E,IAAI,GAAG,KAAK,aAAa,EAAE,CAAC;gBAC1B,MAAM,CAAC,GAAG,CAAC,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACrE,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,KAAgC,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;aAC9B,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;aAC1C,IAAI,EAAE;aACN,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1E,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QACzF,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AACvC,CAAC"}

package/dist/runtime/history-committer.d.ts

@@ -0,0 +1,34 @@
+import type { CancelToken } from "../cancellation.ts";
+import type { AssembledToolCall } from "../model-stream.ts";
+import type { ToolResult } from "../tools.ts";
+export type HistoryMessage = Record<string, unknown>;
+export interface PendingInputBatchLike {
+    content?: string | undefined;
+    eventIds?: readonly string[] | undefined;
+}
+export interface HistoryCommitContext {
+    commitInput?(append: () => void, rollback: () => void): boolean;
+    commitPending?(batch: PendingInputBatchLike, append: () => void, rollback: () => void): boolean;
+    commitIfActive?(callback: () => void): boolean;
+}
+export interface HistoryCommitterOptions {
+    messages: HistoryMessage[];
+    context: HistoryCommitContext | null;
+    cancelToken: CancelToken | null;
+    createCancelled(message: string): Error;
+}
+/** Commits canonical conversation history while preserving session atomicity. */
+export declare class HistoryCommitter {
+    private readonly messages;
+    private readonly context;
+    private readonly cancelToken;
+    private readonly createCancelled;
+    constructor(options: HistoryCommitterOptions);
+    commitInput(content: string): boolean;
+    commitPending(batch: PendingInputBatchLike): boolean;
+    commitAssistant(message: HistoryMessage): boolean;
+    commitToolResults(toolCalls: readonly AssembledToolCall[], toolResults: readonly ToolResult[]): void;
+    snapshot(): HistoryMessage[];
+    private commitContextMessage;
+    private raiseIfCancelled;
+}