@astrosheep/keiyaku 2.8.5 → 2.9.0

package/build/agents/providers/pi.js

@@ -1,557 +0,0 @@
-import { z } from "zod";
-import { FlowError } from "../../flow-error.js";
-import { appendDebugLog } from "../../telemetry/debug-log.js";
-import { SUBAGENT_SESSION_VERSION } from "../session.js";
-import { createProjectionDriver } from "../harness/execution-handle.js";
-import { activityCallFromNamedTool, boundActivityExcerpt, diffstatFromUnifiedPatch } from "../harness/activity-values.js";
-import { AGENT_THOUGHT_TEXT_MAX_CHARS } from "../harness/events.js";
-import { createAbortError, emitProviderEvidence, sha256Utf8 } from "../harness/runtime.js";
-import { SubagentFailure, } from "../types.js";
-/** Opaque Pi resume handle: exact session file path (UUID alone is ambiguous). */
-export const piSessionSchema = z.object({
-    sessionFile: z.string().trim().min(1),
-    sessionId: z.string().trim().min(1).optional(),
-});
-const PI_PROVIDER = "pi";
-/** Exact model shape: `<provider>/<id>` (single slash separating non-empty sides). */
-const PI_MODEL_PATTERN = /^[^/]+\/[^/]+$/;
-const PI_ACKNOWLEDGED_UNPERSISTED = new Set(["message_update", "queue_update"]);
-const PI_MAPPED = new Set([
-    "agent_start",
-    "agent_end",
-    "turn_start",
-    "turn_end",
-    "message_start",
-    "message_end",
-    "tool_execution_start",
-    "tool_execution_update",
-    "tool_execution_end",
-    "compaction_start",
-    "compaction_end",
-    "session_info_changed",
-    "thinking_level_changed",
-    "auto_retry_start",
-    "auto_retry_end",
-]);
-function piIngressDisposition(type) {
-    if (PI_ACKNOWLEDGED_UNPERSISTED.has(type))
-        return "acknowledged-unpersisted";
-    if (PI_MAPPED.has(type))
-        return "mapped";
-    return "unknown";
-}
-function toPiSdkThinkingLevel(effort) {
-    return effort; // opacity boundary
-}
-let piSdkLoader;
-const importPiSdk = new Function("specifier", "return import(specifier)");
-async function loadPiSdk() {
-    if (piSdkLoader) {
-        return piSdkLoader();
-    }
-    return (await importPiSdk("@earendil-works/pi-coding-agent"));
-}
-export function setPiSdkLoaderForTests(loader) {
-    piSdkLoader = loader;
-}
-export function resetPiSdkLoaderForTests() {
-    piSdkLoader = undefined;
-}
-export function parsePiModelRef(model) {
-    const trimmed = model.trim();
-    if (!PI_MODEL_PATTERN.test(trimmed)) {
-        throw new FlowError("INVALID_SETTINGS", `Invalid pi model '${model}'. Expected exact shape '<provider>/<id>'.`);
-    }
-    const slash = trimmed.indexOf("/");
-    return {
-        provider: trimmed.slice(0, slash),
-        id: trimmed.slice(slash + 1),
-    };
-}
-function extractAssistantText(event) {
-    if (event.type === "message_end") {
-        const message = event.message;
-        if (message?.role !== "assistant")
-            return undefined;
-        return extractContentText(message.content);
-    }
-    return undefined;
-}
-function extractContentText(content) {
-    if (typeof content === "string") {
-        const trimmed = content.trim();
-        return trimmed || undefined;
-    }
-    if (!Array.isArray(content))
-        return undefined;
-    const text = content
-        .map((block) => {
-        if (typeof block === "string")
-            return block;
-        if (block &&
-            typeof block === "object" &&
-            block.type === "text") {
-            const candidate = block.text;
-            return typeof candidate === "string" ? candidate : "";
-        }
-        return "";
-    })
-        .join("")
-        .trim();
-    return text || undefined;
-}
-function emitPiThoughts(builder, event, turnOrdinal, messageOrdinal) {
-    if (event.type !== "message_end")
-        return;
-    const message = event.message;
-    if (message?.role !== "assistant" || !Array.isArray(message.content))
-        return;
-    message.content.forEach((block, blockIndex) => {
-        if (!block || typeof block !== "object")
-            return;
-        const record = block;
-        if (record.type !== "thinking" || record.redacted === true)
-            return;
-        if (typeof record.thinking !== "string" || record.thinking.trim().length === 0)
-            return;
-        const truncated = record.thinking.length > AGENT_THOUGHT_TEXT_MAX_CHARS;
-        builder.emit({
-            type: "thought",
-            id: `t${turnOrdinal}:m${messageOrdinal}:b${blockIndex}`,
-            text: record.thinking.slice(0, AGENT_THOUGHT_TEXT_MAX_CHARS),
-            final: true,
-            ...(truncated ? { truncated: true } : {}),
-        });
-    });
-}
-function piRecord(value) {
-    return value !== null && typeof value === "object" && !Array.isArray(value)
-        ? value
-        : undefined;
-}
-function piCallFacts(args) {
-    const value = piRecord(args);
-    if (!value)
-        return {};
-    return {
-        ...(typeof value.command === "string" ? { command: value.command } : {}),
-        ...(typeof value.description === "string" ? { description: value.description } : {}),
-        ...(typeof value.path === "string" ? { path: value.path } : {}),
-        ...(typeof value.offset === "number" && Number.isFinite(value.offset) ? { offset: value.offset } : {}),
-        ...(typeof value.limit === "number" && Number.isFinite(value.limit) ? { limit: value.limit } : {}),
-        ...(typeof value.query === "string" ? { query: value.query } : {}),
-    };
-}
-function piResult(value, isError) {
-    const payload = piRecord(value);
-    const content = Array.isArray(payload?.content)
-        ? payload.content.flatMap((entry) => {
-            const block = piRecord(entry);
-            return block?.type === "text" && typeof block.text === "string" ? [block.text] : [];
-        }).join("\n")
-        : undefined;
-    const details = piRecord(payload?.details);
-    const diffstat = typeof details?.patch === "string" ? diffstatFromUnifiedPatch(details.patch) : undefined;
-    return {
-        status: isError ? "error" : "ok",
-        ...(content ? { message: boundActivityExcerpt(content) } : {}),
-        ...(diffstat ? { diffstat } : {}),
-    };
-}
-function piActivityBody(event, calls) {
-    if (event.type !== "tool_execution_start" &&
-        event.type !== "tool_execution_update" &&
-        event.type !== "tool_execution_end") {
-        return undefined;
-    }
-    if (typeof event.toolCallId !== "string" || typeof event.toolName !== "string")
-        return undefined;
-    const phase = event.type === "tool_execution_start"
-        ? "started"
-        : event.type === "tool_execution_update"
-            ? "updated"
-            : "completed";
-    if (event.type === "tool_execution_start") {
-        calls.set(event.toolCallId, activityCallFromNamedTool(event.toolName, piCallFacts(event.args)));
-    }
-    const call = calls.get(event.toolCallId) ?? (event.type === "tool_execution_end"
-        ? { kind: "other", display: event.toolName }
-        : activityCallFromNamedTool(event.toolName, piCallFacts(event.args)));
-    if (event.type === "tool_execution_end")
-        calls.delete(event.toolCallId);
-    return {
-        type: "activity",
-        phase,
-        id: event.toolCallId,
-        tool: event.toolName,
-        call,
-        ...(event.type === "tool_execution_start"
-            ? {}
-            : { result: piResult(event.type === "tool_execution_update" ? event.partialResult : event.result, event.type === "tool_execution_end" && event.isError) }),
-    };
-}
-function piUsageBody(event) {
-    if (event.type !== "message_end")
-        return undefined;
-    const message = event.message;
-    if (message?.role !== "assistant")
-        return undefined;
-    const usage = message.usage;
-    if (!usage || typeof usage.input !== "number" || typeof usage.output !== "number")
-        return undefined;
-    return {
-        type: "usage",
-        inputTokens: usage.input,
-        outputTokens: usage.output,
-        ...(typeof usage.cacheRead === "number" ? { cachedInputTokens: usage.cacheRead } : {}),
-        ...(typeof usage.cost?.total === "number" ? { costUsd: usage.cost.total } : {}),
-        ...(typeof message.responseModel === "string"
-            ? { model: message.responseModel }
-            : typeof message.model === "string"
-                ? { model: message.model }
-                : {}),
-    };
-}
-/** True when native messages prove the run stopped via abort (stopReason aborted). */
-function agentEndLooksAborted(event) {
-    if (event.type !== "agent_end")
-        return false;
-    const messages = event.messages;
-    for (let i = messages.length - 1; i >= 0; i -= 1) {
-        const message = messages[i];
-        if (!message || message.role !== "assistant")
-            continue;
-        if (message.stopReason === "aborted")
-            return true;
-        // Last assistant message is the terminal one for this run.
-        break;
-    }
-    return false;
-}
-/** Returns the terminal Pi provider error, if the final assistant message ended in error. */
-function agentEndProviderError(event) {
-    if (event.type !== "agent_end")
-        return undefined;
-    const messages = event.messages;
-    for (let i = messages.length - 1; i >= 0; i -= 1) {
-        const message = messages[i];
-        if (!message || message.role !== "assistant")
-            continue;
-        if (message.stopReason !== "error")
-            return undefined;
-        if (typeof message.errorMessage === "string" && message.errorMessage.trim()) {
-            return message.errorMessage;
-        }
-        return `${PI_PROVIDER} provider error (stopReason error)`;
-    }
-    return undefined;
-}
-function normalizeClearQueue(result) {
-    if (Array.isArray(result))
-        return result;
-    return [...(result.steering ?? []), ...(result.followUp ?? [])];
-}
-function buildPiSessionOutcome(session, finalMessage) {
-    const sessionFile = session.sessionFile;
-    if (!sessionFile) {
-        return new SubagentFailure("SUBAGENT_EXEC_ERROR", `${PI_PROVIDER} exec ended without a session file`);
-    }
-    return {
-        finalMessage,
-        session: {
-            provider: "pi",
-            version: SUBAGENT_SESSION_VERSION,
-            handle: {
-                sessionFile,
-                sessionId: session.sessionId,
-            },
-        },
-    };
-}
-function bindPiSessionEvents(session, builder, options, steerControl, state, setTurnOrdinal, settle) {
-    return session.subscribe((event) => {
-        builder.ingress(PI_PROVIDER, event.type, event, piIngressDisposition(event.type), () => {
-            if (event.type === "agent_start") {
-                builder.emit({ type: "turn", phase: "started" });
-            }
-            if (event.type === "turn_start") {
-                state.thoughtTurnOrdinal += 1;
-                state.assistantMessageOrdinal = -1;
-                state.activeAssistantMessageOrdinal = undefined;
-            }
-            if (event.type === "message_start") {
-                const message = event.message;
-                if (message?.role === "assistant") {
-                    state.assistantMessageOrdinal += 1;
-                    state.activeAssistantMessageOrdinal = state.assistantMessageOrdinal;
-                }
-            }
-            const activity = piActivityBody(event, state.activityCalls);
-            if (activity)
-                builder.emit(activity);
-            if (event.type === "message_end") {
-                const message = event.message;
-                if (message?.role === "assistant") {
-                    if (state.activeAssistantMessageOrdinal === undefined) {
-                        state.assistantMessageOrdinal += 1;
-                        state.activeAssistantMessageOrdinal = state.assistantMessageOrdinal;
-                    }
-                    emitPiThoughts(builder, event, state.thoughtTurnOrdinal, state.activeAssistantMessageOrdinal);
-                    state.activeAssistantMessageOrdinal = undefined;
-                }
-            }
-            const assistantText = extractAssistantText(event);
-            if (assistantText) {
-                state.lastAssistantMessage = assistantText;
-                builder.emit({ type: "message", text: assistantText });
-            }
-            const usage = piUsageBody(event);
-            if (usage)
-                builder.emit(usage);
-            if (event.type === "turn_end" && !state.abortIntent) {
-                state.turnOrdinal += 1;
-                setTurnOrdinal(state.turnOrdinal);
-                const checkpoint = { fence: { kind: "ordinal", ordinal: state.turnOrdinal } };
-                // Evidence identifies the native turn_end before the synchronous pump can
-                // consume tells at this checkpoint.
-                if (options.onEvidence) {
-                    emitProviderEvidence(options.onEvidence, { type: "checkpoint", provider: PI_PROVIDER, ...checkpoint, source: "notification" }, (text) => builder.emit({ type: "diagnostic", text }));
-                }
-                // Sync checkpoint window: claim/steer must run before returning from turn_end.
-                options.onSyncCheckpoint?.(steerControl, checkpoint);
-            }
-            if (event.type !== "agent_end" || state.settled) {
-                return;
-            }
-            if (state.abortIntent || agentEndLooksAborted(event)) {
-                settle.reject(createAbortError(`${PI_PROVIDER} exec cancelled by client`));
-                return;
-            }
-            if (event.willRetry) {
-                return;
-            }
-            const providerError = agentEndProviderError(event);
-            if (providerError) {
-                const failure = new SubagentFailure("SUBAGENT_EXEC_ERROR", providerError);
-                builder.emit({ type: "fail", message: failure.message });
-                settle.reject(failure);
-                return;
-            }
-            const outcome = buildPiSessionOutcome(session, state.lastAssistantMessage);
-            if (outcome instanceof SubagentFailure) {
-                builder.emit({ type: "fail", message: outcome.message });
-                settle.reject(outcome);
-                return;
-            }
-            builder.emit({ type: "turn", phase: "completed" });
-            builder.emit({ type: "done", finalMessage: outcome.finalMessage, session: outcome.session });
-            settle.resolve(outcome);
-        });
-    });
-}
-export async function startPi(prompt, cwd, options = {}) {
-    const resumeSessionFile = options.sessionFile?.trim();
-    appendDebugLog(`startPi: mode=${resumeSessionFile ? "resume" : "fresh"} model=${options.model ?? "default"}`, { cwd, section: PI_PROVIDER });
-    const sdk = await loadPiSdk();
-    const sessionManagerApi = readSessionManager(sdk);
-    const sessionManager = resumeSessionFile
-        ? sessionManagerApi.open(resumeSessionFile)
-        : sessionManagerApi.create(cwd);
-    let model;
-    let authStorage;
-    let modelRegistry;
-    if (options.model !== undefined && options.model.trim().length > 0) {
-        const original = options.model.trim();
-        const ref = parsePiModelRef(original);
-        authStorage = sdk.AuthStorage.create();
-        modelRegistry = sdk.ModelRegistry.create(authStorage);
-        model = modelRegistry.find(ref.provider, ref.id);
-        if (!model) {
-            throw new SubagentFailure("SUBAGENT_PROVIDER_UNAVAILABLE", `${PI_PROVIDER} model '${original}' is not available`);
-        }
-    }
-    // With an Akuma body, retain the default loader's discovery and override only its append list.
-    // The override deliberately suppresses discovered APPEND_SYSTEM.md content.
-    if (options.appendSystemPromptOverride && (!sdk.DefaultResourceLoader || !sdk.getAgentDir)) {
-        throw new Error(`${PI_PROVIDER} SDK does not expose DefaultResourceLoader and getAgentDir`);
-    }
-    let emittedAppendEvidence = false;
-    let pendingEvidenceDiagnostics = 0;
-    let reportEvidenceDiagnostic = () => {
-        pendingEvidenceDiagnostics += 1;
-        appendDebugLog("provider evidence callback failed before Pi driver start", { cwd, section: PI_PROVIDER });
-    };
-    const observedAppendOverride = options.appendSystemPromptOverride && options.onEvidence
-        ? () => {
-            const body = options.appendSystemPromptOverride();
-            if (!emittedAppendEvidence) {
-                emittedAppendEvidence = true;
-                emitProviderEvidence(options.onEvidence, { type: "outbound", provider: PI_PROVIDER, direction: "input", kind: "system-prompt", ...sha256Utf8(body.join("\n")) }, reportEvidenceDiagnostic);
-            }
-            return body;
-        }
-        : options.appendSystemPromptOverride;
-    const resourceLoader = observedAppendOverride
-        ? new sdk.DefaultResourceLoader({
-            cwd,
-            agentDir: sdk.getAgentDir(),
-            appendSystemPromptOverride: observedAppendOverride,
-        })
-        : undefined;
-    await resourceLoader?.reload();
-    // Empty bodies retain the SDK's default ResourceLoader path unchanged.
-    const created = await sdk.createAgentSession({
-        cwd,
-        sessionManager,
-        ...(options.thinkingLevel ? { thinkingLevel: toPiSdkThinkingLevel(options.thinkingLevel) } : {}),
-        ...(model ? { model, authStorage, modelRegistry } : {}),
-        ...(resourceLoader ? { resourceLoader } : {}),
-    });
-    const session = created.session;
-    if (options.onEvidence)
-        emitProviderEvidence(options.onEvidence, { type: "session", provider: PI_PROVIDER, sessionId: session.sessionId, ...(session.sessionFile ? { sessionFile: session.sessionFile } : {}) }, reportEvidenceDiagnostic);
-    session.setSteeringMode("one-at-a-time");
-    let submittedAtTurn = 0;
-    let emitTurnStarted;
-    const enqueue = async (tell) => {
-        // Acceptance into volatile queue only. text is core-rendered; send byte-for-byte.
-        if (options.onEvidence)
-            emitProviderEvidence(options.onEvidence, { type: "outbound", provider: PI_PROVIDER, direction: "input", kind: "steer", ...sha256Utf8(tell.text) }, reportEvidenceDiagnostic);
-        if (tell.effort) {
-            if (!session.setThinkingLevel)
-                throw new Error(`${PI_PROVIDER} session does not support thinking level control`);
-            session.setThinkingLevel(toPiSdkThinkingLevel(tell.effort));
-        }
-        await session.steer(tell.text);
-        emitTurnStarted?.();
-        if (options.onEvidence)
-            emitProviderEvidence(options.onEvidence, { type: "ack", provider: PI_PROVIDER, fence: { kind: "ordinal", ordinal: submittedAtTurn }, outbound: "steer" }, reportEvidenceDiagnostic);
-        return { fence: { kind: "ordinal", ordinal: submittedAtTurn } };
-    };
-    const steerControl = { enqueue };
-    let disposed = false;
-    let unsub;
-    const disposeSession = () => {
-        if (disposed)
-            return;
-        disposed = true;
-        unsub?.();
-        try {
-            session.dispose();
-        }
-        catch {
-            // best-effort
-        }
-    };
-    let abortRequested;
-    const requestAbort = async (state) => {
-        state.abortIntent = true;
-        if (abortRequested) {
-            await abortRequested;
-            return;
-        }
-        abortRequested = (async () => {
-            // Publish abortRequested before clearQueue can synchronously emit queue_update.
-            await Promise.resolve();
-            try {
-                session.clearQueue();
-                await session.abort();
-            }
-            finally {
-                if (state.settled) {
-                    disposeSession();
-                }
-            }
-        })();
-        await abortRequested;
-    };
-    return createProjectionDriver({
-        steerMode: "mid-execution",
-        initialTurn: options.initialTurn,
-        enqueue,
-        onIngress: options.onIngress,
-        async abort() {
-            await requestAbort({ abortIntent: true, settled: false });
-        },
-        async run(builder) {
-            reportEvidenceDiagnostic = () => builder.emit({ type: "diagnostic", text: "provider evidence callback failed" });
-            emitTurnStarted = () => builder.emit({ type: "turn", phase: "started" });
-            while (pendingEvidenceDiagnostics > 0) {
-                pendingEvidenceDiagnostics -= 1;
-                reportEvidenceDiagnostic();
-            }
-            const state = {
-                abortIntent: false,
-                lastAssistantMessage: "",
-                settled: false,
-                turnOrdinal: 0,
-                thoughtTurnOrdinal: 0,
-                assistantMessageOrdinal: -1,
-                activityCalls: new Map(),
-            };
-            let settleOutcome;
-            const outcome = new Promise((resolve, reject) => {
-                settleOutcome = {
-                    resolve(value) {
-                        if (state.settled)
-                            return;
-                        state.settled = true;
-                        resolve(value);
-                    },
-                    reject(error) {
-                        if (state.settled)
-                            return;
-                        state.settled = true;
-                        reject(error);
-                    },
-                };
-                unsub = bindPiSessionEvents(session, builder, options, steerControl, state, (ordinal) => { submittedAtTurn = ordinal; }, settleOutcome);
-            });
-            outcome.catch(() => undefined);
-            const onAbort = () => {
-                void requestAbort(state)
-                    .catch((error) => {
-                    settleOutcome.reject(error);
-                })
-                    .finally(() => {
-                    if (!state.settled) {
-                        settleOutcome.reject(createAbortError(`${PI_PROVIDER} exec cancelled by client`));
-                    }
-                });
-            };
-            builder.signal.addEventListener("abort", onAbort, { once: true });
-            options.signal?.addEventListener("abort", onAbort, { once: true });
-            try {
-                if (builder.signal.aborted || options.signal?.aborted) {
-                    await requestAbort(state);
-                    throw createAbortError(`${PI_PROVIDER} exec cancelled by client`);
-                }
-                const promptPromise = session.prompt(prompt).catch((error) => {
-                    if (state.settled) {
-                        return;
-                    }
-                    settleOutcome.reject(state.abortIntent ? createAbortError(`${PI_PROVIDER} exec cancelled by client`) : error);
-                });
-                const result = await outcome;
-                await promptPromise;
-                return result;
-            }
-            catch (error) {
-                if (state.abortIntent) {
-                    throw createAbortError(`${PI_PROVIDER} exec cancelled by client`);
-                }
-                throw error;
-            }
-            finally {
-                builder.signal.removeEventListener("abort", onAbort);
-                options.signal?.removeEventListener("abort", onAbort);
-                await abortRequested?.catch(() => undefined);
-                disposeSession();
-            }
-        },
-    });
-}
-function readSessionManager(sdk) {
-    const store = sdk[`Session${"Man"}${"ager"}`];
-    return store;
-}