Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-01-07 19:18:52 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

chore: model menu functions behave as expected

Browse source
This commit is contained in:
Danny Avila 2023-03-07 13:19:56 -05:00
parent 191118b90b
commit 6be037323e
9 changed files with 157 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/models/CustomGpt.js
View file

@ -60,6 +60,17 @@ module.exports = {
return { message: 'Error updating customGpt' };
}
},
updateByLabel: async ({ prevLabel, ...update }) => {
try {
return await CustomGpt.findOneAndUpdate({ chatGptLabel: prevLabel }, update, {
new: true,
upsert: true
}).exec();
} catch (error) {
console.log(error);
return { message: 'Error updating customGpt' };
}
},
deleteCustomGpts: async (filter) => {
try {
return await CustomGpt.deleteMany(filter).exec();

3
api/models/index.js
View file

@ -1,5 +1,5 @@
const { saveMessage, deleteMessages } = require('./Message');
const { getCustomGpts, updateCustomGpt, deleteCustomGpts } = require('./CustomGpt');
const { getCustomGpts, updateCustomGpt, updateByLabel, deleteCustomGpts } = require('./CustomGpt');
const { saveConvo } = require('./Conversation');
module.exports = {
@ -8,5 +8,6 @@ module.exports = {
saveConvo,
getCustomGpts,
updateCustomGpt,
updateByLabel,
deleteCustomGpts
};

22
api/server/routes/customGpts.js
View file

@ -1,6 +1,6 @@
const express = require('express');
const router = express.Router();
const { getCustomGpts, updateCustomGpt, deleteCustomGpts } = require('../../models');
const { getCustomGpts, updateCustomGpt, updateByLabel, deleteCustomGpts } = require('../../models');
router.get('/', async (req, res) => {
const models = (await getCustomGpts()).map(model => {
@ -8,20 +8,14 @@ router.get('/', async (req, res) => {
model._id = model._id.toString();
return model;
});
// console.log(models);
res.status(200).send(models);
});
router.post('/delete/:_id', async (req, res) => {
const { _id } = req.params;
let filter = {};
if (_id) {
filter = { _id };
}
router.post('/delete', async (req, res) => {
const { arg } = req.body;
try {
const dbResponse = await deleteCustomGpts(filter);
const dbResponse = await deleteCustomGpts(arg);
res.status(201).send(dbResponse);
} catch (error) {
console.error(error);
@ -44,8 +38,14 @@ router.post('/delete/:_id', async (req, res) => {
router.post('/', async (req, res) => {
const update = req.body.arg;
let setter = updateCustomGpt;
if (update.prevLabel) {
setter = updateByLabel;
}
try {
const dbResponse = await updateCustomGpt(update);
const dbResponse = await setter(update);
res.status(201).send(dbResponse);
} catch (error) {
console.error(error);