alex/nova-betterchat
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-betterchat.git synced 2024-11-29 10:14:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

improve useSubmitHook

Browse source
This commit is contained in:
Jing Hua 2023-03-04 13:43:20 +08:00
parent 29f8fe75f3
commit fb910d0a52
2 changed files with 66 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

76
src/components/Chat/ChatContent/Message/MessageContent.tsx
View file

@ -12,10 +12,10 @@ import CrossIcon from '@icon/CrossIcon';
import RefreshIcon from '@icon/RefreshIcon';
import DownChevronArrow from '@icon/DownChevronArrow';
import { MessageInterface } from '@type/chat';
import useSubmit from '@hooks/useSubmit';
import { MessageInterface } from '@type/chat';
const MessageContent = ({
role,
content,
@ -91,6 +91,13 @@ const ContentView = ({
setMessages(updatedMessages);
};
const handleRefresh = () => {
const updatedMessages = JSON.parse(JSON.stringify(messages));
updatedMessages.splice(updatedMessages.length - 1, 1);
setMessages(updatedMessages);
handleSubmit();
};
return (
<>
<div className='markdown prose w-full break-words dark:prose-invert dark'>
@ -161,11 +168,7 @@ const ContentView = ({
{isDelete || (
<>
{role === 'assistant' && messageIndex === messages?.length - 1 && (
<RefreshButton
onClick={() => {
handleSubmit(true);
}}
/>
<RefreshButton onClick={handleRefresh} />
)}
{messageIndex !== 0 && (
<UpButton onClick={() => handleMove('up')} />
@ -289,6 +292,34 @@ const EditView = ({
}
};
const handleSave = () => {
if (_content === '') return;
const updatedMessages: MessageInterface[] = JSON.parse(
JSON.stringify(messages)
);
if (sticky) {
updatedMessages.push({ role: inputRole, content: _content });
_setContent('');
} else {
updatedMessages[messageIndex].content = _content;
setIsEdit(false);
}
setMessages(updatedMessages);
};
// only for sticky forms
const { handleSubmit } = useSubmit();
const handleSaveAndSubmit = () => {
if (_content == '') return;
const updatedMessages: MessageInterface[] = JSON.parse(
JSON.stringify(messages)
);
updatedMessages.push({ role: inputRole, content: _content });
_setContent('');
setMessages(updatedMessages);
handleSubmit();
};
useEffect(() => {
if (textareaRef.current) {
textareaRef.current.style.height = 'auto';
@ -308,28 +339,29 @@ const EditView = ({
}}
value={_content}
onInput={handleInput}
rows={sticky ? 10 : 1}
rows={1}
></textarea>
<div className='text-center mt-2 flex justify-center'>
{sticky && (
<button
className='btn relative btn-primary mr-2'
onClick={() => {
if (_content === '') return;
const updatedMessages: MessageInterface[] = JSON.parse(
JSON.stringify(messages)
);
if (sticky) {
updatedMessages.push({ role: inputRole, content: _content });
_setContent('');
} else {
updatedMessages[messageIndex].content = _content;
setIsEdit(false);
}
setMessages(updatedMessages);
}}
onClick={handleSaveAndSubmit}
>
<div className='flex items-center justify-center gap-2'>
Save and Submit
</div>
</button>
)}
<button
className={`btn relative mr-2 ${
sticky ? 'btn-neutral' : 'btn-primary'
}`}
onClick={handleSave}
>
<div className='flex items-center justify-center gap-2'>Save</div>
</button>
{sticky || (
<button
className='btn relative btn-neutral'

33
src/hooks/useSubmit.ts
View file

@ -7,48 +7,35 @@ import { parseEventSource } from '@api/helper';
const useSubmit = () => {
const [error, setError] = useState<string>('');
const [messages, apiFree, apiKey, setMessages, setGenerating, generating] =
useStore((state) => [
state.messages,
const [apiFree, apiKey, setMessages, setGenerating, generating] = useStore(
(state) => [
state.apiFree,
state.apiKey,
state.setMessages,
state.setGenerating,
state.generating,
]);
]
);
const handleSubmit = async (refresh?: boolean) => {
const handleSubmit = async () => {
if (generating) return;
const messages = useStore.getState().messages;
const updatedMessages: MessageInterface[] = JSON.parse(
JSON.stringify(messages)
);
if (refresh) {
updatedMessages[updatedMessages.length - 1] = {
role: 'assistant',
content: '',
};
} else {
updatedMessages.push({ role: 'assistant', content: '' });
}
setMessages(updatedMessages);
setGenerating(true);
let stream;
try {
if (apiFree) {
if (refresh)
stream = await getChatCompletionStreamFree(
updatedMessages.slice(0, updatedMessages.length - 1)
);
else stream = await getChatCompletionStreamFree(messages);
stream = await getChatCompletionStreamFree(messages);
} else if (apiKey) {
if (refresh)
stream = await getChatCompletionStreamCustom(
apiKey,
updatedMessages.slice(0, updatedMessages.length - 1)
);
else stream = await getChatCompletionStreamFree(messages);
stream = await getChatCompletionStreamFree(messages);
}
if (stream) {