feat: Enhance auto-compression functionality with retry logic and improved error handling

- Added `isCompressing` state to track compression status in chat logic.
- Implemented `performAutoCompression` function with retry mechanism for context compression.
- Updated UI components to reflect compression status and errors, including retry notifications.
- Introduced new messages for compression retry attempts in sub-agent handling.
- Enhanced file creation logic to support overwriting with backup.
- Added translation keys for model count and scroll hints in language files.
- Refactored model selection logic to ensure unique model options are displayed.

Author: MayDay-wpf

source/hooks/conversation/chatLogic/types.ts

@@ -36,6 +36,7 @@ export interface UseChatLogicProps {
 			images?: Array<{type: 'image'; data: string; mimeType: string}>;
 		} | null>
 	>;
+	isCompressing: boolean;
 	setIsCompressing: React.Dispatch<React.SetStateAction<boolean>>;
 	setCompressionError: React.Dispatch<React.SetStateAction<string | null>>;
 	currentContextPercentageRef: React.MutableRefObject<number>;

source/hooks/conversation/chatLogic/useMessageProcessing.ts

@@ -161,7 +161,7 @@ export function useMessageProcessing(props: UseChatLogicProps) {
 
 				const errorMessage: Message = {
 					role: 'assistant',
-					content: `**Auto-compression Failed**\n\n${errorMsg}`,
+					content: `**Auto-compression Failed**`,
 					streaming: false,
 				};
 				setMessages(prev => [...prev, errorMessage]);

source/hooks/conversation/chatLogic/useRemoteEvents.ts

@@ -3,7 +3,7 @@ import type {UseChatLogicProps} from './types.js';
 import type {RollbackMode} from '../../../ui/components/tools/FileRollbackConfirmation.js';
 import {connectionManager} from '../../../utils/connection/ConnectionManager.js';
 import {sessionManager} from '../../../utils/session/sessionManager.js';
-import {executeContextCompression} from '../useCommandHandler.js';
+import {performAutoCompression} from '../../../utils/core/autoCompress.js';
 
 interface UseRemoteEventsHandlers {
 	handleMessageSubmit: (
@@ -318,19 +318,29 @@ export function useRemoteEvents(
 					await connectionManager.notifyCompactStarted();
 
 					const currentSession = sessionManager.getCurrentSession();
-					if (!currentSession) {
-						throw new Error('No active session to compress');
-					}
 
-					const compressionResult = await executeContextCompression(
-						currentSession.id,
-						status => {
+					const compressionResult = await performAutoCompression(
+						currentSession?.id,
+						(status) => {
 							props.onCompressionStatus?.(status);
 						},
 					);
 
+					if (compressionResult && (compressionResult as any).hookFailed) {
+						setCompressionError('Blocked by beforeCompress hook');
+						await connectionManager.notifyCompactCompleted({
+							success: false,
+							error: 'Blocked by beforeCompress hook',
+						});
+						return;
+					}
+
 					if (!compressionResult) {
-						throw new Error('Compression failed');
+						await connectionManager.notifyCompactCompleted({
+							success: false,
+							error: 'Compression failed after retries',
+						});
+						return;
 					}
 
 					props.onCompressionStatus?.(null);
@@ -353,6 +363,9 @@ export function useRemoteEvents(
 						message: errorMsg,
 					});
 					setCompressionError(errorMsg);
+					setTimeout(() => {
+						props.onCompressionStatus?.(null);
+					}, 5000);
 
 					await connectionManager.notifyCompactCompleted({
 						success: false,

source/hooks/conversation/core/autoCompressHandler.ts

@@ -72,7 +72,7 @@ export async function handleAutoCompression(
 		const session = sessionManager.getCurrentSession();
 
 		// Set up status callback for UI display
-		const onStatusUpdate = (status: CompressionStatus) => {
+		const onStatusUpdate = (status: CompressionStatus | null) => {
 			options.onCompressionStatus?.(status);
 		};
 
@@ -81,8 +81,11 @@ export async function handleAutoCompression(
 			onStatusUpdate,
 		);
 
-		// Clear status after completion
-		options.onCompressionStatus?.(null);
+		// Only clear status on success/hookFailed;
+		// failed status will auto-dismiss after 5s (handled by performAutoCompression)
+		if (compressionResult) {
+			options.onCompressionStatus?.(null);
+		}
 
 		// Check if beforeCompress hook failed
 		if (compressionResult && (compressionResult as any).hookFailed) {
@@ -131,6 +134,9 @@ export async function handleAutoCompression(
 			step: 'failed',
 			message: error instanceof Error ? error.message : 'Unknown error',
 		});
+		setTimeout(() => {
+			options.onCompressionStatus?.(null);
+		}, 5000);
 	} finally {
 		compressionCoordinator.releaseLock('main');
 	}

source/hooks/conversation/core/subAgentMessageHandler.ts

@@ -150,6 +150,8 @@ export class SubAgentUIHandler {
 				return this.handleContextUsage(prev, subAgentMessage);
 			case 'context_compressing':
 				return this.handleContextCompressing(prev, subAgentMessage);
+			case 'context_compress_retrying':
+				return this.handleContextCompressRetrying(prev, subAgentMessage);
 			case 'context_compressed':
 				return this.handleContextCompressed(prev, subAgentMessage);
 			case 'inter_agent_sent':
@@ -731,6 +733,27 @@ export class SubAgentUIHandler {
 		];
 	}
 
+	private handleContextCompressRetrying(
+		prev: Message[],
+		subAgentMessage: SubAgentMessage,
+	): Message[] {
+		const msg = subAgentMessage.message as any;
+		return [
+			...prev,
+			{
+				role: 'subagent' as const,
+				content: `\x1b[36m⚇ ${subAgentMessage.agentName}\x1b[0m \x1b[33m⟳ Compression retry (${msg.attempt}/${msg.maxRetries})...\x1b[0m${msg.error ? ` \x1b[90m${msg.error}\x1b[0m` : ''}`,
+				streaming: false,
+				subAgent: {
+					agentId: subAgentMessage.agentId,
+					agentName: subAgentMessage.agentName,
+					isComplete: false,
+				},
+				subAgentInternal: true,
+			},
+		];
+	}
+
 	private handleContextCompressed(
 		prev: Message[],
 		subAgentMessage: SubAgentMessage,

source/hooks/conversation/useChatLogic.ts

@@ -20,6 +20,7 @@ export function useChatLogic(props: UseChatLogicProps) {
 		setPendingMessages,
 		setRestoreInputContent,
 		userInterruptedRef,
+		isCompressing,
 		vscodeState,
 		commandsLoaded,
 		terminalExecutionState,
@@ -211,6 +212,14 @@ export function useChatLogic(props: UseChatLogicProps) {
 				return true;
 			}
 
+			// Block ESC during /compact command compression
+			if (isCompressing) {
+				streamingState.setCompressBlockToast(
+					t.chatScreen.compressionBlockToast,
+				);
+				return true;
+			}
+
 			// Handle scheduler task interruption
 			if (schedulerExecutionState?.state.isRunning) {
 				schedulerExecutionState.resetTask();
@@ -273,6 +282,7 @@ export function useChatLogic(props: UseChatLogicProps) {
 			backgroundProcesses,
 			terminalExecutionState,
 			streamingState,
+			isCompressing,
 			hasFocus,
 			pendingMessages,
 			handleInterrupt,

source/hooks/conversation/useCommandHandler.ts

@@ -478,36 +478,36 @@ export function useCommandHandler(options: CommandHandlerOptions) {
 				options.setCompressionError(null);
 
 				try {
-					// 获取当前会话ID
-					const currentSession = sessionManager.getCurrentSession();
-					if (!currentSession) {
-						throw new Error('No active session to compress');
-					}
+					const {performAutoCompression} = await import(
+						'../../utils/core/autoCompress.js'
+					);
 
-					// 使用提取的压缩函数，传入当前会话ID和状态回调
-					const compressionResult = await executeContextCompression(
-						currentSession.id,
-						status => {
+					const currentSession = sessionManager.getCurrentSession();
+					const compressionResult = await performAutoCompression(
+						currentSession?.id,
+						(status: CompressionStatus | null) => {
 							options.onCompressionStatus?.(status);
 						},
 					);
 
+					if (compressionResult && (compressionResult as any).hookFailed) {
+						const errorMsg = 'Blocked by beforeCompress hook';
+						options.setCompressionError(errorMsg);
+						return;
+					}
+
 					if (!compressionResult) {
-						throw new Error('Compression failed');
+						return;
 					}
 
-					// Clear compression status after completion
 					options.onCompressionStatus?.(null);
 
-					// 更新UI
 					options.clearSavedMessages();
 					options.setMessages(compressionResult.uiMessages);
 					options.setRemountKey(prev => prev + 1);
 
-					// Update token usage with compression result
 					options.setContextUsage(compressionResult.usage);
 				} catch (error) {
-					// Show error message
 					const errorMsg =
 						error instanceof Error
 							? error.message
@@ -517,13 +517,9 @@ export function useCommandHandler(options: CommandHandlerOptions) {
 						message: errorMsg,
 					});
 					options.setCompressionError(errorMsg);
-
-					const errorMessage: Message = {
-						role: 'assistant',
-						content: `**Compression Failed**\\n\\n${errorMsg}`,
-						streaming: false,
-					};
-					options.setMessages(prev => [...prev, errorMessage]);
+					setTimeout(() => {
+						options.onCompressionStatus?.(null);
+					}, 5000);
 				} finally {
 					options.setIsCompressing(false);
 				}

source/i18n/lang/en.ts

@@ -708,6 +708,8 @@ export const en: TranslationKeys = {
 		saveFailed: 'Save failed',
 		modelSaveFailed: 'Model save failed',
 		tipLabel: 'Tip:',
+		modelCount: '{count} models',
+		scrollHint: '↑↓ scroll for more',
 	},
 	profilePanel: {
 		title: 'Select Profile',

source/i18n/lang/zh-TW.ts

@@ -666,6 +666,8 @@ export const zhTW: TranslationKeys = {
 		saveFailed: '儲存失敗',
 		modelSaveFailed: '模型儲存失敗',
 		tipLabel: '提示:',
+		modelCount: '共 {count} 個模型',
+		scrollHint: '↑↓ 捲動瀏覽更多模型',
 	},
 	profilePanel: {
 		title: '選擇設定檔',

source/i18n/lang/zh.ts

@@ -665,6 +665,8 @@ export const zh: TranslationKeys = {
 		saveFailed: '保存失败',
 		modelSaveFailed: '模型保存失败',
 		tipLabel: '提示:',
+		modelCount: '共 {count} 个模型',
+		scrollHint: '↑↓ 滚动浏览更多模型',
 	},
 	profilePanel: {
 		title: '选择配置',