@keyou007/agent-bot 0.1.22 → 0.1.24-alpha.0

package/dist/proxy/ProxySessionController.js

@@ -24,6 +24,7 @@ import { createId } from "../utils/id.js";
 import { formatStorageSize } from "../utils/formatStorageSize.js";
 import { truncateMiddle, truncateText } from "../utils/markdown.js";
 import { normalizeTaskTitle } from "../utils/taskTitle.js";
+import { conversationImportPrompt, exportConversation } from "./ConversationExport.js";
 import { executeShellCommand, } from "../utils/executeShellCommand.js";
 class PresentedRequestError extends Error {
     originalError;
@@ -109,6 +110,8 @@ const HELP_COMMAND_SECTIONS = [
                 usage: "[序号或任务 ID]",
                 description: "从当前或指定任务最近完成轮次创建分支",
             },
+            { command: "/clone", usage: "[title] [--agent <name>]", description: "导出用户 Prompt 和最终回答，交给新 Agent 任务读取" },
+            { command: "/clonegroup", usage: "[title] [--agent <name>]", description: "将对话上下文交给新群中的 Agent 任务" },
             { command: "/turns", usage: "[Turn ID 或序号]", description: "浏览历史轮次，并可 Reset 对话上下文；带参数查看运行详情" },
             {
                 command: "/title",
@@ -601,7 +604,12 @@ export class ProxySessionController {
         await this.outbound.withReplyTarget(contextKey, replyTarget, async () => {
             try {
                 const kind = String(scopedAction.value.action ?? "");
-                if (kind === "help_command") {
+                if (kind === "agentbot_update_cancel") {
+                    if (!this.lifecycle?.cancelAutomaticUpdate)
+                        throw new Error("自动更新功能未启用。");
+                    await this.lifecycle.cancelAutomaticUpdate(scopedAction);
+                }
+                else if (kind === "help_command") {
                     await this.executeHelpCommandAction(scopedAction, contextKey, replyTarget);
                 }
                 else if (kind === "turn_details") {
@@ -1085,6 +1093,12 @@ export class ProxySessionController {
             task: created.task,
         };
     }
+    async controlCloneTask(localSessionId, requestedTitle, requestedAgentName, newGroup = false, userOpenId) {
+        const source = this.requireControlSession(localSessionId);
+        const contextKey = this.controlSessionContextKey(source);
+        const replyTarget = this.controlSessionReplyTarget(source);
+        return this.outbound.withReplyTarget(contextKey, replyTarget, () => this.cloneSession(contextKey, { ...source, contextKey }, requestedTitle, requestedAgentName, newGroup, userOpenId, replyTarget));
+    }
     async controlForkTaskGroup(localSessionId, requestedTitle, userOpenId) {
         const source = this.store.getSession(localSessionId);
         if (!source || source.status === "closed")
@@ -1228,6 +1242,7 @@ export class ProxySessionController {
                 await loaded.runtime.setReasoningEffort(record.localSessionId, nextEffort);
             }
             this.store.updateRuntimeSession(record.localSessionId, { model: nextValue, reasoningEffort: nextEffort });
+            this.outbound.updateSessionModel(record.localSessionId, nextValue);
         }
         else if (setting === "thinking" && nextValue) {
             const models = await loaded.runtime.listModels(loaded.session.modelProvider, loaded.session.model);
@@ -1478,6 +1493,19 @@ export class ProxySessionController {
                 await this.outbound.sendText(contextKey, CREATE_GROUP_STARTED_MESSAGE);
                 await this.createFeishuGroup(contextKey, context.defaultAgent, command.title, userId, command.cwd === undefined ? undefined : resolveUserPath(command.cwd), command.projectless === true);
                 return;
+            case "clone":
+            case "clonegroup": {
+                let source = this.currentSession(contextKey);
+                let throughTurnId;
+                if (incomingMessage?.threadContext && (!source
+                    || !this.store.findLatestCompletedTurnId(source.localSessionId, contextKey))) {
+                    const resolved = this.resolveThreadForkAnchor(incomingMessage, false);
+                    source = resolved.source;
+                    throughTurnId = resolved.anchor.turnId;
+                }
+                await this.cloneSession(contextKey, source ?? this.requireCurrentSession(contextKey), command.title, command.agentName, command.type === "clonegroup", userId, replyTarget, throughTurnId);
+                return;
+            }
             case "forkgroup":
                 await this.forkCurrentSessionToFeishuGroup(contextKey, command.title, userId, incomingMessage);
                 return;
@@ -2297,7 +2325,7 @@ export class ProxySessionController {
             throw error;
         }
     }
-    resolveThreadForkAnchor(message) {
+    resolveThreadForkAnchor(message, includeSnapshot = true) {
         if (!message.threadContext || !message.threadId || !message.chatId) {
             throw new Error("当前消息不属于可识别的飞书话题。");
         }
@@ -2316,8 +2344,9 @@ export class ProxySessionController {
         if (baseChatContextKey(anchor.contextKey ?? source.contextKey) !== `chat_id:${message.chatId}`) {
             throw new Error("话题来源任务不属于当前会话，已拒绝创建分支。");
         }
-        const snapshot = turnViewSnapshot(this.store.getTurnSnapshot(anchor.turnId));
-        if (isTurnStillRunning(snapshot?.status)
+        const snapshot = includeSnapshot ? turnViewSnapshot(this.store.getTurnSnapshot(anchor.turnId)) : undefined;
+        const status = includeSnapshot ? snapshot?.status : this.store.getTurnPromptSummary(anchor.turnId)?.status;
+        if (isTurnStillRunning(status)
             || (source.lastTurnId === anchor.turnId && source.lastTurnStatus === "running")) {
             throw new Error("话题对应的轮次仍在执行，App Server 暂时不能从这一轮 fork。请等待该轮完成后再在话题中发送消息。");
         }
@@ -2914,6 +2943,7 @@ export class ProxySessionController {
             while (true) {
                 let pendingTurnId;
                 try {
+                    this.outbound.updateSessionModel(loaded.record.localSessionId, loaded.runtime.getSession(loaded.record.localSessionId)?.model ?? loaded.session.model);
                     pendingTurnId = await this.outbound.startPendingTurn(loaded.record.localSessionId, loaded.record.contextKey, title, replyTarget, cardPrompt);
                 }
                 catch (error) {
@@ -2985,7 +3015,7 @@ export class ProxySessionController {
         }
         return turnId;
     }
-    async createSession(contextKey, agentName, cwd, announce, prepareTurn, prompt, replyTarget, requestedTitle, executionSettings = {}) {
+    async createSession(contextKey, agentName, cwd, announce, prepareTurn, prompt, replyTarget, requestedTitle, executionSettings = {}, activate = true) {
         const agent = this.ensureAgent(agentName);
         const resolvedExecutionSettings = mergeExecutionSettings(agent.defaults ?? {}, executionSettings);
         const localSessionId = createId("sess");
@@ -3004,7 +3034,8 @@ export class ProxySessionController {
                 permissionMode: resolvedExecutionSettings.permissionMode,
             });
         }
-        this.store.setCurrentSession(contextKey, localSessionId);
+        if (activate)
+            this.store.setCurrentSession(contextKey, localSessionId);
         this.outbound.registerSession(localSessionId, contextKey, initialTitle, sessionCwd, this.agentLabel(agentName), resolvedExecutionSettings.model);
         const runtime = this.runtimes.forAgent(agentName);
         try {
@@ -3085,6 +3116,83 @@ export class ProxySessionController {
         await this.outbound.sendText(sourceContextKey, `已创建飞书群：${group.name}，并创建新任务 ${taskDescription}。`);
         return { group, task };
     }
+    async cloneSession(sourceContextKey, source, requestedTitle, requestedAgentName, newGroup, userId, replyTarget, throughTurnId) {
+        const context = this.store.getOrCreateUserContext(sourceContextKey, this.config.defaults.agent);
+        const agentName = requestedAgentName?.trim() || context.defaultAgent;
+        this.ensureAgent(agentName);
+        if (newGroup && !userId?.startsWith("ou_")) {
+            throw new Error("/clonegroup 只能由具有 open_id 的飞书用户触发。");
+        }
+        await this.outbound.sendText(sourceContextKey, `正在导出对话并创建 ${this.agentLabel(agentName)} 克隆任务，请稍后。`);
+        const exported = await exportConversation(path.join(path.dirname(this.config.storage.sqlitePath), "context-transfers"), this.readCloneConversation(source, throughTurnId));
+        const title = normalizeTaskTitle(requestedTitle) ?? `${source.title ?? "任务"}（克隆）`;
+        const cwd = detectProjectlessWorkspace(source.cwd) ? undefined : source.cwd;
+        const group = newGroup ? await this.createFeishuGroupContext(sourceContextKey, agentName, title, userId, cwd, "/clonegroup") : undefined;
+        const contextKey = group?.contextKey ?? sourceContextKey;
+        let task;
+        try {
+            task = await this.createSession(contextKey, agentName, cwd, false, false, undefined, undefined, title, { permissionMode: source.permissionMode ?? "auto" }, false);
+        }
+        catch (error) {
+            await this.outbound.sendText(contextKey, `${group ? "群已创建，但" : ""}克隆任务创建失败；上下文文件已保留：${exported.filePath}`);
+            throw error;
+        }
+        this.store.audit(contextKey, "session_cloned", {
+            sourceLocalSessionId: source.localSessionId,
+            clonedLocalSessionId: task.localSessionId,
+            contextFile: exported.filePath,
+            turnCount: exported.turnCount,
+        });
+        try {
+            await this.startTurn(await this.loadSession(task), conversationImportPrompt(exported.filePath), group ? undefined : replyTarget, undefined, { displayPrompt: `读取历史上下文（${exported.turnCount} 轮对话）` });
+        }
+        catch (error) {
+            await this.outbound.sendText(contextKey, `克隆任务已创建，但上下文读取启动失败。任务 ID：${task.localSessionId}；上下文文件：${exported.filePath}`);
+            throw error;
+        }
+        this.store.setCurrentSession(contextKey, task.localSessionId);
+        await this.outbound.sendText(sourceContextKey, group
+            ? `已创建克隆群：${group.name}，正在读取 ${exported.turnCount} 轮历史对话。`
+            : `已切换到 ${this.agentLabel(agentName)} 克隆任务：${title}，正在读取 ${exported.turnCount} 轮历史对话。`);
+        return {
+            sourceLocalSessionId: source.localSessionId,
+            contextFile: exported.filePath,
+            turnCount: exported.turnCount,
+            task: this.store.getSession(task.localSessionId) ?? task,
+            ...(group ? { group } : {}),
+        };
+    }
+    async *readCloneConversation(source, throughTurnId) {
+        const runtime = this.runtimes.forAgent(source.agentName);
+        if (!source.remoteSessionId || !runtime.readConversation) {
+            yield* this.store.readConversation(source.localSessionId, throughTurnId);
+            return;
+        }
+        let localSessionId = source.localSessionId;
+        let remoteSessionId = source.remoteSessionId;
+        const visited = new Set();
+        while (!visited.has(remoteSessionId)) {
+            visited.add(remoteSessionId);
+            const fork = localSessionId
+                ? this.store.getForkHistorySource(localSessionId) : undefined;
+            if (fork && runtime.readRemoteForkSource) {
+                const remote = await runtime.readRemoteForkSource(remoteSessionId);
+                const completed = remote.lastCompletedTurnId
+                    ?? (remote.lastTurnStatus === "completed" ? remote.lastTurnId : undefined);
+                if (!completed) {
+                    if (!fork.sourceRemoteSessionId)
+                        throw new Error("找不到分支的原始任务，无法导出继承的上下文。");
+                    throughTurnId ??= fork.sourceTurnId;
+                    localSessionId = fork.sourceLocalSessionId;
+                    remoteSessionId = fork.sourceRemoteSessionId;
+                    continue;
+                }
+            }
+            yield* runtime.readConversation(remoteSessionId, throughTurnId);
+            return;
+        }
+        throw new Error("分支来源包含循环，无法导出上下文。");
+    }
     async forkCurrentSessionToFeishuGroup(sourceContextKey, requestedTitle, userId, incomingMessage) {
         if (!userId?.startsWith("ou_")) {
             throw new Error("/forkgroup 只能由具有 open_id 的飞书用户消息触发。");
@@ -4062,7 +4170,9 @@ export class ProxySessionController {
                 throw error;
             }
         };
-        return loaded.runtime.setExecutionSettings(loaded.record.localSessionId, settings, persist);
+        const session = await loaded.runtime.setExecutionSettings(loaded.record.localSessionId, settings, persist);
+        this.outbound.updateSessionModel(loaded.record.localSessionId, session.model);
+        return session;
     }
     async assertProviderModelSettings(loaded, modelProvider, model, effort) {
         await this.assertModelProvider(loaded, modelProvider);
@@ -4116,6 +4226,7 @@ export class ProxySessionController {
             await loaded.runtime.setReasoningEffort(loaded.record.localSessionId, nextEffort);
         }
         this.store.updateRuntimeSession(loaded.record.localSessionId, { model, reasoningEffort: nextEffort });
+        this.outbound.updateSessionModel(loaded.record.localSessionId, model);
         await this.persistAgentExecutionDefaults(loaded.record.localSessionId);
         const effortMessage = nextEffort && nextEffort !== currentEffort
             ? `，思考强度已自动调整为 ${nextEffort}`