fix: make sure to unset warnings for reused reference content parts (m…

…icrosoft#223539)
mjbvz · Jul 24, 2024 · 06643dd · 06643dd
1 parent 5fe066f
commit 06643dd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/vs/workbench/contrib/chat/browser/chatContentParts/chatReferencesContentPart.ts b/src/vs/workbench/contrib/chat/browser/chatContentParts/chatReferencesContentPart.ts
@@ -312,11 +312,13 @@ class CollapsibleListRenderer implements IListRenderer<IChatCollapsibleListItem,
 			}
 		}
 
-		if (data.options?.status?.kind === ChatResponseReferencePartStatusKind.Omitted || data.options?.status?.kind === ChatResponseReferencePartStatusKind.Partial) {
-			for (const selector of ['.monaco-icon-suffix-container', '.monaco-icon-name-container']) {
-				const element = templateData.label.element.querySelector(selector);
-				if (element) {
+		for (const selector of ['.monaco-icon-suffix-container', '.monaco-icon-name-container']) {
+			const element = templateData.label.element.querySelector(selector);
+			if (element) {
+				if (data.options?.status?.kind === ChatResponseReferencePartStatusKind.Omitted || data.options?.status?.kind === ChatResponseReferencePartStatusKind.Partial) {
 					element.classList.add('warning');
+				} else {
+					element.classList.remove('warning');
 				}
 			}
 		}