highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-30 14:39:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

validate messages (#2139)

Browse Source
This commit is contained in:
pablodanswer
2024-08-14 15:06:48 -07:00
committed by GitHub
parent 2ab192933b
commit d9bcacfae7
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
web/src/app/chat/message/Messages.tsx
View File

@@ -154,10 +154,17 @@ export const AIMessage = ({
return content;
}
const codeBlockRegex = /```[\s\S]*?```|```[\s\S]*?$/g;
const codeBlockRegex = /```(\w*)\n[\s\S]*?```|```[\s\S]*?$/g;
const matches = content.match(codeBlockRegex);
if (matches) {
content = matches.reduce((acc, match) => {
if (!match.match(/```\w+/)) {
return acc.replace(match, match.replace("```", "```plaintext"));
}
return acc;
}, content);
const lastMatch = matches[matches.length - 1];
if (!lastMatch.endsWith("```")) {
return content;
@@ -166,7 +173,6 @@ export const AIMessage = ({
return content + (!isComplete && !toolCallGenerating ? " [*]() " : "");
};
const finalContent = processContent(content as string);
const { isHovering, trackedElementRef, hoverElementRef } = useMouseTracking();
@@ -400,7 +406,6 @@ export const AIMessage = ({
);
}
},
code: (props) => (
<CodeBlock
className="w-full"