Handle partial messages in the codebase search tool

Author: mrubens

src/core/tools/codebaseSearchTool.ts

@@ -29,28 +29,32 @@ export async function codebaseSearchTool(
 	let query: string | undefined = block.params.query
 	let directoryPrefix: string | undefined = block.params.path
 
-	if (!query) {
-		cline.consecutiveMistakeCount++
-		pushToolResult(await cline.sayAndCreateMissingParamError(toolName, "query"))
-		return
-	}
 	query = removeClosingTag("query", query)
 
 	if (directoryPrefix) {
 		directoryPrefix = removeClosingTag("path", directoryPrefix)
 		directoryPrefix = path.normalize(directoryPrefix)
 	}
 
-	// Extract optional sendResultsToUI parameter
-
-	const approvalPayload = {
+	const sharedMessageProps = {
 		tool: "codebaseSearch",
 		query: query,
 		path: directoryPrefix,
 		isOutsideWorkspace: false,
 	}
 
-	const didApprove = await askApproval("tool", JSON.stringify(approvalPayload))
+	if (block.partial) {
+		await cline.ask("tool", JSON.stringify(sharedMessageProps), block.partial).catch(() => {})
+		return
+	}
+
+	if (!query) {
+		cline.consecutiveMistakeCount++
+		pushToolResult(await cline.sayAndCreateMissingParamError(toolName, "query"))
+		return
+	}
+
+	const didApprove = await askApproval("tool", JSON.stringify(sharedMessageProps))
 	if (!didApprove) {
 		pushToolResult(formatResponse.toolDenied())
 		return