sse客户端第二版

Author: Smoke Tester

.snow/notebook/snow-cli.json

@@ -870,4 +870,13 @@
 点击左侧的会话列表中的会话 tab 就是继续此会话，会话的历史记录参考一下那个测试客户端和 TUI
 
 聊天记录区,自动置底,如果用户手动滚动到非底部,则不再自动置底,等当用手动滚到了最底部或用户发送了新消息后,再自动置底到最新一条.
-输入项目绝对路径的地方改成可输入的下拉框,下拉框里有哪些路径,新加一个配置文件吧,然后在启动和停止全部的后面新加一个按钮保存路径删除的话就不在网页里提供按钮了了，就让用户手动去配置里删除好了
+输入项目绝对路径的地方改成可输入的下拉框,下拉框里有哪些路径,新加一个配置文件吧,然后在启动和停止全部的后面新加一个按钮保存路径删除的话就不在网页里提供按钮了了，就让用户手动去配置里删除好了
+有中断会话按钮 和回退到记录点按钮,回退到的那一条消息,把回退点的消息的原文放到输入框中
+用户在主代理工作时也可发送消息 并放入 消息队列,且可撤回编辑参考 tui
+子代理工作完后 子代理自己这个弹窗可手动关闭 
+
+
+现在 sse 客户端整体功能差不多了,但整体界面布局还是有些不好看,操作等不够人性化.特别是聊天记录区的显示和 todo 区的显示.优化下显示效果.用动画显示下当前会话正在工作还是已停止.聊天记录用类似聊天软件那种用户消息在右边助手和工具消息在左边的布局.子代理弹窗也优化下.todo 区也用颜色和图标等展示,并支持树状 todo.底部转换栏也优化下显示.整体界面人性化,风格简洁优美,仍保持深色主题.总之现在功能基本齐了,就是界面和交互体验还可以更上一层楼.
+git 简易查看和管理区也要美化
+当子代理工作时 应显示子代理弹窗,如果是多个子代理被并行调用了，则显示像卡片形式的，可左右滑动查看.多个子代理工作弹窗
+你自行开浏览器检查显示效果

.snow/notebook/snow-cli_snapshots.json

@@ -20,7 +20,8 @@ export type SSEEventType =
 	| 'agent_list'
 	| 'agent_switched'
 	| 'todo_update'
-	| 'todos';
+	| 'todos'
+	| 'sub_agent_message';
 
 /**
  * SSE 事件数据结构
@@ -56,6 +57,7 @@ export interface ClientMessage {
 	agentId?: string; // 目标主代理ID，用于 switch_agent 消息
 	rollback?: {
 		messageIndex: number;
+		snapshotIndex?: number; // UI 消息索引, 用于快照操作(与 messageIndex 属不同索引体系)
 		rollbackFiles: boolean;
 		selectedFiles?: string[];
 		crossSessionRollback?: boolean;
@@ -545,6 +547,9 @@ export class SSEServer {
 				const {hashBasedSnapshotManager} = await import(
 					'../utils/codebase/hashBasedSnapshot.js'
 				);
+				const {convertSessionMessagesToUI} = await import(
+					'../utils/session/sessionConverter.js'
+				);
 
 				const sessionIdRaw = query?.['sessionId'];
 				const sessionId =
@@ -568,6 +573,30 @@ export class SSEServer {
 					return;
 				}
 
+				// 快照系统使用 UI 消息索引(convertSessionMessagesToUI 后的长度)
+				// 而非原始 session.messages 索引, 两者不一致(UI 转换会合并/过滤 tool 等消息)
+				const uiMessages = convertSessionMessagesToUI(session.messages);
+
+				// 构建 "原始 user 消息索引 → 对应 UI 消息索引" 映射
+				// 快照在 tool 执行期间创建, messageIndex = 当时的 uiMessages.length
+				// 即: user 消息保存后, uiMessages 中该 user 位于 uiIdx, 快照记录 uiIdx + 1
+				const rawToUiIndexMap = new Map<number, number>();
+				let uiIdx = 0;
+				for (let i = 0; i < session.messages.length; i++) {
+					const msg: any = session.messages[i];
+					if (!msg || msg.role !== 'user') continue;
+					while (
+						uiIdx < uiMessages.length &&
+						uiMessages[uiIdx]!.role !== 'user'
+					) {
+						uiIdx++;
+					}
+					if (uiIdx < uiMessages.length) {
+						rawToUiIndexMap.set(i, uiIdx + 1);
+						uiIdx++;
+					}
+				}
+
 				const snapshots = await hashBasedSnapshotManager.listSnapshots(
 					sessionId,
 				);
@@ -584,9 +613,11 @@ export class SSEServer {
 
 				const points: Array<{
 					messageIndex: number;
+					snapshotIndex?: number;
 					role: 'user';
 					timestamp: number;
 					summary: string;
+					content: string;
 					hasSnapshot: boolean;
 					snapshot?: {timestamp: number; fileCount: number};
 					filesToRollbackCount: number;
@@ -603,40 +634,31 @@ export class SSEServer {
 							? normalized.slice(0, maxSummaryLen) + '…'
 							: normalized;
 
-					// Snapshot 的 messageIndex 和 session.messages 的索引并不总是一致。
-					// 实测快照通常对应“下一条消息写入前”的索引（例如首条 user 消息后快照会落在 1）。
-					const snapAtNext = snapshotByIndex.get(i + 1);
-					const snapAtCurrent = snapshotByIndex.get(i);
-					const snap = snapAtNext ?? snapAtCurrent;
-					const rollbackIndex = snapAtNext ? i + 1 : i;
+					// 使用 UI 消息索引查找快照(快照系统记录的是 UI 索引)
+					// 映射缺失时(uiIndex === -1)表示该 user 消息无 UI 对应, 跳过快照查找
+					const uiIndex = rawToUiIndexMap.get(i) ?? -1;
+					const snap = uiIndex >= 0 ? snapshotByIndex.get(uiIndex) : undefined;
 
 					let filesToRollbackCount = 0;
 					if (snap && snap.fileCount > 0) {
 						const files = await hashBasedSnapshotManager.getFilesToRollback(
 							sessionId,
-							rollbackIndex,
+							uiIndex,
 						);
 						filesToRollbackCount = Array.isArray(files) ? files.length : 0;
 					}
 
 					points.push({
 						messageIndex: i,
+						...(uiIndex >= 0 ? {snapshotIndex: uiIndex} : {}),
 						role: 'user',
 						timestamp: typeof m.timestamp === 'number' ? m.timestamp : 0,
 						summary,
+						content,
 						hasSnapshot: !!snap && snap.fileCount > 0,
 						snapshot: snap,
 						filesToRollbackCount,
 					});
-
-					// 如果快照存在但落在 i+1（常见），让前端能直接用 messageIndex 作为回滚点索引。
-					if (
-						snapAtNext &&
-						snapAtNext.fileCount > 0 &&
-						i + 1 < session.messages.length
-					) {
-						// 这里不改变 messageIndex 的语义，仅用于确保 hasSnapshot 展示正确。
-					}
 				}
 
 				res.writeHead(200, {
@@ -782,6 +804,7 @@ export class SSEServer {
 					content: string;
 					[key: string]: any;
 				}>;
+				let currentSession: any = null;
 
 				// 支持两种方式：直接传入 messages 或通过 sessionId 获取
 				if (body.messages && Array.isArray(body.messages)) {
@@ -799,6 +822,7 @@ export class SSEServer {
 						return;
 					}
 					messages = session.messages || [];
+					currentSession = session;
 				} else {
 					res.writeHead(400, {
 						'Content-Type': 'application/json',
@@ -856,11 +880,82 @@ export class SSEServer {
 					return;
 				}
 
+				// Build new session messages with compression summary
+				const newSessionMessages: Array<any> = [];
+				let finalContent = `[Context Summary from Previous Conversation]\n\n${result.summary}`;
+
+				if (result.preservedMessages && result.preservedMessages.length > 0) {
+					finalContent +=
+						'\n\n---\n\n[Last Interaction - Preserved for Continuity]\n\n';
+					for (const msg of result.preservedMessages) {
+						if (msg.role === 'user') {
+							finalContent += `**User:**\n${msg.content}\n\n`;
+						} else if (msg.role === 'assistant') {
+							finalContent += `**Assistant:**\n${msg.content}`;
+							if (msg.tool_calls && msg.tool_calls.length > 0) {
+								finalContent += '\n\n**[Tool Calls Initiated]:**\n```json\n';
+								finalContent += JSON.stringify(msg.tool_calls, null, 2);
+								finalContent += '\n```\n\n';
+							} else {
+								finalContent += '\n\n';
+							}
+						} else if (msg.role === 'tool') {
+							finalContent += `**[Tool Result - ${msg.tool_call_id}]:**\n`;
+							try {
+								const parsed = JSON.parse(msg.content);
+								finalContent +=
+									'```json\n' + JSON.stringify(parsed, null, 2) + '\n```\n\n';
+							} catch {
+								finalContent += `${msg.content}\n\n`;
+							}
+						}
+					}
+				}
+
+				newSessionMessages.push({
+					role: 'user',
+					content: finalContent,
+					timestamp: Date.now(),
+				});
+
+				// Create new compressed session
+				const compressedSession = await sessionManager.createNewSession(
+					false,
+					true,
+				);
+
+				// Set session properties
+				compressedSession.messages = newSessionMessages;
+				compressedSession.messageCount = newSessionMessages.length;
+				compressedSession.updatedAt = Date.now();
+				compressedSession.title = currentSession?.title || 'Compressed Session';
+				compressedSession.summary = currentSession?.summary || '';
+
+				// Record compression relationship
+				const sourceSessionId = body.sessionId;
+				compressedSession.compressedFrom = sourceSessionId;
+				compressedSession.compressedAt = Date.now();
+				compressedSession.originalMessageIndex =
+					result.preservedMessageStartIndex;
+
+				// Save new session to disk
+				await sessionManager.saveSession(compressedSession);
+
+				// Return full response with new session info
 				res.writeHead(200, {
 					'Content-Type': 'application/json',
 					'Access-Control-Allow-Origin': '*',
 				});
-				res.end(JSON.stringify({success: true, result}));
+				res.end(
+					JSON.stringify({
+						success: true,
+						result: {
+							...result,
+							sessionId: compressedSession.id,
+							compressedFrom: sourceSessionId,
+						},
+					}),
+				);
 			} catch (error) {
 				res.writeHead(500, {
 					'Content-Type': 'application/json',

requirements/全平台网页版Snow SSE客户端需求.md

@@ -18,6 +18,7 @@ import {
 	type ToolCall,
 	type MCPExecutionContext,
 } from '../../utils/execution/toolExecutor.js';
+import type {SubAgentMessage} from '../../utils/execution/subAgentExecutor.js';
 import {getOpenAiConfig} from '../../utils/config/apiConfig.js';
 import {sessionManager} from '../../utils/session/sessionManager.js';
 import {unifiedHooksExecutor} from '../../utils/execution/unifiedHooksExecutor.js';
@@ -122,6 +123,7 @@ export type ConversationHandlerOptions = {
 	getCurrentContextPercentage?: () => number; // Get current context percentage from ChatInput
 	setCurrentModel?: React.Dispatch<React.SetStateAction<string | null>>; // Set current model name for display
 	setIsStopping?: React.Dispatch<React.SetStateAction<boolean>>; // Control stopping state
+	onRawSubAgentMessage?: (message: SubAgentMessage) => void; // 可选: 原始子代理消息透传回调(SSE 服务端用于转发事件)
 };
 
 /**
@@ -987,6 +989,8 @@ async function executeWithInternalRetry(
 					setStreamTokenCount,
 
 					async subAgentMessage => {
+						// 原始子代理消息透传(SSE 服务端用于转发 sub_agent_message 事件到客户端)
+						options.onRawSubAgentMessage?.(subAgentMessage);
 						// Handle sub-agent messages - display and save to session
 						setMessages(prev => {
 							// Handle sub-agent context usage update

source/api/sse-server.ts

@@ -14,7 +14,7 @@ import type {AgentInfo} from '../api/sse-server.js';
 import {
 	getAgentsPrompt,
 	createSystemContext,
-	getTaskCompletionPrompt,
+	// getTaskCompletionPrompt,
 } from './agentsPromptUtils.js';
 import {getModelSpecificPrompt} from './config/apiConfig.js';
 
@@ -363,10 +363,10 @@ export class MainAgentManager {
 		}
 
 		// 添加任务完成标识提示词
-		const taskCompletionPrompt = getTaskCompletionPrompt();
-		if (taskCompletionPrompt) {
-			prompt += '\n\n' + taskCompletionPrompt;
-		}
+		// const taskCompletionPrompt = getTaskCompletionPrompt();
+		// if (taskCompletionPrompt) {
+		// 	prompt += '\n\n' + taskCompletionPrompt;
+		// }
 
 		return prompt.trim();
 	}

source/hooks/conversation/useConversation.ts

@@ -215,9 +215,9 @@ export function getAgentsPrompt(): string {
  *
  */
 export function getTaskCompletionPrompt(): string {
-	return `若你完成了所有任务,请在你的最终回复的最开头或最结尾添加 \`[Mission_Accomplished!]\` 标记!!!并重新完整做出带标记的最终回复!!!
-你要不停的使用工具直到完成所有任务,才可进行一般回复,且MUST在任务完成后的最终回复的最开头或最结尾中添加 \`[Mission_Accomplished!]\` 标记,即使是简单的打招呼,测试等极简单任务完成后也MUST加上该标记,否则将不会认为此条回复是最终回复,并打回让你继续任务.
-若任务未完成,请继续使用工具直到完成所有任务.特别的不用每完成一个阶段就问一下用户,而是不停继续,去完成所有任务!
+	return `若你完成了所有任务和所有需求,请在你的最终回复的最开头或最结尾添加 \`[Mission_Accomplished!]\` 标记!!!并重新完整做出带标记的最终回复!!!
+你要不停的使用工具直到完成所有任务所有需求,才可进行一般回复,且MUST在任务完成后的最终回复的最开头或最结尾中添加 \`[Mission_Accomplished!]\` 标记,即使是简单的打招呼,测试等极简单任务完成后也MUST加上该标记,否则将不会认为此条回复是最终回复,并打回让你继续任务.
+若任务未完成,请继续使用工具直到完成所有任务所有需求.特别的不用每完成一个阶段就问一下用户,而是不停继续,去完成所有任务!
 若想提问则必须使用专门的提问工具提问,若你无此工具则说明你不可提问,MUST自主决策继续任务.
 `;
 }

source/utils/MainAgentManager.ts

@@ -8,6 +8,12 @@
  * 这些通常是耗时较长的工具，用户需要看到执行进度
  */
 const TWO_STEP_TOOLS = new Set([
+	// 文件读取工具 - 用户需要看到读取操作
+	'filesystem-read',
+	'ace-file_outline',
+	'ace-text_search',
+	'context_engine-codebase-retrieval', // 上下文引擎工具
+
 	// 文件编辑工具 - 耗时较长，需要显示进度
 	'filesystem-edit',
 	'filesystem-edit_search',

source/utils/agentsPromptUtils.ts

@@ -1677,7 +1677,7 @@ export async function executeMCPTool(
 						args.question,
 						args.options,
 						'', //toolCallId will be set by executeToolCall
-						false, // multiSelect 已移除，默认支持单选和多选
+						true, // 默认多选模式, 用户可选一个或多个
 					);
 				default:
 					throw new Error(`Unknown askuser tool: ${actualToolName}`);

source/utils/config/toolDisplayConfig.ts

@@ -6,7 +6,7 @@ import {getSubAgent, getSubAgents} from '../config/subAgentConfig.js';
 import {
 	getAgentsPrompt,
 	createSystemContext,
-	getTaskCompletionPrompt,
+	// getTaskCompletionPrompt,
 } from '../agentsPromptUtils.js';
 import {
 	collectAllMCPTools,
@@ -597,12 +597,12 @@ You have access to these collaboration tools:
 		}
 
 		// 5. 添加任务完成标识提示词
-		const taskCompletionPrompt = getTaskCompletionPrompt();
-		if (taskCompletionPrompt) {
-			finalPrompt = finalPrompt
-				? `${finalPrompt}\n\n${taskCompletionPrompt}`
-				: taskCompletionPrompt;
-		}
+		// const taskCompletionPrompt = getTaskCompletionPrompt();
+		// if (taskCompletionPrompt) {
+		// 	finalPrompt = finalPrompt
+		// 		? `${finalPrompt}\n\n${taskCompletionPrompt}`
+		// 		: taskCompletionPrompt;
+		// }
 
 		// 5.5 注入并行协作上下文
 		if (collaborationContext) {