[Condense] Add a button to condense the task context

Author: canrobins13

src/core/task/Task.ts

@@ -76,7 +76,7 @@ import {
 } from "../checkpoints"
 import { processUserContentMentions } from "../mentions/processUserContentMentions"
 import { ApiMessage } from "../task-persistence/apiMessages"
-import { getMessagesSinceLastSummary } from "../condense"
+import { getMessagesSinceLastSummary, summarizeConversation } from "../condense"
 import { maybeRemoveImageBlocks } from "../../api/transform/image-cleaning"
 
 export type ClineEvents = {
@@ -480,6 +480,13 @@ export class Task extends EventEmitter<ClineEvents> {
 		}
 	}
 
+	public async condenseContext(): Promise<void> {
+		const { messages } = await summarizeConversation(this.apiConversationHistory, this.api)
+		if (messages !== this.apiConversationHistory) {
+			this.overwriteApiConversationHistory(messages)
+		}
+	}
+
 	async say(
 		type: ClineSay,
 		text?: string,

src/core/webview/ClineProvider.ts

@@ -1109,6 +1109,22 @@ export class ClineProvider extends EventEmitter<ClineProviderEvents> implements
 		await downloadTask(historyItem.ts, apiConversationHistory)
 	}
 
+	/* Condenses a task's message history to use fewer tokens. */
+	async condenseTaskContext(taskId: string) {
+		let task = undefined
+		for (let i = this.clineStack.length - 1; i >= 0; i--) {
+			if (this.clineStack[i].taskId === taskId) {
+				task = this.clineStack[i]
+				break
+			}
+		}
+		if (!task) {
+			const { historyItem } = await this.getTaskWithId(taskId)
+			task = await this.initClineWithHistoryItem(historyItem)
+		}
+		await task.condenseContext()
+	}
+
 	// this function deletes a task from task hidtory, and deletes it's checkpoints and delete the task folder
 	async deleteTaskWithId(id: string) {
 		try {

src/core/webview/webviewMessageHandler.ts

@@ -195,6 +195,9 @@ export const webviewMessageHandler = async (provider: ClineProvider, message: We
 		case "showTaskWithId":
 			provider.showTaskWithId(message.text!)
 			break
+		case "condenseTaskContext":
+			provider.condenseTaskContext(message.text!)
+			break
 		case "deleteTaskWithId":
 			provider.deleteTaskWithId(message.text!)
 			break

src/shared/WebviewMessage.ts

@@ -130,6 +130,7 @@ export interface WebviewMessage {
 		| "searchFiles"
 		| "toggleApiConfigPin"
 		| "setHistoryPreviewCollapsed"
+		| "condenseTaskContext"
 	text?: string
 	disabled?: boolean
 	askResponse?: ClineAskResponse

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

@@ -24,6 +24,13 @@ export const TaskActions = ({ item }: { item: HistoryItem | undefined }) => {
 			</Button>
 			{!!item?.size && item.size > 0 && (
 				<>
+					<Button
+						variant="ghost"
+						size="sm"
+						title={t("chat:task.condenseContext")}
+						onClick={() => vscode.postMessage({ type: "condenseTaskContext", text: item?.id })}>
+						<span className="codicon codicon-file-zip" />
+					</Button>
 					<Button
 						variant="ghost"
 						size="sm"

webview-ui/src/i18n/locales/en/chat.json

@@ -7,6 +7,7 @@
 		"tokens": "Tokens:",
 		"cache": "Cache:",
 		"apiCost": "API Cost:",
+		"condenseContext": "Condense task context",
 		"contextWindow": "Context Length:",
 		"closeAndStart": "Close task and start a new one",
 		"export": "Export task history",