feat: create conversation at the beginning then return the userMessage

This commit is contained in:
Wentao Lyu 2023-03-13 13:11:53 +08:00
parent 5a409ccfa6
commit 8773878be2
6 changed files with 50 additions and 40 deletions

View file

@ -11,6 +11,7 @@ import ConvoIcon from '../svg/ConvoIcon';
export default function Conversation({
id,
model,
parentMessageId,
conversationId,
title = 'New conversation',
@ -76,12 +77,12 @@ export default function Conversation({
if (chatGptLabel) {
dispatch(setModel('chatgptCustom'));
} else {
dispatch(setModel(data[1].sender));
dispatch(setModel(model));
}
if (modelMap[data[1].sender.toLowerCase()]) {
console.log('sender', data[1].sender);
dispatch(setCustomModel(data[1].sender.toLowerCase()));
if (modelMap[model.toLowerCase()]) {
console.log('sender', model);
dispatch(setCustomModel(model.toLowerCase()));
} else {
dispatch(setCustomModel(null));
}

View file

@ -26,6 +26,7 @@ export default function Conversations({ conversations, conversationId, showMore
<Conversation
key={convo.conversationId}
id={convo.conversationId}
model={convo.model}
parentMessageId={convo.parentMessageId}
title={convo.title}
conversationId={conversationId}

View file

@ -28,14 +28,24 @@ export default function TextChat({ messages }) {
inputRef.current?.focus();
}, [convo?.conversationId, ])
const messageHandler = (data, currentState) => {
const { messages, currentMsg, message, sender } = currentState;
const messageHandler = (data, currentState, currentMsg) => {
const { messages, _currentMsg, message, sender } = currentState;
dispatch(setMessages([...messages, currentMsg, { sender, text: data }]));
};
const convoHandler = (data, currentState) => {
const createdHandler = (data, currentState, currentMsg) => {
const { conversationId } = currentMsg;
dispatch(
setConversation({
conversationId,
})
);
};
const convoHandler = (data, currentState, currentMsg) => {
const { requestMessage, responseMessage } = data;
const { messages, currentMsg, message, isCustomModel, sender } =
const { messages, _currentMsg, message, isCustomModel, sender } =
currentState;
const { model, chatGptLabel, promptPrefix } = message;
dispatch(
@ -210,6 +220,7 @@ export default function TextChat({ messages }) {
}
const currentState = submission;
let currentMsg = currentState.currentMsg;
const { server, payload } = createPayload(submission);
const onMessage = (e) => {
if (stopStream) {
@ -223,12 +234,15 @@ export default function TextChat({ messages }) {
// }
if (data.final) {
convoHandler(data, currentState);
convoHandler(data, currentState, currentMsg);
console.log('final', data);
} if (data.created) {
currentMsg = data.message;
createdHandler(data, currentState, currentMsg);
} else {
let text = data.text || data.response;
if (data.message) {
messageHandler(text, currentState);
messageHandler(text, currentState, currentMsg);
}
// console.log('dataStream', data);
}