@mupt-ai/dari-router 0.1.0

package/dist/pi_context.js

@@ -0,0 +1,206 @@
+import { RouterFrameworkError } from "./framework_error.js";
+import { encodeOpenAIReasoningSignature } from "./pi_reasoning_signature.js";
+import { isRecord } from "./types.js";
+import { compatibleProviderContinuation, isOpenAIResponsesApi, } from "./continuation_state.js";
+export function piContext(request, model) {
+    const system = [];
+    const messages = [];
+    const calls = new Map();
+    const timestamp = Date.now();
+    let assistant;
+    const flushAssistant = () => {
+        if (assistant === undefined)
+            return;
+        if (assistant.content.length === 0)
+            assistant.content.push({ type: "text", text: "" });
+        assistant.stopReason = assistant.content.some((part) => part.type === "toolCall")
+            ? "toolUse"
+            : "stop";
+        messages.push(assistant);
+        assistant = undefined;
+    };
+    const currentAssistant = () => {
+        assistant ??= {
+            role: "assistant",
+            content: [],
+            api: model.api,
+            provider: model.provider,
+            model: model.id,
+            usage: zeroPiUsage(),
+            stopReason: "stop",
+            timestamp,
+        };
+        return assistant;
+    };
+    for (const item of request.items) {
+        if (item.type === "message" && (item.role === "system" || item.role === "developer")) {
+            flushAssistant();
+            system.push(systemText(item.content));
+            continue;
+        }
+        if (item.type === "message" && item.role === "user") {
+            flushAssistant();
+            messages.push({
+                role: "user",
+                content: piContent(item.content, "user message"),
+                timestamp,
+            });
+            continue;
+        }
+        if (item.type === "message") {
+            const pending = currentAssistant();
+            pending.content.push(...item.content.map((part) => {
+                if (part.type !== "text") {
+                    throw new RouterFrameworkError("invalid_request", "Pi does not support image content in assistant messages.", "pi_assistant_image_unsupported");
+                }
+                return { type: "text", text: part.text };
+            }));
+            continue;
+        }
+        if (item.type === "tool_call") {
+            const argumentsValue = toolArguments(item.arguments, item.id);
+            currentAssistant().content.push({
+                type: "toolCall",
+                id: item.id,
+                name: item.name,
+                arguments: argumentsValue,
+            });
+            calls.set(item.id, item.name);
+            continue;
+        }
+        if (item.type === "reasoning") {
+            const target = {
+                provider: model.provider,
+                api: model.api,
+                model: model.id,
+            };
+            const continuation = compatibleProviderContinuation(item.continuation, target);
+            if (continuation) {
+                currentAssistant().content.push(thinkingContent(item, continuation));
+            }
+            else {
+                const readable = [...(item.summary ?? []), ...item.content].join("\n");
+                if (readable) {
+                    // Cross-provider: the encrypted blob is not replayable, so carry the
+                    // readable reasoning as plain text instead of dropping the prior
+                    // turn's chain-of-thought entirely.
+                    currentAssistant().content.push({ type: "text", text: readable });
+                }
+            }
+            continue;
+        }
+        if (item.type === "hosted_tool_call") {
+            if (!isOpenAIResponsesApi(model.api)) {
+                // Dropping the item would silently lose a conversation turn.
+                throw new RouterFrameworkError("configuration", `Hosted tool calls are not representable on the '${model.api}' API; route hosted-tool conversations to a Responses-API model.`, "hosted_tool_call_unrepresentable");
+            }
+            currentAssistant().content.push({
+                type: "providerItem",
+                item: item.payload,
+            });
+            continue;
+        }
+        flushAssistant();
+        messages.push({
+            role: "toolResult",
+            toolCallId: item.toolCallId,
+            toolName: calls.get(item.toolCallId) ?? "tool",
+            content: piContent(item.content, `tool result '${item.toolCallId}'`),
+            isError: item.isError ?? false,
+            timestamp,
+        });
+    }
+    flushAssistant();
+    const tools = request.tools.map((tool) => ({
+        name: tool.name,
+        description: tool.description ?? "",
+        parameters: tool.inputSchema,
+    }));
+    return {
+        ...(system.length === 0 ? {} : { systemPrompt: system.join("\n\n") }),
+        messages,
+        ...(tools.length === 0 ? {} : { tools }),
+    };
+}
+function thinkingContent(item, continuation) {
+    if (continuation.kind === "openai_reasoning") {
+        const providerItemId = continuation.providerItemId ?? item.id;
+        return {
+            type: "thinking",
+            thinking: [...(item.summary ?? []), ...item.content].join("\n"),
+            thinkingSignature: encodeOpenAIReasoningSignature({
+                ...(providerItemId ? { id: providerItemId } : {}),
+                summary: item.summary ?? [],
+                content: item.content,
+                encryptedContent: continuation.encryptedContent,
+            }),
+        };
+    }
+    if (continuation.kind === "anthropic_thinking") {
+        return {
+            type: "thinking",
+            thinking: continuation.thinking,
+            thinkingSignature: continuation.signature,
+        };
+    }
+    return {
+        type: "thinking",
+        thinking: "",
+        thinkingSignature: continuation.data,
+        redacted: true,
+    };
+}
+function systemText(content) {
+    return content.map((part) => {
+        if (part.type !== "text") {
+            throw new RouterFrameworkError("invalid_request", "Pi does not support image content in system or developer messages.", "pi_system_image_unsupported");
+        }
+        return part.text;
+    }).join("");
+}
+function piContent(content, location) {
+    return content.map((part) => {
+        if (part.type === "text")
+            return { type: "text", text: part.text };
+        const image = dataImage(part.url);
+        if (image === null) {
+            throw new RouterFrameworkError("invalid_request", `Pi ${location} images must use base64 data URLs.`, "pi_remote_image_unsupported");
+        }
+        return { type: "image", data: image.data, mimeType: image.mimeType };
+    });
+}
+function dataImage(url) {
+    const match = /^data:(image\/[a-z0-9.+-]+);base64,([a-z0-9+/=\s]+)$/i.exec(url);
+    if (match === null)
+        return null;
+    return { mimeType: match[1], data: match[2].replace(/\s/g, "") };
+}
+function toolArguments(value, id) {
+    if (typeof value !== "string")
+        return value;
+    // Providers emit "" for zero-argument tool calls; replay it as {}.
+    if (value === "")
+        return {};
+    let parsed;
+    try {
+        parsed = JSON.parse(value);
+    }
+    catch (error) {
+        throw new RouterFrameworkError("invalid_request", `Tool call '${id}' arguments must be valid JSON.`, "pi_tool_arguments_invalid", undefined, { cause: error });
+    }
+    if (!isRecord(parsed)) {
+        throw new RouterFrameworkError("invalid_request", `Tool call '${id}' arguments must be a JSON object.`, "pi_tool_arguments_invalid");
+    }
+    return parsed;
+}
+function zeroPiUsage() {
+    return {
+        input: 0,
+        output: 0,
+        cacheRead: 0,
+        cacheWrite: 0,
+        totalTokens: 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+    };
+}
+//# sourceMappingURL=pi_context.js.map

package/dist/pi_context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_context.js","sourceRoot":"","sources":["../src/pi_context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,8BAA8B,EAAE,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAMtC,OAAO,EACL,8BAA8B,EAC9B,oBAAoB,GAErB,MAAM,yBAAyB,CAAC;AAGjC,MAAM,UAAU,SAAS,CAAC,OAAsB,EAAE,KAAc;IAC9D,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,QAAQ,GAAwB,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,SAAkF,CAAC;IAEvF,MAAM,cAAc,GAAG,GAAG,EAAE;QAC1B,IAAI,SAAS,KAAK,SAAS;YAAE,OAAO;QACpC,IAAI,SAAS,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACvF,SAAS,CAAC,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC;YAC/E,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,MAAM,CAAC;QACX,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzB,SAAS,GAAG,SAAS,CAAC;IACxB,CAAC,CAAC;IACF,MAAM,gBAAgB,GAAG,GAAG,EAAE;QAC5B,SAAS,KAAK;YACZ,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,EAAE;YACf,KAAK,EAAE,WAAW,EAAE;YACpB,UAAU,EAAE,MAAM;YAClB,SAAS;SACV,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,EAAE,CAAC;YACrF,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;YACtC,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACpD,cAAc,EAAE,CAAC;YACjB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC;gBAChD,SAAS;aACV,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,gBAAgB,EAAE,CAAC;YACnC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAChD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACzB,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,0DAA0D,EAC1D,gCAAgC,CACjC,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACpD,CAAC,CAAC,CAAC,CAAC;YACJ,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9D,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;gBAC9B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,cAAc;aAC1B,CAAC,CAAC;YACH,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9B,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,MAAM,MAAM,GAA2B;gBACrC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,KAAK,EAAE,KAAK,CAAC,EAAE;aAChB,CAAC;YACF,MAAM,YAAY,GAAG,8BAA8B,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC/E,IAAI,YAAY,EAAE,CAAC;gBACjB,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvE,IAAI,QAAQ,EAAE,CAAC;oBACb,qEAAqE;oBACrE,iEAAiE;oBACjE,oCAAoC;oBACpC,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;gBACpE,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YACrC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrC,6DAA6D;gBAC7D,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,mDAAmD,KAAK,CAAC,GAAG,kEAAkE,EAC9H,kCAAkC,CACnC,CAAC;YACJ,CAAC;YACA,gBAAgB,EAAE,CAAC,OAAqD,CAAC,IAAI,CAAC;gBAC7E,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE,IAAI,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,cAAc,EAAE,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,YAAY;YAClB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,MAAM;YAC9C,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,IAAI,CAAC,UAAU,GAAG,CAAC;YACpE,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,KAAK;YAC9B,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IACD,cAAc,EAAE,CAAC;IAEjB,MAAM,KAAK,GAAW,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACjD,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;QACnC,UAAU,EAAE,IAAI,CAAC,WAAiC;KACnD,CAAC,CAAC,CAAC;IACJ,OAAO;QACL,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACrE,QAAQ;QACR,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;KACzC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CACtB,IAAoE,EACpE,YAAuC;IAEvC,IAAI,YAAY,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;QAC7C,MAAM,cAAc,GAAG,YAAY,CAAC,cAAc,IAAI,IAAI,CAAC,EAAE,CAAC;QAC9D,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;YAC/D,iBAAiB,EAAE,8BAA8B,CAAC;gBAChD,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACjD,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;gBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;aAChD,CAAC;SACH,CAAC;IACJ,CAAC;IACD,IAAI,YAAY,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;QAC/C,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,iBAAiB,EAAE,YAAY,CAAC,SAAS;SAC1C,CAAC;IACJ,CAAC;IACD,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE,EAAE;QACZ,iBAAiB,EAAE,YAAY,CAAC,IAAI;QACpC,QAAQ,EAAE,IAAI;KACf,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,OAAiC;IACnD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC1B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,oEAAoE,EACpE,6BAA6B,CAC9B,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,SAAS,SAAS,CAChB,OAAiC,EACjC,QAAgB;IAEhB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC1B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;QACnE,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,MAAM,QAAQ,oCAAoC,EAClD,6BAA6B,CAC9B,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;IACvE,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,GAAW;IAC5B,MAAM,KAAK,GAAG,uDAAuD,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChF,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAChC,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAE,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC;AACrE,CAAC;AAED,SAAS,aAAa,CAAC,KAAuC,EAAE,EAAU;IACxE,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,mEAAmE;IACnE,IAAI,KAAK,KAAK,EAAE;QAAE,OAAO,EAAE,CAAC;IAC5B,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,cAAc,EAAE,iCAAiC,EACjD,2BAA2B,EAC3B,SAAS,EACT,EAAE,KAAK,EAAE,KAAK,EAAE,CACjB,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,cAAc,EAAE,oCAAoC,EACpD,2BAA2B,CAC5B,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW;IAClB,OAAO;QACL,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACrE,CAAC;AACJ,CAAC"}

package/dist/pi_options.d.ts

@@ -0,0 +1,3 @@
+import type { SimpleStreamOptions } from "@mupt-ai/pi-ai";
+import type { CreatePiRuntimeOptions, PiExecution, PiModel } from "./pi_types.js";
+export declare function piOptions(runtime: CreatePiRuntimeOptions, execution: PiExecution, model: PiModel): Promise<SimpleStreamOptions>;

package/dist/pi_options.js

@@ -0,0 +1,240 @@
+import { isOpenAIResponsesApi } from "./continuation_state.js";
+import { RouterFrameworkError } from "./framework_error.js";
+import { isRecord } from "./types.js";
+const PAYLOAD_PATCH_APIS = new Set([
+    "anthropic-messages",
+    "azure-openai-responses",
+    "openai-completions",
+    "openai-responses",
+]);
+export async function piOptions(runtime, execution, model) {
+    rejectUnsupportedOptions(execution.request, model);
+    const apiKey = typeof runtime.apiKey === "string"
+        ? runtime.apiKey
+        : await runtime.apiKey({
+            provider: model.provider,
+            model: execution.candidate.id,
+            api: model.api,
+            purpose: execution.purpose,
+        });
+    if (typeof apiKey !== "string" || !apiKey.trim()) {
+        throw new RouterFrameworkError("configuration", `No API key was provided for Pi model '${execution.candidate.id}'.`, "pi_api_key_missing");
+    }
+    const request = execution.request;
+    const options = {
+        apiKey,
+        signal: execution.signal,
+        ...(runtime.timeoutMs === undefined ? {} : { timeoutMs: runtime.timeoutMs }),
+        ...(runtime.maxRetries === undefined ? {} : { maxRetries: runtime.maxRetries }),
+        ...(request.generation.temperature === undefined
+            ? {}
+            : { temperature: request.generation.temperature }),
+        ...(request.generation.topP === undefined ? {} : { topP: request.generation.topP }),
+        ...(request.generation.maxOutputTokens === undefined
+            ? {}
+            : { maxTokens: request.generation.maxOutputTokens }),
+        ...(request.generation.stop === undefined ? {} : { stop: request.generation.stop }),
+        ...(request.cacheKey === undefined ? {} : { sessionId: request.cacheKey }),
+        // Only anthropic-messages reads metadata (and only user_id); no other
+        // fork API consumes metadata fields.
+        ...(request.user !== undefined && model.api === "anthropic-messages"
+            ? { metadata: { user_id: request.user } }
+            : {}),
+        ...(execution.reasoningEffort === "off" ? {} : { reasoning: execution.reasoningEffort }),
+        ...(request.responseFormat === undefined
+            ? {}
+            : { responseFormat: piResponseFormat(request.responseFormat, model.api) }),
+        ...(request.toolChoice === undefined ? {} : { toolChoice: piToolChoice(request.toolChoice) }),
+    };
+    const budget = request.reasoning?.budgetTokens;
+    if (budget !== undefined &&
+        execution.reasoningEffort !== "off" &&
+        execution.reasoningEffort !== "xhigh" &&
+        execution.reasoningEffort !== "max") {
+        options.thinkingBudgets = { [execution.reasoningEffort]: budget };
+    }
+    const patch = payloadPatch(request, model);
+    if (patch !== undefined)
+        options.onPayload = patch;
+    return options;
+}
+// The fork hard-throws for these (api, option) pairs inside streamSimple;
+// reject them up front as client errors instead of surfacing 502s.
+function rejectUnsupportedOptions(request, model) {
+    if (request.generation.stop !== undefined && isOpenAIResponsesApi(model.api)) {
+        throw unsupportedPiOption("stop sequences", model.api);
+    }
+    if (request.generation.topP !== undefined && model.api === "anthropic-messages") {
+        throw unsupportedPiOption("top_p", model.api);
+    }
+    if (typeof request.toolChoice === "object" &&
+        (model.api === "google-generative-ai" || model.api === "google-vertex")) {
+        throw unsupportedPiOption("forcing a specific tool with tool_choice", model.api);
+    }
+    if (request.responseFormat?.type === "json_object" && model.api === "bedrock-converse-stream") {
+        throw unsupportedPiOption("response_format json_object", model.api);
+    }
+}
+function unsupportedPiOption(option, api) {
+    return new RouterFrameworkError("invalid_request", `Pi API '${api}' does not support ${option}.`, "pi_option_unsupported");
+}
+function piResponseFormat(format, api) {
+    if (format.type === "json_object" && api === "anthropic-messages") {
+        // Anthropic has no json_object mode; degrade to a permissive object
+        // schema so the request still produces JSON output.
+        return {
+            type: "json_schema",
+            jsonSchema: { name: "response", schema: { type: "object" } },
+        };
+    }
+    if (format.type !== "json_schema")
+        return format;
+    return {
+        type: "json_schema",
+        jsonSchema: {
+            name: format.name,
+            schema: format.schema,
+            ...(format.description === undefined ? {} : { description: format.description }),
+            ...(format.strict === undefined ? {} : { strict: format.strict }),
+        },
+    };
+}
+function piToolChoice(choice) {
+    if (choice === "required")
+        return "any";
+    if (typeof choice === "string")
+        return choice;
+    return { type: "function", function: { name: choice.name } };
+}
+function payloadPatch(request, model) {
+    const strictTools = request.tools.some((tool) => tool.strict !== undefined);
+    const parallel = request.parallelToolCalls;
+    const imageDetails = imageDetailQueues(request);
+    const thinkingPatch = model.api === "anthropic-messages" && (request.reasoning?.enabled === false || request.reasoning?.budgetTokens !== undefined);
+    const imageDetailPatch = imageDetails.size > 0 &&
+        (isOpenAIResponsesApi(model.api) || model.api === "openai-completions");
+    if (imageDetails.size > 0 &&
+        !imageDetailPatch &&
+        [...imageDetails.values()].some((queue) => queue.some((detail) => detail !== "auto"))) {
+        throw new RouterFrameworkError("invalid_request", `Pi API '${model.api}' cannot preserve image detail controls.`, "pi_image_detail_unsupported");
+    }
+    if (!strictTools && parallel === undefined && !thinkingPatch && !imageDetailPatch) {
+        return undefined;
+    }
+    if ((strictTools || parallel !== undefined) && !PAYLOAD_PATCH_APIS.has(model.api)) {
+        throw new RouterFrameworkError("invalid_request", `Pi API '${model.api}' cannot preserve strict or parallel tool controls.`, "pi_tool_control_unsupported");
+    }
+    return (value) => {
+        if (!isRecord(value))
+            return undefined;
+        let payload = value;
+        if (strictTools)
+            payload = patchStrictTools(payload, request, model.api);
+        if (parallel !== undefined)
+            payload = patchParallelTools(payload, parallel, model.api);
+        if (imageDetailPatch)
+            payload = patchImageDetails(payload, imageDetails);
+        if (thinkingPatch)
+            payload = patchAnthropicThinking(payload, request);
+        return payload;
+    };
+}
+function patchStrictTools(payload, request, api) {
+    if (!Array.isArray(payload.tools))
+        return payload;
+    const tools = new Map(request.tools.map((tool) => [tool.name, tool]));
+    return {
+        ...payload,
+        tools: payload.tools.map((entry) => {
+            if (!isRecord(entry))
+                return entry;
+            if (api === "anthropic-messages") {
+                const tool = typeof entry.name === "string" ? tools.get(entry.name) : undefined;
+                return tool?.strict === undefined ? entry : { ...entry, strict: tool.strict };
+            }
+            const definition = api === "openai-completions" && isRecord(entry.function)
+                ? entry.function
+                : entry;
+            const tool = typeof definition.name === "string" ? tools.get(definition.name) : undefined;
+            if (tool?.strict === undefined)
+                return entry;
+            return api === "openai-completions"
+                ? { ...entry, function: { ...definition, strict: tool.strict } }
+                : { ...entry, strict: tool.strict };
+        }),
+    };
+}
+function patchParallelTools(payload, enabled, api) {
+    if (api !== "anthropic-messages")
+        return { ...payload, parallel_tool_calls: enabled };
+    if (enabled)
+        return payload;
+    const current = isRecord(payload.tool_choice) ? payload.tool_choice : {};
+    return {
+        ...payload,
+        tool_choice: {
+            type: typeof current.type === "string" ? current.type : "auto",
+            ...current,
+            disable_parallel_tool_use: true,
+        },
+    };
+}
+function patchImageDetails(payload, details) {
+    const queues = new Map([...details].map(([url, queue]) => [url, [...queue]]));
+    const patch = (value) => {
+        if (Array.isArray(value))
+            return value.map(patch);
+        if (!isRecord(value))
+            return value;
+        const mapped = Object.fromEntries(Object.entries(value).map(([key, entry]) => [key, patch(entry)]));
+        const responseUrl = mapped.type === "input_image" && typeof mapped.image_url === "string"
+            ? mapped.image_url
+            : undefined;
+        const chatImage = mapped.type === "image_url" && isRecord(mapped.image_url)
+            ? mapped.image_url
+            : undefined;
+        const chatUrl = typeof chatImage?.url === "string" ? chatImage.url : undefined;
+        const detail = queues.get(responseUrl ?? chatUrl ?? "")?.shift();
+        if (detail === undefined)
+            return mapped;
+        return responseUrl === undefined
+            ? { ...mapped, image_url: { ...chatImage, detail } }
+            : { ...mapped, detail };
+    };
+    const patched = patch(payload);
+    return isRecord(patched) ? patched : payload;
+}
+function imageDetailQueues(request) {
+    const details = new Map();
+    const add = (content) => {
+        for (const part of content) {
+            if (part.type !== "image" || part.detail === undefined)
+                continue;
+            const queue = details.get(part.url) ?? [];
+            queue.push(part.detail);
+            details.set(part.url, queue);
+        }
+    };
+    for (const item of request.items) {
+        if (item.type === "message" || item.type === "tool_result")
+            add(item.content);
+    }
+    return details;
+}
+function patchAnthropicThinking(payload, request) {
+    if (request.reasoning?.enabled === false) {
+        return { ...payload, thinking: { type: "disabled" } };
+    }
+    const budget = request.reasoning?.budgetTokens;
+    if (budget === undefined)
+        return payload;
+    return {
+        ...payload,
+        thinking: {
+            ...(isRecord(payload.thinking) ? payload.thinking : {}),
+            type: "enabled",
+            budget_tokens: budget,
+        },
+    };
+}
+//# sourceMappingURL=pi_options.js.map

package/dist/pi_options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pi_options.js","sourceRoot":"","sources":["../src/pi_options.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAQtC,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC;IACjC,oBAAoB;IACpB,wBAAwB;IACxB,oBAAoB;IACpB,kBAAkB;CACnB,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,OAA+B,EAC/B,SAAsB,EACtB,KAAc;IAEd,wBAAwB,CAAC,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ;QAC/C,CAAC,CAAC,OAAO,CAAC,MAAM;QAChB,CAAC,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC;YACnB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,SAAS,CAAC,SAAS,CAAC,EAAE;YAC7B,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,OAAO,EAAE,SAAS,CAAC,OAAO;SAC3B,CAAC,CAAC;IACP,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;QACjD,MAAM,IAAI,oBAAoB,CAC5B,eAAe,EACf,yCAAyC,SAAS,CAAC,SAAS,CAAC,EAAE,IAAI,EACnE,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;IAClC,MAAM,OAAO,GAAwB;QACnC,MAAM;QACN,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;QAC5E,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC;QAC/E,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,KAAK,SAAS;YAC9C,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;QACpD,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QACnF,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,KAAK,SAAS;YAClD,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC;QACtD,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QACnF,GAAG,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC1E,sEAAsE;QACtE,qCAAqC;QACrC,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,GAAG,KAAK,oBAAoB;YAClE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE;YACzC,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,SAAS,CAAC,eAAe,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,SAAS,CAAC,eAAe,EAAE,CAAC;QACxF,GAAG,CAAC,OAAO,CAAC,cAAc,KAAK,SAAS;YACtC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,cAAc,EAAE,gBAAgB,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5E,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,YAAY,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;KAC9F,CAAC;IAEF,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,YAAY,CAAC;IAC/C,IACE,MAAM,KAAK,SAAS;QACpB,SAAS,CAAC,eAAe,KAAK,KAAK;QACnC,SAAS,CAAC,eAAe,KAAK,OAAO;QACrC,SAAS,CAAC,eAAe,KAAK,KAAK,EACnC,CAAC;QACD,OAAO,CAAC,eAAe,GAAG,EAAE,CAAC,SAAS,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC3C,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;IACnD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,0EAA0E;AAC1E,mEAAmE;AACnE,SAAS,wBAAwB,CAAC,OAAsB,EAAE,KAAc;IACtE,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,IAAI,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAC7E,MAAM,mBAAmB,CAAC,gBAAgB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,GAAG,KAAK,oBAAoB,EAAE,CAAC;QAChF,MAAM,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAChD,CAAC;IACD,IACE,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ;QACtC,CAAC,KAAK,CAAC,GAAG,KAAK,sBAAsB,IAAI,KAAK,CAAC,GAAG,KAAK,eAAe,CAAC,EACvE,CAAC;QACD,MAAM,mBAAmB,CAAC,0CAA0C,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACnF,CAAC;IACD,IAAI,OAAO,CAAC,cAAc,EAAE,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,GAAG,KAAK,yBAAyB,EAAE,CAAC;QAC9F,MAAM,mBAAmB,CAAC,6BAA6B,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACtE,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAc,EAAE,GAAW;IACtD,OAAO,IAAI,oBAAoB,CAC7B,iBAAiB,EACjB,WAAW,GAAG,sBAAsB,MAAM,GAAG,EAC7C,uBAAuB,CACxB,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,MAAoD,EACpD,GAAW;IAEX,IAAI,MAAM,CAAC,IAAI,KAAK,aAAa,IAAI,GAAG,KAAK,oBAAoB,EAAE,CAAC;QAClE,oEAAoE;QACpE,oDAAoD;QACpD,OAAO;YACL,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;SAC7D,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,aAAa;QAAE,OAAO,MAAM,CAAC;IACjD,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,UAAU,EAAE;YACV,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;YAChF,GAAG,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;SAClE;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,MAAwB;IAC5C,IAAI,MAAM,KAAK,UAAU;QAAE,OAAO,KAAK,CAAC;IACxC,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,YAAY,CACnB,OAAsB,EACtB,KAAc;IAEd,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IAC5E,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAC3C,MAAM,YAAY,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,KAAK,oBAAoB,IAAI,CAC1D,OAAO,CAAC,SAAS,EAAE,OAAO,KAAK,KAAK,IAAI,OAAO,CAAC,SAAS,EAAE,YAAY,KAAK,SAAS,CACtF,CAAC;IACF,MAAM,gBAAgB,GAAG,YAAY,CAAC,IAAI,GAAG,CAAC;QAC5C,CAAC,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,GAAG,KAAK,oBAAoB,CAAC,CAAC;IAC1E,IACE,YAAY,CAAC,IAAI,GAAG,CAAC;QACrB,CAAC,gBAAgB;QACjB,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,EACrF,CAAC;QACD,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,WAAW,KAAK,CAAC,GAAG,0CAA0C,EAC9D,6BAA6B,CAC9B,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,WAAW,IAAI,QAAQ,KAAK,SAAS,IAAI,CAAC,aAAa,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAClF,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,CAAC,WAAW,IAAI,QAAQ,KAAK,SAAS,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAClF,MAAM,IAAI,oBAAoB,CAC5B,iBAAiB,EACjB,WAAW,KAAK,CAAC,GAAG,qDAAqD,EACzE,6BAA6B,CAC9B,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,KAAK,EAAE,EAAE;QACf,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QACvC,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,WAAW;YAAE,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QACzE,IAAI,QAAQ,KAAK,SAAS;YAAE,OAAO,GAAG,kBAAkB,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QACvF,IAAI,gBAAgB;YAAE,OAAO,GAAG,iBAAiB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACzE,IAAI,aAAa;YAAE,OAAO,GAAG,sBAAsB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACtE,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,OAAgC,EAChC,OAAsB,EACtB,GAAW;IAEX,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAClD,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IACtE,OAAO;QACL,GAAG,OAAO;QACV,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACnC,IAAI,GAAG,KAAK,oBAAoB,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAChF,OAAO,IAAI,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;YAChF,CAAC;YACD,MAAM,UAAU,GAAG,GAAG,KAAK,oBAAoB,IAAI,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACzE,CAAC,CAAC,KAAK,CAAC,QAAQ;gBAChB,CAAC,CAAC,KAAK,CAAC;YACV,MAAM,IAAI,GAAG,OAAO,UAAU,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC1F,IAAI,IAAI,EAAE,MAAM,KAAK,SAAS;gBAAE,OAAO,KAAK,CAAC;YAC7C,OAAO,GAAG,KAAK,oBAAoB;gBACjC,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,EAAE,GAAG,UAAU,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE;gBAChE,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;QACxC,CAAC,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,OAAgC,EAChC,OAAgB,EAChB,GAAW;IAEX,IAAI,GAAG,KAAK,oBAAoB;QAAE,OAAO,EAAE,GAAG,OAAO,EAAE,mBAAmB,EAAE,OAAO,EAAE,CAAC;IACtF,IAAI,OAAO;QAAE,OAAO,OAAO,CAAC;IAC5B,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IACzE,OAAO;QACL,GAAG,OAAO;QACV,WAAW,EAAE;YACX,IAAI,EAAE,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM;YAC9D,GAAG,OAAO;YACV,yBAAyB,EAAE,IAAI;SAChC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CACxB,OAAgC,EAChC,OAAoD;IAEpD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,MAAM,KAAK,GAAG,CAAC,KAAc,EAAW,EAAE;QACxC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACnC,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAC/B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CACjE,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,KAAK,aAAa,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ;YACvF,CAAC,CAAC,MAAM,CAAC,SAAS;YAClB,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,KAAK,WAAW,IAAI,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC;YACzE,CAAC,CAAC,MAAM,CAAC,SAAS;YAClB,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,OAAO,GAAG,OAAO,SAAS,EAAE,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/E,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,WAAW,IAAI,OAAO,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;QACjE,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC;QACxC,OAAO,WAAW,KAAK,SAAS;YAC9B,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,EAAE;YACpD,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,CAAC;IAC5B,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;IAC/B,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;AAC/C,CAAC;AAED,SAAS,iBAAiB,CACxB,OAAsB;IAEtB,MAAM,OAAO,GAAG,IAAI,GAAG,EAA0C,CAAC;IAClE,MAAM,GAAG,GAAG,CAAC,OAAiC,EAAE,EAAE;QAChD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;gBAAE,SAAS;YACjE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAC1C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC,CAAC;IACF,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa;YAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,sBAAsB,CAC7B,OAAgC,EAChC,OAAsB;IAEtB,IAAI,OAAO,CAAC,SAAS,EAAE,OAAO,KAAK,KAAK,EAAE,CAAC;QACzC,OAAO,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC;IACxD,CAAC;IACD,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,YAAY,CAAC;IAC/C,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACzC,OAAO;QACL,GAAG,OAAO;QACV,QAAQ,EAAE;YACR,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;YACvD,IAAI,EAAE,SAAS;YACf,aAAa,EAAE,MAAM;SACtB;KACF,CAAC;AACJ,CAAC"}

package/dist/pi_output.d.ts

@@ -0,0 +1,4 @@
+import type { AssistantMessage, AssistantMessageEvent } from "@mupt-ai/pi-ai";
+import { type RouterCompletion, type RouterStreamEvent } from "./framework_types.js";
+export declare function routerCompletion(message: AssistantMessage, signal: AbortSignal): RouterCompletion;
+export declare function routerEvents(events: AsyncIterable<AssistantMessageEvent>, signal: AbortSignal): AsyncIterable<RouterStreamEvent>;