jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor chat message handling in newChatView

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 11:35:27 -03:00
commit fef68c30a8
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/frontend/src/components/newChatView/chatMessage/index.tsx
View file

@ -24,7 +24,10 @@ export default function ChatMessage({
const template = chat.template; const template = chat.template;
const [promptOpen, setPromptOpen] = useState(false); const [promptOpen, setPromptOpen] = useState(false);
const [streamUrl, setStreamUrl] = useState(chat.stream_url); const [streamUrl, setStreamUrl] = useState(chat.stream_url);
const [chatMessage, setChatMessage] = useState(chat.message.toString()); // We need to check if message is not undefined because
// we need to run .toString() on it
const chatMessageString = chat.message ? chat.message.toString() : "";
const [chatMessage, setChatMessage] = useState(chatMessageString);
const [isStreaming, setIsStreaming] = useState(false); const [isStreaming, setIsStreaming] = useState(false);
// The idea now is that chat.stream_url MAY be a URL if we should stream the output of the chat // The idea now is that chat.stream_url MAY be a URL if we should stream the output of the chat
@ -146,7 +149,7 @@ export default function ChatMessage({
<div className="w-full"> <div className="w-full">
{useMemo( {useMemo(
() => () =>
chat.message.toString() === "" && lockChat ? ( chatMessage === "" && lockChat ? (
<IconComponent <IconComponent
name="MoreHorizontal" name="MoreHorizontal"
className="h-8 w-8 animate-pulse" className="h-8 w-8 animate-pulse"
@ -288,11 +291,11 @@ dark:prose-invert"
} }
return <p>{parts}</p>; return <p>{parts}</p>;
}) })
: chat.message.toString()} : chatMessage}
</span> </span>
</> </>
) : ( ) : (
<span>{chat.message.toString()}</span> <span>{chatMessage}</span>
)} )}
</div> </div>
)} )}