kamp/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2024-11-19 12:40:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

Exception handler on embed chat middleware

Browse Source
This commit is contained in:
timothycarambat 2024-08-27 16:27:58 -07:00
parent 47a5c7126c
commit 548da9ade3
1 changed files with 104 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

193
server/utils/middleware/embedMiddleware.js
View File

@ -41,31 +41,112 @@ async function validEmbedConfigId(request, response, next) {
} }
async function canRespond(request, response, next) { async function canRespond(request, response, next) {
const embed = response.locals.embedConfig; try {
if (!embed) { const embed = response.locals.embedConfig;
response.sendStatus(404).end(); if (!embed) {
return; response.sendStatus(404).end();
} return;
}
// Block if disabled by admin. // Block if disabled by admin.
if (!embed.enabled) { if (!embed.enabled) {
response.status(503).json({ response.status(503).json({
id: uuidv4(), id: uuidv4(),
type: "abort", type: "abort",
textResponse: null, textResponse: null,
sources: [], sources: [],
close: true, close: true,
error: error:
"This chat has been disabled by the administrator - try again later.", "This chat has been disabled by the administrator - try again later.",
}); });
return; return;
} }
// Check if requester hostname is in the valid allowlist of domains. // Check if requester hostname is in the valid allowlist of domains.
const host = request.headers.origin ?? ""; const host = request.headers.origin ?? "";
const allowedHosts = EmbedConfig.parseAllowedHosts(embed); const allowedHosts = EmbedConfig.parseAllowedHosts(embed);
if (allowedHosts !== null && !allowedHosts.includes(host)) { if (allowedHosts !== null && !allowedHosts.includes(host)) {
response.status(401).json({ response.status(401).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error: "Invalid request.",
});
return;
}
const { sessionId, message } = reqBody(request);
if (!message?.length || !VALID_CHAT_MODE.includes(embed.chat_mode)) {
response.status(400).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error: !message?.length
? "Message is empty."
: `${embed.chat_mode} is not a valid mode.`,
});
return;
}
if (
!isNaN(embed.max_chats_per_day) &&
Number(embed.max_chats_per_day) > 0
) {
const dailyChatCount = await EmbedChats.count({
embed_id: embed.id,
createdAt: {
gte: new Date(new Date() - 24 * 60 * 60 * 1000),
},
});
if (dailyChatCount >= Number(embed.max_chats_per_day)) {
response.status(429).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error:
"The quota for this chat has been reached. Try again later or contact the site owner.",
});
return;
}
}
if (
!isNaN(embed.max_chats_per_session) &&
Number(embed.max_chats_per_session) > 0
) {
const dailySessionCount = await EmbedChats.count({
embed_id: embed.id,
session_id: sessionId,
createdAt: {
gte: new Date(new Date() - 24 * 60 * 60 * 1000),
},
});
if (dailySessionCount >= Number(embed.max_chats_per_session)) {
response.status(429).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error:
"Your quota for this chat has been reached. Try again later or contact the site owner.",
});
return;
}
}
next();
} catch (e) {
response.status(500).json({
id: uuidv4(), id: uuidv4(),
type: "abort", type: "abort",
textResponse: null, textResponse: null,
@ -75,72 +156,6 @@ async function canRespond(request, response, next) {
}); });
return; return;
} }
const { sessionId, message } = reqBody(request);
if (!message?.length || !VALID_CHAT_MODE.includes(embed.chat_mode)) {
response.status(400).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error: !message?.length
? "Message is empty."
: `${embed.chat_mode} is not a valid mode.`,
});
return;
}
if (!isNaN(embed.max_chats_per_day) && Number(embed.max_chats_per_day) > 0) {
const dailyChatCount = await EmbedChats.count({
embed_id: embed.id,
createdAt: {
gte: new Date(new Date() - 24 * 60 * 60 * 1000),
},
});
if (dailyChatCount >= Number(embed.max_chats_per_day)) {
response.status(429).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error:
"The quota for this chat has been reached. Try again later or contact the site owner.",
});
return;
}
}
if (
!isNaN(embed.max_chats_per_session) &&
Number(embed.max_chats_per_session) > 0
) {
const dailySessionCount = await EmbedChats.count({
embed_id: embed.id,
session_id: sessionId,
createdAt: {
gte: new Date(new Date() - 24 * 60 * 60 * 1000),
},
});
if (dailySessionCount >= Number(embed.max_chats_per_session)) {
response.status(429).json({
id: uuidv4(),
type: "abort",
textResponse: null,
sources: [],
close: true,
error:
"Your quota for this chat has been reached. Try again later or contact the site owner.",
});
return;
}
}
next();
} }
module.exports = { module.exports = {