From 47a55c20a05e04f165ef8d8022cc220f31b96a31 Mon Sep 17 00:00:00 2001
From: Debanjum <debanjum@gmail.com>
Date: Tue, 18 Nov 2025 14:51:47 -0800
Subject: [PATCH] Associate folder icon with all doc tools use in thinking UX
 on web app

The newer grep_files and list_files should also be associated with
document search in train of thought visualization on the web app.
---
 .../web/app/components/chatMessage/chatMessage.tsx          | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/interface/web/app/components/chatMessage/chatMessage.tsx b/src/interface/web/app/components/chatMessage/chatMessage.tsx
index cf4dc240..4251ca5a 100644
--- a/src/interface/web/app/components/chatMessage/chatMessage.tsx
+++ b/src/interface/web/app/components/chatMessage/chatMessage.tsx
@@ -312,7 +312,11 @@ function chooseIconFromHeader(header: string, iconColor: string) {
         return <Toolbox className={`${classNames}`} />;
     }
 
-    if (compareHeader.includes("notes")) {
+    if (
+        compareHeader.includes("notes") ||
+        compareHeader.includes("documents") ||
+        compareHeader.includes("files")
+    ) {
         return <Folder className={`${classNames}`} />;
     }