KeyArgo
/
llama-gpt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

User message should not render as Markdown

This commit is contained in:
Xiangxuan Liu 2023-03-19 19:37:38 +08:00
parent 240185c11a
commit 0f346a808e
1 changed files with 29 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
components/Chat/ChatMessage.tsx
View File

@ -19,32 +19,35 @@ export const ChatMessage: FC<Props> = ({ message, lightMode }) => {
<div className="mr-1 sm:mr-2 font-bold min-w-[40px]">{message.role === "assistant" ? "AI:" : "You:"}</div> <div className="mr-1 sm:mr-2 font-bold min-w-[40px]">{message.role === "assistant" ? "AI:" : "You:"}</div>
<div className="prose dark:prose-invert mt-[-2px]"> <div className="prose dark:prose-invert mt-[-2px]">
<ReactMarkdown {message.role === "user" ? (
remarkPlugins={[remarkGfm]} <div className="prose dark:prose-invert whitespace-pre">
components={{ {message.content}
code({ node, inline, className, children, ...props }) { </div>
const match = /language-(\w+)/.exec(className || ""); ) : (
return !inline && match ? ( <ReactMarkdown
<CodeBlock remarkPlugins={[remarkGfm]}
key={Math.random()} components={{
language={match[1]} code({ node, inline, className, children, ...props }) {
value={String(children).replace(/\n$/, "")} const match = /language-(\w+)/.exec(className || "");
lightMode={lightMode} return !inline && match ? (
{...props} <CodeBlock
/> key={Math.random()}
) : ( language={match[1]}
<code value={String(children).replace(/\n$/, "")}
className={className} lightMode={lightMode}
{...props} {...props}
> />
{children} ) : (
</code> <code className={className} {...props}>
); {children}
} </code>
}} );
> },
{message.content} }}
</ReactMarkdown> >
{message.content}
</ReactMarkdown>
)}
</div> </div>
</div> </div>
</div> </div>