2023-02-12 16:38:33 -05:00
|
|
|
const express = require('express');
|
|
|
|
const router = express.Router();
|
2023-03-14 01:24:43 +08:00
|
|
|
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');
|
2023-03-14 01:24:43 +08:00
|
|
|
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;
|
2023-03-14 01:24:43 +08:00
|
|
|
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
|
2023-03-15 16:56:47 +08:00
|
|
|
? await getConvoTitle(req?.session?.user?.username, conversationId)
|
2023-03-14 01:24:43 +08:00
|
|
|
: await titleConvo({
|
|
|
|
model: convo?.model,
|
|
|
|
message: firstMessage?.text,
|
|
|
|
response: JSON.stringify(secondMessage?.text || '')
|
|
|
|
});
|
|
|
|
|
2023-03-15 16:56:47 +08:00
|
|
|
await saveConvo(
|
|
|
|
req?.session?.user?.username,
|
2023-03-14 01:24:43 +08:00
|
|
|
{
|
2023-03-15 16:56:47 +08:00
|
|
|
conversationId,
|
|
|
|
title
|
2023-03-14 01:24:43 +08:00
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
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;
|
2023-02-21 21:30:56 -05:00
|
|
|
if (conversationId) {
|
2023-02-12 16:38:33 -05:00
|
|
|
filter = { conversationId };
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
2023-03-14 01:24:43 +08:00
|
|
|
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 {
|
2023-03-14 01:24:43 +08:00
|
|
|
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;
|