Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-17 17:00:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

feat: add regenerate to all response message as official

Browse source
This commit is contained in:
Wentao Lyu 2023-03-17 01:49:09 +08:00
parent e3b0cb7db7
commit 0891566d1e
8 changed files with 211 additions and 177 deletions
Download patch file Download diff file Expand all files Collapse all files

60
client/src/components/Messages/Message.jsx
View file

@ -8,8 +8,9 @@ import { setError } from '~/store/convoSlice';
import { setMessages } from '~/store/messageSlice';
import { setSubmitState, setSubmission } from '~/store/submitSlice';
import { setText } from '~/store/textSlice';
import { setConversation } from '../../store/convoSlice';
import { setConversation, setLatestMessage } from '../../store/convoSlice';
import { getIconOfModel } from '../../utils';
import { useMessageHandler } from '../../utils/handleSubmit'
export default function Message({
message,
@ -32,6 +33,7 @@ export default function Message({
const { error: convoError } = convo;
const last = !message?.children?.length;
const edit = message.messageId == currentEditId;
const { ask } = useMessageHandler();
const dispatch = useDispatch();
// const notUser = !isCreatedByUser; // sender.toLowerCase() !== 'user';
@ -51,7 +53,11 @@ export default function Message({
}, [isSubmitting, text, blinker, scrollToBottom, abortScroll]);
useEffect(() => {
if (last) dispatch(setConversation({ parentMessageId: message?.messageId }));
if (last) {
// TODO: stop using conversation.parentMessageId and remove it.
dispatch(setConversation({ parentMessageId: message?.messageId }));
dispatch(setLatestMessage({ ...message }));
}
}, [last]);
const enterEdit = (cancel) => setCurrentEditId(cancel ? -1 : message.messageId);
@ -87,55 +93,7 @@ export default function Message({
const resubmitMessage = () => {
const text = textEditor.current.innerText;
if (convoError) {
dispatch(setError(false));
}
if (!!isSubmitting || text.trim() === '') {
return;
}
// this is not a real messageId, it is used as placeholder before real messageId returned
const fakeMessageId = crypto.randomUUID();
const isCustomModel = model === 'chatgptCustom' || !initial[model];
const currentMsg = {
sender: 'User',
text: text.trim(),
current: true,
isCreatedByUser: true,
parentMessageId: message?.parentMessageId,
conversationId: message?.conversationId,
messageId: fakeMessageId
};
const sender = model === 'chatgptCustom' ? chatGptLabel : model;
const initialResponse = {
sender,
text: '',
parentMessageId: fakeMessageId,
submitting: true
};
dispatch(setSubmitState(true));
dispatch(setMessages([...messages, currentMsg, initialResponse]));
dispatch(setText(''));
const submission = {
isCustomModel,
message: {
...currentMsg,
model,
chatGptLabel,
promptPrefix
},
messages: messages,
currentMsg,
initialResponse,
sender
};
console.log('User Input:', currentMsg?.text);
// handleSubmit(submission);
dispatch(setSubmission(submission));
ask({ text, parentMessageId: message?.parentMessageId, conversationId: message?.conversationId,});
setSiblingIdx(siblingCount - 1);
enterEdit(true);