@mupt-ai/dari-router 0.1.0

package/dist/pi_output.js

@@ -0,0 +1,328 @@
+import { RouterFrameworkError } from "./framework_error.js";
+import { decodeOpenAIReasoningSignature } from "./pi_reasoning_signature.js";
+import { isRecord } from "./types.js";
+import { isOpenAIResponsesApi, } from "./continuation_state.js";
+import { isRouterHostedToolCallStatus, } from "./framework_types.js";
+export function routerCompletion(message, signal) {
+    assertPiSuccess(message, signal);
+    const content = [];
+    const source = providerIdentity(message);
+    for (const part of message.content) {
+        if (part.type === "text")
+            content.push({ type: "text", text: part.text });
+        if (part.type === "thinking")
+            content.push(routerReasoning(part, source));
+        if (part.type === "toolCall") {
+            content.push({
+                type: "tool_call",
+                id: part.id,
+                name: part.name,
+                arguments: part.arguments,
+            });
+        }
+        const hostedItem = providerItemRecord(part);
+        if (hostedItem !== null)
+            content.push(hostedToolCall(hostedItem, source));
+    }
+    return {
+        ...(message.responseId === undefined ? {} : { id: message.responseId }),
+        createdAtMs: message.timestamp,
+        content,
+        finishReason: routerFinishReason(message.stopReason),
+        usage: routerUsage(message),
+    };
+}
+export async function* routerEvents(events, signal) {
+    const text = new Map();
+    const thinking = new Map();
+    const endedThinking = new Set();
+    const openTools = new Map();
+    const completedTools = new Map();
+    for await (const event of events) {
+        if (event.type === "text_start") {
+            text.set(event.contentIndex, text.get(event.contentIndex) ?? "");
+            continue;
+        }
+        if (event.type === "text_delta") {
+            if (event.delta.length > 0) {
+                text.set(event.contentIndex, (text.get(event.contentIndex) ?? "") + event.delta);
+                yield { type: "text_delta", index: event.contentIndex, delta: event.delta };
+            }
+            continue;
+        }
+        if (event.type === "text_end") {
+            const suffix = remainingText(text.get(event.contentIndex) ?? "", event.content);
+            if (suffix.length > 0)
+                yield { type: "text_delta", index: event.contentIndex, delta: suffix };
+            text.set(event.contentIndex, event.content);
+            continue;
+        }
+        if (event.type === "thinking_start") {
+            thinking.set(event.contentIndex, "");
+            continue;
+        }
+        if (event.type === "thinking_delta") {
+            if (event.delta.length > 0) {
+                thinking.set(event.contentIndex, (thinking.get(event.contentIndex) ?? "") + event.delta);
+                yield { type: "reasoning_delta", index: event.contentIndex, delta: event.delta };
+            }
+            continue;
+        }
+        if (event.type === "thinking_end") {
+            const streamed = thinking.get(event.contentIndex) ?? "";
+            const suffix = remainingText(streamed, event.content);
+            if (suffix.length > 0)
+                yield { type: "reasoning_delta", index: event.contentIndex, delta: suffix };
+            thinking.set(event.contentIndex, event.content);
+            endedThinking.add(event.contentIndex);
+            const part = event.partial.content[event.contentIndex];
+            const source = providerIdentity(event.partial);
+            if (part?.type === "thinking") {
+                const reasoning = routerReasoning(part, source);
+                yield {
+                    type: "reasoning_end",
+                    index: event.contentIndex,
+                    ...(part.redacted === true ? { redacted: true } : {}),
+                    source,
+                    ...(reasoning.continuation ? { continuation: reasoning.continuation } : {}),
+                };
+            }
+            else {
+                yield { type: "reasoning_end", index: event.contentIndex, source };
+            }
+            continue;
+        }
+        if (event.type === "toolcall_start") {
+            const call = event.partial.content[event.contentIndex];
+            if (call?.type !== "toolCall")
+                throw invalidPiStream("tool-call start is missing its tool call.");
+            openTools.set(event.contentIndex, { receivedDelta: false });
+            yield {
+                type: "tool_call_start",
+                index: event.contentIndex,
+                id: call.id,
+                name: call.name,
+            };
+            continue;
+        }
+        if (event.type === "toolcall_delta") {
+            const state = openTools.get(event.contentIndex);
+            if (state === undefined)
+                throw invalidPiStream("tool-call delta arrived before its start.");
+            if (event.delta.length > 0) {
+                state.receivedDelta = true;
+                yield { type: "tool_call_delta", index: event.contentIndex, delta: event.delta };
+            }
+            continue;
+        }
+        if (event.type === "toolcall_end") {
+            const state = openTools.get(event.contentIndex);
+            if (state === undefined) {
+                yield* completeToolCall(event.contentIndex, event.toolCall);
+                completedTools.set(event.contentIndex, event.toolCall);
+                continue;
+            }
+            if (!state.receivedDelta) {
+                yield {
+                    type: "tool_call_delta",
+                    index: event.contentIndex,
+                    delta: JSON.stringify(event.toolCall.arguments),
+                };
+            }
+            openTools.delete(event.contentIndex);
+            completedTools.set(event.contentIndex, event.toolCall);
+            yield { type: "tool_call_end", index: event.contentIndex };
+            continue;
+        }
+        if (event.type === "done") {
+            assertPiSuccess(event.message, signal);
+            if (event.reason !== "stop" && event.reason !== "length" && event.reason !== "toolUse") {
+                throw invalidPiStream(`stream ended with invalid reason '${String(event.reason)}'.`);
+            }
+            if (openTools.size > 0)
+                throw invalidPiStream("stream finished with open tool calls.");
+            yield* reconcileTerminalOutput(event.message, text, thinking, endedThinking, completedTools);
+            yield {
+                type: "finish",
+                finishReason: routerFinishReason(event.reason),
+                usage: routerUsage(event.message),
+            };
+            return;
+        }
+        if (event.type === "error") {
+            if (event.reason === "aborted")
+                throw piCancellation(event.error, signal);
+            throw piProviderError(event.error.errorMessage ?? "Pi provider execution failed.");
+        }
+    }
+    throw invalidPiStream("stream ended without a terminal event.");
+}
+function* reconcileTerminalOutput(message, text, thinking, endedThinking, completedTools) {
+    for (const index of text.keys()) {
+        if (message.content[index]?.type !== "text") {
+            throw invalidPiStream(`terminal output changed text block ${index}.`);
+        }
+    }
+    for (const index of thinking.keys()) {
+        if (message.content[index]?.type !== "thinking") {
+            throw invalidPiStream(`terminal output changed thinking block ${index}.`);
+        }
+    }
+    for (const [index, streamed] of completedTools) {
+        const terminal = message.content[index];
+        if (terminal?.type !== "toolCall" || terminal.id !== streamed.id || terminal.name !== streamed.name) {
+            throw invalidPiStream(`terminal output changed tool call ${index}.`);
+        }
+    }
+    for (const [index, part] of message.content.entries()) {
+        if (part.type === "text") {
+            const suffix = remainingText(text.get(index) ?? "", part.text);
+            if (suffix.length > 0)
+                yield { type: "text_delta", index, delta: suffix };
+            continue;
+        }
+        if (part.type === "thinking") {
+            // Started-but-unended thinking blocks reconcile like unended text:
+            // emit the remaining delta, then the proper reasoning_end.
+            if (!endedThinking.has(index)) {
+                yield* completeReasoning(index, part, providerIdentity(message), thinking.get(index) ?? "");
+            }
+            continue;
+        }
+        if (part.type === "toolCall") {
+            if (!completedTools.has(index))
+                yield* completeToolCall(index, part);
+            continue;
+        }
+        const hostedItem = providerItemRecord(part);
+        if (hostedItem !== null) {
+            yield { ...hostedToolCall(hostedItem, providerIdentity(message)), index };
+        }
+    }
+}
+function* completeToolCall(index, call) {
+    yield { type: "tool_call_start", index, id: call.id, name: call.name };
+    yield { type: "tool_call_delta", index, delta: JSON.stringify(call.arguments) };
+    yield { type: "tool_call_end", index };
+}
+function* completeReasoning(index, part, source, streamed = "") {
+    const suffix = remainingText(streamed, part.thinking);
+    if (suffix.length > 0) {
+        yield { type: "reasoning_delta", index, delta: suffix };
+    }
+    const reasoning = routerReasoning(part, source);
+    yield {
+        type: "reasoning_end",
+        index,
+        ...(part.redacted === true ? { redacted: true } : {}),
+        source,
+        ...(reasoning.continuation ? { continuation: reasoning.continuation } : {}),
+    };
+}
+function providerItemRecord(part) {
+    if (!isRecord(part) || part.type !== "providerItem")
+        return null;
+    if (!isRecord(part.item)) {
+        throw unsupportedHostedTool("the provider item has no payload record");
+    }
+    return part.item;
+}
+function hostedToolCall(item, source) {
+    if (item.type !== "web_search_call") {
+        throw unsupportedHostedTool(`only web_search_call items are replayable, got '${String(item.type)}'`);
+    }
+    if (typeof item.id !== "string" || item.id.length === 0) {
+        throw unsupportedHostedTool("the web_search_call item has no id");
+    }
+    const status = hostedToolCallStatus(item.status);
+    return {
+        type: "hosted_tool_call",
+        id: item.id,
+        tool: "web_search",
+        providerType: "web_search_call",
+        ...(status === undefined ? {} : { status }),
+        payload: item,
+        source,
+    };
+}
+function hostedToolCallStatus(value) {
+    if (value === undefined || isRouterHostedToolCallStatus(value))
+        return value;
+    throw unsupportedHostedTool(`web_search_call status '${String(value)}' is not replayable`);
+}
+function unsupportedHostedTool(detail) {
+    return new RouterFrameworkError("executor", `Pi returned a hosted tool item the router cannot replay: ${detail}.`, "pi_hosted_tool_unsupported");
+}
+function remainingText(streamed, terminal) {
+    if (!terminal.startsWith(streamed)) {
+        throw invalidPiStream("terminal text does not extend the streamed text.");
+    }
+    return terminal.slice(streamed.length);
+}
+function providerIdentity(message) {
+    return { provider: message.provider, api: message.api, model: message.model };
+}
+function routerReasoning(part, source) {
+    const signature = typeof part.thinkingSignature === "string" && part.thinkingSignature.length > 0
+        ? part.thinkingSignature
+        : undefined;
+    let continuation;
+    if (isOpenAIResponsesApi(source.api) && signature) {
+        const native = decodeOpenAIReasoningSignature(signature);
+        if (native?.encryptedContent) {
+            continuation = {
+                kind: "openai_reasoning",
+                source,
+                encryptedContent: native.encryptedContent,
+                ...(native.id ? { providerItemId: native.id } : {}),
+            };
+        }
+    }
+    else if (source.api === "anthropic-messages" && signature) {
+        continuation = part.redacted === true
+            ? { kind: "anthropic_redacted_thinking", source, data: signature }
+            : { kind: "anthropic_thinking", source, thinking: part.thinking, signature };
+    }
+    return {
+        type: "reasoning",
+        text: part.thinking,
+        ...(part.redacted === true ? { redacted: true } : {}),
+        source,
+        ...(continuation ? { continuation } : {}),
+    };
+}
+function assertPiSuccess(message, signal) {
+    if (message.stopReason === "error") {
+        throw piProviderError(message.errorMessage ?? "Pi provider execution failed.");
+    }
+    if (message.stopReason === "aborted")
+        throw piCancellation(message, signal);
+}
+function routerFinishReason(reason) {
+    if (reason === "length")
+        return "length";
+    if (reason === "toolUse")
+        return "tool_calls";
+    return "stop";
+}
+function routerUsage(message) {
+    return {
+        inputTokens: message.usage.input,
+        outputTokens: message.usage.output,
+        cacheReadTokens: message.usage.cacheRead,
+        cacheWriteTokens: message.usage.cacheWrite,
+        totalTokens: message.usage.totalTokens,
+    };
+}
+function piCancellation(message, signal) {
+    return new RouterFrameworkError("cancelled", typeof signal.reason === "string"
+        ? signal.reason
+        : message.errorMessage ?? "Request cancelled.", "request_cancelled");
+}
+function piProviderError(message) {
+    return new RouterFrameworkError("executor", message, "pi_provider_error");
+}
+function invalidPiStream(message) {
+    return new RouterFrameworkError("executor", `Pi ${message}`, "pi_stream_invalid");
+}
+//# sourceMappingURL=pi_output.js.map

package/dist/pi_output.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_output.js","sourceRoot":"","sources":["../src/pi_output.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,8BAA8B,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EACL,oBAAoB,GAGrB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,4BAA4B,GAK7B,MAAM,sBAAsB,CAAC;AAE9B,MAAM,UAAU,gBAAgB,CAC9B,OAAyB,EACzB,MAAmB;IAEnB,eAAe,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,OAAO,GAAgC,EAAE,CAAC;IAChD,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACzC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1E,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;QAC1E,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,UAAU,KAAK,IAAI;YAAE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;IAC5E,CAAC;IACD,OAAO;QACL,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;QACvE,WAAW,EAAE,OAAO,CAAC,SAAS;QAC9B,OAAO;QACP,YAAY,EAAE,kBAAkB,CAAC,OAAO,CAAC,UAAU,CAAC;QACpD,KAAK,EAAE,WAAW,CAAC,OAAO,CAAC;KAC5B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,YAAY,CACjC,MAA4C,EAC5C,MAAmB;IAEnB,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC3C,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAsC,CAAC;IAChE,MAAM,cAAc,GAAG,IAAI,GAAG,EAAyC,CAAC;IACxE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAChC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;YACjE,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAChC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;gBACjF,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;YAC9E,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAChF,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;YAC9F,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACpC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACrC,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;gBACzF,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;YACnF,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACtD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;YACnG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAChD,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,IAAI,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC9B,MAAM,SAAS,GAAG,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAChD,MAAM;oBACJ,IAAI,EAAE,eAAe;oBACrB,KAAK,EAAE,KAAK,CAAC,YAAY;oBACzB,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACrD,MAAM;oBACN,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5E,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,MAAM,EAAE,CAAC;YACrE,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACvD,IAAI,IAAI,EAAE,IAAI,KAAK,UAAU;gBAAE,MAAM,eAAe,CAAC,2CAA2C,CAAC,CAAC;YAClG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5D,MAAM;gBACJ,IAAI,EAAE,iBAAiB;gBACvB,KAAK,EAAE,KAAK,CAAC,YAAY;gBACzB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB,CAAC;YACF,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAChD,IAAI,KAAK,KAAK,SAAS;gBAAE,MAAM,eAAe,CAAC,2CAA2C,CAAC,CAAC;YAC5F,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC3B,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;YACnF,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAChD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,KAAK,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC5D,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACvD,SAAS;YACX,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM;oBACJ,IAAI,EAAE,iBAAiB;oBACvB,KAAK,EAAE,KAAK,CAAC,YAAY;oBACzB,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;iBAChD,CAAC;YACJ,CAAC;YACD,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACrC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;YAC3D,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,eAAe,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACvC,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACvF,MAAM,eAAe,CAAC,qCAAqC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACvF,CAAC;YACD,IAAI,SAAS,CAAC,IAAI,GAAG,CAAC;gBAAE,MAAM,eAAe,CAAC,uCAAuC,CAAC,CAAC;YACvF,KAAK,CAAC,CAAC,uBAAuB,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;YAC7F,MAAM;gBACJ,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC;gBAC9C,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;aAClC,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS;gBAAE,MAAM,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC1E,MAAM,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,IAAI,+BAA+B,CAAC,CAAC;QACrF,CAAC;IACH,CAAC;IACD,MAAM,eAAe,CAAC,wCAAwC,CAAC,CAAC;AAClE,CAAC;AAED,QAAQ,CAAC,CAAC,uBAAuB,CAC/B,OAAyB,EACzB,IAAiC,EACjC,QAAqC,EACrC,aAAkC,EAClC,cAAkE;IAElE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5C,MAAM,eAAe,CAAC,sCAAsC,KAAK,GAAG,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IACD,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;QACpC,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;YAChD,MAAM,eAAe,CAAC,0CAA0C,KAAK,GAAG,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IACD,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,cAAc,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,QAAQ,EAAE,IAAI,KAAK,UAAU,IAAI,QAAQ,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpG,MAAM,eAAe,CAAC,qCAAqC,KAAK,GAAG,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QACtD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;YAC1E,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,mEAAmE;YACnE,2DAA2D;YAC3D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9B,KAAK,CAAC,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9F,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,KAAK,CAAC,CAAC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACrE,SAAS;QACX,CAAC;QACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;YACxB,MAAM,EAAE,GAAG,cAAc,CAAC,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;QAC5E,CAAC;IACH,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,CAAC,gBAAgB,CAAC,KAAa,EAAE,IAAc;IACtD,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IACvE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;IAChF,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC;AACzC,CAAC;AAED,QAAQ,CAAC,CAAC,iBAAiB,CACzB,KAAa,EACb,IAAqB,EACrB,MAAwB,EACxB,QAAQ,GAAG,EAAE;IAEb,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IAC1D,CAAC;IACD,MAAM,SAAS,GAAG,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAChD,MAAM;QACJ,IAAI,EAAE,eAAe;QACrB,KAAK;QACL,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,MAAM;QACN,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5E,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAa;IACvC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc;QAAE,OAAO,IAAI,CAAC;IACjE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,MAAM,qBAAqB,CAAC,yCAAyC,CAAC,CAAC;IACzE,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC;AACnB,CAAC;AAED,SAAS,cAAc,CACrB,IAA6B,EAC7B,MAAwB;IAExB,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QACpC,MAAM,qBAAqB,CACzB,mDAAmD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CACxE,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ,IAAI,IAAI,CAAC,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxD,MAAM,qBAAqB,CAAC,oCAAoC,CAAC,CAAC;IACpE,CAAC;IACD,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjD,OAAO;QACL,IAAI,EAAE,kBAAkB;QACxB,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,IAAI,EAAE,YAAY;QAClB,YAAY,EAAE,iBAAiB;QAC/B,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3C,OAAO,EAAE,IAAI;QACb,MAAM;KACP,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,IAAI,KAAK,KAAK,SAAS,IAAI,4BAA4B,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7E,MAAM,qBAAqB,CAAC,2BAA2B,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;AAC7F,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAc;IAC3C,OAAO,IAAI,oBAAoB,CAC7B,UAAU,EACV,4DAA4D,MAAM,GAAG,EACrE,4BAA4B,CAC7B,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,QAAgB;IACvD,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,MAAM,eAAe,CAAC,kDAAkD,CAAC,CAAC;IAC5E,CAAC;IACD,OAAO,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,gBAAgB,CACvB,OAA6D;IAE7D,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;AAChF,CAAC;AAED,SAAS,eAAe,CACtB,IAAqB,EACrB,MAAwB;IAExB,MAAM,SAAS,GAAG,OAAO,IAAI,CAAC,iBAAiB,KAAK,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC;QAC/F,CAAC,CAAC,IAAI,CAAC,iBAAiB;QACxB,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,YAAmD,CAAC;IACxD,IAAI,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,SAAS,EAAE,CAAC;QAClD,MAAM,MAAM,GAAG,8BAA8B,CAAC,SAAS,CAAC,CAAC;QACzD,IAAI,MAAM,EAAE,gBAAgB,EAAE,CAAC;YAC7B,YAAY,GAAG;gBACb,IAAI,EAAE,kBAAkB;gBACxB,MAAM;gBACN,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;gBACzC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpD,CAAC;QACJ,CAAC;IACH,CAAC;SAAM,IAAI,MAAM,CAAC,GAAG,KAAK,oBAAoB,IAAI,SAAS,EAAE,CAAC;QAC5D,YAAY,GAAG,IAAI,CAAC,QAAQ,KAAK,IAAI;YACnC,CAAC,CAAC,EAAE,IAAI,EAAE,6BAA6B,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE;YAClE,CAAC,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;IACjF,CAAC;IACD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,IAAI,CAAC,QAAQ;QACnB,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,MAAM;QACN,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1C,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAyB,EAAE,MAAmB;IACrE,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;QACnC,MAAM,eAAe,CAAC,OAAO,CAAC,YAAY,IAAI,+BAA+B,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;QAAE,MAAM,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,kBAAkB,CACzB,MAAsE;IAEtE,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IACzC,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,YAAY,CAAC;IAC9C,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAClB,OAAwC;IAExC,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK;QAChC,YAAY,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM;QAClC,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,SAAS;QACxC,gBAAgB,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;QAC1C,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,WAAW;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CACrB,OAA+C,EAC/C,MAAmB;IAEnB,OAAO,IAAI,oBAAoB,CAC7B,WAAW,EACX,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;QAC/B,CAAC,CAAC,MAAM,CAAC,MAAM;QACf,CAAC,CAAC,OAAO,CAAC,YAAY,IAAI,oBAAoB,EAChD,mBAAmB,CACpB,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,OAAO,IAAI,oBAAoB,CAAC,UAAU,EAAE,OAAO,EAAE,mBAAmB,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,OAAO,IAAI,oBAAoB,CAAC,UAAU,EAAE,MAAM,OAAO,EAAE,EAAE,mBAAmB,CAAC,CAAC;AACpF,CAAC"}

package/dist/pi_reasoning_signature.d.ts

@@ -0,0 +1,10 @@
+export declare function encodeOpenAIReasoningSignature(state: {
+    id?: string;
+    summary: string[];
+    content: string[];
+    encryptedContent: string;
+}): string;
+export declare function decodeOpenAIReasoningSignature(value: string): {
+    id?: string;
+    encryptedContent?: string;
+} | null;

package/dist/pi_reasoning_signature.js

@@ -0,0 +1,33 @@
+import { isRecord } from "./types.js";
+// Wire codec for the JSON blob Pi carries in thinkingSignature on
+// openai-responses APIs: a serialized Responses reasoning item whose
+// encrypted_content holds the replayable state.
+export function encodeOpenAIReasoningSignature(state) {
+    return JSON.stringify({
+        type: "reasoning",
+        ...(state.id ? { id: state.id } : {}),
+        summary: state.summary.map((text) => ({ type: "summary_text", text })),
+        ...(state.content.length
+            ? { content: state.content.map((text) => ({ type: "reasoning_text", text })) }
+            : {}),
+        encrypted_content: state.encryptedContent,
+    });
+}
+export function decodeOpenAIReasoningSignature(value) {
+    let parsed;
+    try {
+        parsed = JSON.parse(value);
+    }
+    catch {
+        return null;
+    }
+    if (!isRecord(parsed) || parsed.type !== "reasoning")
+        return null;
+    return {
+        ...(typeof parsed.id === "string" && parsed.id.length > 0 ? { id: parsed.id } : {}),
+        ...(typeof parsed.encrypted_content === "string" && parsed.encrypted_content.length > 0
+            ? { encryptedContent: parsed.encrypted_content }
+            : {}),
+    };
+}
+//# sourceMappingURL=pi_reasoning_signature.js.map

package/dist/pi_reasoning_signature.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_reasoning_signature.js","sourceRoot":"","sources":["../src/pi_reasoning_signature.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAEtC,kEAAkE;AAClE,qEAAqE;AACrE,gDAAgD;AAEhD,MAAM,UAAU,8BAA8B,CAAC,KAK9C;IACC,OAAO,IAAI,CAAC,SAAS,CAAC;QACpB,IAAI,EAAE,WAAW;QACjB,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM;YACtB,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;YAC9E,CAAC,CAAC,EAAE,CAAC;QACP,iBAAiB,EAAE,KAAK,CAAC,gBAAgB;KAC1C,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,KAAa;IAI1D,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,IAAI,CAAC;IAClE,OAAO;QACL,GAAG,CAAC,OAAO,MAAM,CAAC,EAAE,KAAK,QAAQ,IAAI,MAAM,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,GAAG,CAAC,OAAO,MAAM,CAAC,iBAAiB,KAAK,QAAQ,IAAI,MAAM,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC;YACrF,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,iBAAiB,EAAE;YAChD,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;AACJ,CAAC"}

package/dist/pi_runtime.d.ts

@@ -0,0 +1,3 @@
+import type { CreatePiRuntimeOptions, PiRuntime } from "./pi_types.js";
+export type { CreatePiRuntimeOptions, PiApiKey, PiCredentialInput, PiModelRegistry, PiRouterModelOptions, PiRuntime, } from "./pi_types.js";
+export declare function createPiRuntime(options: CreatePiRuntimeOptions): Promise<PiRuntime>;

package/dist/pi_runtime.js

@@ -0,0 +1,155 @@
+import { RouterFrameworkError } from "./framework_error.js";
+import { nativeModelId, providerForModel } from "./model_ids.js";
+import { piContext } from "./pi_context.js";
+import { piOptions } from "./pi_options.js";
+import { routerCompletion, routerEvents } from "./pi_output.js";
+import { openAIChatRequest } from "./protocol_openai_chat.js";
+const STRUCTURED_OUTPUT_APIS = new Set([
+    "anthropic-messages",
+    "azure-openai-responses",
+    "google-generative-ai",
+    "google-vertex",
+    "mistral-conversations",
+    "openai-completions",
+    "openai-responses",
+]);
+export async function createPiRuntime(options) {
+    validateRuntimeOptions(options);
+    const dependencies = await piDependencies(options.registry);
+    const model = (id, overrides = {}) => {
+        const provider = modelProvider(id);
+        const registered = registeredModel(dependencies.registry, provider, id);
+        const reasoningEfforts = overrides.reasoningEfforts === undefined
+            ? dependencies.supportedThinkingLevels(registered)
+            : [...overrides.reasoningEfforts];
+        const defaultReasoningEffort = overrides.defaultReasoningEffort
+            ?? (reasoningEfforts.includes("off") ? "off" : reasoningEfforts[0]);
+        return {
+            id,
+            executor: "pi",
+            provider,
+            api: registered.api,
+            reasoningEfforts,
+            ...(defaultReasoningEffort === undefined ? {} : { defaultReasoningEffort }),
+            capabilities: {
+                imageInput: registered.input.includes("image"),
+                toolUse: true,
+                structuredOutput: STRUCTURED_OUTPUT_APIS.has(registered.api),
+                streaming: true,
+                ...overrides.capabilities,
+            },
+            ...(overrides.metadata === undefined ? {} : { metadata: overrides.metadata }),
+        };
+    };
+    const run = async (execution) => {
+        const registered = registeredModel(dependencies.registry, execution.candidate.provider, execution.candidate.id);
+        if (registered.api !== execution.candidate.api) {
+            throw new RouterFrameworkError("configuration", `Model '${execution.candidate.id}' declares API '${execution.candidate.api}' but Pi uses '${registered.api}'.`, "pi_model_api_mismatch");
+        }
+        const context = piContext(execution.request, registered);
+        const streamOptions = await piOptions(options, execution, registered);
+        if (execution.request.stream) {
+            const events = dependencies.registry.streamSimple(registered, context, streamOptions);
+            return { type: "stream", events: routerEvents(events, execution.signal) };
+        }
+        const message = await dependencies.registry.completeSimple(registered, context, streamOptions);
+        return { type: "complete", output: routerCompletion(message, execution.signal) };
+    };
+    return {
+        model,
+        execute: async (input) => run({
+            request: input.request,
+            candidate: input.model,
+            reasoningEffort: input.decision.reasoningEffort,
+            signal: input.signal,
+            purpose: "execution",
+        }),
+        select: async (request, signal = new AbortController().signal) => {
+            let normalized;
+            try {
+                normalized = openAIChatRequest(request);
+            }
+            catch (error) {
+                // The selector request is constructed internally; an invalid request
+                // here is a router bug, not the end user's fault.
+                if (error instanceof RouterFrameworkError && error.kind === "invalid_request") {
+                    throw new RouterFrameworkError("configuration", error.message, "pi_selector_request_invalid", error.param, { cause: error });
+                }
+                throw error;
+            }
+            const selectorModel = requiredSelectorModel(request.model);
+            const candidate = model(selectorModel);
+            const result = await run({
+                request: { ...normalized, stream: false },
+                candidate: {
+                    id: candidate.id,
+                    provider: candidate.provider,
+                    api: candidate.api,
+                },
+                reasoningEffort: normalized.reasoning?.effort ?? candidate.defaultReasoningEffort ?? "off",
+                signal,
+                purpose: "selector",
+            });
+            if (result.type !== "complete") {
+                throw new RouterFrameworkError("configuration", "Pi returned a stream for a non-streaming selector request.", "pi_selector_mode_mismatch");
+            }
+            return result.output.content
+                .filter((item) => item.type === "text")
+                .map((item) => item.text)
+                .join("");
+        },
+    };
+}
+async function piDependencies(registry) {
+    const pi = await import("@mupt-ai/pi-ai");
+    const resolvedRegistry = registry ?? (await import("@mupt-ai/pi-ai/providers/all")).builtinModels();
+    return {
+        registry: resolvedRegistry,
+        supportedThinkingLevels: (model) => pi.getSupportedThinkingLevels(model),
+    };
+}
+function validateRuntimeOptions(options) {
+    if (!options || (typeof options.apiKey !== "string" && typeof options.apiKey !== "function")) {
+        throw new RouterFrameworkError("configuration", "createPiRuntime requires an API key or API-key resolver.", "pi_api_key_missing");
+    }
+    if (typeof options.apiKey === "string" && !options.apiKey.trim()) {
+        throw new RouterFrameworkError("configuration", "createPiRuntime API keys must be non-empty strings.", "pi_api_key_missing");
+    }
+    for (const [name, value] of [
+        ["timeoutMs", options.timeoutMs],
+        ["maxRetries", options.maxRetries],
+    ]) {
+        if (value !== undefined && (!Number.isSafeInteger(value) || value < 0)) {
+            throw new RouterFrameworkError("configuration", `createPiRuntime ${name} must be a non-negative integer.`, "pi_options_invalid");
+        }
+    }
+}
+function modelProvider(id) {
+    try {
+        return providerForModel(id);
+    }
+    catch (error) {
+        throw new RouterFrameworkError("configuration", `Pi model '${id}' must include a provider prefix.`, "pi_model_provider_missing", undefined, { cause: error });
+    }
+}
+function registeredModel(registry, provider, id) {
+    let nativeId;
+    try {
+        nativeId = nativeModelId(id);
+    }
+    catch {
+        nativeId = id;
+    }
+    const registered = registry.getModel(provider, nativeId) ?? registry.getModel(provider, id);
+    if (registered === undefined) {
+        throw new RouterFrameworkError("configuration", `Pi does not know model '${id}' for provider '${provider}'.`, "pi_model_not_found");
+    }
+    return registered;
+}
+function requiredSelectorModel(value) {
+    if (typeof value !== "string" || !value.trim()) {
+        throw new RouterFrameworkError("configuration", "The Pi selector request requires a model.", "pi_selector_model_missing");
+    }
+    return value;
+}
+//# sourceMappingURL=pi_runtime.js.map

package/dist/pi_runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_runtime.js","sourceRoot":"","sources":["../src/pi_runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AA8B9D,MAAM,sBAAsB,GAAG,IAAI,GAAG,CAAC;IACrC,oBAAoB;IACpB,wBAAwB;IACxB,sBAAsB;IACtB,eAAe;IACf,uBAAuB;IACvB,oBAAoB;IACpB,kBAAkB;CACnB,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,OAA+B;IAE/B,sBAAsB,CAAC,OAAO,CAAC,CAAC;IAChC,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAE5D,MAAM,KAAK,GAAG,CACZ,EAAU,EACV,YAAqD,EAAE,EAChC,EAAE;QACzB,MAAM,QAAQ,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,eAAe,CAAC,YAAY,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QACxE,MAAM,gBAAgB,GAAG,SAAS,CAAC,gBAAgB,KAAK,SAAS;YAC/D,CAAC,CAAC,YAAY,CAAC,uBAAuB,CAAC,UAAU,CAAC;YAClD,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACpC,MAAM,sBAAsB,GAAG,SAAS,CAAC,sBAAsB;eAC1D,CAAC,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;QACtE,OAAO;YACL,EAAE;YACF,QAAQ,EAAE,IAAI;YACd,QAAQ;YACR,GAAG,EAAE,UAAU,CAAC,GAAG;YACnB,gBAAgB;YAChB,GAAG,CAAC,sBAAsB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,sBAAsB,EAAE,CAAC;YAC3E,YAAY,EAAE;gBACZ,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC9C,OAAO,EAAE,IAAI;gBACb,gBAAgB,EAAE,sBAAsB,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC;gBAC5D,SAAS,EAAE,IAAI;gBACf,GAAG,SAAS,CAAC,YAAY;aAC1B;YACD,GAAG,CAAC,SAAS,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC;SAC9E,CAAC;IACJ,CAAC,CAAC;IAEF,MAAM,GAAG,GAAG,KAAK,EAAE,SAAsB,EAAiC,EAAE;QAC1E,MAAM,UAAU,GAAG,eAAe,CAChC,YAAY,CAAC,QAAQ,EACrB,SAAS,CAAC,SAAS,CAAC,QAAQ,EAC5B,SAAS,CAAC,SAAS,CAAC,EAAE,CACvB,CAAC;QACF,IAAI,UAAU,CAAC,GAAG,KAAK,SAAS,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;YAC/C,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,UAAU,SAAS,CAAC,SAAS,CAAC,EAAE,mBAAmB,SAAS,CAAC,SAAS,CAAC,GAAG,kBAAkB,UAAU,CAAC,GAAG,IAAI,EAC9G,uBAAuB,CACxB,CAAC;QACJ,CAAC;QACD,MAAM,OAAO,GAAG,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACzD,MAAM,aAAa,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QACtE,IAAI,SAAS,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;YACtF,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5E,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAC/F,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;IACnF,CAAC,CAAC;IAEF,OAAO;QACL,KAAK;QACL,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC;YAC5B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS,EAAE,KAAK,CAAC,KAAK;YACtB,eAAe,EAAE,KAAK,CAAC,QAAQ,CAAC,eAAe;YAC/C,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,OAAO,EAAE,WAAW;SACrB,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,EAAE,EAAE;YAC/D,IAAI,UAAyB,CAAC;YAC9B,IAAI,CAAC;gBACH,UAAU,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,qEAAqE;gBACrE,kDAAkD;gBAClD,IAAI,KAAK,YAAY,oBAAoB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC9E,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,KAAK,CAAC,OAAO,EACb,6BAA6B,EAC7B,KAAK,CAAC,KAAK,EACX,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;gBACJ,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,aAAa,GAAG,qBAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3D,MAAM,SAAS,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC;gBACvB,OAAO,EAAE,EAAE,GAAG,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE;gBACzC,SAAS,EAAE;oBACT,EAAE,EAAE,SAAS,CAAC,EAAE;oBAChB,QAAQ,EAAE,SAAS,CAAC,QAAS;oBAC7B,GAAG,EAAE,SAAS,CAAC,GAAI;iBACpB;gBACD,eAAe,EAAE,UAAU,CAAC,SAAS,EAAE,MAAM,IAAI,SAAS,CAAC,sBAAsB,IAAI,KAAK;gBAC1F,MAAM;gBACN,OAAO,EAAE,UAAU;aACpB,CAAC,CAAC;YACH,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,4DAA4D,EAC5D,2BAA2B,CAC5B,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO;iBACzB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;iBACtC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxB,IAAI,CAAC,EAAE,CAAC,CAAC;QACd,CAAC;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,QAAqC;IACjE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAC1C,MAAM,gBAAgB,GAAG,QAAQ,IAAI,CAAC,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACpG,OAAO;QACL,QAAQ,EAAE,gBAAgB;QAC1B,uBAAuB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,0BAA0B,CAAC,KAAK,CAAsB;KAC9F,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,OAA+B;IAC7D,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,UAAU,CAAC,EAAE,CAAC;QAC7F,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,0DAA0D,EAC1D,oBAAoB,CACrB,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;QACjE,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,qDAAqD,EACrD,oBAAoB,CACrB,CAAC;IACJ,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI;QAC1B,CAAC,WAAW,EAAE,OAAO,CAAC,SAAS,CAAC;QAChC,CAAC,YAAY,EAAE,OAAO,CAAC,UAAU,CAAC;KAC1B,EAAE,CAAC;QACX,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC;YACvE,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,mBAAmB,IAAI,kCAAkC,EACzD,oBAAoB,CACrB,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,EAAU;IAC/B,IAAI,CAAC;QACH,OAAO,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,aAAa,EAAE,mCAAmC,EAClD,2BAA2B,EAC3B,SAAS,EACT,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,QAAyB,EAAE,QAAgB,EAAE,EAAU;IAC9E,IAAI,QAAgB,CAAC;IACrB,IAAI,CAAC;QACH,QAAQ,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,QAAQ,GAAG,EAAE,CAAC;IAChB,CAAC;IACD,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC5F,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,2BAA2B,EAAE,mBAAmB,QAAQ,IAAI,EAC5D,oBAAoB,CACrB,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;QAC/C,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,2CAA2C,EAC3C,2BAA2B,CAC5B,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/pi_types.d.ts

@@ -0,0 +1,35 @@
+import type { Api, AssistantMessage, AssistantMessageEvent, Context, Model, SimpleStreamOptions } from "@mupt-ai/pi-ai";
+import type { ChatCompletionRequest, ReasoningEffort } from "./types.js";
+import type { RouterCandidate, RouterExecutorInput, RouterExecutorResult, RouterModel, RouterRequest } from "./framework_types.js";
+export type PiModel = Model<Api>;
+export type PiCredentialInput = {
+    provider: string;
+    model: string;
+    api: string;
+    purpose: "execution" | "selector";
+};
+export type PiApiKey = string | ((input: PiCredentialInput) => string | Promise<string>);
+export type PiModelRegistry = {
+    getModel(provider: string, id: string): PiModel | undefined;
+    completeSimple(model: PiModel, context: Context, options?: SimpleStreamOptions): Promise<AssistantMessage>;
+    streamSimple(model: PiModel, context: Context, options?: SimpleStreamOptions): AsyncIterable<AssistantMessageEvent>;
+};
+export type CreatePiRuntimeOptions = {
+    apiKey: PiApiKey;
+    registry?: PiModelRegistry;
+    timeoutMs?: number;
+    maxRetries?: number;
+};
+export type PiRouterModelOptions<Metadata = unknown> = Omit<RouterModel<Metadata>, "id" | "executor" | "provider" | "api">;
+export type PiRuntime = {
+    model<Metadata = unknown>(id: string, options?: Partial<PiRouterModelOptions<Metadata>>): RouterModel<Metadata>;
+    execute<Metadata>(input: RouterExecutorInput<Metadata>): Promise<RouterExecutorResult>;
+    select(request: ChatCompletionRequest, signal?: AbortSignal): Promise<string>;
+};
+export type PiExecution = {
+    request: RouterRequest;
+    candidate: Pick<RouterCandidate, "id" | "provider" | "api">;
+    reasoningEffort: ReasoningEffort;
+    signal: AbortSignal;
+    purpose: PiCredentialInput["purpose"];
+};

package/dist/pi_types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=pi_types.js.map

package/dist/pi_types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_types.js","sourceRoot":"","sources":["../src/pi_types.ts"],"names":[],"mappings":""}

package/dist/portable_reasoning_state.d.ts

@@ -0,0 +1,7 @@
+import type { ProviderIdentity } from "./continuation_state.js";
+export type PortableReasoningState = {
+    source?: ProviderIdentity;
+    itemId?: string;
+};
+export declare function encodePortableReasoningState(state: PortableReasoningState): string;
+export declare function decodeUntrustedPortableReasoningState(value: string): PortableReasoningState | null;