wip

Author: canrobins13

src/core/task/Task.ts

@@ -481,10 +481,35 @@ 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)
+		const {
+			messages,
+			summary,
+			cost,
+			newContextTokens = 0,
+		} = await summarizeConversation(this.apiConversationHistory, this.api)
+		if (!summary) {
+			return
 		}
+		const lastMessageContent = this.apiConversationHistory.at(-1)?.content
+		await this.overwriteApiConversationHistory(messages)
+		const { contextTokens } = this.getTokenUsage()
+		const lastContent =
+			typeof lastMessageContent === "string"
+				? [{ type: "text" as const, text: lastMessageContent }]
+				: lastMessageContent
+		const lastMessageTokens = lastContent ? await this.api.countTokens(lastContent) : 0
+		const prevContextTokens = contextTokens + lastMessageTokens
+		const contextCondense: ContextCondense = { summary, cost, newContextTokens, prevContextTokens }
+		await this.say(
+			"condense_context",
+			undefined /* text */,
+			undefined /* images */,
+			false /* partial */,
+			undefined /* checkpoint */,
+			undefined /* progressStatus */,
+			undefined /* options */,
+			contextCondense,
+		)
 	}
 
 	async say(

src/core/webview/ClineProvider.ts

@@ -1123,6 +1123,7 @@ export class ClineProvider extends EventEmitter<ClineProviderEvents> implements
 			task = await this.initClineWithHistoryItem(historyItem)
 		}
 		await task.condenseContext()
+		await this.postMessageToWebview({ type: "condenseTaskContextResponse" })
 	}
 
 	// this function deletes a task from task hidtory, and deletes it's checkpoints and delete the task folder

src/core/webview/webviewMessageHandler.ts

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

src/shared/ExtensionMessage.ts

@@ -69,6 +69,7 @@ export interface ExtensionMessage {
 		| "setHistoryPreviewCollapsed"
 		| "commandExecutionStatus"
 		| "vsCodeSetting"
+		| "condenseTaskContextResponse"
 	text?: string
 	action?:
 		| "chatButtonClicked"

src/shared/WebviewMessage.ts

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

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

@@ -131,6 +131,7 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 	const lastTtsRef = useRef<string>("")
 	const [wasStreaming, setWasStreaming] = useState<boolean>(false)
 	const [showCheckpointWarning, setShowCheckpointWarning] = useState<boolean>(false)
+	const [isCondensing, setIsCondensing] = useState<boolean>(false)
 
 	// UI layout depends on the last 2 messages
 	// (since it relies on the content of these messages, we are deep comparing. i.e. the button state after hitting button sets enableButtons to false, and this effect otherwise would have to true again even if messages didn't change
@@ -581,12 +582,20 @@ const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewPro
 							handleSecondaryButtonClick(message.text ?? "", message.images ?? [])
 							break
 					}
+					break
+				case "condenseTaskContextResponse":
+					if (isCondensing) {
+						setSendingDisabled(false)
+					}
+					setIsCondensing(false)
+					break
 			}
 			// textAreaRef.current is not explicitly required here since React
 			// guarantees that ref will be stable across re-renders, and we're
 			// not using its value but its reference.
 		},
 		[
+			isCondensing,
 			isHidden,
 			sendingDisabled,
 			enableButtons,

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

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