Mamadou DICKO 2023-11-30 10:31:48 +01:00 committed by GitHub
parent a84219f193
commit 00f11f93f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 11 deletions

View File

@ -79,11 +79,13 @@ export const useMentionConfig = ({
}); });
}, },
onUpdate: (props) => { onUpdate: (props) => {
reactRenderer?.updateProps(props); reactRenderer?.updateProps({
...props,
if (!props.clientRect) { suggestionData: {
return; ...suggestionData,
} items: props.items,
},
});
}, },
onKeyDown: (props) => { onKeyDown: (props) => {
if (props.event.key === "Escape") { if (props.event.key === "Escape") {

View File

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