Handle LanguageModelPromptTsxPart or complex nested structures

Author: complyue

src/core/tools/vsclmt.ts

@@ -21,7 +21,7 @@ export async function useVSCLMT(
 			arguments: removeClosingTag("arguments", toolArgs),
 		})
 
-		await cline.ask("tool", partialMessage, toolUse.partial).catch(() => { })
+		await cline.ask("tool", partialMessage, toolUse.partial).catch(() => {})
 		return
 	}
 
@@ -85,17 +85,59 @@ export async function useVSCLMT(
 		const result = await vsclmtService.invokeTool(tool_name, parsedArgs)
 
 		// Format the result for display
+		const extractText = (obj: unknown): string => {
+			if (typeof obj === "string") return obj
+			if (!obj || typeof obj !== "object") return ""
+			const asRecord = obj as Record<string, unknown>
+			if ("text" in asRecord && typeof asRecord.text === "string") return asRecord.text
+			if ("value" in asRecord && typeof asRecord.value === "string") return asRecord.value
+			// Special handling for PromptTsx node structure
+			if ("node" in asRecord && asRecord.node && typeof asRecord.node === "object") {
+				const node = asRecord.node as Record<string, unknown>
+				if ("children" in node && Array.isArray(node.children)) {
+					return node.children.map((child: unknown) => extractText(child)).join("")
+				}
+			}
+			if (Array.isArray(obj)) return obj.map((item) => extractText(item)).join("")
+			return Object.values(asRecord)
+				.map((v) => extractText(v))
+				.join("")
+		}
+
 		const resultText = result.content
-			.map(part =>
-				typeof part === 'object' && part !== null && 'value' in part && typeof part.value === 'string' ?
-					(part as { value: string }).value :
-					String(part)
-			)
-			.join('\n\n')
+			.map((part): string => {
+				if (typeof part === "object" && part !== null) {
+					// Handle MarkdownString specifically
+					if (part instanceof vscode.MarkdownString) {
+						return part.value
+					}
+					// Handle objects with value property
+					if ("value" in part) {
+						const value = part.value
+						if (value instanceof vscode.MarkdownString) {
+							return value.value
+						}
+						if (typeof value === "string") {
+							return value
+						}
+						// Handle LanguageModelPromptTsxPart or complex nested structures
+						if (value && typeof value === "object") {
+							return extractText(value)
+						}
+					}
+					return JSON.stringify(part, null, 2)
+				}
+				return String(part)
+			})
+			.join("\n\n")
 
 		pushToolResult(resultText)
 	} catch (error) {
 		await handleError("VS Code LM tool invocation", error)
-		pushToolResult(formatResponse.toolError(`Failed to invoke VS Code LM tool '${tool_name}': ${error.stack || error.message}`))
+		pushToolResult(
+			formatResponse.toolError(
+				`Failed to invoke VS Code LM tool '${tool_name}': ${error.stack || error.message}`,
+			),
+		)
 	}
 }