Improve manual summarization UX with visual feedback

Author: shariqriazz

src/core/Cline.ts

@@ -370,7 +370,7 @@ export class Cline extends EventEmitter<ClineEvents> {
 		this.emit("message", { action: "updated", message: partialMessage })
 	}
 
-	private async saveClineMessages() {
+	public async saveClineMessages() {
 		try {
 			await saveTaskMessages({
 				messages: this.clineMessages,

src/core/webview/webviewMessageHandler.ts

@@ -882,18 +882,61 @@ export const webviewMessageHandler = async (provider: ClineProvider, message: We
 			const currentCline = provider.getCurrentCline()
 			if (currentCline) {
 				try {
+					// First send a message to the webview to show a progress indicator
+					provider.postMessageToWebview({
+						type: "summarizationStatus",
+						status: "started",
+						text: t("common:info.summarizing_context"),
+					})
+
 					// Notify user that summarization is in progress
 					vscode.window.showInformationMessage(t("common:info.summarizing_context"))
 
+					// Add a "summarizing" message to the chat
+					await currentCline.say("summarizing", t("common:info.summarizing_context"), undefined, true)
+
 					// Trigger the summarization process
 					await currentCline.summarizeConversationContext(true) // true indicates manual trigger
 
+					// Update the "summarizing" message to show completion
+					const lastMessage = currentCline.clineMessages.at(-1)
+					if (lastMessage && lastMessage.say === "summarizing" && lastMessage.partial) {
+						lastMessage.text = t("common:info.summarization_complete")
+						lastMessage.partial = false
+						await currentCline.saveClineMessages()
+						await provider.postStateToWebview()
+					}
+
+					// Send a message to the webview to hide the progress indicator
+					provider.postMessageToWebview({
+						type: "summarizationStatus",
+						status: "completed",
+						text: t("common:info.summarization_complete"),
+					})
+
 					// 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)}`,
 					)
+
+					// Update the UI to show the error
+					provider.postMessageToWebview({
+						type: "summarizationStatus",
+						status: "failed",
+						text: t("common:errors.summarization_failed"),
+					})
+
+					// Update any partial message
+					const lastMessage = currentCline.clineMessages.at(-1)
+					if (lastMessage && lastMessage.say === "summarizing" && lastMessage.partial) {
+						lastMessage.text = t("common:errors.summarization_failed")
+						lastMessage.partial = false
+						await currentCline.saveClineMessages()
+						await provider.postStateToWebview()
+					}
+
 					vscode.window.showErrorMessage(t("common:errors.summarization_failed"))
 				}
 			}

src/exports/roo-code.d.ts

@@ -342,6 +342,7 @@ type ClineMessage = {
 				| "checkpoint_saved"
 				| "rooignore_error"
 				| "diff_error"
+				| "summarizing"
 		  )
 		| undefined
 	text?: string | undefined
@@ -423,6 +424,7 @@ type RooCodeEvents = {
 							| "checkpoint_saved"
 							| "rooignore_error"
 							| "diff_error"
+							| "summarizing"
 					  )
 					| undefined
 				text?: string | undefined

src/exports/types.ts

@@ -347,6 +347,7 @@ type ClineMessage = {
 				| "checkpoint_saved"
 				| "rooignore_error"
 				| "diff_error"
+				| "summarizing"
 		  )
 		| undefined
 	text?: string | undefined
@@ -432,6 +433,7 @@ type RooCodeEvents = {
 							| "checkpoint_saved"
 							| "rooignore_error"
 							| "diff_error"
+							| "summarizing"
 					  )
 					| undefined
 				text?: string | undefined

src/schemas/index.ts

@@ -789,6 +789,7 @@ export const clineSays = [
 	"checkpoint_saved",
 	"rooignore_error",
 	"diff_error",
+	"summarizing",
 ] as const
 
 export const clineSaySchema = z.enum(clineSays)

src/shared/ExtensionMessage.ts

@@ -67,6 +67,7 @@ export interface ExtensionMessage {
 		| "toggleApiConfigPin"
 		| "acceptInput"
 		| "setHistoryPreviewCollapsed"
+		| "summarizationStatus"
 	text?: string
 	action?:
 		| "chatButtonClicked"
@@ -103,6 +104,7 @@ export interface ExtensionMessage {
 	promptText?: string
 	results?: { path: string; type: "file" | "folder"; label?: string }[]
 	error?: string
+	status?: "started" | "completed" | "failed"
 }
 
 export type ExtensionState = Pick<

src/shared/WebviewMessage.ts

@@ -133,6 +133,7 @@ export interface WebviewMessage {
 		| "contextSummarizationInitialStaticTurns"
 		| "contextSummarizationRecentTurns"
 		| "manualSummarize"
+		| "summarizationStatus"
 	text?: string
 	disabled?: boolean
 	askResponse?: ClineAskResponse
@@ -161,6 +162,7 @@ export interface WebviewMessage {
 	hasSystemPromptOverride?: boolean
 	terminalOperation?: "continue" | "abort"
 	historyPreviewCollapsed?: boolean
+	status?: "started" | "completed" | "failed"
 }
 
 export const checkoutDiffPayloadSchema = z.object({