feat: ✨ add eraseAnnotations to clean text in chat

microsoft · Oct 11, 2024 · 860e082 · 860e082
1 parent 643740c
commit 860e082
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/packages/core/src/annotations.ts b/packages/core/src/annotations.ts
@@ -64,6 +64,14 @@ export function parseAnnotations(text: string): Diagnostic[] {
     return Array.from(annotations.values()) // Convert the set to an array
 }
 
+export function eraseAnnotations(text: string) {
+    return [
+        TYPESCRIPT_ANNOTATIONS_RX,
+        GITHUB_ANNOTATIONS_RX,
+        AZURE_DEVOPS_ANNOTATIONS_RX,
+    ].reduce((t, rx) => t.replace(rx, ""), text)
+}
+
 /**
  * Converts a `Diagnostic` to a GitHub Action command string.
  *

diff --git a/packages/vscode/src/chatparticipant.ts b/packages/vscode/src/chatparticipant.ts
@@ -3,6 +3,7 @@ import { ExtensionState } from "./state"
 import { TOOL_ID } from "../../core/src/constants"
 import { Fragment } from "../../core/src/generation"
 import { prettifyMarkdown } from "../../core/src/markdown"
+import { eraseAnnotations } from "../../core/src/annotations"
 
 export async function activateChatParticipant(state: ExtensionState) {
     const { context } = state
@@ -52,9 +53,12 @@ export async function activateChatParticipant(state: ExtensionState) {
 
             if (token.isCancellationRequested) return
 
-            const { text } = res || {}
+            const { text = "", annotations = [] } = res || {}
             response.markdown(
-                new vscode.MarkdownString(prettifyMarkdown(text || ""), true)
+                new vscode.MarkdownString(
+                    prettifyMarkdown(eraseAnnotations(text)),
+                    true
+                )
             )
         }
     )