LibreChat/api/server/routes/convos.js

68 lines
1.9 KiB
JavaScript
Raw Normal View History

2023-02-12 16:38:33 -05:00
const express = require('express');
const router = express.Router();
const { titleConvo } = require('../../app/');
const { getConvo, saveConvo, getConvoTitle } = require('../../models');
2023-03-15 04:05:14 +08:00
const { getConvosByPage, deleteConvos, updateConvo } = require('../../models/Conversation');
const { getMessages } = require('../../models/Message');
2023-02-12 16:38:33 -05:00
router.get('/', async (req, res) => {
2023-03-05 14:41:50 -05:00
const pageNumber = req.query.pageNumber || 1;
res.status(200).send(await getConvosByPage(req?.session?.user?.username, pageNumber));
});
router.post('/gen_title', async (req, res) => {
const { conversationId } = req.body.arg;
const convo = await getConvo(req?.session?.user?.username, conversationId)
const firstMessage = (await getMessages({ conversationId }))[0]
const secondMessage = (await getMessages({ conversationId }))[1]
const title = convo.jailbreakConversationId
? await getConvoTitle(req?.session?.user?.username, conversationId)
: await titleConvo({
model: convo?.model,
message: firstMessage?.text,
response: JSON.stringify(secondMessage?.text || '')
});
await saveConvo(
req?.session?.user?.username,
{
conversationId,
title
}
)
res.status(200).send(title);
2023-02-12 16:38:33 -05:00
});
router.post('/clear', async (req, res) => {
let filter = {};
const { conversationId } = req.body.arg;
if (conversationId) {
2023-02-12 16:38:33 -05:00
filter = { conversationId };
}
try {
const dbResponse = await deleteConvos(req?.session?.user?.username, filter);
2023-02-12 16:38:33 -05:00
res.status(201).send(dbResponse);
} catch (error) {
console.error(error);
res.status(500).send(error);
}
});
router.post('/update', async (req, res) => {
const update = req.body.arg;
try {
const dbResponse = await updateConvo(req?.session?.user?.username, update);
2023-02-12 16:38:33 -05:00
res.status(201).send(dbResponse);
} catch (error) {
console.error(error);
res.status(500).send(error);
}
});
module.exports = router;