Support llmGenerated property on CodeAction

client-node-tests/src/converter.test.ts

@@ -1547,6 +1547,26 @@ suite('Code Converter', () => {
 		strictEqual(result.triggerKind, proto.CodeActionTriggerKind.Automatic);
 	});
 
+	test('CodeAction - LLMGenerated tag c2p', () => {
+		const item: vscode.CodeAction = {
+			title: 'title',
+			isAI: true
+		};
+
+		const result = c2p.asCodeActionSync(item);
+		strictEqual(result.tags![0], proto.CodeActionTag.LLMGenerated);
+	});
+
+	test('CodeAction - LLMGenerated tag p2c', async () => {
+		const item: proto.CodeAction = {
+			title: 'title',
+			tags: [proto.CodeActionTag.LLMGenerated]
+		};
+
+		const result = await p2c.asCodeAction(item);
+		strictEqual(result.isAI, true);
+	});
+
 	test('Uri Rewrite', () => {
 		const converter = codeConverter.createConverter((value: vscode.Uri) => {
 			return `${value.toString()}.vscode`;

client/package.json

@@ -15,7 +15,7 @@
 	"bugs": {
 		"url": "https://github.com/Microsoft/vscode-languageserver-node/issues"
 	},
-	"enabledApiProposals": [],
+	"enabledApiProposals": ["codeActionAI"],
 	"exports": {
 		".": {
 			"types": "./lib/common/api.d.ts",

client/src/common/client.ts

@@ -2,6 +2,8 @@
  * Copyright (c) Microsoft Corporation. All rights reserved.
  * Licensed under the MIT License. See License.txt in the project root for license information.
  * ------------------------------------------------------------------------------------------ */
+/// <reference path="../../typings/vscode.proposed.codeActionAI.d.ts" />
+
 import {
 	workspace as Workspace, window as Window, languages as Languages, version as VSCodeVersion, TextDocument, Disposable, OutputChannel,
 	FileSystemWatcher as VFileSystemWatcher, DiagnosticCollection, Diagnostic as VDiagnostic, Uri, CancellationToken, WorkspaceEdit as VWorkspaceEdit,

client/src/common/codeConverter.ts

@@ -742,6 +742,10 @@ export function createConverter(uriConverter?: URIConverter): Converter {
 		if (item.command !== undefined) { result.command = asCommand(item.command); }
 		if (item.isPreferred !== undefined) { result.isPreferred = item.isPreferred; }
 		if (item.disabled !== undefined) { result.disabled = { reason: item.disabled.reason }; }
+		if (item.isAI) {
+			result.tags ??= [];
+			result.tags.push(proto.CodeActionTag.LLMGenerated);
+		}
 		return result;
 	}
 
@@ -756,6 +760,10 @@ export function createConverter(uriConverter?: URIConverter): Converter {
 		if (item.command !== undefined) { result.command = asCommand(item.command); }
 		if (item.isPreferred !== undefined) { result.isPreferred = item.isPreferred; }
 		if (item.disabled !== undefined) { result.disabled = { reason: item.disabled.reason }; }
+		if (item.isAI) {
+			result.tags ??= [];
+			result.tags.push(proto.CodeActionTag.LLMGenerated);
+		}
 		return result;
 	}
 

client/src/common/protocolConverter.ts

@@ -1020,6 +1020,10 @@ export function createConverter(
 		if (item.command !== undefined) { result.command = asCommand(item.command); }
 		if (item.isPreferred !== undefined) { result.isPreferred = item.isPreferred; }
 		if (item.disabled !== undefined) { result.disabled = { reason: item.disabled.reason }; }
+		if (item.tags?.includes(ls.CodeActionTag.LLMGenerated)) {
+			result.isAI = true;
+		}
+
 		return result;
 	}
 

client/typings/vscode.proposed.codeActionAI.d.ts

@@ -0,0 +1,16 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+declare module 'vscode' {
+
+	export interface CodeAction {
+		/**
+		 * Marks this as an AI action.
+		 *
+		 * Ex: A quick fix should be marked AI if it invokes AI.
+		 */
+		isAI?: boolean;
+	}
+}

protocol/metaModel.json

@@ -5812,6 +5812,20 @@
 					"optional": true,
 					"documentation": "A data entry field that is preserved on a code action between\na `textDocument/codeAction` and a `codeAction/resolve` request.\n\n@since 3.16.0",
 					"since": "3.16.0"
+				},
+				{
+					"name": "tags",
+					"type": {
+						"kind": "array",
+						"element": {
+							"kind": "reference",
+							"name": "CodeActionTag"
+						}
+					},
+					"optional": true,
+					"documentation": "Tags for this code action.\n\n@since 3.18.0 - proposed",
+					"since": "3.18.0 - proposed",
+					"proposed": true
 				}
 			],
 			"documentation": "A code action represents a change that can be performed in code, e.g. to fix a problem or\nto refactor code.\n\nA CodeAction must set either `edit` and/or a `command`. If both are supplied, the `edit` is applied first, then the `command` is executed."
@@ -14772,6 +14786,23 @@
 			"supportsCustomValues": true,
 			"documentation": "A set of predefined code action kinds"
 		},
+		{
+			"name": "CodeActionTag",
+			"type": {
+				"kind": "base",
+				"name": "uinteger"
+			},
+			"values": [
+				{
+					"name": "LLMGenerated",
+					"value": 1,
+					"documentation": "Render a code action as LLM-generated, usually using a sparkle icon."
+				}
+			],
+			"documentation": "Code action tags are extra annotations that tweak the rendering of a code action.\n\n@since 3.18.0",
+			"since": "3.18.0",
+			"proposed": true
+		},
 		{
 			"name": "TraceValue",
 			"type": {

types/src/main.ts

@@ -3372,6 +3372,26 @@ export type CodeActionDisabled = {
 	reason: string;
 };
 
+/**
+ * Code action tags are extra annotations that tweak the rendering of a code action.
+ *
+ * @since 3.18.0 - proposed
+ */
+export namespace CodeActionTag {
+	/**
+	 * Render a code action as LLM-generated, usually using a sparkle icon.
+	 */
+	export const LLMGenerated = 1;
+
+	/**
+	 * Checks whether the given literal conforms to the {@link CodeActionTag} interface.
+	 */
+	export function is(value: any): value is CodeActionTag {
+		return Is.defined(value) && value === CodeActionTag.LLMGenerated;
+	}
+}
+export type CodeActionTag = 1;
+
 /**
  * A code action represents a change that can be performed in code, e.g. to fix a problem or
  * to refactor code.
@@ -3446,6 +3466,13 @@ export interface CodeAction {
 	 * @since 3.16.0
 	 */
 	data?: LSPAny;
+
+	/**
+ 	 * Tags for this code action.
+	 *
+	 * @since 3.18.0 - proposed
+	 */
+	tags?: CodeActionTag[];
 }
 
 export namespace CodeAction {
@@ -3499,7 +3526,8 @@ export namespace CodeAction {
 			(candidate.edit !== undefined || candidate.command !== undefined) &&
 			(candidate.command === undefined || Command.is(candidate.command)) &&
 			(candidate.isPreferred === undefined || Is.boolean(candidate.isPreferred)) &&
-			(candidate.edit === undefined || WorkspaceEdit.is(candidate.edit));
+			(candidate.edit === undefined || WorkspaceEdit.is(candidate.edit)) &&
+			(candidate.tags === undefined || Is.typedArray(candidate.tags, CodeActionTag.is));
 	}
 }