@nervekit/agent 0.1.0

package/dist/runtime/loop/tool-execution.js

@@ -0,0 +1,271 @@
+import { validateToolArguments, } from "@earendil-works/pi-ai";
+import { isAgentToolSuspension } from "../../suspension.js";
+/**
+ * Execute tool calls from an assistant message.
+ */
+export async function executeToolCalls(currentContext, assistantMessage, config, signal, emit) {
+    const toolCalls = assistantMessage.content.filter((c) => c.type === "toolCall");
+    const hasSequentialToolCall = toolCalls.some((tc) => currentContext.tools?.find((t) => t.name === tc.name)?.executionMode ===
+        "sequential");
+    if (config.toolExecution === "sequential" || hasSequentialToolCall) {
+        return executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit);
+    }
+    return executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit);
+}
+export async function executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit) {
+    const finalizedCalls = [];
+    const messages = [];
+    for (const toolCall of toolCalls) {
+        await emit({
+            type: "tool_execution_start",
+            toolCallId: toolCall.id,
+            toolName: toolCall.name,
+            args: toolCall.arguments,
+        });
+        const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
+        let finalized;
+        if (preparation.kind === "immediate") {
+            finalized = {
+                toolCall,
+                result: preparation.result,
+                isError: preparation.isError,
+            };
+        }
+        else {
+            try {
+                const executed = await executePreparedToolCall(preparation, signal, emit);
+                finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
+            }
+            catch (error) {
+                if (isAgentToolSuspension(error)) {
+                    const index = toolCalls.findIndex((candidate) => candidate.id === toolCall.id);
+                    throw error.withContext({
+                        assistantMessage,
+                        toolCall,
+                        remainingToolCalls: index === -1 ? [] : toolCalls.slice(index + 1),
+                    });
+                }
+                throw error;
+            }
+        }
+        await emitToolExecutionEnd(finalized, emit);
+        const toolResultMessage = createToolResultMessage(finalized);
+        await emitToolResultMessage(toolResultMessage, emit);
+        finalizedCalls.push(finalized);
+        messages.push(toolResultMessage);
+        if (signal?.aborted) {
+            break;
+        }
+    }
+    return {
+        messages,
+        terminate: shouldTerminateToolBatch(finalizedCalls),
+    };
+}
+export async function executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit) {
+    const finalizedCalls = [];
+    for (const toolCall of toolCalls) {
+        await emit({
+            type: "tool_execution_start",
+            toolCallId: toolCall.id,
+            toolName: toolCall.name,
+            args: toolCall.arguments,
+        });
+        const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
+        if (preparation.kind === "immediate") {
+            const finalized = {
+                toolCall,
+                result: preparation.result,
+                isError: preparation.isError,
+            };
+            await emitToolExecutionEnd(finalized, emit);
+            finalizedCalls.push(finalized);
+            if (signal?.aborted) {
+                break;
+            }
+            continue;
+        }
+        finalizedCalls.push(async () => {
+            const executed = await executePreparedToolCall(preparation, signal, emit);
+            const finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
+            await emitToolExecutionEnd(finalized, emit);
+            return finalized;
+        });
+        if (signal?.aborted) {
+            break;
+        }
+    }
+    const orderedFinalizedCalls = await Promise.all(finalizedCalls.map((entry) => typeof entry === "function" ? entry() : Promise.resolve(entry)));
+    const messages = [];
+    for (const finalized of orderedFinalizedCalls) {
+        const toolResultMessage = createToolResultMessage(finalized);
+        await emitToolResultMessage(toolResultMessage, emit);
+        messages.push(toolResultMessage);
+    }
+    return {
+        messages,
+        terminate: shouldTerminateToolBatch(orderedFinalizedCalls),
+    };
+}
+function shouldTerminateToolBatch(finalizedCalls) {
+    return (finalizedCalls.length > 0 &&
+        finalizedCalls.every((finalized) => finalized.result.terminate === true));
+}
+function prepareToolCallArguments(tool, toolCall) {
+    if (!tool.prepareArguments) {
+        return toolCall;
+    }
+    const preparedArguments = tool.prepareArguments(toolCall.arguments);
+    if (preparedArguments === toolCall.arguments) {
+        return toolCall;
+    }
+    return {
+        ...toolCall,
+        arguments: preparedArguments,
+    };
+}
+async function prepareToolCall(currentContext, assistantMessage, toolCall, config, signal) {
+    const tool = currentContext.tools?.find((t) => t.name === toolCall.name);
+    if (!tool) {
+        return {
+            kind: "immediate",
+            result: createErrorToolResult(`Tool ${toolCall.name} not found`),
+            isError: true,
+        };
+    }
+    try {
+        const preparedToolCall = prepareToolCallArguments(tool, toolCall);
+        const validatedArgs = validateToolArguments(tool, preparedToolCall);
+        if (config.beforeToolCall) {
+            const beforeResult = await config.beforeToolCall({
+                assistantMessage,
+                toolCall,
+                args: validatedArgs,
+                context: currentContext,
+            }, signal);
+            if (signal?.aborted) {
+                return {
+                    kind: "immediate",
+                    result: createErrorToolResult("Operation aborted"),
+                    isError: true,
+                };
+            }
+            if (beforeResult?.block) {
+                return {
+                    kind: "immediate",
+                    result: createErrorToolResult(beforeResult.reason || "Tool execution was blocked"),
+                    isError: true,
+                };
+            }
+        }
+        if (signal?.aborted) {
+            return {
+                kind: "immediate",
+                result: createErrorToolResult("Operation aborted"),
+                isError: true,
+            };
+        }
+        return {
+            kind: "prepared",
+            toolCall,
+            tool,
+            args: validatedArgs,
+        };
+    }
+    catch (error) {
+        return {
+            kind: "immediate",
+            result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
+            isError: true,
+        };
+    }
+}
+async function executePreparedToolCall(prepared, signal, emit) {
+    const updateEvents = [];
+    try {
+        const result = await prepared.tool.execute(prepared.toolCall.id, prepared.args, signal, (partialResult) => {
+            updateEvents.push(Promise.resolve(emit({
+                type: "tool_execution_update",
+                toolCallId: prepared.toolCall.id,
+                toolName: prepared.toolCall.name,
+                args: prepared.toolCall.arguments,
+                partialResult,
+            })));
+        });
+        await Promise.all(updateEvents);
+        return { result, isError: false };
+    }
+    catch (error) {
+        await Promise.all(updateEvents);
+        if (isAgentToolSuspension(error))
+            throw error;
+        return {
+            result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
+            isError: true,
+        };
+    }
+}
+async function finalizeExecutedToolCall(currentContext, assistantMessage, prepared, executed, config, signal) {
+    let result = executed.result;
+    let isError = executed.isError;
+    if (config.afterToolCall) {
+        try {
+            const afterResult = await config.afterToolCall({
+                assistantMessage,
+                toolCall: prepared.toolCall,
+                args: prepared.args,
+                result,
+                isError,
+                context: currentContext,
+            }, signal);
+            if (afterResult) {
+                result = {
+                    content: afterResult.content ?? result.content,
+                    details: afterResult.details ?? result.details,
+                    terminate: afterResult.terminate ?? result.terminate,
+                };
+                isError = afterResult.isError ?? isError;
+            }
+        }
+        catch (error) {
+            result = createErrorToolResult(error instanceof Error ? error.message : String(error));
+            isError = true;
+        }
+    }
+    return {
+        toolCall: prepared.toolCall,
+        result,
+        isError,
+    };
+}
+function createErrorToolResult(message) {
+    return {
+        content: [{ type: "text", text: message }],
+        details: {},
+    };
+}
+async function emitToolExecutionEnd(finalized, emit) {
+    await emit({
+        type: "tool_execution_end",
+        toolCallId: finalized.toolCall.id,
+        toolName: finalized.toolCall.name,
+        result: finalized.result,
+        isError: finalized.isError,
+    });
+}
+function createToolResultMessage(finalized) {
+    return {
+        role: "toolResult",
+        toolCallId: finalized.toolCall.id,
+        toolName: finalized.toolCall.name,
+        content: finalized.result.content,
+        details: finalized.result.details,
+        isError: finalized.isError,
+        timestamp: Date.now(),
+    };
+}
+async function emitToolResultMessage(toolResultMessage, emit) {
+    await emit({ type: "message_start", message: toolResultMessage });
+    await emit({ type: "message_end", message: toolResultMessage });
+}
+//# sourceMappingURL=tool-execution.js.map

package/dist/runtime/loop/tool-execution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-execution.js","sourceRoot":"","sources":["../../../src/runtime/loop/tool-execution.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAU5D;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,cAA4B,EAC5B,gBAAkC,EAClC,MAAuB,EACvB,MAA+B,EAC/B,IAAoB;IAEpB,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAC7B,CAAC;IACF,MAAM,qBAAqB,GAAG,SAAS,CAAC,IAAI,CAC1C,CAAC,EAAE,EAAE,EAAE,CACL,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,aAAa;QACpE,YAAY,CACf,CAAC;IACF,IAAI,MAAM,CAAC,aAAa,KAAK,YAAY,IAAI,qBAAqB,EAAE,CAAC;QACnE,OAAO,0BAA0B,CAC/B,cAAc,EACd,gBAAgB,EAChB,SAAS,EACT,MAAM,EACN,MAAM,EACN,IAAI,CACL,CAAC;IACJ,CAAC;IACD,OAAO,wBAAwB,CAC7B,cAAc,EACd,gBAAgB,EAChB,SAAS,EACT,MAAM,EACN,MAAM,EACN,IAAI,CACL,CAAC;AACJ,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,cAA4B,EAC5B,gBAAkC,EAClC,SAA0B,EAC1B,MAAuB,EACvB,MAA+B,EAC/B,IAAoB;IAEpB,MAAM,cAAc,GAA+B,EAAE,CAAC;IACtD,MAAM,QAAQ,GAAwB,EAAE,CAAC;IAEzC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,IAAI,CAAC;YACT,IAAI,EAAE,sBAAsB;YAC5B,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,IAAI,EAAE,QAAQ,CAAC,SAAS;SACzB,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,eAAe,CACvC,cAAc,EACd,gBAAgB,EAChB,QAAQ,EACR,MAAM,EACN,MAAM,CACP,CAAC;QACF,IAAI,SAAmC,CAAC;QACxC,IAAI,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,SAAS,GAAG;gBACV,QAAQ;gBACR,MAAM,EAAE,WAAW,CAAC,MAAM;gBAC1B,OAAO,EAAE,WAAW,CAAC,OAAO;aAC7B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAC5C,WAAW,EACX,MAAM,EACN,IAAI,CACL,CAAC;gBACF,SAAS,GAAG,MAAM,wBAAwB,CACxC,cAAc,EACd,gBAAgB,EAChB,WAAW,EACX,QAAQ,EACR,MAAM,EACN,MAAM,CACP,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,MAAM,KAAK,GAAG,SAAS,CAAC,SAAS,CAC/B,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAC5C,CAAC;oBACF,MAAM,KAAK,CAAC,WAAW,CAAC;wBACtB,gBAAgB;wBAChB,QAAQ;wBACR,kBAAkB,EAAE,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;qBACnE,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,MAAM,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC5C,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC;QAC7D,MAAM,qBAAqB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACrD,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAEjC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ;QACR,SAAS,EAAE,wBAAwB,CAAC,cAAc,CAAC;KACpD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,cAA4B,EAC5B,gBAAkC,EAClC,SAA0B,EAC1B,MAAuB,EACvB,MAA+B,EAC/B,IAAoB;IAEpB,MAAM,cAAc,GAA6B,EAAE,CAAC;IAEpD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,IAAI,CAAC;YACT,IAAI,EAAE,sBAAsB;YAC5B,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,IAAI,EAAE,QAAQ,CAAC,SAAS;SACzB,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,eAAe,CACvC,cAAc,EACd,gBAAgB,EAChB,QAAQ,EACR,MAAM,EACN,MAAM,CACP,CAAC;QACF,IAAI,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,MAAM,SAAS,GAAG;gBAChB,QAAQ;gBACR,MAAM,EAAE,WAAW,CAAC,MAAM;gBAC1B,OAAO,EAAE,WAAW,CAAC,OAAO;aACM,CAAC;YACrC,MAAM,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC5C,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/B,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM;YACR,CAAC;YACD,SAAS;QACX,CAAC;QAED,cAAc,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;YAC7B,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;YAC1E,MAAM,SAAS,GAAG,MAAM,wBAAwB,CAC9C,cAAc,EACd,gBAAgB,EAChB,WAAW,EACX,QAAQ,EACR,MAAM,EACN,MAAM,CACP,CAAC;YACF,MAAM,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC5C,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QACH,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM;QACR,CAAC;IACH,CAAC;IAED,MAAM,qBAAqB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7C,cAAc,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC3B,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAC/D,CACF,CAAC;IACF,MAAM,QAAQ,GAAwB,EAAE,CAAC;IACzC,KAAK,MAAM,SAAS,IAAI,qBAAqB,EAAE,CAAC;QAC9C,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC;QAC7D,MAAM,qBAAqB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACrD,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACnC,CAAC;IAED,OAAO;QACL,QAAQ;QACR,SAAS,EAAE,wBAAwB,CAAC,qBAAqB,CAAC;KAC3D,CAAC;AACJ,CAAC;AA8BD,SAAS,wBAAwB,CAC/B,cAA0C;IAE1C,OAAO,CACL,cAAc,CAAC,MAAM,GAAG,CAAC;QACzB,cAAc,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,KAAK,IAAI,CAAC,CACzE,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAC/B,IAAe,EACf,QAAuB;IAEvB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC3B,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpE,IAAI,iBAAiB,KAAK,QAAQ,CAAC,SAAS,EAAE,CAAC;QAC7C,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO;QACL,GAAG,QAAQ;QACX,SAAS,EAAE,iBAA4C;KACxD,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,cAA4B,EAC5B,gBAAkC,EAClC,QAAuB,EACvB,MAAuB,EACvB,MAA+B;IAE/B,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;IACzE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,qBAAqB,CAAC,QAAQ,QAAQ,CAAC,IAAI,YAAY,CAAC;YAChE,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAClE,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;QACpE,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YAC1B,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,cAAc,CAC9C;gBACE,gBAAgB;gBAChB,QAAQ;gBACR,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,cAAc;aACxB,EACD,MAAM,CACP,CAAC;YACF,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,MAAM,EAAE,qBAAqB,CAAC,mBAAmB,CAAC;oBAClD,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YACD,IAAI,YAAY,EAAE,KAAK,EAAE,CAAC;gBACxB,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,MAAM,EAAE,qBAAqB,CAC3B,YAAY,CAAC,MAAM,IAAI,4BAA4B,CACpD;oBACD,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,MAAM,EAAE,qBAAqB,CAAC,mBAAmB,CAAC;gBAClD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QACD,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,QAAQ;YACR,IAAI;YACJ,IAAI,EAAE,aAAa;SACpB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,qBAAqB,CAC3B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,QAA0B,EAC1B,MAA+B,EAC/B,IAAoB;IAEpB,MAAM,YAAY,GAAoB,EAAE,CAAC;IAEzC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,CACxC,QAAQ,CAAC,QAAQ,CAAC,EAAE,EACpB,QAAQ,CAAC,IAAa,EACtB,MAAM,EACN,CAAC,aAAa,EAAE,EAAE;YAChB,YAAY,CAAC,IAAI,CACf,OAAO,CAAC,OAAO,CACb,IAAI,CAAC;gBACH,IAAI,EAAE,uBAAuB;gBAC7B,UAAU,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE;gBAChC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;gBAChC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;gBACjC,aAAa;aACd,CAAC,CACH,CACF,CAAC;QACJ,CAAC,CACF,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAChC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IACpC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAChC,IAAI,qBAAqB,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;QAC9C,OAAO;YACL,MAAM,EAAE,qBAAqB,CAC3B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,cAA4B,EAC5B,gBAAkC,EAClC,QAA0B,EAC1B,QAAiC,EACjC,MAAuB,EACvB,MAA+B;IAE/B,IAAI,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC7B,IAAI,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;IAE/B,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,aAAa,CAC5C;gBACE,gBAAgB;gBAChB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM;gBACN,OAAO;gBACP,OAAO,EAAE,cAAc;aACxB,EACD,MAAM,CACP,CAAC;YACF,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,GAAG;oBACP,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO;oBAC9C,OAAO,EAAE,WAAW,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO;oBAC9C,SAAS,EAAE,WAAW,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS;iBACrD,CAAC;gBACF,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,OAAO,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,qBAAqB,CAC5B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;YACF,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,MAAM;QACN,OAAO;KACR,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAe;IAC5C,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAC1C,OAAO,EAAE,EAAE;KACZ,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,oBAAoB,CACjC,SAAmC,EACnC,IAAoB;IAEpB,MAAM,IAAI,CAAC;QACT,IAAI,EAAE,oBAAoB;QAC1B,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,EAAE;QACjC,QAAQ,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI;QACjC,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,OAAO,EAAE,SAAS,CAAC,OAAO;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,uBAAuB,CAC9B,SAAmC;IAEnC,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,UAAU,EAAE,SAAS,CAAC,QAAQ,CAAC,EAAE;QACjC,QAAQ,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI;QACjC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,OAAO;QACjC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,OAAO;QACjC,OAAO,EAAE,SAAS,CAAC,OAAO;QAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,iBAAoC,EACpC,IAAoB;IAEpB,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC,CAAC;IAClE,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC,CAAC;AAClE,CAAC"}

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

@@ -0,0 +1,7 @@
+import type { AgentContext, AgentLoopConfig, AgentMessage, StreamFn } from "../../types.js";
+import type { AgentEventSink } from "./events.js";
+/**
+ * Main loop logic shared by agentLoop and agentLoopContinue.
+ */
+export declare function runLoop(initialContext: AgentContext, newMessages: AgentMessage[], initialConfig: AgentLoopConfig, signal: AbortSignal | undefined, emit: AgentEventSink, streamFn?: StreamFn): Promise<void>;
+//# sourceMappingURL=turn-loop.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"turn-loop.d.ts","sourceRoot":"","sources":["../../../src/runtime/loop/turn-loop.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,YAAY,EACZ,eAAe,EACf,YAAY,EACZ,QAAQ,EACT,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAIlD;;GAEG;AACH,wBAAsB,OAAO,CAC3B,cAAc,EAAE,YAAY,EAC5B,WAAW,EAAE,YAAY,EAAE,EAC3B,aAAa,EAAE,eAAe,EAC9B,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,IAAI,EAAE,cAAc,EACpB,QAAQ,CAAC,EAAE,QAAQ,GAClB,OAAO,CAAC,IAAI,CAAC,CAuGf"}

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

@@ -0,0 +1,84 @@
+import { streamAssistantResponse } from "./stream-assistant.js";
+import { executeToolCalls } from "./tool-execution.js";
+/**
+ * Main loop logic shared by agentLoop and agentLoopContinue.
+ */
+export async function runLoop(initialContext, newMessages, initialConfig, signal, emit, streamFn) {
+    let currentContext = initialContext;
+    let config = initialConfig;
+    let firstTurn = true;
+    let pendingMessages = (await config.getSteeringMessages?.()) || [];
+    while (true) {
+        if (!firstTurn) {
+            await emit({ type: "turn_start" });
+        }
+        else {
+            firstTurn = false;
+        }
+        if (pendingMessages.length > 0) {
+            for (const message of pendingMessages) {
+                await emit({ type: "message_start", message });
+                await emit({ type: "message_end", message });
+                currentContext.messages.push(message);
+                newMessages.push(message);
+            }
+            pendingMessages = [];
+        }
+        const message = await streamAssistantResponse(currentContext, config, signal, emit, streamFn);
+        newMessages.push(message);
+        if (message.stopReason === "error" || message.stopReason === "aborted") {
+            await emit({ type: "turn_end", message, toolResults: [] });
+            await emit({ type: "agent_end", messages: newMessages });
+            return;
+        }
+        const toolCalls = message.content.filter((c) => c.type === "toolCall");
+        const toolResults = [];
+        let hasMoreToolCalls = false;
+        if (toolCalls.length > 0) {
+            const executedToolBatch = await executeToolCalls(currentContext, message, config, signal, emit);
+            toolResults.push(...executedToolBatch.messages);
+            hasMoreToolCalls = !executedToolBatch.terminate;
+            for (const result of toolResults) {
+                currentContext.messages.push(result);
+                newMessages.push(result);
+            }
+        }
+        await emit({ type: "turn_end", message, toolResults });
+        const nextTurnContext = {
+            message,
+            toolResults,
+            context: currentContext,
+            newMessages,
+        };
+        const nextTurnSnapshot = await config.prepareNextTurn?.(nextTurnContext);
+        if (nextTurnSnapshot) {
+            currentContext = nextTurnSnapshot.context ?? currentContext;
+            config = {
+                ...config,
+                model: nextTurnSnapshot.model ?? config.model,
+                reasoning: nextTurnSnapshot.thinkingLevel === undefined
+                    ? config.reasoning
+                    : nextTurnSnapshot.thinkingLevel === "off"
+                        ? undefined
+                        : nextTurnSnapshot.thinkingLevel,
+            };
+        }
+        pendingMessages = (await config.getSteeringMessages?.()) || [];
+        if (pendingMessages.length > 0)
+            continue;
+        if (await config.shouldStopAfterTurn?.({
+            message,
+            toolResults,
+            context: currentContext,
+            newMessages,
+        })) {
+            await emit({ type: "agent_end", messages: newMessages });
+            return;
+        }
+        if (hasMoreToolCalls)
+            continue;
+        break;
+    }
+    await emit({ type: "agent_end", messages: newMessages });
+}
+//# sourceMappingURL=turn-loop.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"turn-loop.js","sourceRoot":"","sources":["../../../src/runtime/loop/turn-loop.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAC3B,cAA4B,EAC5B,WAA2B,EAC3B,aAA8B,EAC9B,MAA+B,EAC/B,IAAoB,EACpB,QAAmB;IAEnB,IAAI,cAAc,GAAG,cAAc,CAAC;IACpC,IAAI,MAAM,GAAG,aAAa,CAAC;IAC3B,IAAI,SAAS,GAAG,IAAI,CAAC;IACrB,IAAI,eAAe,GACjB,CAAC,MAAM,MAAM,CAAC,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;IAE/C,OAAO,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;QAED,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC/C,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC7C,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACtC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;YACD,eAAe,GAAG,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAC3C,cAAc,EACd,MAAM,EACN,MAAM,EACN,IAAI,EACJ,QAAQ,CACT,CAAC;QACF,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE1B,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACvE,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;YAC3D,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;YACzD,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QACvE,MAAM,WAAW,GAAwB,EAAE,CAAC;QAC5C,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAE7B,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,iBAAiB,GAAG,MAAM,gBAAgB,CAC9C,cAAc,EACd,OAAO,EACP,MAAM,EACN,MAAM,EACN,IAAI,CACL,CAAC;YACF,WAAW,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,gBAAgB,GAAG,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAEhD,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;gBACjC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAEvD,MAAM,eAAe,GAAG;YACtB,OAAO;YACP,WAAW;YACX,OAAO,EAAE,cAAc;YACvB,WAAW;SACZ,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,MAAM,CAAC,eAAe,EAAE,CAAC,eAAe,CAAC,CAAC;QACzE,IAAI,gBAAgB,EAAE,CAAC;YACrB,cAAc,GAAG,gBAAgB,CAAC,OAAO,IAAI,cAAc,CAAC;YAC5D,MAAM,GAAG;gBACP,GAAG,MAAM;gBACT,KAAK,EAAE,gBAAgB,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK;gBAC7C,SAAS,EACP,gBAAgB,CAAC,aAAa,KAAK,SAAS;oBAC1C,CAAC,CAAC,MAAM,CAAC,SAAS;oBAClB,CAAC,CAAC,gBAAgB,CAAC,aAAa,KAAK,KAAK;wBACxC,CAAC,CAAC,SAAS;wBACX,CAAC,CAAC,gBAAgB,CAAC,aAAa;aACvC,CAAC;QACJ,CAAC;QAED,eAAe,GAAG,CAAC,MAAM,MAAM,CAAC,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAC/D,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC;YAAE,SAAS;QAEzC,IACE,MAAM,MAAM,CAAC,mBAAmB,EAAE,CAAC;YACjC,OAAO;YACP,WAAW;YACX,OAAO,EAAE,cAAc;YACvB,WAAW;SACZ,CAAC,EACF,CAAC;YACD,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;YACzD,OAAO;QACT,CAAC;QAED,IAAI,gBAAgB;YAAE,SAAS;QAC/B,MAAM;IACR,CAAC;IAED,MAAM,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;AAC3D,CAAC"}

package/dist/runtime.d.ts

@@ -0,0 +1,63 @@
+import { type AssistantMessageEvent, type Message, type Model } from "@earendil-works/pi-ai";
+import type { ThinkingLevel } from "./types.js";
+export interface AgentModelSelection {
+    provider: string;
+    modelId: string;
+}
+export interface AgentModelInfo extends AgentModelSelection {
+    name: string;
+    reasoning: boolean;
+    supportedThinkingLevels: ThinkingLevel[];
+    contextWindow: number;
+    maxOutputTokens: number;
+}
+/**
+ * A user-defined model (custom provider or manually-added model). The
+ * orchestrator owns persistence; the runtime turns these into pi-ai `Model`
+ * objects. Custom providers supply explicit connection settings; manual models
+ * under built-in providers can inherit settings from pi-ai's provider catalog.
+ */
+export interface AgentCustomModel {
+    provider: string;
+    modelId: string;
+    name: string;
+    api?: string;
+    baseUrl?: string;
+    reasoning: boolean;
+    supportedThinkingLevels?: ThinkingLevel[];
+    thinkingLevelMap?: Record<string, string | null>;
+    input?: ("text" | "image")[];
+    cost?: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+    };
+    contextWindow?: number;
+    maxTokens?: number;
+    headers?: Record<string, string>;
+    compat?: Record<string, unknown>;
+}
+/**
+ * Register a source of user-defined models (custom providers / manual models).
+ * When set, resolution and listing functions consult it unless an explicit
+ * `customModels` argument is passed. The orchestrator wires this to its
+ * persisted provider catalog so unknown providers resolve correctly.
+ */
+export declare function setCustomModelProvider(provider: (() => AgentCustomModel[]) | undefined): void;
+export interface AgentPromptInput {
+    systemPrompt?: string;
+    messages: Message[];
+    model?: AgentModelSelection;
+    apiKey?: string;
+    headers?: Record<string, string>;
+    signal?: AbortSignal;
+}
+export declare function resolveAgentModel(selection?: AgentModelSelection, customModels?: AgentCustomModel[]): Model<string>;
+export declare function getAgentModelInfo(model: Model<string>): AgentModelInfo;
+/** Resolve the context window for a model selection (0 when unknown). */
+export declare function getModelContextWindow(selection?: AgentModelSelection, customModels?: AgentCustomModel[]): number;
+export declare function clampAgentThinkingLevel(selection: AgentModelSelection | undefined, requested: ThinkingLevel | undefined, customModels?: AgentCustomModel[]): ThinkingLevel;
+export declare function streamAgentPrompt(input: AgentPromptInput): AsyncIterable<AssistantMessageEvent>;
+export declare function listAvailableModels(customModels?: AgentCustomModel[]): AgentModelInfo[];
+//# sourceMappingURL=runtime.d.ts.map

package/dist/runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,qBAAqB,EAS1B,KAAK,OAAO,EACZ,KAAK,KAAK,EAGX,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAEhD,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAe,SAAQ,mBAAmB;IACzD,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;IACnB,uBAAuB,EAAE,aAAa,EAAE,CAAC;IACzC,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;;;;GAKG;AACH,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,OAAO,CAAC;IACnB,uBAAuB,CAAC,EAAE,aAAa,EAAE,CAAC;IAC1C,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,CAAC;IACjD,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;IAC7B,IAAI,CAAC,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AA6CD;;;;;GAKG;AACH,wBAAgB,sBAAsB,CACpC,QAAQ,EAAE,CAAC,MAAM,gBAAgB,EAAE,CAAC,GAAG,SAAS,GAC/C,IAAI,CAEN;AA6BD,MAAM,WAAW,gBAAgB;IAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,mBAAmB,CAAC;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAmED,wBAAgB,iBAAiB,CAC/B,SAAS,CAAC,EAAE,mBAAmB,EAC/B,YAAY,CAAC,EAAE,gBAAgB,EAAE,GAChC,KAAK,CAAC,MAAM,CAAC,CAEf;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,cAAc,CAYtE;AAED,yEAAyE;AACzE,wBAAgB,qBAAqB,CACnC,SAAS,CAAC,EAAE,mBAAmB,EAC/B,YAAY,CAAC,EAAE,gBAAgB,EAAE,GAChC,MAAM,CAIR;AAED,wBAAgB,uBAAuB,CACrC,SAAS,EAAE,mBAAmB,GAAG,SAAS,EAC1C,SAAS,EAAE,aAAa,GAAG,SAAS,EACpC,YAAY,CAAC,EAAE,gBAAgB,EAAE,GAChC,aAAa,CAGf;AAED,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,gBAAgB,GACtB,aAAa,CAAC,qBAAqB,CAAC,CAYtC;AAgBD,wBAAgB,mBAAmB,CACjC,YAAY,CAAC,EAAE,gBAAgB,EAAE,GAChC,cAAc,EAAE,CAalB"}

package/dist/runtime.js

@@ -0,0 +1,170 @@
+import { clampThinkingLevel, fauxAssistantMessage, getModel, getModels, getProviders, getSupportedThinkingLevels, registerFauxProvider, streamSimple, } from "@earendil-works/pi-ai";
+function templateForCustomModel(model) {
+    if (!isKnownProvider(model.provider))
+        return undefined;
+    return (getModel(model.provider, model.modelId) ??
+        getModels(model.provider)[0]);
+}
+function toPiModel(model) {
+    const template = model.api && model.baseUrl ? undefined : templateForCustomModel(model);
+    const api = model.api ?? template?.api;
+    const baseUrl = model.baseUrl ?? template?.baseUrl;
+    if (!api || !baseUrl)
+        return undefined;
+    return {
+        id: model.modelId,
+        name: model.name,
+        api,
+        provider: model.provider,
+        baseUrl,
+        reasoning: model.reasoning,
+        thinkingLevelMap: model.thinkingLevelMap ?? template?.thinkingLevelMap,
+        input: model.input ?? template?.input ?? ["text"],
+        cost: model.cost ??
+            template?.cost ?? {
+            input: 0,
+            output: 0,
+            cacheRead: 0,
+            cacheWrite: 0,
+        },
+        contextWindow: model.contextWindow ?? template?.contextWindow ?? 0,
+        maxTokens: model.maxTokens ?? template?.maxTokens ?? 0,
+        headers: { ...(template?.headers ?? {}), ...(model.headers ?? {}) },
+        compat: (model.compat ?? template?.compat),
+    };
+}
+let customModelProvider;
+/**
+ * Register a source of user-defined models (custom providers / manual models).
+ * When set, resolution and listing functions consult it unless an explicit
+ * `customModels` argument is passed. The orchestrator wires this to its
+ * persisted provider catalog so unknown providers resolve correctly.
+ */
+export function setCustomModelProvider(provider) {
+    customModelProvider = provider;
+}
+function activeCustomModels(explicit) {
+    return explicit ?? customModelProvider?.();
+}
+function findCustomModel(customModels, selection) {
+    if (!customModels || !selection)
+        return undefined;
+    return customModels.find((model) => model.provider === selection.provider &&
+        model.modelId === selection.modelId);
+}
+function customModelInfo(model) {
+    const resolved = toPiModel(model);
+    if (!resolved)
+        return undefined;
+    const info = getAgentModelInfo(resolved);
+    return model.supportedThinkingLevels
+        ? { ...info, supportedThinkingLevels: model.supportedThinkingLevels }
+        : info;
+}
+let fauxProvider;
+const fauxResponseFactory = (context) => {
+    const latest = [...context.messages]
+        .reverse()
+        .find((message) => message.role === "user");
+    const prompt = latest ? userMessageText(latest) : "";
+    return fauxAssistantMessage([
+        "I’m the temporary Nerve agent runtime.",
+        "",
+        prompt
+            ? `I received your prompt: “${prompt.slice(0, 240)}${prompt.length > 240 ? "…" : ""}”`
+            : "I did not receive a user prompt.",
+        "",
+        "The orchestrator, HTTP API, WebSocket event stream, and prompt plumbing are connected. Real provider execution can use a configured model/API key; otherwise this faux model keeps local development deterministic.",
+    ].join("\n"));
+};
+function getFauxProvider() {
+    if (!fauxProvider) {
+        fauxProvider = registerFauxProvider({
+            provider: "nerve-faux",
+            models: [{ id: "faux-fast", name: "Nerve Faux Fast" }],
+            tokensPerSecond: 80,
+            tokenSize: { min: 10, max: 22 },
+        });
+    }
+    return fauxProvider;
+}
+function userMessageText(message) {
+    if (typeof message.content === "string")
+        return message.content;
+    return message.content
+        .filter((block) => block.type === "text")
+        .map((block) => block.text)
+        .join("\n");
+}
+function isKnownProvider(provider) {
+    return getProviders().includes(provider);
+}
+function resolveAgentModelInternal(selection, appendFauxResponse, customModels) {
+    const custom = findCustomModel(activeCustomModels(customModels), selection);
+    const customResolved = custom ? toPiModel(custom) : undefined;
+    if (customResolved)
+        return customResolved;
+    if (selection && isKnownProvider(selection.provider)) {
+        return getModel(selection.provider, selection.modelId);
+    }
+    const faux = getFauxProvider();
+    if (appendFauxResponse)
+        faux.appendResponses([fauxResponseFactory]);
+    return faux.getModel();
+}
+export function resolveAgentModel(selection, customModels) {
+    return resolveAgentModelInternal(selection, true, customModels);
+}
+export function getAgentModelInfo(model) {
+    return {
+        provider: model.provider,
+        modelId: model.id,
+        name: model.name || model.id,
+        reasoning: model.reasoning,
+        supportedThinkingLevels: getSupportedThinkingLevels(model),
+        contextWindow: model.contextWindow ?? 0,
+        maxOutputTokens: model.maxTokens ?? 0,
+    };
+}
+/** Resolve the context window for a model selection (0 when unknown). */
+export function getModelContextWindow(selection, customModels) {
+    return (resolveAgentModelInternal(selection, false, customModels).contextWindow ?? 0);
+}
+export function clampAgentThinkingLevel(selection, requested, customModels) {
+    const model = resolveAgentModelInternal(selection, false, customModels);
+    return clampThinkingLevel(model, requested ?? "off");
+}
+export function streamAgentPrompt(input) {
+    const model = resolveAgentModel(input.model);
+    const context = {
+        systemPrompt: input.systemPrompt,
+        messages: input.messages,
+    };
+    return streamSimple(model, context, {
+        signal: input.signal,
+        apiKey: input.apiKey,
+        headers: input.headers,
+        maxTokens: 4096,
+    });
+}
+const COMPLETE_MODEL_LIST_PROVIDERS = new Set([
+    // Nerve supports Anthropic subscription OAuth, and newer Opus/Sonnet
+    // models sort after older Claude 3.x IDs. Truncating the provider list hides
+    // currently available subscription models such as claude-opus-4-8.
+    "anthropic",
+]);
+function visibleModelsForProvider(provider) {
+    const models = getModels(provider);
+    return COMPLETE_MODEL_LIST_PROVIDERS.has(provider)
+        ? models
+        : models.slice(0, 8);
+}
+export function listAvailableModels(customModels) {
+    const faux = getFauxProvider().models.map((model) => getAgentModelInfo(model));
+    const configured = getProviders().flatMap((provider) => visibleModelsForProvider(provider).map((model) => getAgentModelInfo(model)));
+    const custom = (activeCustomModels(customModels) ?? [])
+        .map((model) => customModelInfo(model))
+        .filter((model) => Boolean(model));
+    return [...faux, ...configured, ...custom];
+}
+//# sourceMappingURL=runtime.js.map

package/dist/runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,kBAAkB,EAClB,oBAAoB,EACpB,QAAQ,EACR,SAAS,EACT,YAAY,EACZ,0BAA0B,EAI1B,oBAAoB,EACpB,YAAY,GACb,MAAM,uBAAuB,CAAC;AA4C/B,SAAS,sBAAsB,CAC7B,KAAuB;IAEvB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC;QAAE,OAAO,SAAS,CAAC;IACvD,OAAO,CACJ,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAgB,CAElC;QACb,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAA+B,CAC5D,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,KAAuB;IACxC,MAAM,QAAQ,GACZ,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;IACzE,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,IAAI,QAAQ,EAAE,GAAG,CAAC;IACvC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,QAAQ,EAAE,OAAO,CAAC;IACnD,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO;QAAE,OAAO,SAAS,CAAC;IACvC,OAAO;QACL,EAAE,EAAE,KAAK,CAAC,OAAO;QACjB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,GAAG;QACH,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,OAAO;QACP,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB,IAAI,QAAQ,EAAE,gBAAgB;QACtE,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,CAAC,MAAM,CAAC;QACjD,IAAI,EAAE,KAAK,CAAC,IAAI;YACd,QAAQ,EAAE,IAAI,IAAI;YAChB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;SACd;QACH,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,QAAQ,EAAE,aAAa,IAAI,CAAC;QAClE,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,QAAQ,EAAE,SAAS,IAAI,CAAC;QACtD,OAAO,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE;QACnE,MAAM,EAAE,CAAC,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,MAAM,CAAU;KACnC,CAAC;AACrB,CAAC;AAED,IAAI,mBAA2D,CAAC;AAEhE;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAgD;IAEhD,mBAAmB,GAAG,QAAQ,CAAC;AACjC,CAAC;AAED,SAAS,kBAAkB,CACzB,QAA6B;IAE7B,OAAO,QAAQ,IAAI,mBAAmB,EAAE,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,eAAe,CACtB,YAA4C,EAC5C,SAA0C;IAE1C,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAClD,OAAO,YAAY,CAAC,IAAI,CACtB,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,QAAQ;QACrC,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,OAAO,CACtC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAuB;IAC9C,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAClC,IAAI,CAAC,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChC,MAAM,IAAI,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO,KAAK,CAAC,uBAAuB;QAClC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,uBAAuB,EAAE,KAAK,CAAC,uBAAuB,EAAE;QACrE,CAAC,CAAC,IAAI,CAAC;AACX,CAAC;AAWD,IAAI,YAAiE,CAAC;AAEtE,MAAM,mBAAmB,GAEV,CAAC,OAAO,EAAE,EAAE;IACzB,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;SACjC,OAAO,EAAE;SACT,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAC9C,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACrD,OAAO,oBAAoB,CACzB;QACE,wCAAwC;QACxC,EAAE;QACF,MAAM;YACJ,CAAC,CAAC,4BAA4B,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG;YACtF,CAAC,CAAC,kCAAkC;QACtC,EAAE;QACF,qNAAqN;KACtN,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;AACJ,CAAC,CAAC;AAEF,SAAS,eAAe;IACtB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,oBAAoB,CAAC;YAClC,QAAQ,EAAE,YAAY;YACtB,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;YACtD,eAAe,EAAE,EAAE;YACnB,SAAS,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;SAChC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,eAAe,CAAC,OAA2C;IAClE,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,OAAO,CAAC;IAChE,OAAO,OAAO,CAAC,OAAO;SACnB,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;SACxC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;SAC1B,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACvC,OAAQ,YAAY,EAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,yBAAyB,CAChC,SAA0C,EAC1C,kBAA2B,EAC3B,YAAiC;IAEjC,MAAM,MAAM,GAAG,eAAe,CAAC,kBAAkB,CAAC,YAAY,CAAC,EAAE,SAAS,CAAC,CAAC;IAC5E,MAAM,cAAc,GAAG,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9D,IAAI,cAAc;QAAE,OAAO,cAAc,CAAC;IAC1C,IAAI,SAAS,IAAI,eAAe,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,OAAO,QAAQ,CACb,SAAS,CAAC,QAAQ,EAClB,SAAS,CAAC,OAAgB,CACV,CAAC;IACrB,CAAC;IACD,MAAM,IAAI,GAAG,eAAe,EAAE,CAAC;IAC/B,IAAI,kBAAkB;QAAE,IAAI,CAAC,eAAe,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;IACpE,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,SAA+B,EAC/B,YAAiC;IAEjC,OAAO,yBAAyB,CAAC,SAAS,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAoB;IACpD,OAAO;QACL,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,OAAO,EAAE,KAAK,CAAC,EAAE;QACjB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE;QAC5B,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,uBAAuB,EAAE,0BAA0B,CACjD,KAAK,CACa;QACpB,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC;QACvC,eAAe,EAAE,KAAK,CAAC,SAAS,IAAI,CAAC;KACtC,CAAC;AACJ,CAAC;AAED,yEAAyE;AACzE,MAAM,UAAU,qBAAqB,CACnC,SAA+B,EAC/B,YAAiC;IAEjC,OAAO,CACL,yBAAyB,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,aAAa,IAAI,CAAC,CAC7E,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,SAA0C,EAC1C,SAAoC,EACpC,YAAiC;IAEjC,MAAM,KAAK,GAAG,yBAAyB,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;IACxE,OAAO,kBAAkB,CAAC,KAAK,EAAE,SAAS,IAAI,KAAK,CAAkB,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,KAAuB;IAEvB,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC7C,MAAM,OAAO,GAAY;QACvB,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,QAAQ,EAAE,KAAK,CAAC,QAAQ;KACzB,CAAC;IACF,OAAO,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE;QAClC,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,SAAS,EAAE,IAAI;KAChB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAS;IACpD,qEAAqE;IACrE,6EAA6E;IAC7E,mEAAmE;IACnE,WAAW;CACZ,CAAC,CAAC;AAEH,SAAS,wBAAwB,CAAC,QAAuB;IACvD,MAAM,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;IACnC,OAAO,6BAA6B,CAAC,GAAG,CAAC,QAAQ,CAAC;QAChD,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,YAAiC;IAEjC,MAAM,IAAI,GAAG,eAAe,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAClD,iBAAiB,CAAC,KAAK,CAAC,CACzB,CAAC;IACF,MAAM,UAAU,GAAG,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CACrD,wBAAwB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC/C,iBAAiB,CAAC,KAAsB,CAAC,CAC1C,CACF,CAAC;IACF,MAAM,MAAM,GAAG,CAAC,kBAAkB,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;SACpD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;SACtC,MAAM,CAAC,CAAC,KAAK,EAA2B,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG,IAAI,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC,CAAC;AAC7C,CAAC"}

package/dist/suspension.d.ts

@@ -0,0 +1,18 @@
+import type { AssistantMessage } from "@earendil-works/pi-ai";
+import type { AgentToolCall } from "./types.js";
+export type AgentToolSuspensionData = {
+    toolCallId: string;
+    toolName: string;
+    reason: string;
+    assistantMessage?: AssistantMessage;
+    toolCall?: AgentToolCall;
+    remainingToolCalls?: AgentToolCall[];
+};
+export declare class AgentToolSuspension extends Error {
+    readonly kind = "agent_tool_suspension";
+    readonly data: AgentToolSuspensionData;
+    constructor(data: AgentToolSuspensionData);
+    withContext(patch: Partial<Pick<AgentToolSuspensionData, "assistantMessage" | "toolCall" | "remainingToolCalls">>): AgentToolSuspension;
+}
+export declare function isAgentToolSuspension(value: unknown): value is AgentToolSuspension;
+//# sourceMappingURL=suspension.d.ts.map