From 9dc23bf3e7103b260df9f4b16c386b40bf7293ec Mon Sep 17 00:00:00 2001
From: pablodanswer <pablo@danswer.ai>
Date: Fri, 22 Nov 2024 15:26:53 -0800
Subject: [PATCH] revert to previous doc select logic (#3217)

* revert to previous doc select logic

* k
---
 web/src/app/chat/ChatPage.tsx                        | 2 --
 web/src/app/chat/documentSidebar/DocumentSidebar.tsx | 3 ++-
 web/src/app/chat/message/Messages.tsx                | 5 +++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/web/src/app/chat/ChatPage.tsx b/web/src/app/chat/ChatPage.tsx
index bd7c39646fd..0bb3ebfa965 100644
--- a/web/src/app/chat/ChatPage.tsx
+++ b/web/src/app/chat/ChatPage.tsx
@@ -2461,7 +2461,6 @@ export function ChatPage({
                               textAreaRef={textAreaRef}
                               chatSessionId={chatSessionIdRef.current!}
                             />
-
                             {enterpriseSettings &&
                               enterpriseSettings.custom_lower_disclaimer_content && (
                                 <div className="mobile:hidden mt-4 flex items-center justify-center relative w-[95%] mx-auto">
@@ -2474,7 +2473,6 @@ export function ChatPage({
                                   </div>
                                 </div>
                               )}
-
                             {enterpriseSettings &&
                               enterpriseSettings.use_custom_logotype && (
                                 <div className="hidden lg:block absolute right-0 bottom-0">
diff --git a/web/src/app/chat/documentSidebar/DocumentSidebar.tsx b/web/src/app/chat/documentSidebar/DocumentSidebar.tsx
index 5a6ed956fe6..021c2398157 100644
--- a/web/src/app/chat/documentSidebar/DocumentSidebar.tsx
+++ b/web/src/app/chat/documentSidebar/DocumentSidebar.tsx
@@ -75,7 +75,8 @@ export const DocumentSidebar = forwardRef<HTMLDivElement, DocumentSidebarProps>(
           <div className="pb-6 flex-initial overflow-y-hidden flex flex-col h-screen">
             {popup}
             <div className="pl-3 mx-2 pr-6 mt-3 flex text-text-800 flex-col text-2xl text-emphasis flex font-semibold">
-              {dedupedDocuments.length} Documents
+              {dedupedDocuments.length} Document
+              {dedupedDocuments.length > 1 ? "s" : ""}
               <p className="text-sm font-semibold flex flex-wrap gap-x-2 text-text-600 mt-1">
                 Select to add to continuous context
                 <a
diff --git a/web/src/app/chat/message/Messages.tsx b/web/src/app/chat/message/Messages.tsx
index c420a3f53e9..cc4f9c9cac8 100644
--- a/web/src/app/chat/message/Messages.tsx
+++ b/web/src/app/chat/message/Messages.tsx
@@ -482,9 +482,10 @@ export const AIMessage = ({
                               ))}
                             <div
                               onClick={() => {
-                                if (toggleDocumentSelection) {
-                                  toggleDocumentSelection();
+                                if (messageId) {
+                                  onMessageSelection?.(messageId);
                                 }
+                                toggleDocumentSelection?.();
                               }}
                               key={-1}
                               className="cursor-pointer w-[200px] rounded-lg flex-none transition-all duration-500 hover:bg-background-125 bg-text-100 px-4 py-2 border-b"