mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 08:50:15 +01:00
379e470e38
* chore: bump langchain to v0.0.213 from v0.0.186 * fix: handle abort edge cases: - abort message server-side if response experienced error mid-generation - attempt to recover message if aborting resulted in error - if abortKey is not provided, use conversationId if it exists - if headers were already sent, send an Event stream message - issue warning for possible Google censor/filter refactor(streamResponse): for `sendError`, allow passing overrides so that error can include partial generation, improve typing for `sendMessage` * chore(MessageContent): remove eslint warning for unused `i`, rephrase unfinished message text * fix(useSSE): avoid invoking cancelHandler if the abort response was 404 * chore(TMessage): remove unnecessary, unused legacy message property `submitting` * chore(TMessage): remove unnecessary legacy message property `cancelled` * chore(abortMiddleware): remove unused `errorText` property to avoid confusion
84 lines
2.3 KiB
JavaScript
84 lines
2.3 KiB
JavaScript
const crypto = require('crypto');
|
|
const { saveMessage, getMessages } = require('~/models/Message');
|
|
const { getConvo } = require('~/models/Conversation');
|
|
|
|
/**
|
|
* Sends error data in Server Sent Events format and ends the response.
|
|
* @param {object} res - The server response.
|
|
* @param {string} message - The error message.
|
|
*/
|
|
const handleError = (res, message) => {
|
|
res.write(`event: error\ndata: ${JSON.stringify(message)}\n\n`);
|
|
res.end();
|
|
};
|
|
|
|
/**
|
|
* Sends message data in Server Sent Events format.
|
|
* @param {object} res - - The server response.
|
|
* @param {string} message - The message to be sent.
|
|
* @param {'message' | 'error' | 'cancel'} event - [Optional] The type of event. Default is 'message'.
|
|
*/
|
|
const sendMessage = (res, message, event = 'message') => {
|
|
if (message.length === 0) {
|
|
return;
|
|
}
|
|
res.write(`event: ${event}\ndata: ${JSON.stringify(message)}\n\n`);
|
|
};
|
|
|
|
/**
|
|
* Processes an error with provided options, saves the error message and sends a corresponding SSE response
|
|
* @async
|
|
* @param {object} res - The server response.
|
|
* @param {object} options - The options for handling the error containing message properties.
|
|
* @param {function} callback - [Optional] The callback function to be executed.
|
|
*/
|
|
const sendError = async (res, options, callback) => {
|
|
const {
|
|
user,
|
|
sender,
|
|
conversationId,
|
|
messageId,
|
|
parentMessageId,
|
|
text,
|
|
shouldSaveMessage,
|
|
overrideProps = {},
|
|
} = options;
|
|
const errorMessage = {
|
|
sender,
|
|
messageId: messageId ?? crypto.randomUUID(),
|
|
conversationId,
|
|
parentMessageId,
|
|
unfinished: false,
|
|
error: true,
|
|
final: true,
|
|
text,
|
|
isCreatedByUser: false,
|
|
...overrideProps,
|
|
};
|
|
if (callback && typeof callback === 'function') {
|
|
await callback();
|
|
}
|
|
|
|
if (shouldSaveMessage) {
|
|
await saveMessage({ ...errorMessage, user });
|
|
}
|
|
|
|
if (!errorMessage.error) {
|
|
const requestMessage = { messageId: parentMessageId, conversationId };
|
|
const query = await getMessages(requestMessage);
|
|
return sendMessage(res, {
|
|
final: true,
|
|
requestMessage: query?.[0] ? query[0] : requestMessage,
|
|
responseMessage: errorMessage,
|
|
conversation: await getConvo(user, conversationId),
|
|
});
|
|
}
|
|
|
|
handleError(res, errorMessage);
|
|
};
|
|
|
|
module.exports = {
|
|
handleError,
|
|
sendMessage,
|
|
sendError,
|
|
};
|