chatlab-cli 0.22.1 → 0.23.0

package/dist/{chunk-SQKT45MM.mjs → chunk-G7Q7EQ3V.mjs}

@@ -2022,6 +2022,25 @@ function getSessionSummaries(db, options) {
   });
 }
 var DEFAULT_SESSION_GAP_THRESHOLD = 1800;
+function getSessionIndexStats(db) {
+  let sessionCount = 0;
+  if (hasTable(db, "chat_session")) {
+    try {
+      const row = db.prepare("SELECT COUNT(*) as count FROM chat_session").get();
+      sessionCount = row?.count ?? 0;
+    } catch {
+    }
+  }
+  let gapThreshold = DEFAULT_SESSION_GAP_THRESHOLD;
+  try {
+    const meta = db.prepare("SELECT session_gap_threshold FROM meta LIMIT 1").get();
+    if (meta?.session_gap_threshold) {
+      gapThreshold = meta.session_gap_threshold;
+    }
+  } catch {
+  }
+  return { sessionCount, hasIndex: sessionCount > 0, gapThreshold };
+}
 function getChatSessionList(db) {
   if (!hasTable(db, "chat_session")) return [];
   try {
@@ -5774,6 +5793,138 @@ function needsStreamingThinkParsing(provider, _modelId) {
   return provider === "minimax";
 }
 
+// ../../packages/core/src/ai/thinking.ts
+var TYPE_LEVEL_MAP = {
+  gpt5: {
+    minimal: "minimal",
+    low: "low",
+    medium: "medium",
+    high: "high"
+  },
+  gpt5_1: {
+    off: "none",
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: null
+  },
+  gpt5_2plus: {
+    off: "none",
+    minimal: "minimal",
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: "xhigh"
+  },
+  grok: {
+    minimal: null,
+    low: "low",
+    medium: null,
+    high: "high",
+    xhigh: null
+  },
+  kimi: {
+    off: "none",
+    auto: "auto",
+    minimal: null,
+    low: "low",
+    medium: "medium",
+    high: "high",
+    xhigh: null
+  },
+  doubao: {
+    off: null,
+    auto: "auto",
+    high: "high"
+  },
+  deepseek_hybrid: {
+    off: null,
+    auto: "auto",
+    high: "high"
+  }
+};
+var REASONING_FALLBACK_REGEX = /^(?!.*-non-reasoning\b)(o\d+(?:-[\w-]+)?|.*\b(?:reasoning|reasoner|thinking|think)\b.*|.*-[rR]\d+.*|.*\bqwq(?:-[\w-]+)?\b.*|.*\bhunyuan-t1(?:-[\w-]+)?\b.*|.*\bgrok-(?:3-mini|4|4-fast|build)(?:-[\w-]+)?\b.*)$/i;
+function classifyThinkingType(provider, modelId) {
+  const id = modelId.toLowerCase();
+  const prov = provider.toLowerCase();
+  if (id.includes("-non-reasoning")) return "none";
+  if (prov === "anthropic") return "none";
+  if (prov === "gemini") {
+    if (/gemini-(?:2\.5|3(?:\.\d+)?-(?:flash|pro)|flash-latest|pro-latest)/i.test(id)) return "gemini";
+    return "none";
+  }
+  if (prov === "qwen" || /\bqwen|qwq/i.test(id)) return "qwen";
+  if (/deepseek[_-]v([4-9]|\d{2,})/i.test(id) || /deepseek-ai\/deepseek-r1/i.test(id)) return "deepseek_v4";
+  if (/deepseek[_-]chat|deepseek[_-]v3/i.test(id)) return "deepseek_hybrid";
+  if (prov === "openai" || prov === "openai-compatible") {
+    if (/gpt-5\.[2-9]|gpt-5\.[1-9]\d/.test(id)) return "gpt5_2plus";
+    if (/gpt-5\.1/.test(id)) return "gpt5_1";
+    if (/gpt-5/.test(id)) return "gpt5";
+    if (/^o\d/.test(id)) return "o_series";
+  }
+  if (/^o\d/.test(id)) return "o_series";
+  if (/\bgrok-(?:3-mini|4|4-fast|build)\b/i.test(id)) return "grok";
+  if (/\bkimi-k(?:2-thinking|2\.[5-9]|[3-9])/i.test(id)) return "kimi";
+  if (prov === "kimi" && /thinking/i.test(id)) return "kimi";
+  if (/\bdoubao-.*seed/i.test(id) || prov === "doubao" && /seed/i.test(id)) return "doubao";
+  if (/\bhunyuan-(?:t1|a13b)/i.test(id)) return "hunyuan";
+  if (prov === "glm" || /\bglm-?(?:5|4\.[5-7]|z1|zero)/i.test(id)) return "qwen";
+  if (prov === "siliconflow" && /r[1-9]/i.test(id)) return "deepseek_v4";
+  if (prov === "openrouter") {
+    if (/deepseek.*(v[4-9]|r\d)/i.test(id)) return "deepseek_v4";
+    if (/deepseek[_-]?(chat|v3)/i.test(id)) return "deepseek_hybrid";
+    if (/\bgrok-(?:3-mini|4|4-fast|build)\b/i.test(id)) return "grok";
+    if (/\bkimi-k(?:2-thinking|2\.[5-9]|[3-9])/i.test(id)) return "kimi";
+    if (REASONING_FALLBACK_REGEX.test(id)) return "default";
+    return "none";
+  }
+  if (/\bmimo-v2/i.test(id)) return "deepseek_hybrid";
+  if (/\bminimax-m[123]/i.test(id)) return "deepseek_hybrid";
+  if (/\bseed-oss/i.test(id)) return "deepseek_hybrid";
+  if (/\bbaichuan-m[23]/i.test(id)) return "default";
+  if (/\bmagistral/i.test(id)) return "default";
+  if (/\bmistral-small-2603/i.test(id)) return "default";
+  if (/\bstep-(?:3|r1-v-mini)/i.test(id)) return "default";
+  if (/\bgemma-?4/i.test(id)) return "default";
+  if (/\bpangu-pro-moe/i.test(id)) return "default";
+  if (/\bsonar-deep-research/i.test(id)) return "default";
+  if (/\bring-(?:1t|mini|flash)/i.test(id)) return "default";
+  if (/\bqwen|qwq/i.test(id)) return "qwen";
+  if (/deepseek[_-]?r\d|deepseek[_-]?v[4-9]/i.test(id)) return "deepseek_v4";
+  if (/^o\d/i.test(id)) return "o_series";
+  if (REASONING_FALLBACK_REGEX.test(id)) return "default";
+  return "none";
+}
+function isReasoningModel(provider, modelId) {
+  return classifyThinkingType(provider, modelId) !== "none";
+}
+function getThinkingCompat(provider, modelId) {
+  const type = classifyThinkingType(provider, modelId);
+  if (type === "none") return {};
+  if (type === "gemini") return {};
+  if (type === "qwen" || type === "hunyuan") {
+    return { thinkingFormat: "qwen" };
+  }
+  if (type === "deepseek_v4") {
+    return {
+      thinkingFormat: "deepseek",
+      thinkingLevelMap: { high: "high", xhigh: "max", minimal: null, low: null, medium: null }
+    };
+  }
+  if (type === "deepseek_hybrid") {
+    return {
+      thinkingFormat: "deepseek",
+      thinkingLevelMap: { auto: "auto", high: "high", minimal: null, low: null, medium: null, xhigh: null }
+    };
+  }
+  const compat = { supportsReasoningEffort: true };
+  const levelMap = TYPE_LEVEL_MAP[type];
+  if (levelMap) {
+    compat.thinkingLevelMap = levelMap;
+  }
+  return compat;
+}
+
 // ../../packages/core/src/import/dedup.ts
 function fnv1a64(input) {
   let h0 = 2166136261;
@@ -11657,6 +11808,7 @@ var AIConversationManager = class {
         id TEXT PRIMARY KEY,
         session_id TEXT NOT NULL,
         title TEXT,
+        active_message_id TEXT,
         created_at INTEGER NOT NULL,
         updated_at INTEGER NOT NULL
       );
@@ -11670,6 +11822,9 @@ var AIConversationManager = class {
         data_keywords TEXT,
         data_message_count INTEGER,
         content_blocks TEXT,
+        parent_id TEXT,
+        sibling_group_id TEXT,
+        branch_index INTEGER DEFAULT 0,
         FOREIGN KEY(conversation_id) REFERENCES ai_conversation(id) ON DELETE CASCADE
       );
 
@@ -11683,6 +11838,7 @@ var AIConversationManager = class {
     try {
       const messageTableInfo = db.pragma("table_info(ai_message)");
       const messageColumns = messageTableInfo.map((col) => col.name);
+      const needsMessageTreeBackfill = !messageColumns.includes("parent_id") || !messageColumns.includes("sibling_group_id") || !messageColumns.includes("branch_index");
       if (!messageColumns.includes("content_blocks")) {
         db.exec("ALTER TABLE ai_message ADD COLUMN content_blocks TEXT");
       }
@@ -11692,18 +11848,149 @@ var AIConversationManager = class {
       if (!messageColumns.includes("debug_context")) {
         db.exec("ALTER TABLE ai_message ADD COLUMN debug_context TEXT");
       }
+      if (!messageColumns.includes("parent_id")) {
+        db.exec("ALTER TABLE ai_message ADD COLUMN parent_id TEXT");
+      }
+      if (!messageColumns.includes("sibling_group_id")) {
+        db.exec("ALTER TABLE ai_message ADD COLUMN sibling_group_id TEXT");
+      }
+      if (!messageColumns.includes("branch_index")) {
+        db.exec("ALTER TABLE ai_message ADD COLUMN branch_index INTEGER DEFAULT 0");
+      }
       const convTableInfo = db.pragma("table_info(ai_conversation)");
       const convColumns = convTableInfo.map((col) => col.name);
+      const needsConversationBackfill = !convColumns.includes("active_message_id");
       if (!convColumns.includes("assistant_id")) {
         db.exec(`ALTER TABLE ai_conversation ADD COLUMN assistant_id TEXT DEFAULT '${DEFAULT_GENERAL_ID}'`);
       }
+      if (!convColumns.includes("active_message_id")) {
+        db.exec("ALTER TABLE ai_conversation ADD COLUMN active_message_id TEXT");
+      }
+      if (needsMessageTreeBackfill || needsConversationBackfill || this.hasUnbackfilledMessageTree(db)) {
+        this.backfillMessageTree(db);
+      }
+      db.exec(`
+        CREATE INDEX IF NOT EXISTS idx_ai_message_parent ON ai_message(parent_id);
+        CREATE INDEX IF NOT EXISTS idx_ai_message_sibling ON ai_message(sibling_group_id);
+      `);
     } catch (error) {
       console.error("[AI DB Migration] Migration failed:", error);
     }
   }
+  hasUnbackfilledMessageTree(db) {
+    const row = db.prepare(
+      `SELECT 1
+         FROM ai_conversation c
+         WHERE EXISTS (
+           SELECT 1 FROM ai_message m
+           WHERE m.conversation_id = c.id
+         )
+         AND (
+           c.active_message_id IS NULL
+           OR EXISTS (
+             SELECT 1 FROM ai_message m
+             WHERE m.conversation_id = c.id
+               AND (m.sibling_group_id IS NULL OR m.branch_index IS NULL)
+           )
+         )
+         LIMIT 1`
+    ).get();
+    return !!row;
+  }
+  backfillMessageTree(db) {
+    const conversations = db.prepare("SELECT id FROM ai_conversation").all();
+    const updateMessage = db.prepare(
+      "UPDATE ai_message SET parent_id = ?, sibling_group_id = COALESCE(sibling_group_id, ?), branch_index = COALESCE(branch_index, 0) WHERE id = ?"
+    );
+    const updateConversation = db.prepare("UPDATE ai_conversation SET active_message_id = ? WHERE id = ?");
+    const tx = db.transaction(() => {
+      for (const conversation of conversations) {
+        const messages = db.prepare(
+          `SELECT id, parent_id as parentId, sibling_group_id as siblingGroupId
+             FROM ai_message WHERE conversation_id = ? ORDER BY timestamp ASC, id ASC`
+        ).all(conversation.id);
+        let previousId = null;
+        for (const message of messages) {
+          const parentId = message.parentId === void 0 ? previousId : message.parentId ?? previousId;
+          updateMessage.run(parentId, message.siblingGroupId ?? message.id, message.id);
+          previousId = message.id;
+        }
+        const conversationRow = db.prepare("SELECT active_message_id as activeMessageId FROM ai_conversation WHERE id = ?").get(conversation.id);
+        if (!conversationRow?.activeMessageId && previousId) {
+          updateConversation.run(previousId, conversation.id);
+        }
+      }
+    });
+    tx();
+  }
   generateId(prefix) {
     return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
   }
+  parseMessageRow(row) {
+    return {
+      id: row.id,
+      conversationId: row.conversationId,
+      role: row.role,
+      content: row.content,
+      timestamp: row.timestamp,
+      parentId: row.parentId ?? null,
+      dataKeywords: row.dataKeywords ? JSON.parse(row.dataKeywords) : void 0,
+      dataMessageCount: row.dataMessageCount ?? void 0,
+      contentBlocks: row.contentBlocks ? JSON.parse(row.contentBlocks) : void 0,
+      tokenUsage: row.tokenUsage ? JSON.parse(row.tokenUsage) : void 0
+    };
+  }
+  getMessageRow(messageId) {
+    const db = this.getDb();
+    const row = db.prepare(
+      `SELECT id, conversation_id as conversationId, role, content, timestamp,
+                parent_id as parentId, sibling_group_id as siblingGroupId, branch_index as branchIndex,
+                data_keywords as dataKeywords, data_message_count as dataMessageCount,
+                content_blocks as contentBlocks, token_usage as tokenUsage
+         FROM ai_message WHERE id = ?`
+    ).get(messageId);
+    return row ?? null;
+  }
+  getActiveMessageId(conversationId) {
+    const db = this.getDb();
+    const row = db.prepare("SELECT active_message_id as activeMessageId FROM ai_conversation WHERE id = ?").get(conversationId);
+    if (row?.activeMessageId) {
+      const activeExists = db.prepare("SELECT 1 FROM ai_message WHERE id = ?").get(row.activeMessageId);
+      if (activeExists) return row.activeMessageId;
+    }
+    const fallback = db.prepare("SELECT id FROM ai_message WHERE conversation_id = ? ORDER BY timestamp DESC, id DESC LIMIT 1").get(conversationId);
+    if (fallback?.id) {
+      db.prepare("UPDATE ai_conversation SET active_message_id = ? WHERE id = ?").run(fallback.id, conversationId);
+      return fallback.id;
+    }
+    return null;
+  }
+  getAllMessageRows(conversationId) {
+    return this.getDb().prepare(
+      `SELECT id, conversation_id as conversationId, role, content, timestamp,
+                parent_id as parentId, sibling_group_id as siblingGroupId, branch_index as branchIndex,
+                data_keywords as dataKeywords, data_message_count as dataMessageCount,
+                content_blocks as contentBlocks, token_usage as tokenUsage
+         FROM ai_message WHERE conversation_id = ? ORDER BY timestamp ASC, id ASC`
+    ).all(conversationId);
+  }
+  getActivePathRows(conversationId, leafMessageId) {
+    if (leafMessageId === null) return [];
+    const allRows = this.getAllMessageRows(conversationId);
+    if (allRows.length === 0) return [];
+    const rowMap = new Map(allRows.map((row) => [row.id, row]));
+    let currentId = leafMessageId ?? this.getActiveMessageId(conversationId);
+    const path37 = [];
+    const seen = /* @__PURE__ */ new Set();
+    while (currentId && !seen.has(currentId)) {
+      seen.add(currentId);
+      const row = rowMap.get(currentId);
+      if (!row) break;
+      path37.push(row);
+      currentId = row.parentId;
+    }
+    return path37.length > 0 ? path37.reverse() : allRows;
+  }
   // ==================== 生命周期 ====================
   close() {
     if (this.db) {
@@ -11766,7 +12053,7 @@ var AIConversationManager = class {
       `INSERT INTO ai_conversation (id, session_id, title, assistant_id, created_at, updated_at)
        VALUES (?, ?, ?, ?, ?, ?)`
     ).run(id, sessionId, title || null, assistantId, now, now);
-    return { id, sessionId, title: title || null, assistantId, createdAt: now, updatedAt: now };
+    return { id, sessionId, title: title || null, assistantId, activeMessageId: null, createdAt: now, updatedAt: now };
   }
   getConversationCountsBySession() {
     const result = /* @__PURE__ */ new Map();
@@ -11784,6 +12071,7 @@ var AIConversationManager = class {
     const db = this.getDb();
     return db.prepare(
       `SELECT id, session_id as sessionId, title, assistant_id as assistantId,
+                active_message_id as activeMessageId,
                 created_at as createdAt, updated_at as updatedAt
          FROM ai_conversation WHERE session_id = ? ORDER BY updated_at DESC`
     ).all(sessionId);
@@ -11792,6 +12080,7 @@ var AIConversationManager = class {
     const db = this.getDb();
     const row = db.prepare(
       `SELECT id, session_id as sessionId, title, assistant_id as assistantId,
+                active_message_id as activeMessageId,
                 created_at as createdAt, updated_at as updatedAt
          FROM ai_conversation WHERE id = ?`
     ).get(conversationId);
@@ -11814,13 +12103,19 @@ var AIConversationManager = class {
     const db = this.getDb();
     const now = Math.floor(Date.now() / 1e3);
     const id = this.generateId("msg");
+    const parentId = this.getActiveMessageId(conversationId);
+    const siblingGroupId = id;
+    const branchIndex = 0;
     const pendingDebug = role === "assistant" ? this.pendingDebugContextMap.get(conversationId) : void 0;
     if (pendingDebug) {
       this.pendingDebugContextMap.delete(conversationId);
     }
     db.prepare(
-      `INSERT INTO ai_message (id, conversation_id, role, content, timestamp, data_keywords, data_message_count, content_blocks, token_usage, debug_context)
-       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+      `INSERT INTO ai_message (
+         id, conversation_id, role, content, timestamp, data_keywords, data_message_count,
+         content_blocks, token_usage, debug_context, parent_id, sibling_group_id, branch_index
+       )
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       id,
       conversationId,
@@ -11831,15 +12126,23 @@ var AIConversationManager = class {
       dataMessageCount ?? null,
       contentBlocks ? JSON.stringify(contentBlocks) : null,
       tokenUsage ? JSON.stringify(tokenUsage) : null,
-      pendingDebug ?? null
+      pendingDebug ?? null,
+      parentId,
+      siblingGroupId,
+      branchIndex
+    );
+    db.prepare("UPDATE ai_conversation SET active_message_id = ?, updated_at = ? WHERE id = ?").run(
+      id,
+      now,
+      conversationId
     );
-    db.prepare("UPDATE ai_conversation SET updated_at = ? WHERE id = ?").run(now, conversationId);
     return {
       id,
       conversationId,
       role,
       content,
       timestamp: now,
+      parentId,
       dataKeywords,
       dataMessageCount,
       contentBlocks,
@@ -11847,40 +12150,181 @@ var AIConversationManager = class {
     };
   }
   getMessages(conversationId) {
-    const db = this.getDb();
-    const rows = db.prepare(
-      `SELECT id, conversation_id as conversationId, role, content, timestamp,
-                data_keywords as dataKeywords, data_message_count as dataMessageCount,
-                content_blocks as contentBlocks, token_usage as tokenUsage
-         FROM ai_message WHERE conversation_id = ? ORDER BY timestamp ASC`
-    ).all(conversationId);
-    return rows.map((row) => ({
-      id: row.id,
-      conversationId: row.conversationId,
-      role: row.role,
-      content: row.content,
-      timestamp: row.timestamp,
-      dataKeywords: row.dataKeywords ? JSON.parse(row.dataKeywords) : void 0,
-      dataMessageCount: row.dataMessageCount ?? void 0,
-      contentBlocks: row.contentBlocks ? JSON.parse(row.contentBlocks) : void 0,
-      tokenUsage: row.tokenUsage ? JSON.parse(row.tokenUsage) : void 0
-    }));
+    return this.getActivePathRows(conversationId).map((row) => this.parseMessageRow(row));
   }
   deleteMessage(messageId) {
     const db = this.getDb();
     const result = db.prepare("DELETE FROM ai_message WHERE id = ?").run(messageId);
     return result.changes > 0;
   }
-  getConversationTokenUsage(conversationId) {
+  deleteMessagesFrom(conversationId, messageId) {
     const db = this.getDb();
-    const row = db.prepare(
-      `SELECT
-           COALESCE(SUM(json_extract(token_usage, '$.promptTokens')), 0) as promptTokens,
-           COALESCE(SUM(json_extract(token_usage, '$.completionTokens')), 0) as completionTokens,
-           COALESCE(SUM(json_extract(token_usage, '$.totalTokens')), 0) as totalTokens
-         FROM ai_message WHERE conversation_id = ? AND token_usage IS NOT NULL`
-    ).get(conversationId);
-    return { promptTokens: row.promptTokens, completionTokens: row.completionTokens, totalTokens: row.totalTokens };
+    const target = this.getMessageRow(messageId);
+    if (!target || target.conversationId !== conversationId) {
+      throw new Error("Message not found in conversation");
+    }
+    const activePath = this.getActivePathRows(conversationId);
+    const targetIndex = activePath.findIndex((row) => row.id === messageId);
+    if (targetIndex < 0) {
+      throw new Error("Message not on active path");
+    }
+    const idsToDelete = activePath.slice(targetIndex).map((row) => row.id);
+    const placeholders = idsToDelete.map(() => "?").join(", ");
+    db.prepare(`DELETE FROM ai_message WHERE id IN (${placeholders})`).run(...idsToDelete);
+    const newLeafId = targetIndex > 0 ? activePath[targetIndex - 1].id : null;
+    const now = Math.floor(Date.now() / 1e3);
+    db.prepare("UPDATE ai_conversation SET active_message_id = ?, updated_at = ? WHERE id = ?").run(
+      newLeafId,
+      now,
+      conversationId
+    );
+  }
+  forkConversation(sourceConversationId, upToMessageId, title) {
+    const db = this.getDb();
+    const source = this.getConversation(sourceConversationId);
+    if (!source) {
+      throw new Error("Source conversation not found");
+    }
+    const activePath = this.getActivePathRows(sourceConversationId);
+    const cutIndex = activePath.findIndex((row) => row.id === upToMessageId);
+    if (cutIndex < 0) {
+      throw new Error("Message not on active path");
+    }
+    const messagesToCopy = activePath.slice(0, cutIndex + 1);
+    const now = Math.floor(Date.now() / 1e3);
+    const newConvId = this.generateId("conv");
+    const forkTitle = title || `${source.title || "Untitled"} (fork)`;
+    db.prepare(
+      `INSERT INTO ai_conversation (id, session_id, title, assistant_id, active_message_id, created_at, updated_at)
+       VALUES (?, ?, ?, ?, NULL, ?, ?)`
+    ).run(newConvId, source.sessionId, forkTitle, source.assistantId, now, now);
+    const idMap = /* @__PURE__ */ new Map();
+    let lastNewId = null;
+    for (const row of messagesToCopy) {
+      const newMsgId = this.generateId("msg");
+      idMap.set(row.id, newMsgId);
+      const newParentId = row.parentId ? idMap.get(row.parentId) ?? null : null;
+      db.prepare(
+        `INSERT INTO ai_message (
+           id, conversation_id, role, content, timestamp, data_keywords, data_message_count,
+           content_blocks, token_usage, debug_context, parent_id, sibling_group_id, branch_index
+         )
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, NULL, ?, ?, 0)`
+      ).run(
+        newMsgId,
+        newConvId,
+        row.role,
+        row.content,
+        row.timestamp,
+        row.dataKeywords,
+        row.dataMessageCount,
+        row.contentBlocks,
+        row.tokenUsage,
+        newParentId,
+        newMsgId
+      );
+      lastNewId = newMsgId;
+    }
+    if (lastNewId) {
+      db.prepare("UPDATE ai_conversation SET active_message_id = ? WHERE id = ?").run(lastNewId, newConvId);
+    }
+    return {
+      id: newConvId,
+      sessionId: source.sessionId,
+      title: forkTitle,
+      assistantId: source.assistantId,
+      activeMessageId: lastNewId,
+      createdAt: now,
+      updatedAt: now
+    };
+  }
+  updateMessageContent(messageId, newContent) {
+    const db = this.getDb();
+    const result = db.prepare("UPDATE ai_message SET content = ? WHERE id = ?").run(newContent, messageId);
+    if (result.changes === 0) throw new Error("Message not found");
+  }
+  deleteAndRelinkMessage(conversationId, messageId) {
+    const db = this.getDb();
+    const target = this.getMessageRow(messageId);
+    if (!target || target.conversationId !== conversationId) {
+      throw new Error("Message not found in conversation");
+    }
+    db.prepare("UPDATE ai_message SET parent_id = ? WHERE parent_id = ? AND conversation_id = ?").run(
+      target.parentId,
+      messageId,
+      conversationId
+    );
+    const conv = this.getConversation(conversationId);
+    if (conv?.activeMessageId === messageId) {
+      const now = Math.floor(Date.now() / 1e3);
+      db.prepare("UPDATE ai_conversation SET active_message_id = ?, updated_at = ? WHERE id = ?").run(
+        target.parentId,
+        now,
+        conversationId
+      );
+    }
+    db.prepare("DELETE FROM ai_message WHERE id = ?").run(messageId);
+  }
+  insertMessageAfter(conversationId, afterMessageId, role, content, contentBlocks, tokenUsage) {
+    const db = this.getDb();
+    const now = Math.floor(Date.now() / 1e3);
+    const id = this.generateId("msg");
+    const pendingDebug = role === "assistant" ? this.pendingDebugContextMap.get(conversationId) : void 0;
+    if (pendingDebug) {
+      this.pendingDebugContextMap.delete(conversationId);
+    }
+    const childRow = db.prepare("SELECT id FROM ai_message WHERE parent_id = ? AND conversation_id = ? LIMIT 1").get(afterMessageId, conversationId);
+    db.prepare(
+      `INSERT INTO ai_message (
+         id, conversation_id, role, content, timestamp, data_keywords, data_message_count,
+         content_blocks, token_usage, debug_context, parent_id, sibling_group_id, branch_index
+       )
+       VALUES (?, ?, ?, ?, ?, NULL, NULL, ?, ?, ?, ?, ?, 0)`
+    ).run(
+      id,
+      conversationId,
+      role,
+      content,
+      now,
+      contentBlocks ? JSON.stringify(contentBlocks) : null,
+      tokenUsage ? JSON.stringify(tokenUsage) : null,
+      pendingDebug ?? null,
+      afterMessageId,
+      id
+    );
+    if (childRow) {
+      db.prepare("UPDATE ai_message SET parent_id = ? WHERE id = ?").run(id, childRow.id);
+      db.prepare("UPDATE ai_conversation SET updated_at = ? WHERE id = ?").run(now, conversationId);
+    } else {
+      db.prepare("UPDATE ai_conversation SET active_message_id = ?, updated_at = ? WHERE id = ?").run(
+        id,
+        now,
+        conversationId
+      );
+    }
+    return {
+      id,
+      conversationId,
+      role,
+      content,
+      timestamp: now,
+      parentId: afterMessageId,
+      contentBlocks,
+      tokenUsage
+    };
+  }
+  getConversationTokenUsage(conversationId) {
+    const rows = this.getActivePathRows(conversationId);
+    const result = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+    for (const row of rows) {
+      if (row.tokenUsage) {
+        const usage = JSON.parse(row.tokenUsage);
+        result.promptTokens += usage.promptTokens;
+        result.completionTokens += usage.completionTokens;
+        result.totalTokens += usage.totalTokens;
+      }
+    }
+    return result;
   }
   // ==================== Debug context ====================
   setPendingDebugContext(conversationId, debugContext) {
@@ -11896,8 +12340,8 @@ var AIConversationManager = class {
     return result.changes;
   }
   // ==================== Agent 专用 ====================
-  getHistoryForAgent(conversationId, maxMessages) {
-    const messages = this.getMessages(conversationId);
+  getHistoryForAgent(conversationId, maxMessages, leafMessageId) {
+    const messages = this.getActivePathRows(conversationId, leafMessageId).map((row) => this.parseMessageRow(row));
     const validMessages = messages.filter(
       (m) => (m.role === "user" || m.role === "assistant" || m.role === "summary") && m.content?.trim()
     );
@@ -11940,8 +12384,6 @@ var AIConversationManager = class {
   }
   // ==================== Summary / 压缩专用 ====================
   addSummaryMessage(conversationId, content, meta) {
-    const db = this.getDb();
-    db.prepare("DELETE FROM ai_message WHERE conversation_id = ? AND role = 'summary'").run(conversationId);
     const contentBlocks = [
       {
         type: "summary_meta",
@@ -11952,60 +12394,27 @@ var AIConversationManager = class {
     return this.addMessage(conversationId, "summary", content, void 0, void 0, contentBlocks);
   }
   getLatestSummary(conversationId) {
-    const db = this.getDb();
-    const row = db.prepare(
-      `SELECT id, conversation_id as conversationId, role, content, timestamp,
-                data_keywords as dataKeywords, data_message_count as dataMessageCount,
-                content_blocks as contentBlocks
-         FROM ai_message WHERE conversation_id = ? AND role = 'summary'
-         ORDER BY timestamp DESC LIMIT 1`
-    ).get(conversationId);
-    if (!row) return null;
-    return {
-      id: row.id,
-      conversationId: row.conversationId,
-      role: row.role,
-      content: row.content,
-      timestamp: row.timestamp,
-      dataKeywords: row.dataKeywords ? JSON.parse(row.dataKeywords) : void 0,
-      dataMessageCount: row.dataMessageCount ?? void 0,
-      contentBlocks: row.contentBlocks ? JSON.parse(row.contentBlocks) : void 0
-    };
+    const row = [...this.getActivePathRows(conversationId)].reverse().find((message) => message.role === "summary");
+    return row ? this.parseMessageRow(row) : null;
   }
   getMessagesAfterSummary(conversationId, summaryTimestamp) {
-    const db = this.getDb();
-    const rows = db.prepare(
-      `SELECT role, content, timestamp FROM ai_message
-         WHERE conversation_id = ? AND timestamp > ? AND role IN ('user', 'assistant')
-         ORDER BY timestamp ASC`
-    ).all(conversationId, summaryTimestamp);
-    return rows.map((r) => ({ role: r.role, content: r.content, timestamp: r.timestamp }));
+    return this.getActivePathRows(conversationId).filter((row) => row.timestamp > summaryTimestamp && (row.role === "user" || row.role === "assistant")).map((row) => ({ role: row.role, content: row.content, timestamp: row.timestamp }));
   }
   getAllUserAssistantMessages(conversationId) {
-    const db = this.getDb();
-    const rows = db.prepare(
-      `SELECT role, content, timestamp FROM ai_message
-         WHERE conversation_id = ? AND role IN ('user', 'assistant')
-         ORDER BY timestamp ASC`
-    ).all(conversationId);
-    return rows.map((r) => ({ role: r.role, content: r.content, timestamp: r.timestamp }));
+    return this.getActivePathRows(conversationId).filter((row) => row.role === "user" || row.role === "assistant").map((row) => ({ role: row.role, content: row.content, timestamp: row.timestamp }));
   }
   getMessageCountAfterSummary(conversationId) {
     const summary = this.getLatestSummary(conversationId);
     if (!summary) {
-      const db2 = this.getDb();
-      const row2 = db2.prepare("SELECT COUNT(*) as count FROM ai_message WHERE conversation_id = ? AND role IN ('user', 'assistant')").get(conversationId);
-      return row2.count;
+      return this.getActivePathRows(conversationId).filter((row) => row.role === "user" || row.role === "assistant").length;
     }
     const metaBlock = summary.contentBlocks?.find(
       (b) => b.type === "summary_meta"
     );
     const boundary = metaBlock?.bufferBoundaryTimestamp ?? summary.timestamp;
-    const db = this.getDb();
-    const row = db.prepare(
-      "SELECT COUNT(*) as count FROM ai_message WHERE conversation_id = ? AND timestamp >= ? AND role IN ('user', 'assistant')"
-    ).get(conversationId, boundary);
-    return row.count;
+    return this.getActivePathRows(conversationId).filter(
+      (row) => row.timestamp >= boundary && (row.role === "user" || row.role === "assistant")
+    ).length;
   }
 };
 
@@ -13177,7 +13586,8 @@ function applyPreprocessingPipeline(options) {
 // ../../packages/node-runtime/src/ai/agent/core.ts
 import { Agent as PiAgentCore } from "@earendil-works/pi-agent-core";
 import {
-  streamSimple as defaultStreamSimple
+  streamSimple as defaultStreamSimple,
+  clampThinkingLevel
 } from "@earendil-works/pi-ai";
 function createEmptyPiUsage() {
   return {
@@ -13241,11 +13651,34 @@ async function runAgentCore(options) {
   if (abortSignal?.aborted) {
     return { usage: totalUsage, finalMessages: [], toolsUsed: [], toolRounds: 0 };
   }
+  const isDefault = !options.thinkingLevel || options.thinkingLevel === "default";
+  const effectiveModel = isDefault ? { ...piModel, reasoning: false, compat: void 0, thinkingLevelMap: void 0 } : piModel;
+  const resolvedThinkingLevel = (() => {
+    if (isDefault) return "off";
+    const level = options.thinkingLevel;
+    if (level === "auto") {
+      if (!piModel.reasoning) return "off";
+      const compat = piModel.compat;
+      if (compat?.thinkingFormat) return "high";
+      if (piModel.thinkingLevelMap) return "high";
+      return void 0;
+    }
+    return clampThinkingLevel(piModel, level);
+  })();
+  const finalThinkingLevel = resolvedThinkingLevel ?? (piModel.reasoning ? void 0 : "off");
+  console.debug("[agent-core] thinkingLevel resolution:", {
+    input: options.thinkingLevel,
+    modelId: piModel.id,
+    modelReasoning: effectiveModel.reasoning,
+    isDefault,
+    resolved: resolvedThinkingLevel,
+    final: finalThinkingLevel
+  });
   const coreAgent = new PiAgentCore({
     initialState: {
       systemPrompt,
-      model: piModel,
-      thinkingLevel: piModel.reasoning ? "medium" : "off",
+      model: effectiveModel,
+      thinkingLevel: finalThinkingLevel,
       tools: maxToolRounds > 0 ? tools : [],
       messages: toPiHistoryMessages(history)
     },
@@ -15388,6 +15821,20 @@ var BUILTIN_PROVIDER_API = {
   gemini: "google-generative-ai",
   anthropic: "anthropic-messages"
 };
+function inferReasoning(provider, modelId, modelDef) {
+  const reasoning = modelDef?.capabilities.includes("reasoning") || !modelDef?.builtin && isReasoningModel(provider, modelId);
+  if (!reasoning) return { reasoning: false, compat: void 0 };
+  const thinkingCompat = getThinkingCompat(provider, modelId);
+  if (Object.keys(thinkingCompat).length === 0) {
+    return { reasoning: true, compat: void 0 };
+  }
+  const { thinkingLevelMap, ...compatFields } = thinkingCompat;
+  return {
+    reasoning: true,
+    compat: Object.keys(compatFields).length > 0 ? compatFields : void 0,
+    thinkingLevelMap
+  };
+}
 function buildPiModel(config, options) {
   const providerDef = BUILTIN_PROVIDERS.find((p) => p.id === config.provider);
   const baseUrl = config.baseUrl || providerDef?.defaultBaseUrl || "";
@@ -15403,7 +15850,7 @@ function buildPiModel(config, options) {
       api: "google-generative-ai",
       provider: "google",
       baseUrl,
-      reasoning: false,
+      reasoning: isReasoningModel(config.provider, modelId),
       input: ["text"],
       cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
       contextWindow,
@@ -15425,6 +15872,7 @@ function buildPiModel(config, options) {
     };
   }
   const resolvedBaseUrl = config.provider === "openai-compatible" && (apiFormat === "openai-completions" || apiFormat === "openai-responses") ? normalizeOpenAICompatibleBaseUrl(baseUrl) : baseUrl;
+  const { reasoning, compat, thinkingLevelMap } = inferReasoning(config.provider, modelId, modelDef);
   return {
     id: modelId,
     name: modelId,
@@ -15432,12 +15880,13 @@ function buildPiModel(config, options) {
     provider: config.provider,
     baseUrl: resolvedBaseUrl,
     headers: options?.headers,
-    reasoning: config.isReasoningModel ?? false,
+    reasoning,
+    thinkingLevelMap,
     input: ["text"],
     cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
     contextWindow,
     maxTokens: config.maxTokens ?? 4096,
-    compat: config.disableThinking ? { thinkingFormat: "qwen" } : void 0
+    compat
   };
 }
 
@@ -18804,6 +19253,7 @@ __export(session_index_service_exports, {
   clearIndex: () => clearIndex,
   generateIncrementalIndex: () => generateIncrementalIndex,
   generateIndex: () => generateIndex,
+  getAllIndexStats: () => getAllIndexStats,
   getFtsStatus: () => getFtsStatus,
   rebuildFts: () => rebuildFts,
   searchFts: () => searchFts
@@ -18832,6 +19282,18 @@ function rebuildFts(adapter, sessionId) {
   const db = adapter.ensureWritable(sessionId);
   return rebuildFtsIndex(db);
 }
+function getAllIndexStats(adapter) {
+  const ids = adapter.listSessionIds();
+  return ids.map((sessionId) => {
+    try {
+      const db = adapter.ensureReadonly(sessionId);
+      const stats = getSessionIndexStats(db);
+      return { sessionId, hasIndex: stats.hasIndex, sessionCount: stats.sessionCount };
+    } catch {
+      return { sessionId, hasIndex: false, sessionCount: 0 };
+    }
+  });
+}
 
 // ../../packages/node-runtime/src/services/summary-service.ts
 var summary_service_exports = {};
@@ -19201,4 +19663,4 @@ export {
   export_service_exports,
   createDatabaseManagerAdapter
 };
-//# sourceMappingURL=chunk-SQKT45MM.mjs.map
+//# sourceMappingURL=chunk-G7Q7EQ3V.mjs.map