chatlab-cli 0.22.1 → 0.23.1

package/dist/{chunk-XPBPKH6B.mjs → chunk-L37WJ33K.mjs}

@@ -7,21 +7,20 @@ const __dirname = __pathDirname(__filename);
 import {
   importData,
   parseFile
-} from "./chunk-OC2W5Q5T.mjs";
+} from "./chunk-3MX7VF2B.mjs";
 import {
   analyzeIncrementalImport,
   analyzeNewImport,
   incrementalImport,
   streamImport
-} from "./chunk-QRC3RSER.mjs";
+} from "./chunk-643S2N2U.mjs";
 import {
   AGENT_TOOL_REGISTRY,
   CoreDataProvider
-} from "./chunk-Z2LYMKCC.mjs";
+} from "./chunk-WTQ7LAZR.mjs";
 import {
-  getVersion,
-  resolveCliPath
-} from "./chunk-6NB475AA.mjs";
+  getVersion
+} from "./chunk-5ADKYYNA.mjs";
 import {
   AIConversationManager,
   ALL_MIGRATIONS,
@@ -115,7 +114,10 @@ import {
   verifyCliDataPath,
   writeAuthProfile,
   writeConfigField
-} from "./chunk-SQKT45MM.mjs";
+} from "./chunk-G7Q7EQ3V.mjs";
+import {
+  resolveCliPath
+} from "./chunk-FJ35V3TE.mjs";
 
 // src/http/index.ts
 import * as fs14 from "fs";
@@ -847,6 +849,9 @@ function registerSqlRoutes(server2, adapter) {
 
 // src/http/routes/web/session-index.ts
 function registerSessionIndexRoutes(server2, adapter) {
+  server2.get("/_web/sessions/index-stats", async () => {
+    return session_index_service_exports.getAllIndexStats(adapter);
+  });
   server2.post("/_web/sessions/:id/generate-index", async (request) => {
     const gapThreshold = request.body?.gapThreshold ?? 1800;
     const sessionCount = session_index_service_exports.generateIndex(adapter, request.params.id, gapThreshold);
@@ -1349,9 +1354,16 @@ data: ${JSON.stringify(eventData)}
       cleanupTemp(tmpPath, tmpDir);
     }
   });
+  const DEMO_FILES = [
+    "demo-group.json",
+    "demo-private-A-cuilan.json",
+    "demo-private-B-wukong.json",
+    "demo-private-C-spider.json"
+  ];
   server2.post("/_web/demo/import", async (request, reply) => {
     const locale = request.body?.locale || "en";
     const nativeBinding = resolveNativeBinding();
+    const total = DEMO_FILES.length;
     reply.raw.writeHead(200, {
       "Content-Type": "text/event-stream",
       "Cache-Control": "no-cache",
@@ -1364,36 +1376,37 @@ data: ${JSON.stringify(eventData)}
 `);
     }
     const tmpDir = fs4.mkdtempSync(path3.join(os.tmpdir(), "chatlab-demo-"));
-    const groupPath = path3.join(tmpDir, "demo-group.json");
-    const privatePath = path3.join(tmpDir, "demo-private.json");
     try {
-      sendEvent("progress", { stage: "downloading", current: 1, total: 2 });
-      const groupResp = await fetch(`${DEMO_BASE_URL}/${locale}/demo-group.json`, {
-        signal: AbortSignal.timeout(6e4)
-      });
-      if (!groupResp.ok) throw new Error(`Download group demo failed: ${groupResp.status}`);
-      fs4.writeFileSync(groupPath, Buffer.from(await groupResp.arrayBuffer()));
-      sendEvent("progress", { stage: "downloading", current: 2, total: 2 });
-      const privateResp = await fetch(`${DEMO_BASE_URL}/${locale}/demo-private.json`, {
-        signal: AbortSignal.timeout(6e4)
-      });
-      if (!privateResp.ok) throw new Error(`Download private demo failed: ${privateResp.status}`);
-      fs4.writeFileSync(privatePath, Buffer.from(await privateResp.arrayBuffer()));
-      sendEvent("progress", { stage: "importing", current: 1, total: 2 });
-      const groupResult = await streamImport(dbManager2, groupPath, { nativeBinding });
+      const localPaths = [];
+      for (let i = 0; i < total; i++) {
+        sendEvent("progress", { stage: "downloading", current: i + 1, total });
+        const localPath = path3.join(tmpDir, DEMO_FILES[i]);
+        const resp = await fetch(`${DEMO_BASE_URL}/${locale}/${DEMO_FILES[i]}`, {
+          signal: AbortSignal.timeout(6e4)
+        });
+        if (!resp.ok) throw new Error(`Download demo failed (${DEMO_FILES[i]}): ${resp.status}`);
+        fs4.writeFileSync(localPath, Buffer.from(await resp.arrayBuffer()));
+        localPaths.push(localPath);
+      }
+      sendEvent("progress", { stage: "importing", current: 1, total });
+      const groupResult = await streamImport(dbManager2, localPaths[0], { nativeBinding });
       if (!groupResult.success) throw new Error(groupResult.error || "Failed to import group demo");
-      sendEvent("progress", { stage: "importing", current: 2, total: 2 });
-      const privateResult = await streamImport(dbManager2, privatePath, { nativeBinding });
-      if (!privateResult.success) throw new Error(privateResult.error || "Failed to import private demo");
-      sendEvent("progress", { stage: "done", current: 2, total: 2 });
+      const privateSessionIds = [];
+      for (let i = 1; i < localPaths.length; i++) {
+        sendEvent("progress", { stage: "importing", current: i + 1, total });
+        const result = await streamImport(dbManager2, localPaths[i], { nativeBinding });
+        if (!result.success) throw new Error(result.error || `Failed to import private demo: ${DEMO_FILES[i]}`);
+        if (result.sessionId) privateSessionIds.push(result.sessionId);
+      }
+      sendEvent("progress", { stage: "done", current: total, total });
       sendEvent("result", {
         success: true,
         groupSessionId: groupResult.sessionId,
-        privateSessionId: privateResult.sessionId
+        privateSessionIds
       });
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
-      sendEvent("progress", { stage: "error", current: 0, total: 2, message });
+      sendEvent("progress", { stage: "error", current: 0, total, message });
       sendEvent("result", { success: false, error: message });
     } finally {
       reply.raw.end();
@@ -1485,7 +1498,7 @@ function registerMergeRoutes(server2, dbManager2, mergeCache) {
       const merged = buildMergedOutput(dataSources, outputName);
       let sessionId;
       if (andImport) {
-        const { streamImport: serverStreamImport } = await import("./stream-import-ZTEW3YVL.mjs");
+        const { streamImport: serverStreamImport } = await import("./stream-import-QM5R5FDG.mjs");
         const jsonData = JSON.stringify(merged.chatLabData);
         const tmpDir = fs5.mkdtempSync(path4.join(os2.tmpdir(), "chatlab-merged-"));
         const tmpPath = path4.join(tmpDir, `${outputName}.json`);
@@ -1538,7 +1551,7 @@ function registerMergeRoutes(server2, dbManager2, mergeCache) {
       const db = ensureDb2(dbManager2, sid);
       const exported = exportSessionToJson(db);
       const { db: tempDb, tempDbPath } = mergeCache.createTempDatabase(exported.meta.name);
-      const { TempDbWriter } = await import("./src-FVJL5ENA.mjs");
+      const { TempDbWriter } = await import("./src-EGXULYFQ.mjs");
       const writer = new TempDbWriter(tempDb);
       writer.writeMeta({
         name: exported.meta.name,
@@ -2096,6 +2109,7 @@ async function runServerAgent(options) {
   const {
     userMessage,
     conversationId,
+    historyLeafMessageId,
     chatType = "group",
     locale = "zh-CN",
     assistantSystemPrompt,
@@ -2108,7 +2122,8 @@ async function runServerAgent(options) {
     abortSignal,
     ownerInfo,
     mentionedMembers,
-    dataSnapshot
+    dataSnapshot,
+    thinkingLevel
   } = options;
   const aiLogger = getServerAiLogger();
   const llmConfig = getDefaultAssistantConfig(aiDataDir);
@@ -2138,7 +2153,7 @@ async function runServerAgent(options) {
     context: {},
     systemPrompt
   });
-  if (compressionConfig?.enabled) {
+  if (compressionConfig?.enabled && historyLeafMessageId === void 0) {
     const llmAdapter = createCompressionLlmAdapter({
       piModel,
       apiKey: llmConfig.apiKey,
@@ -2163,6 +2178,11 @@ async function runServerAgent(options) {
         }
       });
     }
+  } else if (compressionConfig?.enabled && historyLeafMessageId !== void 0) {
+    aiLogger?.info?.("Compression", "Skipping compression for edited branch request", {
+      conversationId,
+      historyLeafMessageId
+    });
   }
   if (abortSignal?.aborted) {
     handler.emitStatus("aborted", [], { force: true });
@@ -2171,7 +2191,7 @@ async function runServerAgent(options) {
   }
   let history = [];
   try {
-    history = convManager2.getHistoryForAgent(conversationId);
+    history = convManager2.getHistoryForAgent(conversationId, void 0, historyLeafMessageId);
   } catch {
   }
   handler.emitStatus("preparing", [], { pendingUserMessage: userMessage, force: true });
@@ -2188,6 +2208,7 @@ async function runServerAgent(options) {
       maxToolRounds: 5,
       abortSignal,
       steerMessage,
+      thinkingLevel,
       onConvertToLlm: (filteredMessages) => {
         cachedMessages = filteredMessages;
       },
@@ -2625,6 +2646,51 @@ data: ${JSON.stringify(data)}
   server2.get("/_web/ai/conversations/:id/messages", async (request) => {
     return convManager2.getMessages(request.params.id);
   });
+  server2.post("/_web/ai/conversations/:id/messages/:messageId/delete-from", async (request, reply) => {
+    const { id, messageId } = request.params;
+    if (!id || typeof id !== "string" || !messageId || typeof messageId !== "string") {
+      return reply.code(400).send({ error: "conversation id and messageId are required" });
+    }
+    convManager2.deleteMessagesFrom(id, messageId);
+    return { success: true };
+  });
+  server2.post("/_web/ai/conversations/:id/fork", async (request, reply) => {
+    const { upToMessageId, title } = request.body;
+    if (!upToMessageId || typeof upToMessageId !== "string") {
+      return reply.code(400).send({ error: "upToMessageId is required" });
+    }
+    return convManager2.forkConversation(request.params.id, upToMessageId, title);
+  });
+  server2.put("/_web/ai/messages/:messageId/content", async (request, reply) => {
+    const { content } = request.body;
+    if (!content || typeof content !== "string") {
+      return reply.code(400).send({ error: "content is required" });
+    }
+    convManager2.updateMessageContent(request.params.messageId, content);
+    return { success: true };
+  });
+  server2.post("/_web/ai/conversations/:id/messages/:messageId/delete-relink", async (request, reply) => {
+    const { id, messageId } = request.params;
+    if (!id || !messageId) {
+      return reply.code(400).send({ error: "conversation id and messageId are required" });
+    }
+    convManager2.deleteAndRelinkMessage(id, messageId);
+    return { success: true };
+  });
+  server2.post("/_web/ai/conversations/:id/messages/insert-after", async (request, reply) => {
+    const { afterMessageId, role, content, contentBlocks, tokenUsage } = request.body;
+    if (!afterMessageId || typeof afterMessageId !== "string") {
+      return reply.code(400).send({ error: "afterMessageId is required" });
+    }
+    return convManager2.insertMessageAfter(
+      request.params.id,
+      afterMessageId,
+      role,
+      content,
+      contentBlocks,
+      tokenUsage
+    );
+  });
   server2.get("/_web/ai/conversations/:id/token-usage", async (request) => {
     return convManager2.getConversationTokenUsage(request.params.id);
   });
@@ -2633,13 +2699,15 @@ data: ${JSON.stringify(data)}
     const {
       userMessage,
       conversationId,
+      historyLeafMessageId,
       sessionId,
       chatType,
       locale,
       assistantId,
       compressionConfig,
       ownerInfo,
-      mentionedMembers
+      mentionedMembers,
+      thinkingLevel
     } = request.body;
     const requestId = `req_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
     const abortController = new AbortController();
@@ -2726,6 +2794,7 @@ data: ${JSON.stringify(data)}
       await runServerAgent({
         userMessage,
         conversationId,
+        historyLeafMessageId,
         chatType,
         locale,
         assistantSystemPrompt,
@@ -2738,7 +2807,8 @@ data: ${JSON.stringify(data)}
         abortSignal: abortController.signal,
         ownerInfo,
         mentionedMembers,
-        dataSnapshot
+        dataSnapshot,
+        thinkingLevel
       });
     } catch (error) {
       const msg = error instanceof Error ? error.message : String(error);
@@ -2830,6 +2900,32 @@ function registerPreferencesRoutes(server2, pathProvider) {
   });
 }
 
+// src/http/routes/proxy.ts
+var PROXY_TARGET = "https://chatlab.fun";
+function registerProxyRoutes(server2) {
+  server2.get("/_proxy/chatlab.fun/*", async (request, reply) => {
+    const subPath = request.params["*"] ?? "";
+    const queryString = new URLSearchParams(
+      request.query
+    ).toString();
+    const targetUrl = queryString ? `${PROXY_TARGET}/${subPath}?${queryString}` : `${PROXY_TARGET}/${subPath}`;
+    try {
+      const upstream = await fetch(targetUrl, {
+        headers: { Accept: request.headers.accept ?? "*/*" },
+        signal: AbortSignal.timeout(1e4)
+      });
+      const contentType = upstream.headers.get("content-type") ?? "application/octet-stream";
+      const body = Buffer.from(await upstream.arrayBuffer());
+      reply.code(upstream.status).header("content-type", contentType).send(body);
+    } catch (err) {
+      reply.code(502).send({
+        error: "Proxy error",
+        message: err instanceof Error ? err.message : String(err)
+      });
+    }
+  });
+}
+
 // ../../packages/sync/src/types.ts
 var noop = () => {
 };
@@ -3765,6 +3861,7 @@ async function startHttpServer(options) {
   initSync(server, dbManager, pathProvider, { port, host, token });
   if (options?.webRoot && fs14.existsSync(options.webRoot)) {
     setWebMode(true);
+    registerProxyRoutes(server);
     const fastifyStatic = await import("@fastify/static");
     await server.register(fastifyStatic.default, {
       root: options.webRoot,
@@ -3808,4 +3905,4 @@ export {
   startHttpServer,
   stopHttpServer
 };
-//# sourceMappingURL=chunk-XPBPKH6B.mjs.map
+//# sourceMappingURL=chunk-L37WJ33K.mjs.map