@oni.bot/core 0.6.2 → 0.8.0

package/dist/models/openrouter.js

@@ -0,0 +1,322 @@
+/* ------------------------------------------------------------------ */
+/*  Message conversion                                                 */
+/* ------------------------------------------------------------------ */
+function contentToString(content) {
+    if (typeof content === "string")
+        return content;
+    return content
+        .filter((p) => p.type === "text")
+        .map((p) => p.text ?? "")
+        .join("");
+}
+function convertMessages(messages, systemPrompt) {
+    const converted = [];
+    if (systemPrompt) {
+        converted.push({ role: "system", content: systemPrompt });
+    }
+    for (const msg of messages) {
+        if (msg.role === "system") {
+            converted.push({ role: "system", content: contentToString(msg.content) });
+            continue;
+        }
+        if (msg.role === "tool") {
+            converted.push({
+                role: "tool",
+                content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
+                tool_call_id: msg.toolCallId,
+            });
+            continue;
+        }
+        if (msg.role === "assistant" && msg.toolCalls && msg.toolCalls.length > 0) {
+            const text = contentToString(msg.content);
+            const toolCalls = msg.toolCalls.map((tc) => ({
+                id: tc.id,
+                type: "function",
+                function: { name: tc.name, arguments: JSON.stringify(tc.args) },
+            }));
+            converted.push({
+                role: "assistant",
+                content: text || null,
+                tool_calls: toolCalls,
+            });
+            continue;
+        }
+        converted.push({
+            role: msg.role,
+            content: contentToString(msg.content),
+        });
+    }
+    return converted;
+}
+function convertTools(tools) {
+    return tools.map((t) => ({
+        type: "function",
+        function: {
+            name: t.name,
+            description: t.description,
+            parameters: t.parameters,
+        },
+    }));
+}
+function mapFinishReason(reason) {
+    if (reason === "tool_calls")
+        return "tool_use";
+    if (reason === "length")
+        return "max_tokens";
+    if (reason === "stop")
+        return "end";
+    return "end";
+}
+/* ------------------------------------------------------------------ */
+/*  SSE line parser                                                    */
+/* ------------------------------------------------------------------ */
+async function* parseSSE(body) {
+    const reader = body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = "";
+    try {
+        for (;;) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split("\n");
+            buffer = lines.pop();
+            for (const line of lines) {
+                if (line.startsWith("data: ")) {
+                    const data = line.slice(6).trim();
+                    if (data && data !== "[DONE]") {
+                        yield data;
+                    }
+                    if (data === "[DONE]")
+                        return;
+                }
+            }
+        }
+        if (buffer.trim()) {
+            const lines = buffer.split("\n");
+            for (const line of lines) {
+                if (line.startsWith("data: ")) {
+                    const data = line.slice(6).trim();
+                    if (data && data !== "[DONE]") {
+                        yield data;
+                    }
+                }
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+}
+/* ------------------------------------------------------------------ */
+/*  Factory                                                            */
+/* ------------------------------------------------------------------ */
+export function openrouter(modelId, opts) {
+    const apiKey = opts?.apiKey ?? process.env["OPENROUTER_API_KEY"] ?? "";
+    const baseUrl = (opts?.baseUrl ?? "https://openrouter.ai/api").replace(/\/$/, "");
+    const defaultMaxTokens = opts?.defaultMaxTokens;
+    const defaultTemperature = opts?.defaultTemperature;
+    const referer = opts?.referer ?? process.env["OPENROUTER_REFERER"] ?? "";
+    const appTitle = opts?.appTitle ?? process.env["OPENROUTER_APP_TITLE"] ?? "";
+    const providerPrefs = opts?.provider;
+    const transforms = opts?.transforms;
+    function headers() {
+        const h = {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${apiKey}`,
+        };
+        if (referer)
+            h["HTTP-Referer"] = referer;
+        if (appTitle)
+            h["X-Title"] = appTitle;
+        return h;
+    }
+    function buildBody(params, stream) {
+        const messages = convertMessages(params.messages, params.systemPrompt);
+        const body = {
+            model: modelId,
+            messages,
+        };
+        const maxTokens = params.maxTokens ?? defaultMaxTokens;
+        if (maxTokens !== undefined) {
+            body.max_tokens = maxTokens;
+        }
+        if (params.temperature !== undefined || defaultTemperature !== undefined) {
+            body.temperature = params.temperature ?? defaultTemperature;
+        }
+        if (params.stopSequences && params.stopSequences.length > 0) {
+            body.stop = params.stopSequences;
+        }
+        if (params.tools && params.tools.length > 0) {
+            body.tools = convertTools(params.tools);
+        }
+        if (providerPrefs) {
+            body.provider = providerPrefs;
+        }
+        if (transforms) {
+            body.transforms = transforms;
+        }
+        if (stream) {
+            body.stream = true;
+        }
+        return body;
+    }
+    /* ---- chat -------------------------------------------------------- */
+    async function chat(params) {
+        const body = buildBody(params, false);
+        const res = await fetch(`${baseUrl}/v1/chat/completions`, {
+            method: "POST",
+            headers: headers(),
+            body: JSON.stringify(body),
+            signal: params.signal,
+        });
+        if (!res.ok) {
+            const text = await res.text().catch(() => "");
+            throw new Error(`OpenRouter API error ${res.status}: ${text}`);
+        }
+        const json = (await res.json());
+        const choice = json.choices[0];
+        const message = choice.message;
+        const content = message.content ?? "";
+        let toolCalls;
+        if (message.tool_calls && message.tool_calls.length > 0) {
+            toolCalls = message.tool_calls.map((tc) => ({
+                id: tc.id,
+                name: tc.function.name,
+                args: (() => { try {
+                    return JSON.parse(tc.function.arguments);
+                }
+                catch {
+                    return {};
+                } })(),
+            }));
+        }
+        return {
+            content,
+            toolCalls,
+            usage: {
+                inputTokens: json.usage.prompt_tokens,
+                outputTokens: json.usage.completion_tokens,
+            },
+            stopReason: mapFinishReason(choice.finish_reason),
+            raw: json,
+        };
+    }
+    /* ---- stream ------------------------------------------------------ */
+    async function* stream(params) {
+        const body = buildBody(params, true);
+        const res = await fetch(`${baseUrl}/v1/chat/completions`, {
+            method: "POST",
+            headers: headers(),
+            body: JSON.stringify(body),
+            signal: params.signal,
+        });
+        if (!res.ok) {
+            const text = await res.text().catch(() => "");
+            throw new Error(`OpenRouter API error ${res.status}: ${text}`);
+        }
+        if (!res.body) {
+            throw new Error("OpenRouter API returned no body for stream");
+        }
+        const activeToolCalls = new Map();
+        for await (const data of parseSSE(res.body)) {
+            let parsed;
+            try {
+                parsed = JSON.parse(data);
+            }
+            catch {
+                continue;
+            }
+            const choices = parsed["choices"];
+            if (!choices || choices.length === 0)
+                continue;
+            const choice = choices[0];
+            const delta = choice["delta"];
+            if (!delta)
+                continue;
+            // Text content
+            if (delta["content"] && typeof delta["content"] === "string") {
+                yield { type: "text", text: delta["content"] };
+            }
+            // Tool calls
+            const toolCalls = delta["tool_calls"];
+            if (toolCalls) {
+                for (const tc of toolCalls) {
+                    const index = tc["index"];
+                    const fn = tc["function"];
+                    if (tc["id"]) {
+                        activeToolCalls.set(index, {
+                            id: tc["id"],
+                            name: fn?.["name"] ?? "",
+                            args: fn?.["arguments"] ?? "",
+                        });
+                        yield {
+                            type: "tool_call_start",
+                            toolCall: {
+                                id: tc["id"],
+                                name: fn?.["name"] ?? "",
+                                args: {},
+                            },
+                        };
+                    }
+                    else if (fn?.["arguments"]) {
+                        const active = activeToolCalls.get(index);
+                        if (active) {
+                            active.args += fn["arguments"];
+                        }
+                        yield {
+                            type: "tool_call_delta",
+                            toolCall: {
+                                args: { __partial: fn["arguments"] },
+                            },
+                        };
+                    }
+                }
+            }
+            // Finish reason → emit tool_call_end
+            const finishReason = choice["finish_reason"];
+            if (finishReason === "tool_calls") {
+                for (const [, active] of activeToolCalls) {
+                    let args = {};
+                    try {
+                        args = JSON.parse(active.args);
+                    }
+                    catch {
+                        // partial args
+                    }
+                    yield {
+                        type: "tool_call_end",
+                        toolCall: { id: active.id, name: active.name, args },
+                    };
+                }
+                activeToolCalls.clear();
+            }
+            // Usage
+            const usage = parsed["usage"];
+            if (usage) {
+                yield {
+                    type: "usage",
+                    usage: {
+                        inputTokens: usage.prompt_tokens ?? 0,
+                        outputTokens: usage.completion_tokens ?? 0,
+                    },
+                };
+            }
+        }
+    }
+    /* ---- model object ------------------------------------------------ */
+    return {
+        chat,
+        stream,
+        provider: "openrouter",
+        modelId,
+        capabilities: {
+            tools: true,
+            vision: true,
+            streaming: true,
+            embeddings: false,
+        },
+    };
+}
+//# sourceMappingURL=openrouter.js.map

package/dist/models/openrouter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openrouter.js","sourceRoot":"","sources":["../../src/models/openrouter.ts"],"names":[],"mappings":"AAoFA,wEAAwE;AACxE,yEAAyE;AACzE,wEAAwE;AAExE,SAAS,eAAe,CAAC,OAA+B;IACtD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC;IAChD,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;SACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,SAAS,eAAe,CACtB,QAA2B,EAC3B,YAAqB;IAErB,MAAM,SAAS,GAAoB,EAAE,CAAC;IAEtC,IAAI,YAAY,EAAE,CAAC;QACjB,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1E,SAAS;QACX,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,SAAS,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;gBACpF,YAAY,EAAE,GAAG,CAAC,UAAU;aAC7B,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,GAAG,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC1C,MAAM,SAAS,GAAqB,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC7D,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE;aAChE,CAAC,CAAC,CAAC;YACJ,SAAS,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI,IAAI,IAAI;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,GAAG,CAAC,IAA4B;YACtC,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC;SACtC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,YAAY,CAAC,KAAmB;IACvC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACvB,IAAI,EAAE,UAAmB;QACzB,QAAQ,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;SACzB;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,eAAe,CACtB,MAAqB;IAErB,IAAI,MAAM,KAAK,YAAY;QAAE,OAAO,UAAU,CAAC;IAC/C,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,YAAY,CAAC;IAC7C,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,KAAK,CAAC;IACpC,OAAO,KAAK,CAAC;AACf,CAAC;AAED,wEAAwE;AACxE,yEAAyE;AACzE,wEAAwE;AAExE,KAAK,SAAS,CAAC,CAAC,QAAQ,CACtB,IAAgC;IAEhC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,SAAS,CAAC;YACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAChB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAElD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAG,CAAC;YAEtB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAClC,IAAI,IAAI,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;wBAC9B,MAAM,IAAI,CAAC;oBACb,CAAC;oBACD,IAAI,IAAI,KAAK,QAAQ;wBAAE,OAAO;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAClB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBAClC,IAAI,IAAI,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;wBAC9B,MAAM,IAAI,CAAC;oBACb,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC;AAED,wEAAwE;AACxE,yEAAyE;AACzE,wEAAwE;AAExE,MAAM,UAAU,UAAU,CACxB,OAAe,EACf,IAAwB;IAExB,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,EAAE,CAAC;IACvE,MAAM,OAAO,GAAG,CAAC,IAAI,EAAE,OAAO,IAAI,2BAA2B,CAAC,CAAC,OAAO,CACpE,KAAK,EACL,EAAE,CACH,CAAC;IACF,MAAM,gBAAgB,GAAG,IAAI,EAAE,gBAAgB,CAAC;IAChD,MAAM,kBAAkB,GAAG,IAAI,EAAE,kBAAkB,CAAC;IACpD,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,EAAE,CAAC;IACzE,MAAM,QAAQ,GAAG,IAAI,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,EAAE,CAAC;IAC7E,MAAM,aAAa,GAAG,IAAI,EAAE,QAAQ,CAAC;IACrC,MAAM,UAAU,GAAG,IAAI,EAAE,UAAU,CAAC;IAEpC,SAAS,OAAO;QACd,MAAM,CAAC,GAA2B;YAChC,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,MAAM,EAAE;SACpC,CAAC;QACF,IAAI,OAAO;YAAE,CAAC,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC;QACzC,IAAI,QAAQ;YAAE,CAAC,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;QACtC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,SAAS,SAAS,CAAC,MAAkB,EAAE,MAAe;QACpD,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAEvE,MAAM,IAAI,GAA0B;YAClC,KAAK,EAAE,OAAO;YACd,QAAQ;SACT,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,gBAAgB,CAAC;QACvD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QAED,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;YACzE,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,kBAAkB,CAAC;QAC9D,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5D,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC;QACnC,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,GAAG,aAAa,CAAC;QAChC,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,wEAAwE;IAExE,KAAK,UAAU,IAAI,CAAC,MAAkB;QACpC,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,sBAAsB,EAAE;YACxD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,OAAO,EAAE;YAClB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAE,CAAC;QAChC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAE/B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;QAEtC,IAAI,SAAyC,CAAC;QAC9C,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC1C,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBACtB,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;oBAAC,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAAC,CAAC;gBAAC,MAAM,CAAC;oBAAC,OAAO,EAAE,CAAC;gBAAC,CAAC,CAAC,CAAC,CAAC,EAA6B;aACtH,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO;YACL,OAAO;YACP,SAAS;YACT,KAAK,EAAE;gBACL,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;gBACrC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;aAC3C;YACD,UAAU,EAAE,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC;YACjD,GAAG,EAAE,IAAI;SACV,CAAC;IACJ,CAAC;IAED,wEAAwE;IAExE,KAAK,SAAS,CAAC,CAAC,MAAM,CAAC,MAAkB;QACvC,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAErC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,sBAAsB,EAAE;YACxD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,OAAO,EAAE;YAClB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,GAAG,EAAsD,CAAC;QAEtF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,MAA+B,CAAC;YACpC,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B,CAAC;YACvD,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAA+C,CAAC;YAChF,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAE/C,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAwC,CAAC;YACrE,IAAI,CAAC,KAAK;gBAAE,SAAS;YAErB,eAAe;YACf,IAAI,KAAK,CAAC,SAAS,CAAC,IAAI,OAAO,KAAK,CAAC,SAAS,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAC7D,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;YACjD,CAAC;YAED,aAAa;YACb,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,CAA+C,CAAC;YACpF,IAAI,SAAS,EAAE,CAAC;gBACd,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;oBAC3B,MAAM,KAAK,GAAG,EAAE,CAAC,OAAO,CAAW,CAAC;oBACpC,MAAM,EAAE,GAAG,EAAE,CAAC,UAAU,CAAwC,CAAC;oBAEjE,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;wBACb,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE;4BACzB,EAAE,EAAE,EAAE,CAAC,IAAI,CAAW;4BACtB,IAAI,EAAE,EAAE,EAAE,CAAC,MAAM,CAAW,IAAI,EAAE;4BAClC,IAAI,EAAG,EAAE,EAAE,CAAC,WAAW,CAAY,IAAI,EAAE;yBAC1C,CAAC,CAAC;wBACH,MAAM;4BACJ,IAAI,EAAE,iBAAiB;4BACvB,QAAQ,EAAE;gCACR,EAAE,EAAE,EAAE,CAAC,IAAI,CAAW;gCACtB,IAAI,EAAE,EAAE,EAAE,CAAC,MAAM,CAAW,IAAI,EAAE;gCAClC,IAAI,EAAE,EAAE;6BACT;yBACF,CAAC;oBACJ,CAAC;yBAAM,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;wBAC7B,MAAM,MAAM,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;wBAC1C,IAAI,MAAM,EAAE,CAAC;4BACX,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,CAAW,CAAC;wBAC3C,CAAC;wBACD,MAAM;4BACJ,IAAI,EAAE,iBAAiB;4BACvB,QAAQ,EAAE;gCACR,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,WAAW,CAAW,EAAwC;6BACrF;yBACF,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,qCAAqC;YACrC,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,CAAkB,CAAC;YAC9D,IAAI,YAAY,KAAK,YAAY,EAAE,CAAC;gBAClC,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,eAAe,EAAE,CAAC;oBACzC,IAAI,IAAI,GAA4B,EAAE,CAAC;oBACvC,IAAI,CAAC;wBACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAA4B,CAAC;oBAC5D,CAAC;oBAAC,MAAM,CAAC;wBACP,eAAe;oBACjB,CAAC;oBACD,MAAM;wBACJ,IAAI,EAAE,eAAe;wBACrB,QAAQ,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE;qBACrD,CAAC;gBACJ,CAAC;gBACD,eAAe,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC;YAED,QAAQ;YACR,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAuE,CAAC;YACpG,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM;oBACJ,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,WAAW,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC;wBACrC,YAAY,EAAE,KAAK,CAAC,iBAAiB,IAAI,CAAC;qBAC3C;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,wEAAwE;IAExE,OAAO;QACL,IAAI;QACJ,MAAM;QACN,QAAQ,EAAE,YAAY;QACtB,OAAO;QACP,YAAY,EAAE;YACZ,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,KAAK;SAClB;KACF,CAAC;AACJ,CAAC"}

package/dist/prebuilt/tool-node.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-node.d.ts","sourceRoot":"","sources":["../../src/prebuilt/tool-node.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,UAAU,EAAe,MAAM,aAAa,CAAC;AAM3D,MAAM,WAAW,OAAO;IACtB,IAAI,EAAS,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,EAAE,EAAW,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;CAC5E;AAMD,wBAAgB,cAAc,CAC5B,KAAK,EAAE,OAAO,EAAE,GACf,MAAM,CAAC;IAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;CAAE,CAAC,CA8CpC;AAOD,wBAAgB,cAAc,CAC5B,KAAK,EAAE;IAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;CAAE,GAChC,OAAO,GAAG,SAAS,CAIrB"}
+{"version":3,"file":"tool-node.d.ts","sourceRoot":"","sources":["../../src/prebuilt/tool-node.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,UAAU,EAAe,MAAM,aAAa,CAAC;AAM3D,MAAM,WAAW,OAAO;IACtB,IAAI,EAAS,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,EAAE,EAAW,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;CAC5E;AAMD,wBAAgB,cAAc,CAC5B,KAAK,EAAE,OAAO,EAAE,GACf,MAAM,CAAC;IAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;CAAE,CAAC,CA6CpC;AAOD,wBAAgB,cAAc,CAC5B,KAAK,EAAE;IAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;CAAE,GAChC,OAAO,GAAG,SAAS,CAIrB"}

package/dist/prebuilt/tool-node.js

@@ -13,7 +13,6 @@ export function createToolNode(tools) {
     return async (state) => {
         const lastMsg = state.messages.at(-1);
         if (!lastMsg || lastMsg.role !== "assistant" || !lastMsg.tool_calls?.length) {
-            console.warn("[ToolNode] No tool calls found in last assistant message.");
             return {};
         }
         const results = await Promise.all(lastMsg.tool_calls.map(async (tc) => {

package/dist/prebuilt/tool-node.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-node.js","sourceRoot":"","sources":["../../src/prebuilt/tool-node.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,oCAAoC;AACpC,+DAA+D;AAC/D,0DAA0D;AAC1D,8DAA8D;AAC9D,wDAAwD;AACxD,+DAA+D;AAgB/D,mEAAmE;AACnE,mBAAmB;AACnB,mEAAmE;AAEnE,MAAM,UAAU,cAAc,CAC5B,KAAgB;IAEhB,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvD,OAAO,KAAK,EAAE,KAAK,EAAE,EAAE;QACrB,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;YAC5E,OAAO,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;YAC1E,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAe,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YAElC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,gBAAgB,EAAE,CAAC,IAAI,cAAc;oBACpD,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;gBACtC,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC3E,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,uBAAuB,EAAE,CAAC,IAAI,MAAM,GAAG,EAAE;oBACxD,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC,CAAC;AACJ,CAAC;AAED,mEAAmE;AACnE,iEAAiE;AACjE,+CAA+C;AAC/C,mEAAmE;AAEnE,MAAM,UAAU,cAAc,CAC5B,KAAiC;IAEjC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,IAAI,IAAI,EAAE,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,UAAU,EAAE,MAAM;QAAE,OAAO,OAAO,CAAC;IAC1E,OAAO,SAAS,CAAC;AACnB,CAAC"}
+{"version":3,"file":"tool-node.js","sourceRoot":"","sources":["../../src/prebuilt/tool-node.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,oCAAoC;AACpC,+DAA+D;AAC/D,0DAA0D;AAC1D,8DAA8D;AAC9D,wDAAwD;AACxD,+DAA+D;AAgB/D,mEAAmE;AACnE,mBAAmB;AACnB,mEAAmE;AAEnE,MAAM,UAAU,cAAc,CAC5B,KAAgB;IAEhB,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvD,OAAO,KAAK,EAAE,KAAK,EAAE,EAAE;QACrB,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;YAC5E,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAe,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YAElC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,gBAAgB,EAAE,CAAC,IAAI,cAAc;oBACpD,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;gBACtC,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC3E,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO;oBACL,IAAI,EAAW,MAAe;oBAC9B,OAAO,EAAQ,uBAAuB,EAAE,CAAC,IAAI,MAAM,GAAG,EAAE;oBACxD,IAAI,EAAW,EAAE,CAAC,IAAI;oBACtB,YAAY,EAAG,EAAE,CAAC,EAAE;iBACrB,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC,CAAC;AACJ,CAAC;AAED,mEAAmE;AACnE,iEAAiE;AACjE,+CAA+C;AAC/C,mEAAmE;AAEnE,MAAM,UAAU,cAAc,CAC5B,KAAiC;IAEjC,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,IAAI,IAAI,EAAE,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,UAAU,EAAE,MAAM;QAAE,OAAO,OAAO,CAAC;IAC1E,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/pregel.d.ts

@@ -1,4 +1,5 @@
 import { type NodeDefinition, type Edge, type ChannelSchema, type ONIConfig, type ONIStreamEvent, type StreamMode, type InterruptConfig, type ONICheckpointer, type ONICheckpoint } from "./types.js";
+import { DeadLetterQueue, type DeadLetter } from "./dlq.js";
 import type { BaseStore } from "./store/index.js";
 import { HITLSessionStore } from "./hitl/index.js";
 import { EventBus } from "./events/bus.js";
@@ -6,6 +7,7 @@ import type { GuardrailsConfig } from "./guardrails/types.js";
 import type { EventListeners } from "./events/types.js";
 import { AuditLog } from "./guardrails/audit.js";
 import { BudgetTracker } from "./guardrails/budget.js";
+import { ONITracer, type TracerLike } from "./telemetry.js";
 export declare class ONIPregelRunner<S extends Record<string, unknown>> {
     private readonly nodes;
     private readonly edges;
@@ -13,8 +15,11 @@ export declare class ONIPregelRunner<S extends Record<string, unknown>> {
     private readonly interruptConfig;
     private readonly checkpointer;
     private readonly store;
+    private readonly defaults?;
+    private readonly dlq;
     private hitlStore;
     private nodeCache;
+    private circuitBreakers;
     /** Set to true when this runner is being invoked as a subgraph */
     private _isSubgraph;
     /** Accumulated parent updates from Command.PARENT during subgraph execution */
@@ -22,7 +27,10 @@ export declare class ONIPregelRunner<S extends Record<string, unknown>> {
     readonly eventBus: EventBus;
     readonly auditLog: AuditLog | null;
     readonly budgetTracker: BudgetTracker | null;
-    constructor(nodes: Map<string, NodeDefinition<S>>, edges: Edge<S>[], channels: ChannelSchema<S>, interruptConfig?: InterruptConfig, checkpointer?: ONICheckpointer<S> | null, store?: BaseStore | null, guardrails?: GuardrailsConfig, listeners?: EventListeners);
+    readonly tracer: ONITracer;
+    constructor(nodes: Map<string, NodeDefinition<S>>, edges: Edge<S>[], channels: ChannelSchema<S>, interruptConfig?: InterruptConfig, checkpointer?: ONICheckpointer<S> | null, store?: BaseStore | null, guardrails?: GuardrailsConfig, listeners?: EventListeners, defaults?: {
+        nodeTimeout?: number;
+    } | undefined, dlq?: DeadLetterQueue | null, tracer?: TracerLike | null);
     private buildInitialState;
     private applyUpdate;
     private resetEphemeral;
@@ -42,6 +50,8 @@ export declare class ONIPregelRunner<S extends Record<string, unknown>> {
     forkFrom(threadId: string, step: number, newThreadId: string): Promise<void>;
     getPendingInterrupts(threadId: string): import("./hitl/resume.js").HITLSession<S>[];
     hitlSessionStore(): HITLSessionStore<S>;
+    private getCircuitBreaker;
+    getDeadLetters(threadId: string): DeadLetter[];
     private saveCheckpoint;
     private evt;
 }

package/dist/pregel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pregel.d.ts","sourceRoot":"","sources":["../src/pregel.ts"],"names":[],"mappings":"AAKA,OAAO,EAEU,KAAK,cAAc,EAAE,KAAK,IAAI,EAAE,KAAK,aAAa,EACjE,KAAK,SAAS,EAAE,KAAK,cAAc,EAAE,KAAK,UAAU,EAAE,KAAK,eAAe,EAC1E,KAAK,eAAe,EAAE,KAAK,aAAa,EAEzC,MAAM,YAAY,CAAC;AAMpB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAGlD,OAAO,EACwC,gBAAgB,EAG9D,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AASvD,qBAAa,eAAe,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAa1D,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,KAAK;IAjBxB,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,SAAS,CAAmE;IACpF,kEAAkE;IAClE,OAAO,CAAC,WAAW,CAAS;IAC5B,+EAA+E;IAC/E,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAM;IAE7C,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACnC,QAAQ,CAAC,aAAa,EAAE,aAAa,GAAG,IAAI,CAAC;gBAG1B,KAAK,EAAY,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,EAC/C,KAAK,EAAY,IAAI,CAAC,CAAC,CAAC,EAAE,EAC1B,QAAQ,EAAS,aAAa,CAAC,CAAC,CAAC,EACjC,eAAe,GAAE,eAAoB,EACrC,YAAY,GAAK,eAAe,CAAC,CAAC,CAAC,GAAG,IAAW,EACjD,KAAK,GAAY,SAAS,GAAG,IAAW,EACzD,UAAU,CAAC,EAAuB,gBAAgB,EAClD,SAAS,CAAC,EAAwB,cAAc;IAWlD,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,WAAW;IAanB,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,YAAY;YA0BN,WAAW;IAoEzB,OAAO,CAAC,qBAAqB;IActB,OAAO,CACZ,KAAK,EAAO,OAAO,CAAC,CAAC,CAAC,EACtB,MAAM,CAAC,EAAK,SAAS,EACrB,UAAU,GAAE,UAAU,GAAG,UAAU,EAAc,GAChD,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAmW9B,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC;IAQxD,MAAM,CACX,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,CAAC,EAAE,SAAS,GAAG;QAAE,UAAU,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE,CAAA;KAAE,GAC9D,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAI9B,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAa7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAK7C,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAShE,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAM7D,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAKzD,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAelF,oBAAoB,CAAC,QAAQ,EAAE,MAAM;IAIrC,gBAAgB,IAAI,gBAAgB,CAAC,CAAC,CAAC;YAMzB,cAAc;IAe5B,OAAO,CAAC,GAAG;CAMZ"}
+{"version":3,"file":"pregel.d.ts","sourceRoot":"","sources":["../src/pregel.ts"],"names":[],"mappings":"AAKA,OAAO,EAEU,KAAK,cAAc,EAAE,KAAK,IAAI,EAAE,KAAK,aAAa,EACjE,KAAK,SAAS,EAAE,KAAK,cAAc,EAAE,KAAK,UAAU,EAAE,KAAK,eAAe,EAC1E,KAAK,eAAe,EAAE,KAAK,aAAa,EAEzC,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,eAAe,EAAE,KAAK,UAAU,EAAE,MAAM,UAAU,CAAC;AAK5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAGlD,OAAO,EACwC,gBAAgB,EAG9D,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,KAAK,UAAU,EAAiB,MAAM,gBAAgB,CAAC;AAS3E,qBAAa,eAAe,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAe1D,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,YAAY;IAC7B,OAAO,CAAC,QAAQ,CAAC,KAAK;IAGtB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAC,GAAG;IAvBtB,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,SAAS,CAAmE;IACpF,OAAO,CAAC,eAAe,CAAqC;IAC5D,kEAAkE;IAClE,OAAO,CAAC,WAAW,CAAS;IAC5B,+EAA+E;IAC/E,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAM;IAE7C,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,QAAQ,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACnC,QAAQ,CAAC,aAAa,EAAE,aAAa,GAAG,IAAI,CAAC;IAC7C,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;gBAGR,KAAK,EAAY,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,EAC/C,KAAK,EAAY,IAAI,CAAC,CAAC,CAAC,EAAE,EAC1B,QAAQ,EAAS,aAAa,CAAC,CAAC,CAAC,EACjC,eAAe,GAAE,eAAoB,EACrC,YAAY,GAAK,eAAe,CAAC,CAAC,CAAC,GAAG,IAAW,EACjD,KAAK,GAAY,SAAS,GAAG,IAAW,EACzD,UAAU,CAAC,EAAuB,gBAAgB,EAClD,SAAS,CAAC,EAAwB,cAAc,EAC/B,QAAQ,CAAC,EAAQ;QAAE,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,YAAA,EACzC,GAAG,GAAc,eAAe,GAAG,IAAW,EAC/D,MAAM,CAAC,EAA2B,UAAU,GAAG,IAAI;IAYrD,OAAO,CAAC,iBAAiB;IAQzB,OAAO,CAAC,WAAW;IAanB,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,YAAY;YA0BN,WAAW;IAsGzB,OAAO,CAAC,qBAAqB;IActB,OAAO,CACZ,KAAK,EAAO,OAAO,CAAC,CAAC,CAAC,EACtB,MAAM,CAAC,EAAK,SAAS,EACrB,UAAU,GAAE,UAAU,GAAG,UAAU,EAAc,GAChD,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAyX9B,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC;IAQxD,MAAM,CACX,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,CAAC,EAAE,SAAS,GAAG;QAAE,UAAU,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE,CAAA;KAAE,GAC9D,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAI9B,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAa7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAK7C,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAShE,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAM7D,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAKzD,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAelF,oBAAoB,CAAC,QAAQ,EAAE,MAAM;IAIrC,gBAAgB,IAAI,gBAAgB,CAAC,CAAC,CAAC;IAMvC,OAAO,CAAC,iBAAiB;IAkBzB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,EAAE;YAMhC,cAAc;IAiB5B,OAAO,CAAC,GAAG;CAMZ"}

package/dist/pregel.js

@@ -3,7 +3,8 @@
 // New: interrupt() context management, HITL resume, token streaming
 // ============================================================
 import { START, END, Send, Command, } from "./types.js";
-import { RecursionLimitError, NodeNotFoundError, ONIInterrupt } from "./errors.js";
+import { RecursionLimitError, NodeNotFoundError, ONIInterrupt, NodeTimeoutError } from "./errors.js";
+import { CircuitBreaker } from "./circuit-breaker.js";
 import { withRetry } from "./retry.js";
 import { NamespacedCheckpointer } from "./checkpointers/namespaced.js";
 import { _runWithContext } from "./context.js";
@@ -12,6 +13,7 @@ import { NodeInterruptSignal, HITLInterruptException, HITLSessionStore, _install
 import { EventBus } from "./events/bus.js";
 import { AuditLog } from "./guardrails/audit.js";
 import { BudgetTracker } from "./guardrails/budget.js";
+import { ONITracer } from "./telemetry.js";
 const DEFAULT_RECURSION_LIMIT = 25;
 export class ONIPregelRunner {
     nodes;
@@ -20,8 +22,11 @@ export class ONIPregelRunner {
     interruptConfig;
     checkpointer;
     store;
+    defaults;
+    dlq;
     hitlStore = new HITLSessionStore();
     nodeCache = new Map();
+    circuitBreakers = new Map();
     /** Set to true when this runner is being invoked as a subgraph */
     _isSubgraph = false;
     /** Accumulated parent updates from Command.PARENT during subgraph execution */
@@ -29,16 +34,20 @@ export class ONIPregelRunner {
     eventBus;
     auditLog;
     budgetTracker;
-    constructor(nodes, edges, channels, interruptConfig = {}, checkpointer = null, store = null, guardrails, listeners) {
+    tracer;
+    constructor(nodes, edges, channels, interruptConfig = {}, checkpointer = null, store = null, guardrails, listeners, defaults, dlq = null, tracer) {
         this.nodes = nodes;
         this.edges = edges;
         this.channels = channels;
         this.interruptConfig = interruptConfig;
         this.checkpointer = checkpointer;
         this.store = store;
+        this.defaults = defaults;
+        this.dlq = dlq;
         this.eventBus = new EventBus(listeners);
         this.auditLog = guardrails?.audit ? new AuditLog() : null;
         this.budgetTracker = guardrails?.budget ? new BudgetTracker(guardrails.budget) : null;
+        this.tracer = new ONITracer(tracer ?? null);
     }
     // ----------------------------------------------------------------
     // State helpers
@@ -130,11 +139,43 @@ export class ONIPregelRunner {
             });
             try {
                 const run = () => Promise.resolve(nodeDef.fn(state, config));
+                // Core execute call: retry-aware
+                const executeCall = async () => {
+                    if (nodeDef.retry)
+                        return withRetry(run, nodeDef.name, nodeDef.retry);
+                    return run();
+                };
+                // Wrap with timeout if configured (per-node > global default > none)
+                const timeoutMs = nodeDef.timeout ?? this.defaults?.nodeTimeout;
+                const executeWithTimeout = async () => {
+                    if (timeoutMs != null && timeoutMs > 0) {
+                        const ac = new AbortController();
+                        const timer = setTimeout(() => ac.abort(), timeoutMs);
+                        try {
+                            return await Promise.race([
+                                executeCall(),
+                                new Promise((_, reject) => {
+                                    ac.signal.addEventListener("abort", () => {
+                                        reject(new NodeTimeoutError(nodeDef.name, timeoutMs));
+                                    });
+                                }),
+                            ]);
+                        }
+                        finally {
+                            clearTimeout(timer);
+                        }
+                    }
+                    return executeCall();
+                };
+                // Wrap with circuit breaker if configured
+                const cb = this.getCircuitBreaker(nodeDef);
                 let result;
-                if (nodeDef.retry)
-                    result = await withRetry(run, nodeDef.name, nodeDef.retry);
-                else
-                    result = await run();
+                if (cb) {
+                    result = await cb.execute(executeWithTimeout);
+                }
+                else {
+                    result = await executeWithTimeout();
+                }
                 // Store in cache
                 if (nodeDef.cache) {
                     const policy = typeof nodeDef.cache === "object" ? nodeDef.cache : {};
@@ -172,6 +213,8 @@ export class ONIPregelRunner {
         const modes = new Set(Array.isArray(streamMode) ? streamMode : [streamMode]);
         const isMultiMode = Array.isArray(streamMode);
         const tag = (evt, mode) => isMultiMode ? { ...evt, mode } : evt;
+        // Telemetry: graph-level span
+        const graphSpan = this.tracer.startGraphSpan("invoke", { threadId, agentId });
         // Load resume values from config (set by resume() call)
         const resumeMap = config?.__resumeValues ?? {};
         // Load or init state
@@ -206,6 +249,7 @@ export class ONIPregelRunner {
         while (true) {
             const nextNodes = [];
             const nextSends = [];
+            state = this.resetEphemeral(state);
             // Drain sends — group by target node for parallel fan-out execution
             const sendGroups = new Map();
             for (const send of pendingSends) {
@@ -262,7 +306,6 @@ export class ONIPregelRunner {
                 break;
             if (step >= recursionLimit)
                 throw new RecursionLimitError(recursionLimit);
-            state = this.resetEphemeral(state);
             // Emit debug node_start events before parallel execution
             if (modes.has("debug")) {
                 for (const nodeName of executableNodes) {
@@ -304,6 +347,8 @@ export class ONIPregelRunner {
                 // Emit agent.start lifecycle event
                 const nodeStartTime = Date.now();
                 this.eventBus.emit({ type: "agent.start", agent: name, timestamp: nodeStartTime, step });
+                // Telemetry: node-level span
+                const nodeSpan = this.tracer.startNodeSpan(name, { threadId, step, agentId });
                 let result;
                 let subParentUpdates = [];
                 try {
@@ -368,8 +413,18 @@ export class ONIPregelRunner {
                         }
                         throw new HITLInterruptException(threadId, iv, state);
                     }
+                    // Record to DLQ before re-throwing
+                    if (this.dlq && err instanceof Error) {
+                        this.dlq.record(threadId, name, state, err, nodeDef.retry?.maxAttempts ?? 1);
+                    }
+                    // Telemetry: record error on node span
+                    if (err instanceof Error)
+                        this.tracer.recordError(nodeSpan, err);
+                    this.tracer.endSpan(nodeSpan);
                     throw err;
                 }
+                // Telemetry: end node span
+                this.tracer.endSpan(nodeSpan);
                 // Emit agent.end lifecycle event
                 this.eventBus.emit({ type: "agent.end", agent: name, timestamp: Date.now(), step, duration: Date.now() - nodeStartTime });
                 // Collect events for yielding after parallel execution
@@ -497,6 +552,9 @@ export class ONIPregelRunner {
         }
         if (modes.has("values"))
             yield tag(this.evt("state_update", state, step, agentId), "values");
+        // Telemetry: end graph span
+        graphSpan.setAttribute("oni.steps", step);
+        this.tracer.endSpan(graphSpan);
     }
     // ----------------------------------------------------------------
     // Public API
@@ -565,16 +623,39 @@ export class ONIPregelRunner {
     hitlSessionStore() {
         return this.hitlStore;
     }
+    // ---- Circuit Breaker ----
+    getCircuitBreaker(nodeDef) {
+        if (!nodeDef.circuitBreaker)
+            return null;
+        let cb = this.circuitBreakers.get(nodeDef.name);
+        if (!cb) {
+            cb = new CircuitBreaker({
+                threshold: nodeDef.circuitBreaker.threshold,
+                resetAfter: nodeDef.circuitBreaker.resetAfter,
+                fallback: nodeDef.circuitBreaker.fallback
+                    ? () => nodeDef.circuitBreaker.fallback(undefined, undefined)
+                    : undefined,
+            }, nodeDef.name);
+            this.circuitBreakers.set(nodeDef.name, cb);
+        }
+        return cb;
+    }
+    // ---- Dead Letter Queue ----
+    getDeadLetters(threadId) {
+        return this.dlq?.getAll(threadId) ?? [];
+    }
     // ---- Helpers ----
     async saveCheckpoint(threadId, step, state, nextNodes, pendingSends, agentId, metadata, pendingWrites) {
         if (!this.checkpointer)
             return;
+        const cpSpan = this.tracer.startCheckpointSpan("put", { threadId });
         await this.checkpointer.put({
             threadId, step, state, agentId, metadata, pendingWrites,
             nextNodes: nextNodes.map(String),
             pendingSends: pendingSends,
             timestamp: Date.now(),
         });
+        this.tracer.endSpan(cpSpan);
     }
     evt(event, data, step, agentId, node) {
         return { event, data, step, timestamp: Date.now(), agentId, node };