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

@ -17,11 +17,13 @@ router.use('/bing', askBing);
router.use('/sydney', askSydney);
router.post('/', async (req, res) => {
let { model, text, parentMessageId, conversationId, chatGptLabel, promptPrefix } = req.body;
let { model, text, parentMessageId, conversationId: oldConversationId , chatGptLabel, promptPrefix } = req.body;
if (text.length === 0) {
return handleError(res, 'Prompt empty or too short');
}
const conversationId = oldConversationId || crypto.randomUUID();
const userMessageId = crypto.randomUUID();
const userParentMessageId = parentMessageId || '00000000-0000-0000-0000-000000000000'
let userMessage = {
@ -36,8 +38,6 @@ router.post('/', async (req, res) => {
console.log('ask log', {
model,
...userMessage,
parentMessageId: userParentMessageId,
conversationId,
chatGptLabel,
promptPrefix
});
@ -61,12 +61,6 @@ router.post('/', async (req, res) => {
}
}
// if (messageId) {
// // existing conversation
// await saveMessage(userMessage);
// await deleteMessagesSince(userMessage);
// } else {}
res.writeHead(200, {
Connection: 'keep-alive',
'Content-Type': 'text/event-stream',
@ -75,6 +69,10 @@ router.post('/', async (req, res) => {
'X-Accel-Buffering': 'no'
});
await saveMessage(userMessage);
await saveConvo({ ...userMessage, model, chatGptLabel, promptPrefix });
sendMessage(res, { message: userMessage, created: true });
try {
let i = 0;
let tokens = '';
@ -165,7 +163,7 @@ router.post('/', async (req, res) => {
res.end();
} catch (error) {
console.log(error);
await deleteMessages({ messageId: userMessageId });
// await deleteMessages({ messageId: userMessageId });
handleError(res, error.message);
}
});