Add context summarization feature with manual trigger option

Author: shariqriazz

src/core/Cline.ts

@@ -2676,4 +2676,83 @@ export class Cline extends EventEmitter<ClineEvents> {
 		}
 		return totalTokens
 	}
+
+	/**
+	 * Manually triggers summarization of the conversation context.
+	 * @param isManualTrigger Whether this summarization was manually triggered by the user.
+	 * @returns A promise that resolves when summarization is complete.
+	 */
+	public async summarizeConversationContext(isManualTrigger: boolean = false): Promise<void> {
+		// Skip if summarization is disabled
+		if (!this.enableContextSummarization) {
+			this.providerRef.deref()?.log("[Summarization] Context summarization is disabled.")
+			return
+		}
+
+		const initialMessagesToKeep = this.contextSummarizationInitialStaticTurns
+		const recentMessagesToKeep = this.contextSummarizationRecentTurns
+
+		// Ensure we have enough messages to summarize
+		if (this.apiConversationHistory.length <= initialMessagesToKeep + recentMessagesToKeep) {
+			this.providerRef
+				.deref()
+				?.log(
+					`[Summarization] Not enough messages to summarize. Need more than ${initialMessagesToKeep + recentMessagesToKeep} messages.`,
+				)
+			return
+		}
+
+		// Calculate slice points
+		const initialSliceEnd = initialMessagesToKeep
+		const recentSliceStart = this.apiConversationHistory.length - recentMessagesToKeep
+
+		// Ensure slice points don't overlap
+		if (initialSliceEnd >= recentSliceStart) {
+			this.providerRef
+				.deref()
+				?.log(
+					`[Summarization] Skipping: initialSliceEnd (${initialSliceEnd}) >= recentSliceStart (${recentSliceStart}). Not enough messages between initial/recent turns.`,
+				)
+			return
+		}
+
+		// Slice the conversation history
+		const initialMessages = this.apiConversationHistory.slice(0, initialSliceEnd)
+		const recentMessages = this.apiConversationHistory.slice(recentSliceStart)
+		const messagesToSummarize = this.apiConversationHistory.slice(initialSliceEnd, recentSliceStart)
+
+		this.providerRef
+			.deref()
+			?.log(
+				`[Summarization] Slicing: Keep Initial ${initialMessages.length}, Summarize ${messagesToSummarize.length}, Keep Recent ${recentMessages.length}`,
+			)
+
+		// Create summarizer and generate summary
+		const summarizer = new ContextSummarizer(this.api)
+		const summaryMessage = await summarizer.summarize(messagesToSummarize)
+
+		if (!summaryMessage) {
+			this.providerRef.deref()?.log(`[Summarization] Failed to generate summary.`)
+			return
+		}
+
+		// Create new history with summary
+		const newHistory = [...initialMessages, summaryMessage, ...recentMessages]
+
+		// Add a system message to notify the user in the UI
+		if (isManualTrigger) {
+			await this.say("text", "[Conversation history has been summarized to preserve context]")
+		} else {
+			await this.say("text", "[Older conversation turns summarized to preserve context]")
+		}
+
+		// Update the conversation history
+		await this.overwriteApiConversationHistory(newHistory)
+
+		this.providerRef
+			.deref()
+			?.log(
+				`[Summarization] Successfully summarized ${messagesToSummarize.length} messages. New history length: ${newHistory.length}`,
+			)
+	}
 }

src/core/webview/webviewMessageHandler.ts

@@ -877,6 +877,27 @@ export const webviewMessageHandler = async (provider: ClineProvider, message: We
 			await updateGlobalState("contextSummarizationRecentTurns", message.value ?? 10)
 			await provider.postStateToWebview()
 			break
+		case "manualSummarize":
+			// Trigger manual summarization of the conversation context
+			const currentCline = provider.getCurrentCline()
+			if (currentCline) {
+				try {
+					// Notify user that summarization is in progress
+					vscode.window.showInformationMessage(t("common:info.summarizing_context"))
+
+					// Trigger the summarization process
+					await currentCline.summarizeConversationContext(true) // true indicates manual trigger
+
+					// Notify user that summarization is complete
+					vscode.window.showInformationMessage(t("common:info.summarization_complete"))
+				} catch (error) {
+					provider.log(
+						`Error during manual summarization: ${JSON.stringify(error, Object.getOwnPropertyNames(error), 2)}`,
+					)
+					vscode.window.showErrorMessage(t("common:errors.summarization_failed"))
+				}
+			}
+			break
 		// --- End Context Summarization ---
 		case "toggleApiConfigPin":
 			if (message.text) {

src/i18n/locales/en/common.json

@@ -47,6 +47,7 @@
 		"create_mcp_json": "Failed to create or open .roo/mcp.json: {{error}}",
 		"hmr_not_running": "Local development server is not running, HMR will not work. Please run 'npm run dev' before launching the extension to enable HMR.",
 		"retrieve_current_mode": "Error: failed to retrieve current mode from state.",
+		"summarization_failed": "Failed to summarize conversation context.",
 		"failed_delete_repo": "Failed to delete associated shadow repository or branch: {{error}}",
 		"failed_remove_directory": "Failed to remove task directory: {{error}}",
 		"custom_storage_path_unusable": "Custom storage path \"{{path}}\" is unusable, will use default path",
@@ -67,7 +68,9 @@
 		"mcp_server_not_found": "Server \"{{serverName}}\" not found in configuration",
 		"custom_storage_path_set": "Custom storage path set: {{path}}",
 		"default_storage_path": "Reverted to using default storage path",
-		"settings_imported": "Settings imported successfully."
+		"settings_imported": "Settings imported successfully.",
+		"summarizing_context": "Summarizing conversation context...",
+		"summarization_complete": "Conversation context summarization complete."
 	},
 	"answers": {
 		"yes": "Yes",

src/shared/WebviewMessage.ts

@@ -132,6 +132,7 @@ export interface WebviewMessage {
 		| "contextSummarizationTriggerThreshold"
 		| "contextSummarizationInitialStaticTurns"
 		| "contextSummarizationRecentTurns"
+		| "manualSummarize"
 	text?: string
 	disabled?: boolean
 	askResponse?: ClineAskResponse

src/shared/context-mentions.ts

@@ -50,11 +50,11 @@ Mention regex:
 
 */
 export const mentionRegex =
-	/@((?:\/|\w+:\/\/)[^\s]+?|[a-f0-9]{7,40}\b|problems\b|git-changes\b|terminal\b)(?=[.,;:!?]?(?=[\s\r\n]|$))/
+	/@((?:\/|\w+:\/\/)[^\s]+?|[a-f0-9]{7,40}\b|problems\b|git-changes\b|terminal\b|summarize\b)(?=[.,;:!?]?(?=[\s\r\n]|$))/
 export const mentionRegexGlobal = new RegExp(mentionRegex.source, "g")
 
 export interface MentionSuggestion {
-	type: "file" | "folder" | "git" | "problems"
+	type: "file" | "folder" | "git" | "problems" | "summarize"
 	label: string
 	description?: string
 	value: string

webview-ui/src/components/chat/ChatTextArea.tsx

@@ -252,6 +252,14 @@ const ChatTextArea = forwardRef<HTMLTextAreaElement, ChatTextAreaProps>(
 				setShowContextMenu(false)
 				setSelectedType(null)
 
+				if (type === ContextMenuOptionType.Summarize) {
+					// Handle summarize action - trigger manual summarization
+					vscode.postMessage({ type: "manualSummarize" })
+					setShowContextMenu(false)
+					setSelectedType(null)
+					return
+				}
+
 				if (textAreaRef.current) {
 					let insertValue = value || ""
 
@@ -267,6 +275,8 @@ const ChatTextArea = forwardRef<HTMLTextAreaElement, ChatTextAreaProps>(
 						insertValue = value || ""
 					}
 
+					// Note: The Summarize case is handled above before this block
+
 					const { newValue, mentionIndex } = insertMention(
 						textAreaRef.current.value,
 						cursorPosition,

webview-ui/src/components/chat/ContextMenu.tsx

@@ -89,6 +89,8 @@ const ContextMenu: React.FC<ContextMenuProps> = ({
 				return <span>Problems</span>
 			case ContextMenuOptionType.Terminal:
 				return <span>Terminal</span>
+			case ContextMenuOptionType.Summarize:
+				return <span>{option.label || "Summarize"}</span>
 			case ContextMenuOptionType.URL:
 				return <span>Paste URL to fetch contents</span>
 			case ContextMenuOptionType.NoResults:
@@ -175,6 +177,8 @@ const ContextMenu: React.FC<ContextMenuProps> = ({
 				return "link"
 			case ContextMenuOptionType.Git:
 				return "git-commit"
+			case ContextMenuOptionType.Summarize:
+				return "archive"
 			case ContextMenuOptionType.NoResults:
 				return "info"
 			default:

webview-ui/src/utils/context-mentions.ts

@@ -77,6 +77,7 @@ export enum ContextMenuOptionType {
 	Git = "git",
 	NoResults = "noResults",
 	Mode = "mode", // Add mode type
+	Summarize = "summarize", // Add summarize type
 }
 
 export interface ContextMenuQueryItem {
@@ -170,6 +171,7 @@ export function getContextMenuOptions(
 			{ type: ContextMenuOptionType.Folder },
 			{ type: ContextMenuOptionType.File },
 			{ type: ContextMenuOptionType.Git },
+			{ type: ContextMenuOptionType.Summarize, label: "Summarize", description: "Compress conversation history" },
 		]
 	}
 
@@ -193,6 +195,13 @@ export function getContextMenuOptions(
 	if ("terminal".startsWith(lowerQuery)) {
 		suggestions.push({ type: ContextMenuOptionType.Terminal })
 	}
+	if ("summarize".startsWith(lowerQuery)) {
+		suggestions.push({
+			type: ContextMenuOptionType.Summarize,
+			label: "Summarize",
+			description: "Compress conversation history",
+		})
+	}
 	if (query.startsWith("http")) {
 		suggestions.push({ type: ContextMenuOptionType.URL, value: query })
 	}