pi-kot 0.1.21 → 0.1.22

package/server/routes/prompt.js

@@ -0,0 +1,311 @@
+import { getSession, autoNameSession } from "../session-store.js";
+import { writeFile, mkdir } from "node:fs/promises";
+import { join } from "node:path";
+import { randomUUID } from "node:crypto";
+import { expandFileReferences } from "../path-refs.js";
+import { getProject } from "../workspace-store.js";
+const ABORT_TIMEOUT_MS = 10_000;
+const VISION_TEMP_DIR = "/tmp/pi-kot-vision";
+/**
+ * If the session's model doesn't support images natively (no "image" in
+ * model.input), save the image data to temp files and inject file-path
+ * markers into the text. This lets text-only models use tools like
+ * describe_image that accept file paths.
+ *
+ * If the model supports images, passes through unchanged.
+ */
+async function maybeSaveImagesToFiles(model, text, images) {
+    if (!images || images.length === 0)
+        return { text, images };
+    if (model?.input?.includes("image"))
+        return { text, images };
+    try {
+        await mkdir(VISION_TEMP_DIR, { recursive: true });
+        const paths = [];
+        for (const img of images) {
+            const ext = img.mimeType.split("/")[1] ?? "png";
+            const filepath = join(VISION_TEMP_DIR, `${randomUUID()}.${ext}`);
+            await writeFile(filepath, Buffer.from(img.data, "base64"));
+            paths.push(filepath);
+        }
+        const pathMarkers = paths.map((p) => `[Image: ${p}]`).join("\n");
+        return { text: `${text}\n\n${pathMarkers}`, images: undefined };
+    }
+    catch (err) {
+        console.error("[prompt] failed to save images to files, dropping:", err);
+        // Fall back to stripping images silently
+        return { text, images: undefined };
+    }
+}
+/**
+ * Pre-flight checks shared by prompt + steer routes.
+ * Returns the LiveSession or undefined (after sending a 4xx reply).
+ */
+async function preflight(req, reply) {
+    const sessionId = req.params.id;
+    const live = getSession(sessionId);
+    if (live === undefined) {
+        await reply.code(404).send({ error: "session_not_found", message: "no live session with that id" });
+        return undefined;
+    }
+    const model = live.session.model;
+    if (model === undefined) {
+        await reply.code(400).send({
+            error: "no_model_configured",
+            message: "no model is configured for this session",
+        });
+        return undefined;
+    }
+    // Check auth is configured for the current model
+    if (!live.session.modelRegistry.hasConfiguredAuth(model)) {
+        await reply.code(400).send({
+            error: "auth_not_configured",
+            message: `No API key configured for provider "${model.provider}". Add one via PUT /api/v1/config/auth/${model.provider}.`,
+        });
+        return undefined;
+    }
+    return live;
+}
+export const promptRoutes = async (fastify) => {
+    // POST /api/v1/sessions/:id/prompt — fire-and-forget, returns 202
+    fastify.post("/sessions/:id/prompt", {
+        schema: {
+            description: "Send a prompt to the session. Returns 202 immediately; the agent " +
+                "response streams over GET /sessions/:id/stream.\n\n" +
+                "Pi SDK key fact: session.prompt() resolves only after the FULL " +
+                "agent run finishes (including retries + compaction). Routes MUST " +
+                "NOT await it — call without await and return 202 immediately. " +
+                "Output streams over SSE.",
+            tags: ["sessions"],
+            params: {
+                type: "object",
+                required: ["id"],
+                properties: { id: { type: "string" } },
+            },
+            body: {
+                type: "object",
+                required: ["text"],
+                additionalProperties: false,
+                properties: {
+                    text: { type: "string", minLength: 1 },
+                    streamingBehavior: { type: "string", enum: ["steer", "followUp"] },
+                    images: {
+                        type: "array",
+                        items: {
+                            type: "object",
+                            required: ["type", "data", "mimeType"],
+                            properties: {
+                                type: { type: "string", const: "image" },
+                                data: { type: "string" },
+                                mimeType: { type: "string" },
+                            },
+                        },
+                    },
+                },
+            },
+            response: {
+                202: {
+                    type: "object",
+                    required: ["accepted"],
+                    properties: { accepted: { type: "boolean", const: true } },
+                },
+                400: {
+                    type: "object",
+                    properties: {
+                        error: { type: "string" },
+                        message: { type: "string" },
+                    },
+                },
+                404: {
+                    type: "object",
+                    properties: {
+                        error: { type: "string" },
+                        message: { type: "string" },
+                    },
+                },
+            },
+        },
+    }, async (req, reply) => {
+        const live = await preflight(req, reply);
+        if (live === undefined)
+            return reply;
+        let { text, streamingBehavior, images } = req.body;
+        // Expand @-file references before sending to the LLM
+        const project = await getProject(live.projectId);
+        if (project !== undefined) {
+            text = await expandFileReferences(text, project.path);
+        }
+        // Save images to files if the model doesn't support vision natively
+        let transformed = await maybeSaveImagesToFiles(live.session.model, text, images);
+        text = transformed.text;
+        images = transformed.images;
+        const opts = {};
+        if (streamingBehavior !== undefined)
+            opts.streamingBehavior = streamingBehavior;
+        if (images !== undefined && images.length > 0)
+            opts.images = images;
+        // Auto-name from the first prompt if no name is set yet
+        autoNameSession(req.params.id, text);
+        // Synthesize a failure event if prompt() rejects without emitting agent_end.
+        const synthesizeFailureEvent = (err) => {
+            const errorMessage = err instanceof Error ? err.message : String(err);
+            for (const client of live.clients) {
+                try {
+                    client.send({
+                        type: "agent_end",
+                        sessionId: req.params.id,
+                        errorMessage,
+                    });
+                }
+                catch {
+                    // single client send-failure shouldn't stop fan-out
+                }
+            }
+        };
+        // Fire-and-forget — the promise resolves only after the ENTIRE
+        // agent run is complete, so we DO NOT await it.
+        live.session.prompt(text, opts).catch((err) => {
+            synthesizeFailureEvent(err);
+        });
+        return reply.code(202).send({ accepted: true });
+    });
+    // POST /api/v1/sessions/:id/abort — abort current run
+    fastify.post("/sessions/:id/abort", {
+        schema: {
+            description: "Abort the current agent run.",
+            tags: ["sessions"],
+            params: {
+                type: "object",
+                required: ["id"],
+                properties: { id: { type: "string" } },
+            },
+            response: {
+                200: {
+                    type: "object",
+                    required: ["aborted"],
+                    properties: { aborted: { type: "boolean" } },
+                },
+                404: {
+                    type: "object",
+                    properties: { error: { type: "string" } },
+                },
+            },
+        },
+    }, async (req, reply) => {
+        const live = getSession(req.params.id);
+        if (live === undefined) {
+            return reply.code(404).send({ error: "session_not_found" });
+        }
+        try {
+            // Fire the abort signal immediately, then wait for the agent to
+            // become idle with a timeout. The SDK's session.abort() calls
+            // agent.abort() (fires the AbortController) and then awaits
+            // agent.waitForIdle(). If an MCP tool (e.g. ctx_execute) is
+            // in-flight, the abort signal propagates to the MCP client, but
+            // waitForIdle() may hang if the transport doesn't reject the
+            // pending call in time. A timeout ensures the route returns to
+            // the client promptly — the agent loop will clean up in the
+            // background.
+            await withTimeout(live.session.abort(), ABORT_TIMEOUT_MS, "session.abort");
+        }
+        catch (err) {
+            if (err instanceof TimeoutError) {
+                req.log.warn({ sessionId: req.params.id }, "abort timed out after " + ABORT_TIMEOUT_MS + "ms");
+            }
+            // best-effort — abort signal was already fired
+        }
+        return { aborted: true };
+    });
+    // POST /api/v1/sessions/:id/steer — steer during streaming
+    fastify.post("/sessions/:id/steer", {
+        schema: {
+            description: "Steer or follow-up during an active streaming session.",
+            tags: ["sessions"],
+            params: {
+                type: "object",
+                required: ["id"],
+                properties: { id: { type: "string" } },
+            },
+            body: {
+                type: "object",
+                required: ["text"],
+                properties: {
+                    text: { type: "string", minLength: 1 },
+                    mode: { type: "string", enum: ["steer", "followUp"], default: "steer" },
+                    images: {
+                        type: "array",
+                        items: {
+                            type: "object",
+                            required: ["type", "data", "mimeType"],
+                            properties: {
+                                type: { type: "string", const: "image" },
+                                data: { type: "string" },
+                                mimeType: { type: "string" },
+                            },
+                        },
+                    },
+                },
+            },
+            response: {
+                202: {
+                    type: "object",
+                    required: ["accepted"],
+                    properties: { accepted: { type: "boolean" } },
+                },
+            },
+        },
+    }, async (req, reply) => {
+        const live = getSession(req.params.id);
+        if (live === undefined) {
+            return reply.code(404).send({ error: "session_not_found" });
+        }
+        let { text, mode, images } = req.body;
+        // Expand @-file references
+        const project = await getProject(live.projectId);
+        if (project !== undefined) {
+            text = await expandFileReferences(text, project.path);
+        }
+        // Save images to files if the model doesn't support vision natively
+        const transformed = await maybeSaveImagesToFiles(live.session.model, text, images);
+        text = transformed.text;
+        images = transformed.images;
+        try {
+            if (mode === "followUp") {
+                await live.session.followUp(text, images);
+            }
+            else {
+                await live.session.steer(text, images);
+            }
+        }
+        catch {
+            // best-effort; the session handles invalid-state calls gracefully
+        }
+        return reply.code(202).send({ accepted: true });
+    });
+};
+/**
+ * Wrap a Promise in a timeout.
+ * Pattern from control.ts withTimeout.
+ */
+async function withTimeout(promise, ms, label) {
+    let timer;
+    try {
+        return await Promise.race([
+            promise,
+            new Promise((_, reject) => {
+                timer = setTimeout(() => reject(new TimeoutError(`${label} timed out after ${ms}ms`)), ms);
+            }),
+        ]);
+    }
+    finally {
+        if (timer !== undefined)
+            clearTimeout(timer);
+    }
+}
+class TimeoutError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "TimeoutError";
+    }
+}
+//# sourceMappingURL=prompt.js.map

package/server/routes/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../src/routes/prompt.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAElE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnD,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEhC,MAAM,eAAe,GAAG,oBAAoB,CAAC;AAS7C;;;;;;;GAOG;AACH,KAAK,UAAU,sBAAsB,CACnC,KAAuC,EACvC,IAAY,EACZ,MAAkC;IAElC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAC5D,IAAI,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAE7D,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,GAAG,UAAU,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YACjE,MAAM,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,OAAO,EAAE,IAAI,EAAE,GAAG,IAAI,OAAO,WAAW,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAClE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,oDAAoD,EAAE,GAAG,CAAC,CAAC;QACzE,yCAAyC;QACzC,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACrC,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,SAAS,CACtB,GAAmB,EACnB,KAAmB;IAEnB,MAAM,SAAS,GAAI,GAAG,CAAC,MAAyB,CAAC,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IACnC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC,CAAC;QACpG,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;IACjC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACzB,KAAK,EAAE,qBAAqB;YAC5B,OAAO,EAAE,yCAAyC;SACnD,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,iDAAiD;IACjD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QACzD,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACzB,KAAK,EAAE,qBAAqB;YAC5B,OAAO,EAAE,uCAAuC,KAAK,CAAC,QAAQ,0CAA0C,KAAK,CAAC,QAAQ,GAAG;SAC1H,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAuB,KAAK,EAAE,OAAO,EAAE,EAAE;IAChE,kEAAkE;IAClE,OAAO,CAAC,IAAI,CAIV,sBAAsB,EACtB;QACE,MAAM,EAAE;YACN,WAAW,EACT,mEAAmE;gBACnE,qDAAqD;gBACrD,iEAAiE;gBACjE,mEAAmE;gBACnE,gEAAgE;gBAChE,0BAA0B;YAC5B,IAAI,EAAE,CAAC,UAAU,CAAC;YAClB,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,CAAC;gBAChB,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACvC;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,MAAM,CAAC;gBAClB,oBAAoB,EAAE,KAAK;gBAC3B,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAAE;oBACtC,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE;oBAClE,MAAM,EAAE;wBACN,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE;4BACL,IAAI,EAAE,QAAQ;4BACd,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC;4BACtC,UAAU,EAAE;gCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE;gCACxC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gCACxB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC7B;yBACF;qBACF;iBACF;aACF;YACD,QAAQ,EAAE;gBACR,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC,UAAU,CAAC;oBACtB,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;iBAC3D;gBACD,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACzB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC5B;iBACF;gBACD,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACzB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC5B;iBACF;aACF;SACF;KACF,EACD,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QACnB,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACzC,IAAI,IAAI,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QAErC,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;QAEnD,qDAAqD;QACrD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,IAAI,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACxD,CAAC;QAED,oEAAoE;QACpE,IAAI,WAAW,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACjF,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;QACxB,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAE5B,MAAM,IAAI,GAA8C,EAAE,CAAC;QAC3D,IAAI,iBAAiB,KAAK,SAAS;YAAE,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAChF,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAEpE,wDAAwD;QACxD,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAErC,6EAA6E;QAC7E,MAAM,sBAAsB,GAAG,CAAC,GAAY,EAAQ,EAAE;YACpD,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClC,IAAI,CAAC;oBACH,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,WAAW;wBACjB,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE;wBACxB,YAAY;qBACb,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,oDAAoD;gBACtD,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,+DAA+D;QAC/D,gDAAgD;QAChD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YACrD,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC,CACF,CAAC;IAEF,sDAAsD;IACtD,OAAO,CAAC,IAAI,CAGV,qBAAqB,EACrB;QACE,MAAM,EAAE;YACN,WAAW,EAAE,8BAA8B;YAC3C,IAAI,EAAE,CAAC,UAAU,CAAC;YAClB,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,CAAC;gBAChB,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACvC;YACD,QAAQ,EAAE;gBACR,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC,SAAS,CAAC;oBACrB,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;iBAC7C;gBACD,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;iBAC1C;aACF;SACF;KACF,EACD,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QACnB,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC;YACH,gEAAgE;YAChE,8DAA8D;YAC9D,4DAA4D;YAC5D,4DAA4D;YAC5D,gEAAgE;YAChE,6DAA6D;YAC7D,+DAA+D;YAC/D,4DAA4D;YAC5D,cAAc;YACd,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,gBAAgB,EAAE,eAAe,CAAC,CAAC;QAC7E,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,YAAY,EAAE,CAAC;gBAChC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,wBAAwB,GAAG,gBAAgB,GAAG,IAAI,CAAC,CAAC;YACjG,CAAC;YACD,+CAA+C;QACjD,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC,CACF,CAAC;IAEF,2DAA2D;IAC3D,OAAO,CAAC,IAAI,CAIV,qBAAqB,EACrB;QACE,MAAM,EAAE;YACN,WAAW,EAAE,wDAAwD;YACrE,IAAI,EAAE,CAAC,UAAU,CAAC;YAClB,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,CAAC;gBAChB,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACvC;YACD,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,MAAM,CAAC;gBAClB,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAAE;oBACtC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE;oBACvE,MAAM,EAAE;wBACN,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE;4BACL,IAAI,EAAE,QAAQ;4BACd,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC;4BACtC,UAAU,EAAE;gCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE;gCACxC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gCACxB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC7B;yBACF;qBACF;iBACF;aACF;YACD,QAAQ,EAAE;gBACR,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC,UAAU,CAAC;oBACtB,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;iBAC9C;aACF;SACF;KACF,EACD,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QACnB,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;QAEtC,2BAA2B;QAC3B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,IAAI,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACxD,CAAC;QAED,oEAAoE;QACpE,MAAM,WAAW,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACnF,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;QACxB,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAE5B,IAAI,CAAC;YACH,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC,CACF,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,KAAK,UAAU,WAAW,CAAI,OAAmB,EAAE,EAAU,EAAE,KAAa;IAC1E,IAAI,KAAiC,CAAC;IACtC,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC;YACxB,OAAO;YACP,IAAI,OAAO,CAAI,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC3B,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,GAAG,KAAK,oBAAoB,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7F,CAAC,CAAC;SACH,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,IAAI,KAAK,KAAK,SAAS;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,MAAM,YAAa,SAAQ,KAAK;IAC9B,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC;IAC7B,CAAC;CACF"}

package/server/routes/session-extensions.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Session Extension Routes — expose runtime extension state from the
+ * ExtensionRunner to the web UI.
+ *
+ * These routes give the GUI visibility into what extensions are active
+ * in a session, what commands and tools they registered, so the
+ * Extensions page shows "Active in Session" instead of just "Installed".
+ */
+import type { FastifyPluginAsync } from "fastify";
+export declare const sessionExtensionRoutes: FastifyPluginAsync;
+//# sourceMappingURL=session-extensions.d.ts.map

package/server/routes/session-extensions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-extensions.d.ts","sourceRoot":"","sources":["../../src/routes/session-extensions.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAGlD,eAAO,MAAM,sBAAsB,EAAE,kBA+IpC,CAAC"}

package/server/routes/session-extensions.js

@@ -0,0 +1,171 @@
+/**
+ * Session Extension Routes — expose runtime extension state from the
+ * ExtensionRunner to the web UI.
+ *
+ * These routes give the GUI visibility into what extensions are active
+ * in a session, what commands and tools they registered, so the
+ * Extensions page shows "Active in Session" instead of just "Installed".
+ */
+import { getSession } from "../session-store.js";
+export const sessionExtensionRoutes = async (fastify) => {
+    // ── GET /sessions/:id/extensions — runtime extension state ──────────
+    fastify.get("/sessions/:id/extensions", {
+        schema: {
+            description: "List extensions currently active in a live session, " +
+                "including their registered commands, tools, and event handlers.",
+            tags: ["extensions"],
+            params: {
+                type: "object",
+                required: ["id"],
+                properties: { id: { type: "string" } },
+            },
+            response: {
+                200: {
+                    type: "object",
+                    required: ["activeExtensions", "commands", "registeredTools"],
+                    properties: {
+                        activeExtensions: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    path: { type: "string" },
+                                    displayPath: { type: "string" },
+                                },
+                            },
+                        },
+                        commands: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    name: { type: "string" },
+                                    invocationName: { type: "string" },
+                                    description: { type: "string" },
+                                },
+                            },
+                        },
+                        registeredTools: {
+                            type: "array",
+                            items: {
+                                type: "object",
+                                properties: {
+                                    name: { type: "string" },
+                                    description: { type: "string" },
+                                },
+                            },
+                        },
+                        eventHandlers: {
+                            type: "array",
+                            items: { type: "string" },
+                        },
+                    },
+                },
+                404: {
+                    type: "object",
+                    properties: { error: { type: "string" } },
+                },
+            },
+        },
+    }, async (req, reply) => {
+        const sessionId = req.params.id;
+        const live = getSession(sessionId);
+        if (live === undefined) {
+            return reply.code(404).send({ error: "session_not_found" });
+        }
+        const runner = live.session.extensionRunner;
+        // Active extension paths
+        const activeExtensions = runner.getExtensionPaths().map((p) => ({
+            path: p,
+            // Derive a user-friendly display name from the path
+            displayPath: friendlyExtensionName(p),
+        }));
+        // Registered commands
+        const commands = runner.getRegisteredCommands().map((cmd) => ({
+            name: cmd.name,
+            invocationName: cmd.invocationName,
+            description: cmd.description ?? "",
+        }));
+        // Registered tools
+        const registeredTools = runner.getAllRegisteredTools().map((t) => ({
+            name: t.definition.name,
+            description: typeof t.definition.description === "string"
+                ? t.definition.description
+                : "",
+        }));
+        // Event handlers — enumerate all event types this runner handles
+        const allEventTypes = [
+            "project_trust",
+            "resources_discover",
+            "session_start",
+            "session_before_switch",
+            "session_before_fork",
+            "session_before_compact",
+            "session_compact",
+            "session_shutdown",
+            "session_before_tree",
+            "session_tree",
+            "context",
+            "before_provider_request",
+            "after_provider_response",
+            "before_agent_start",
+            "agent_start",
+            "agent_end",
+            "turn_start",
+            "turn_end",
+            "message_start",
+            "message_update",
+            "message_end",
+            "tool_execution_start",
+            "tool_execution_update",
+            "tool_execution_end",
+            "model_select",
+            "thinking_level_select",
+            "tool_call",
+            "tool_result",
+            "user_bash",
+            "input",
+        ];
+        // biome-ignore lint/complexity/noExcessiveCognitiveComplexity: simple filter
+        const eventHandlers = allEventTypes.filter((et) => runner.hasHandlers(et));
+        return reply.send({
+            activeExtensions,
+            commands,
+            registeredTools,
+            eventHandlers,
+        });
+    });
+};
+/**
+ * Derive a user-friendly display name from an extension path.
+ * e.g. "/home/user/.pi/agent/npm/node_modules/@ayulab/pi-rewind/dist/index.js"
+ *   → "@ayulab/pi-rewind"
+ */
+function friendlyExtensionPath(absolutePath) {
+    // Try to extract npm package name from node_modules path
+    const nmIndex = absolutePath.lastIndexOf("node_modules/");
+    if (nmIndex !== -1) {
+        const afterNm = absolutePath.slice(nmIndex + "node_modules/".length);
+        // The package name is the first path segment (which may contain @scope/)
+        const parts = afterNm.split("/");
+        if (parts[0]?.startsWith("@")) {
+            // Scoped package: @scope/name
+            return `${parts[0]}/${parts[1] ?? ""}`;
+        }
+        return parts[0] ?? absolutePath;
+    }
+    // Fallback: return the last two path segments
+    const segments = absolutePath.split("/");
+    return segments.slice(-2).join("/");
+}
+/**
+ * Derive a user-friendly display name from an extension path.
+ * Includes a fallback if the path is inline or temporary.
+ */
+function friendlyExtensionName(absolutePath) {
+    if (absolutePath.startsWith("<inline:")) {
+        return `inline-extension${absolutePath.replace("<inline:", "").replace(">", "")}`;
+    }
+    return friendlyExtensionPath(absolutePath);
+}
+//# sourceMappingURL=session-extensions.js.map

package/server/routes/session-extensions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-extensions.js","sourceRoot":"","sources":["../../src/routes/session-extensions.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,MAAM,CAAC,MAAM,sBAAsB,GAAuB,KAAK,EAAE,OAAO,EAAE,EAAE;IAC1E,uEAAuE;IAEvE,OAAO,CAAC,GAAG,CAGT,0BAA0B,EAC1B;QACE,MAAM,EAAE;YACN,WAAW,EACT,sDAAsD;gBACtD,iEAAiE;YACnE,IAAI,EAAE,CAAC,YAAY,CAAC;YACpB,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,CAAC,IAAI,CAAC;gBAChB,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACvC;YACD,QAAQ,EAAE;gBACR,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,CAAC,kBAAkB,EAAE,UAAU,EAAE,iBAAiB,CAAC;oBAC7D,UAAU,EAAE;wBACV,gBAAgB,EAAE;4BAChB,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,UAAU,EAAE;oCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oCACxB,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iCAChC;6BACF;yBACF;wBACD,QAAQ,EAAE;4BACR,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,UAAU,EAAE;oCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oCACxB,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oCAClC,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iCAChC;6BACF;yBACF;wBACD,eAAe,EAAE;4BACf,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,UAAU,EAAE;oCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oCACxB,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iCAChC;6BACF;yBACF;wBACD,aAAa,EAAE;4BACb,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;yBAC1B;qBACF;iBACF;gBACD,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;iBAC1C;aACF;SACF;KACF,EACD,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QACnB,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;QACnC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC;QAE5C,yBAAyB;QACzB,MAAM,gBAAgB,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9D,IAAI,EAAE,CAAC;YACP,oDAAoD;YACpD,WAAW,EAAE,qBAAqB,CAAC,CAAC,CAAC;SACtC,CAAC,CAAC,CAAC;QAEJ,sBAAsB;QACtB,MAAM,QAAQ,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC5D,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,cAAc,EAAE,GAAG,CAAC,cAAc;YAClC,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE;SACnC,CAAC,CAAC,CAAC;QAEJ,mBAAmB;QACnB,MAAM,eAAe,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjE,IAAI,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI;YACvB,WAAW,EACT,OAAO,CAAC,CAAC,UAAU,CAAC,WAAW,KAAK,QAAQ;gBAC1C,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW;gBAC1B,CAAC,CAAC,EAAE;SACT,CAAC,CAAC,CAAC;QAEJ,iEAAiE;QACjE,MAAM,aAAa,GAAG;YACpB,eAAe;YACf,oBAAoB;YACpB,eAAe;YACf,uBAAuB;YACvB,qBAAqB;YACrB,wBAAwB;YACxB,iBAAiB;YACjB,kBAAkB;YAClB,qBAAqB;YACrB,cAAc;YACd,SAAS;YACT,yBAAyB;YACzB,yBAAyB;YACzB,oBAAoB;YACpB,aAAa;YACb,WAAW;YACX,YAAY;YACZ,UAAU;YACV,eAAe;YACf,gBAAgB;YAChB,aAAa;YACb,sBAAsB;YACtB,uBAAuB;YACvB,oBAAoB;YACpB,cAAc;YACd,uBAAuB;YACvB,WAAW;YACX,aAAa;YACb,WAAW;YACX,OAAO;SACR,CAAC;QACF,6EAA6E;QAC7E,MAAM,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;QAE3E,OAAO,KAAK,CAAC,IAAI,CAAC;YAChB,gBAAgB;YAChB,QAAQ;YACR,eAAe;YACf,aAAa;SACd,CAAC,CAAC;IACL,CAAC,CACF,CAAC;AACJ,CAAC,CAAC;AAEF;;;;GAIG;AACH,SAAS,qBAAqB,CAAC,YAAoB;IACjD,yDAAyD;IACzD,MAAM,OAAO,GAAG,YAAY,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IAC1D,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;QACnB,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QACrE,yEAAyE;QACzE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC9B,8BAA8B;YAC9B,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QACzC,CAAC;QACD,OAAO,KAAK,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC;IAClC,CAAC;IACD,8CAA8C;IAC9C,MAAM,QAAQ,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzC,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,SAAS,qBAAqB,CAAC,YAAoB;IACjD,IAAI,YAAY,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACxC,OAAO,mBAAmB,YAAY,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC;IACpF,CAAC;IACD,OAAO,qBAAqB,CAAC,YAAY,CAAC,CAAC;AAC7C,CAAC"}

package/server/routes/sessions.d.ts

@@ -0,0 +1,3 @@
+import type { FastifyPluginAsync } from "fastify";
+export declare const sessionRoutes: FastifyPluginAsync;
+//# sourceMappingURL=sessions.d.ts.map

package/server/routes/sessions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessions.d.ts","sourceRoot":"","sources":["../../src/routes/sessions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAqBlD,eAAO,MAAM,aAAa,EAAE,kBAkwB3B,CAAC"}