@mupt-ai/dari-router 0.2.5 → 1.0.290

package/dist/pi_options.js

@@ -1,300 +0,0 @@
-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.maxOutputTokens === undefined
-            ? {}
-            : { maxTokens: request.generation.maxOutputTokens }),
-        ...(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 }),
-    };
-    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, execution);
-    if (patch !== undefined)
-        options.onPayload = patch;
-    return options;
-}
-// Reject controls the provider API cannot represent before starting a request.
-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.toolChoice === "none" && model.api === "bedrock-converse-stream") {
-        throw unsupportedPiOption("disabling tools 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 (api === "anthropic-messages") {
-        // Anthropic has no json_object mode; use a permissive object schema.
-        return {
-            type: "json_schema",
-            schema: format.type === "json_schema" ? format.schema : { type: "object" },
-        };
-    }
-    if (format.type !== "json_schema")
-        return format;
-    const { type: _, ...schema } = format;
-    return isOpenAIResponsesApi(api)
-        ? { type: "json_schema", ...schema }
-        : { type: "json_schema", json_schema: schema };
-}
-function piToolChoice(choice, api) {
-    if (api === "anthropic-messages") {
-        return typeof choice === "string"
-            ? { type: choice === "required" ? "any" : choice }
-            : { type: "tool", name: choice.name };
-    }
-    if (typeof choice === "string")
-        return choice;
-    return isOpenAIResponsesApi(api)
-        ? { type: "function", name: choice.name }
-        : { type: "function", function: { name: choice.name } };
-}
-function payloadPatch(request, model, execution) {
-    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 &&
-        request.responseFormat === undefined &&
-        request.toolChoice === undefined &&
-        request.generation.topP === undefined &&
-        request.generation.stop === undefined) {
-        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;
-        const api = model.api;
-        const google = api === "google-generative-ai" || api === "google-vertex";
-        const config = isRecord(payload.config) ? payload.config : {};
-        if (request.generation.topP !== undefined || request.generation.stop !== undefined) {
-            const stop = request.generation.stop;
-            const sequences = typeof stop === "string" ? [stop] : stop;
-            const controls = {
-                ...(request.generation.topP === undefined ? {} : { [google || api === "bedrock-converse-stream" ? "topP" : "top_p"]: request.generation.topP }),
-                ...(stop === undefined ? {} : { [google || api === "bedrock-converse-stream" ? "stopSequences" : api === "anthropic-messages" ? "stop_sequences" : "stop"]: api === "openai-completions" ? stop : sequences }),
-            };
-            payload = google ? { ...payload, config: { ...config, ...controls } }
-                : api === "bedrock-converse-stream" ? { ...payload, inferenceConfig: { ...(isRecord(payload.inferenceConfig) ? payload.inferenceConfig : {}), ...controls } }
-                    : { ...payload, ...controls };
-        }
-        if (request.responseFormat !== undefined &&
-            (request.responseFormat.type !== "text" || isOpenAIResponsesApi(api) || api === "openai-completions")) {
-            const format = piResponseFormat(request.responseFormat, api);
-            if (google) {
-                payload = { ...payload, config: { ...(isRecord(payload.config) ? payload.config : {}), responseMimeType: "application/json", ...(request.responseFormat.type === "json_schema" ? { responseJsonSchema: request.responseFormat.schema } : {}) } };
-            }
-            else if (api === "bedrock-converse-stream") {
-                const response = request.responseFormat;
-                if (response.type === "json_schema")
-                    payload = { ...payload, outputConfig: { textFormat: { type: "json_schema", structure: { jsonSchema: { name: response.name, schema: JSON.stringify(response.schema), ...(response.description === undefined ? {} : { description: response.description }) } } } } };
-            }
-            else if (isOpenAIResponsesApi(api)) {
-                payload = { ...payload, text: { ...(isRecord(payload.text) ? payload.text : {}), format } };
-            }
-            else if (api === "anthropic-messages") {
-                payload = { ...payload, output_config: { ...(isRecord(payload.output_config) ? payload.output_config : {}), format } };
-            }
-            else {
-                payload = { ...payload, response_format: format };
-            }
-        }
-        if (request.toolChoice !== undefined) {
-            const choice = request.toolChoice;
-            if (google) {
-                const mode = choice === "none" ? "NONE" : choice === "required" ? "ANY" : "AUTO";
-                payload = { ...payload, config: { ...(isRecord(payload.config) ? payload.config : {}), toolConfig: { functionCallingConfig: { mode } } } };
-            }
-            else if (api === "bedrock-converse-stream") {
-                payload = { ...payload, toolConfig: { ...(isRecord(payload.toolConfig) ? payload.toolConfig : {}), toolChoice: typeof choice === "string" ? choice === "required" ? { any: {} } : { auto: {} } : { tool: { name: choice.name } } } };
-            }
-            else if (Array.isArray(payload.tools) && payload.tools.length > 0) {
-                payload = { ...payload, tool_choice: piToolChoice(choice, api) };
-            }
-            else {
-                const { tool_choice: _, ...withoutToolChoice } = payload;
-                payload = withoutToolChoice;
-            }
-        }
-        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);
-        if (api !== "bedrock-converse-stream" &&
-            !google &&
-            (!Array.isArray(payload.tools) || payload.tools.length === 0)) {
-            const { tool_choice: _, ...withoutToolChoice } = payload;
-            payload = withoutToolChoice;
-        }
-        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

@@ -1 +0,0 @@
-{"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,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,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;KACzF,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,EAAE,SAAS,CAAC,CAAC;IACtD,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;IACnD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,+EAA+E;AAC/E,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,UAAU,KAAK,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,yBAAyB,EAAE,CAAC;QAC7E,MAAM,mBAAmB,CAAC,kCAAkC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAC3E,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,GAAG,KAAK,oBAAoB,EAAE,CAAC;QACjC,qEAAqE;QACrE,OAAO;YACL,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,MAAM,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC3E,CAAC;IACJ,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,aAAa;QAAE,OAAO,MAAM,CAAC;IACjD,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;IACtC,OAAO,oBAAoB,CAAC,GAAG,CAAC;QAC9B,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,MAAM,EAAE;QACpC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,YAAY,CAAC,MAAwB,EAAE,GAAW;IACzD,IAAI,GAAG,KAAK,oBAAoB,EAAE,CAAC;QACjC,OAAO,OAAO,MAAM,KAAK,QAAQ;YAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE;YAClD,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;IAC1C,CAAC;IACD,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,OAAO,oBAAoB,CAAC,GAAG,CAAC;QAC9B,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;QACzC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,YAAY,CACnB,OAAsB,EACtB,KAAc,EACd,SAAsB;IAEtB,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,IACE,CAAC,WAAW;QACZ,QAAQ,KAAK,SAAS;QACtB,CAAC,aAAa;QACd,CAAC,gBAAgB;QACjB,OAAO,CAAC,cAAc,KAAK,SAAS;QACpC,OAAO,CAAC,UAAU,KAAK,SAAS;QAChC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS;QACrC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,EACrC,CAAC;QACD,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,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;QACtB,MAAM,MAAM,GAAG,GAAG,KAAK,sBAAsB,IAAI,GAAG,KAAK,eAAe,CAAC;QACzE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YACnF,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;YACrC,MAAM,SAAS,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC3D,MAAM,QAAQ,GAAG;gBACf,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG,KAAK,yBAAyB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBAC/I,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG,KAAK,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,KAAK,oBAAoB,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;aAC/M,CAAC;YACF,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,GAAG,QAAQ,EAAE,EAAE;gBACnE,CAAC,CAAC,GAAG,KAAK,yBAAyB,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,eAAe,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,QAAQ,EAAE,EAAE;oBAC7J,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;QAClC,CAAC;QACD,IACE,OAAO,CAAC,cAAc,KAAK,SAAS;YACpC,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,KAAK,MAAM,IAAI,oBAAoB,CAAC,GAAG,CAAC,IAAI,GAAG,KAAK,oBAAoB,CAAC,EACrG,CAAC;YACD,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YAC7D,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,OAAO,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACnP,CAAC;iBAAM,IAAI,GAAG,KAAK,yBAAyB,EAAE,CAAC;gBAC7C,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,CAAC;gBACxC,IAAI,QAAQ,CAAC,IAAI,KAAK,aAAa;oBAAE,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,YAAY,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,QAAQ,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;YAC1S,CAAC;iBAAM,IAAI,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrC,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;YAC9F,CAAC;iBAAM,IAAI,GAAG,KAAK,oBAAoB,EAAE,CAAC;gBACxC,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,aAAa,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;YACzH,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC;YACpD,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;YAClC,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,IAAI,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;gBACjF,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,qBAAqB,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YAC7I,CAAC;iBAAM,IAAI,GAAG,KAAK,yBAAyB,EAAE,CAAC;gBAC7C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACvO,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpE,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;YACnE,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,GAAG,iBAAiB,EAAE,GAAG,OAAO,CAAC;gBACzD,OAAO,GAAG,iBAAiB,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,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,IACE,GAAG,KAAK,yBAAyB;YACjC,CAAC,MAAM;YACP,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,EAC7D,CAAC;YACD,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,GAAG,iBAAiB,EAAE,GAAG,OAAO,CAAC;YACzD,OAAO,GAAG,iBAAiB,CAAC;QAC9B,CAAC;QACD,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

@@ -1,4 +0,0 @@
-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>;

package/dist/pi_output.js

@@ -1,328 +0,0 @@
-import { RouterFrameworkError } from "./framework_error.js";
-import { decodeOpenAIReasoningSignature } from "./pi_reasoning_signature.js";
-import { isRecord } from "./types.js";
-import { isOpenAIReasoningContinuationApi, } 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 (isOpenAIReasoningContinuationApi(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