revert to previous doc select logic

danswer-ai · Nov 22, 2024 · 2c7c9fb · 2c7c9fb
1 parent 5dc07d4
commit 2c7c9fb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/web/src/app/chat/ChatPage.tsx b/web/src/app/chat/ChatPage.tsx
@@ -2198,6 +2198,10 @@ export function ChatPage({
                                         const newCompleteMessageMap = new Map(
                                           messageMap
                                         );
+                                        console.log(
+                                          "newCompleteMessageMap",
+                                          newCompleteMessageMap
+                                        );
                                         newCompleteMessageMap.get(
                                           message.parentMessageId!
                                         )!.latestChildMessageId = messageId;
@@ -2461,7 +2465,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 +2477,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
@@ -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
@@ -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"