@docyrus/docyrus 0.0.70 → 0.0.72

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@docyrus/docyrus",
-  "version": "0.0.70",
+  "version": "0.0.72",
   "private": false,
   "description": "Docyrus API CLI",
   "main": "./main.js",
@@ -14,8 +14,8 @@
     "@clack/prompts": "^0.11.0",
     "@ff-labs/fff-node": "0.6.4",
     "@hono/node-server": "^1.19.13",
-    "@mariozechner/pi-ai": "0.71.0",
-    "@mariozechner/pi-coding-agent": "0.71.0",
+    "@mariozechner/pi-ai": "0.71.1",
+    "@mariozechner/pi-coding-agent": "0.71.1",
     "@modelcontextprotocol/ext-apps": "^1.2.2",
     "@modelcontextprotocol/sdk": "^1.25.1",
     "@mozilla/readability": "^0.6.0",

package/server-loader.js

@@ -35692,12 +35692,27 @@ function createEventBridge(params) {
         handleToolExecutionEnd(event);
         break;
       }
+      case "message_end": {
+        handleMessageEnd(event);
+        break;
+      }
       case "agent_end": {
         onDone();
         break;
       }
     }
   }
+  function handleMessageEnd(event) {
+    const message = event.message;
+    if (!message) {
+      return;
+    }
+    const stopReason = message.stopReason;
+    if (stopReason !== "error" && stopReason !== "aborted") {
+      return;
+    }
+    onError(extractAssistantEventErrorMessage(message));
+  }
   function handleMessageUpdate(event) {
     const assistantEvent = event.assistantMessageEvent;
     if (!assistantEvent) {
@@ -45351,88 +45366,11 @@ async function createAgentServer(params) {
       modelIdsByProvider: getModelIdsByProvider()
     });
   }
-  app.use("/*", cors({ origin: "*" }));
-  app.use("/*", async (c, next) => {
-    const unauthorizedResponse = authorizeServerRequest(c.req.raw, authToken);
-    if (unauthorizedResponse) {
-      return unauthorizedResponse;
-    }
-    await next();
-  });
-  app.get("/api/health", (c) => {
-    return c.json({ ok: true });
-  });
-  app.get("/api/status", (c) => {
-    return c.json({
-      isStreaming: activeSession.isStreaming,
-      model: activeSession.model ? { provider: activeSession.model.provider, id: activeSession.model.id } : null,
-      mode: sessionMode
-    });
-  });
-  app.post("/api/chat", async (c) => {
-    const body2 = await c.req.json();
-    const messages = body2.messages ?? [];
-    if (body2.sessionId && body2.sessionId.trim() !== activeSession.id?.trim()) {
-      try {
-        activeSession = await onResumeSession(body2.sessionId);
-        sessionMode = "normal";
-      } catch (error48) {
-        const msg = error48 instanceof Error ? error48.message : String(error48);
-        return c.json({ error: `Failed to resume session: ${msg}` }, 400);
-      }
-    }
-    const chatUnavailableReason = getChatUnavailableReason({ session: activeSession });
-    if (chatUnavailableReason) {
-      return c.json({ error: chatUnavailableReason }, 409);
-    }
-    const sessionId = body2.sessionId?.trim() || activeSession.id?.trim() || "active";
-    const askUserResponse = extractAskUserToolResponse(messages);
-    const pendingAskUser = pendingAskUserRequests.get(sessionId);
-    const userMessage = askUserResponse ? pendingAskUser && pendingAskUser.toolCallId === askUserResponse.toolCallId ? formatAskUserResponsePrompt(askUserResponse.response) : void 0 : extractLastUserText(messages);
-    if (!userMessage) {
-      return c.json({
-        error: askUserResponse ? "No matching pending ask_user request found" : "No user message found"
-      }, 400);
-    }
-    if (askUserResponse) {
-      pendingAskUserRequests.delete(sessionId);
-    }
-    let promptText = userMessage;
-    if (!askUserResponse) {
-      const modeCmd = parseModeSlashCommand(userMessage);
-      if (modeCmd) {
-        if (activeSession.isStreaming) {
-          await activeSession.abort();
-          await waitForIdle(activeSession);
-        }
-        await activeSession.prompt(`/${modeCmd.command}`);
-        sessionMode = SESSION_MODE_COMMANDS[modeCmd.command] ?? sessionMode;
-        if (!modeCmd.remainder) {
-          const notifEncoder = new TextEncoder();
-          const notifMessageId = generateMessageId();
-          return c.body(
-            makeModeNotificationStream({ command: modeCmd.command, messageId: notifMessageId, encoder: notifEncoder }),
-            {
-              headers: {
-                "Content-Type": "text/event-stream",
-                "Cache-Control": "no-cache",
-                "Connection": "keep-alive",
-                "x-vercel-ai-ui-message-stream": "v1"
-              }
-            }
-          );
-        }
-        promptText = modeCmd.remainder;
-      }
-    }
-    if (activeSession.isStreaming) {
-      await activeSession.abort();
-      await waitForIdle(activeSession);
-    }
+  function createPromptStreamResponse(promptText, sessionId) {
     const messageId = generateMessageId();
     const encoder = new TextEncoder();
     const stream = new ReadableStream({
-      start(controller) {
+      async start(controller) {
         let closed = false;
         function writeChunk(chunk) {
           if (closed) {
@@ -45456,6 +45394,10 @@ async function createAgentServer(params) {
           } catch {
           }
         }
+        if (activeSession.isStreaming) {
+          await activeSession.abort();
+          await waitForIdle(activeSession);
+        }
         writeChunk({ type: "start" });
         writeChunk({ type: "start-step" });
         let extensionUICleanup;
@@ -45535,7 +45477,7 @@ async function createAgentServer(params) {
         });
       }
     });
-    return c.body(stream, {
+    return new Response(stream, {
       headers: {
         "Content-Type": "text/event-stream",
         "Cache-Control": "no-cache",
@@ -45543,6 +45485,119 @@ async function createAgentServer(params) {
         "x-vercel-ai-ui-message-stream": "v1"
       }
     });
+  }
+  app.use("/*", cors({ origin: "*" }));
+  app.use("/*", async (c, next) => {
+    const unauthorizedResponse = authorizeServerRequest(c.req.raw, authToken);
+    if (unauthorizedResponse) {
+      return unauthorizedResponse;
+    }
+    await next();
+  });
+  app.get("/api/health", (c) => {
+    return c.json({ ok: true });
+  });
+  app.get("/api/status", (c) => {
+    return c.json({
+      isStreaming: activeSession.isStreaming,
+      model: activeSession.model ? { provider: activeSession.model.provider, id: activeSession.model.id } : null,
+      mode: sessionMode
+    });
+  });
+  app.post("/api/chat", async (c) => {
+    const body2 = await c.req.json();
+    const messages = body2.messages ?? [];
+    if (body2.sessionId && body2.sessionId.trim() !== activeSession.id?.trim()) {
+      try {
+        activeSession = await onResumeSession(body2.sessionId);
+        sessionMode = "normal";
+      } catch (error48) {
+        const msg = error48 instanceof Error ? error48.message : String(error48);
+        return c.json({ error: `Failed to resume session: ${msg}` }, 400);
+      }
+    }
+    const chatUnavailableReason = getChatUnavailableReason({ session: activeSession });
+    if (chatUnavailableReason) {
+      return c.json({ error: chatUnavailableReason }, 409);
+    }
+    const sessionId = body2.sessionId?.trim() || activeSession.id?.trim() || "active";
+    const askUserResponse = extractAskUserToolResponse(messages);
+    const pendingAskUser = pendingAskUserRequests.get(sessionId);
+    const userMessage = askUserResponse ? pendingAskUser && pendingAskUser.toolCallId === askUserResponse.toolCallId ? formatAskUserResponsePrompt(askUserResponse.response) : void 0 : extractLastUserText(messages);
+    if (!userMessage) {
+      return c.json({
+        error: askUserResponse ? "No matching pending ask_user request found" : "No user message found"
+      }, 400);
+    }
+    if (askUserResponse) {
+      pendingAskUserRequests.delete(sessionId);
+    }
+    let promptText = userMessage;
+    if (!askUserResponse) {
+      const modeCmd = parseModeSlashCommand(userMessage);
+      if (modeCmd) {
+        if (activeSession.isStreaming) {
+          await activeSession.abort();
+          await waitForIdle(activeSession);
+        }
+        await activeSession.prompt(`/${modeCmd.command}`);
+        sessionMode = SESSION_MODE_COMMANDS[modeCmd.command] ?? sessionMode;
+        if (!modeCmd.remainder) {
+          const notifEncoder = new TextEncoder();
+          const notifMessageId = generateMessageId();
+          return c.body(
+            makeModeNotificationStream({ command: modeCmd.command, messageId: notifMessageId, encoder: notifEncoder }),
+            {
+              headers: {
+                "Content-Type": "text/event-stream",
+                "Cache-Control": "no-cache",
+                "Connection": "keep-alive",
+                "x-vercel-ai-ui-message-stream": "v1"
+              }
+            }
+          );
+        }
+        promptText = modeCmd.remainder;
+      }
+    }
+    return createPromptStreamResponse(promptText, sessionId);
+  });
+  app.post("/api/chat/retry", async (c) => {
+    const body2 = await c.req.json().catch(() => ({}));
+    if (body2.sessionId && body2.sessionId.trim() !== activeSession.id?.trim()) {
+      try {
+        activeSession = await onResumeSession(body2.sessionId);
+        sessionMode = "normal";
+      } catch (error48) {
+        const msg = error48 instanceof Error ? error48.message : String(error48);
+        return c.json({ error: `Failed to resume session: ${msg}` }, 400);
+      }
+    }
+    const chatUnavailableReason = getChatUnavailableReason({ session: activeSession });
+    if (chatUnavailableReason) {
+      return c.json({ error: chatUnavailableReason }, 409);
+    }
+    const userMessages = activeSession.getUserMessagesForForking();
+    const lastUserMessage = userMessages[userMessages.length - 1];
+    if (!lastUserMessage) {
+      return c.json({ error: "Session has no user message to retry" }, 404);
+    }
+    if (activeSession.isStreaming) {
+      await activeSession.abort();
+      await waitForIdle(activeSession);
+    }
+    try {
+      const navigation = await activeSession.navigateTree(lastUserMessage.entryId);
+      if (navigation.cancelled) {
+        return c.json({ error: "Retry cancelled by session" }, 409);
+      }
+    } catch (error48) {
+      const msg = error48 instanceof Error ? error48.message : String(error48);
+      return c.json({ error: `Failed to rewind session for retry: ${msg}` }, 500);
+    }
+    const sessionId = body2.sessionId?.trim() || activeSession.id?.trim() || "active";
+    pendingAskUserRequests.delete(sessionId);
+    return createPromptStreamResponse(lastUserMessage.text, sessionId);
   });
   app.post("/api/chat/abort", async (c) => {
     if (!activeSession.isStreaming) {
@@ -45644,6 +45699,35 @@ async function createAgentServer(params) {
       return c.json({ error: message }, 500);
     }
   });
+  app.delete("/api/sessions/:sessionId", async (c) => {
+    const sessionId = c.req.param("sessionId");
+    try {
+      const activeId = activeSession.id?.trim();
+      if (activeId && activeId === sessionId) {
+        return c.json({
+          error: "Cannot delete the active session. Create or switch to a different session first."
+        }, 409);
+      }
+      const sessions = await sessionManager.list();
+      const match2 = sessions.find((s) => s.id === sessionId);
+      if (!match2) {
+        return c.json({ error: "Session not found" }, 404);
+      }
+      if (match2.path) {
+        await (0, import_promises18.rm)(match2.path, { force: true });
+      }
+      const sessionConfigPath = (0, import_node_path22.join)(
+        resolveDocyrusTrackedSessionConfigRootPath(context.cwd),
+        `${sessionId}.json`
+      );
+      await (0, import_promises18.rm)(sessionConfigPath, { force: true }).catch(() => {
+      });
+      return c.body(null, 204);
+    } catch (error48) {
+      const message = error48 instanceof Error ? error48.message : String(error48);
+      return c.json({ error: message }, 500);
+    }
+  });
   app.get("/api/sessions/:sessionId/messages", async (c) => {
     const sessionId = c.req.param("sessionId");
     try {
@@ -47362,6 +47446,8 @@ async function createAgentServer(params) {
 
 `);
     process.stderr.write(`  POST /api/chat                          \u2014 send chat messages (SSE UIMessage stream)
+`);
+    process.stderr.write(`  POST /api/chat/retry                    \u2014 retry the last user message after an error
 `);
     process.stderr.write(`  POST /api/chat/abort                    \u2014 abort active chat stream
 `);
@@ -47372,6 +47458,8 @@ async function createAgentServer(params) {
     process.stderr.write(`  GET  /api/sessions                      \u2014 list sessions
 `);
     process.stderr.write(`  POST /api/sessions                      \u2014 create a new session
+`);
+    process.stderr.write(`  DEL  /api/sessions/:sessionId           \u2014 delete a session
 `);
     process.stderr.write(`  GET  /api/sessions/:sessionId/messages  \u2014 session messages
 `);
@@ -47605,6 +47693,12 @@ var SERVER_MODE_ALLOWED_MODEL_IDS_BY_PROVIDER = {
   minimax: [
     "MiniMax-M2.7",
     "MiniMax-M2.7-highspeed"
+  ],
+  moonshotai: [
+    "kimi-k2.6"
+  ],
+  deepseek: [
+    "deepseek-v4-pro"
   ]
 };
 var ALLOWED_MODEL_IDS_BY_PROVIDER = Object.fromEntries(
@@ -47689,6 +47783,17 @@ function createServerSessionAdapter(params) {
     listCommands() {
       const getCommands = params.extensionsResult.runtime.getCommands;
       return typeof getCommands === "function" ? getCommands() : [];
+    },
+    getUserMessagesForForking() {
+      const fn = params.session.getUserMessagesForForking;
+      return typeof fn === "function" ? fn.call(params.session) : [];
+    },
+    navigateTree(targetId, options) {
+      const fn = params.session.navigateTree;
+      if (typeof fn !== "function") {
+        return Promise.resolve({ cancelled: true });
+      }
+      return fn.call(params.session, targetId, options);
     }
   };
 }
@@ -47990,7 +48095,7 @@ async function main() {
     process.env.PI_SKIP_VERSION_CHECK = "1";
     settingsManager.setQuietStartup(true);
     settingsManager.setCollapseChangelog(true);
-    settingsManager.setLastChangelogVersion(version2);
+    settingsManager.setLastChangelogVersion(pi.VERSION || version2);
   }
   renderStartupSplash(version2);
   const spinner = createSpinner("Loading agent...");