fix: update mention suggestion filtering logic (#1763)

Issue: QuivrHQ/quivr#1762 Demo: https://github.com/StanGirard/quivr/assets/63923024/c578b9f0-5b1d-4196-85cf-042a93acd9c6
coolCatalyst · Nov 30, 2023 · 1ab1b34 · 1ab1b34
1 parent 71c54a6
commit 1ab1b34
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/...ar/components/ChatInput/components/ChatEditor/components/Editor/hooks/useMentionConfig.ts b/...ar/components/ChatInput/components/ChatEditor/components/Editor/hooks/useMentionConfig.ts
@@ -79,11 +79,13 @@ export const useMentionConfig = ({
             });
           },
           onUpdate: (props) => {
-            reactRenderer?.updateProps(props);
-
-            if (!props.clientRect) {
-              return;
-            }
+            reactRenderer?.updateProps({
+              ...props,
+              suggestionData: {
+                ...suggestionData,
+                items: props.items,
+              },
+            });
           },
           onKeyDown: (props) => {
             if (props.event.key === "Escape") {

diff --git a/...gueArea/components/ChatDialogue/components/QADisplay/components/MessageRow/MessageRow.tsx b/...gueArea/components/ChatDialogue/components/QADisplay/components/MessageRow/MessageRow.tsx
@@ -4,10 +4,9 @@ import { CopyButton } from "./components/CopyButton";
 import { MessageContent } from "./components/MessageContent";
 import { QuestionBrain } from "./components/QuestionBrain";
 import { QuestionPrompt } from "./components/QuestionPrompt";
-import { SourcesButton } from './components/SourcesButton'; // Import the new component
+import { SourcesButton } from "./components/SourcesButton";
 import { useMessageRow } from "./hooks/useMessageRow";
 
-
 type MessageRowProps = {
   speaker: "user" | "assistant";
   text?: string;
@@ -38,9 +37,11 @@ export const MessageRow = React.forwardRef(
 
     const sourcesIndex = messageContent.lastIndexOf("**Sources:**");
     const hasSources = sourcesIndex !== -1;
-    
+
     if (hasSources) {
-      sourcesContent = messageContent.substring(sourcesIndex + "**Sources:**".length).trim();
+      sourcesContent = messageContent
+        .substring(sourcesIndex + "**Sources:**".length)
+        .trim();
       messageContent = messageContent.substring(0, sourcesIndex).trim();
     }
 
@@ -49,7 +50,7 @@ export const MessageRow = React.forwardRef(
         <div ref={ref} className={containerClasses}>
           <div className="flex justify-between items-start w-full">
             {/* Left section for the question and prompt */}
-            <div className="flex">
+            <div className="flex gap-1">
               <QuestionBrain brainName={brainName} />
               <QuestionPrompt promptName={promptName} />
             </div>
@@ -74,5 +75,5 @@ export const MessageRow = React.forwardRef(
     );
   }
 );
-    
+
 MessageRow.displayName = "MessageRow";