diff --git a/ui/components/MessageBox.tsx b/ui/components/MessageBox.tsx
index 9ee7245..9712a23 100644
--- a/ui/components/MessageBox.tsx
+++ b/ui/components/MessageBox.tsx
@@ -34,15 +34,13 @@ const MessageBox = ({
const [speechMessage, setSpeechMessage] = useState(message.content);
useEffect(() => {
+ const regex = /\[(\d+)\]/g;
+
if (
message.role === 'assistant' &&
message?.sources &&
message.sources.length > 0
) {
- const regex = /\[(\d+)\]/g;
-
- setSpeechMessage(message.content.replace(regex, ''));
-
return setParsedMessage(
message.content.replace(
regex,
@@ -51,6 +49,8 @@ const MessageBox = ({
),
);
}
+
+ setSpeechMessage(message.content.replace(regex, ''));
setParsedMessage(message.content);
}, [message.content, message.sources, message.role]);
@@ -95,7 +95,7 @@ const MessageBox = ({