Inovcation message

Author: lramos15

src/vs/workbench/api/common/extHostChatAgents2.ts

@@ -296,11 +296,11 @@ export class ChatAgentResponseStream {
 					_report(dto);
 					return this;
 				},
-				prepareToolInvocation(toolName, streamData) {
+				prepareToolInvocation(toolName, invocationMessage) {
 					throwIfDone(this.prepareToolInvocation);
 					checkProposedApiEnabled(that._extension, 'chatParticipantAdditions');
 
-					const part = new extHostTypes.ChatPrepareToolInvocationPart(toolName, streamData);
+					const part = new extHostTypes.ChatPrepareToolInvocationPart(toolName, invocationMessage);
 					const dto = typeConvert.ChatPrepareToolInvocationPart.from(part);
 					_report(dto);
 					return this;

src/vs/workbench/api/common/extHostTypeConverters.ts

@@ -2816,14 +2816,17 @@ export namespace ChatPrepareToolInvocationPart {
 		return {
 			kind: 'prepareToolInvocation',
 			toolName: part.toolName,
-			streamData: part.streamData ? {
-				partialInput: part.streamData.partialInput
-			} : undefined
+			invocationMessage: part.invocationMessage && typeof part.invocationMessage !== 'string'
+				? MarkdownString.fromStrict(part.invocationMessage)
+				: part.invocationMessage
 		};
 	}
 
 	export function to(part: IChatPrepareToolInvocationPart): vscode.ChatPrepareToolInvocationPart {
-		return new types.ChatPrepareToolInvocationPart(part.toolName, part.streamData);
+		const invocationMessage = part.invocationMessage && typeof part.invocationMessage !== 'string'
+			? MarkdownString.to(part.invocationMessage)
+			: part.invocationMessage;
+		return new types.ChatPrepareToolInvocationPart(part.toolName, invocationMessage);
 	}
 }
 

src/vs/workbench/api/common/extHostTypes.ts

@@ -3339,16 +3339,14 @@ export class ChatResponseNotebookEditPart implements vscode.ChatResponseNotebook
 
 export class ChatPrepareToolInvocationPart {
 	toolName: string;
-	streamData?: {
-		readonly partialInput?: unknown;
-	};
+	invocationMessage?: string | vscode.MarkdownString;
 	/**
 	 * @param toolName The name of the tool being prepared for invocation.
-	 * @param streamData Optional streaming data with partial arguments.
+	 * @param invocationMessage Optional message to display while preparing the invocation.
 	 */
-	constructor(toolName: string, streamData?: { readonly partialInput?: unknown }) {
+	constructor(toolName: string, invocationMessage?: string | vscode.MarkdownString) {
 		this.toolName = toolName;
-		this.streamData = streamData;
+		this.invocationMessage = invocationMessage;
 	}
 }
 

src/vs/workbench/contrib/chat/browser/chatContentParts/toolInvocationParts/chatPrepareToolInvocationPart.ts

@@ -24,17 +24,15 @@ export class ChatPrepareToolInvocationPart extends ChatProgressContentPart imple
 		@IConfigurationService configurationService: IConfigurationService
 	) {
 		let messageContent: MarkdownString;
-		if (prepareToolPart.streamData?.partialInput !== undefined) {
-			// Show partial input if available
-			const partialInputStr = typeof prepareToolPart.streamData.partialInput === 'string'
-				? prepareToolPart.streamData.partialInput
-				: JSON.stringify(prepareToolPart.streamData.partialInput, null, 2);
-			messageContent = new MarkdownString().appendText(`Preparing to call ${prepareToolPart.toolName}...`);
-			if (partialInputStr && partialInputStr.length > 0 && partialInputStr !== '{}') {
-				messageContent.appendMarkdown(`\n\n\`\`\`json\n${partialInputStr}\n\`\`\``);
+		if (prepareToolPart.invocationMessage) {
+			// Use custom invocation message if provided
+			if (typeof prepareToolPart.invocationMessage === 'string') {
+				messageContent = new MarkdownString().appendText(prepareToolPart.invocationMessage);
+			} else {
+				messageContent = new MarkdownString(prepareToolPart.invocationMessage.value, prepareToolPart.invocationMessage);
 			}
 		} else {
-			// No streaming data yet, just show that we're preparing
+			// Default message if no custom message provided
 			messageContent = new MarkdownString().appendText(`Preparing to call ${prepareToolPart.toolName}...`);
 		}
 

src/vs/workbench/contrib/chat/common/chatService.ts

@@ -665,9 +665,7 @@ export class ChatMcpServersStarting implements IChatMcpServersStarting {
 export interface IChatPrepareToolInvocationPart {
 	readonly kind: 'prepareToolInvocation';
 	readonly toolName: string;
-	readonly streamData?: {
-		readonly partialInput?: unknown;
-	};
+	readonly invocationMessage?: string | IMarkdownString;
 }
 
 export type IChatProgress =

src/vscode-dts/vscode.proposed.chatParticipantAdditions.d.ts

@@ -81,11 +81,8 @@ declare module 'vscode' {
 
 	export class ChatPrepareToolInvocationPart {
 		toolName: string;
-		/**
-		 * Partial arguments that have streamed in for the tool invocation.
-		 */
-		streamData?: ChatToolInvocationStreamData;
-		constructor(toolName: string, streamData?: ChatToolInvocationStreamData);
+		invocationMessage?: string | MarkdownString;
+		constructor(toolName: string, invocationMessage?: string | MarkdownString);
 	}
 
 	export interface ChatToolInvocationStreamData {
@@ -360,10 +357,10 @@ declare module 'vscode' {
 		codeCitation(value: Uri, license: string, snippet: string): void;
 
 		/**
-		 * Notifies the UI that a tool invocation is being prepared. Optional streaming data can be
-		 * provided to render partial arguments while the invocation input is still being generated.
+		 * Notifies the UI that a tool invocation is being prepared. Optional invocation message can be
+		 * provided to render a custom message while the invocation input is still being generated.
 		 */
-		prepareToolInvocation(toolName: string, streamData?: ChatToolInvocationStreamData): void;
+		prepareToolInvocation(toolName: string, invocationMessage?: string | MarkdownString): void;
 
 		push(part: ExtendedChatResponsePart): void;