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

wip: refactor: new abort message handling

Browse source
This commit is contained in:
Daniel Avila 2023-04-09 09:23:03 -04:00
parent b59588c6ee
commit 6246ffff1e
2 changed files with 45 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

28
api/server/routes/ask/askOpenAI.js
View file

@ -8,18 +8,18 @@ const { handleError, sendMessage, createOnProgress, handleText } = require('./ha
const abortControllers = new Map();
router.get('/abort', (req, res) => {
const requestId = req.query.requestId;
if (abortControllers.has(requestId)) {
const abortController = abortControllers.get(requestId);
abortController.abort();
abortControllers.delete(requestId);
console.log('Aborted request', requestId);
res.status(200).send('Aborted');
} else {
res.status(404).send('Request not found');
router.post('/abort', (req, res) => {
const { abortKey, message } = req.body;
if (!abortControllers.has(abortKey)) {
return res.status(404).send('Request not found');
}
const { abortController, userMessage } = abortControllers.get(abortKey);
abortController.abort();
abortControllers.delete(abortKey);
console.log('Aborted request', abortKey, userMessage);
res.status(200).send('Aborted');
});
router.post('/', async (req, res) => {
@ -116,11 +116,11 @@ const ask = async ({
try {
const progressCallback = createOnProgress();
const abortController = new AbortController();
const abortKey = userMessage.messageId;
abortControllers.set(abortKey, abortController);
const abortKey = conversationId;
console.log('conversationId -----> ', conversationId);
abortControllers.set(abortKey, { abortController, userMessage });
res.on('close', () => {
console.log('stopped message, aborting');
abortController.abort();
return res.end();
});