mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02:00
a2fc7d312a
* refactor: move model definitions and database-related methods to packages/data-schemas * ci: update tests due to new DB structure fix: disable mocking `librechat-data-provider` feat: Add schema exports to data-schemas package - Introduced a new schema module that exports various schemas including action, agent, and user schemas. - Updated index.ts to include the new schema exports for better modularity and organization. ci: fix appleStrategy tests fix: Agent.spec.js ci: refactor handleTools tests to use MongoMemoryServer for in-memory database fix: getLogStores imports ci: update banViolation tests to use MongoMemoryServer and improve session mocking test: refactor samlStrategy tests to improve mock configurations and user handling ci: fix crypto mock in handleText tests for improved accuracy ci: refactor spendTokens tests to improve model imports and setup ci: refactor Message model tests to use MongoMemoryServer and improve database interactions * refactor: streamline IMessage interface and move feedback properties to types/message.ts * refactor: use exported initializeRoles from `data-schemas`, remove api workspace version (this serves as an example of future migrations that still need to happen) * refactor: update model imports to use destructuring from `~/db/models` for consistency and clarity * refactor: remove unused mongoose imports from model files for cleaner code * refactor: remove unused mongoose imports from Share, Prompt, and Transaction model files for cleaner code * refactor: remove unused import in Transaction model for cleaner code * ci: update deploy workflow to reference new Docker Dev Branch Images Build and add new workflow for building Docker images on dev branch * chore: cleanup imports
82 lines
2.6 KiB
JavaScript
82 lines
2.6 KiB
JavaScript
const { logger } = require('@librechat/data-schemas');
|
|
const { Preset } = require('~/db/models');
|
|
|
|
const getPreset = async (user, presetId) => {
|
|
try {
|
|
return await Preset.findOne({ user, presetId }).lean();
|
|
} catch (error) {
|
|
logger.error('[getPreset] Error getting single preset', error);
|
|
return { message: 'Error getting single preset' };
|
|
}
|
|
};
|
|
|
|
module.exports = {
|
|
getPreset,
|
|
getPresets: async (user, filter) => {
|
|
try {
|
|
const presets = await Preset.find({ ...filter, user }).lean();
|
|
const defaultValue = 10000;
|
|
|
|
presets.sort((a, b) => {
|
|
let orderA = a.order !== undefined ? a.order : defaultValue;
|
|
let orderB = b.order !== undefined ? b.order : defaultValue;
|
|
|
|
if (orderA !== orderB) {
|
|
return orderA - orderB;
|
|
}
|
|
|
|
return b.updatedAt - a.updatedAt;
|
|
});
|
|
|
|
return presets;
|
|
} catch (error) {
|
|
logger.error('[getPresets] Error getting presets', error);
|
|
return { message: 'Error retrieving presets' };
|
|
}
|
|
},
|
|
savePreset: async (user, { presetId, newPresetId, defaultPreset, ...preset }) => {
|
|
try {
|
|
const setter = { $set: {} };
|
|
const { user: _, ...cleanPreset } = preset;
|
|
const update = { presetId, ...cleanPreset };
|
|
if (preset.tools && Array.isArray(preset.tools)) {
|
|
update.tools =
|
|
preset.tools
|
|
.map((tool) => tool?.pluginKey ?? tool)
|
|
.filter((toolName) => typeof toolName === 'string') ?? [];
|
|
}
|
|
if (newPresetId) {
|
|
update.presetId = newPresetId;
|
|
}
|
|
|
|
if (defaultPreset) {
|
|
update.defaultPreset = defaultPreset;
|
|
update.order = 0;
|
|
|
|
const currentDefault = await Preset.findOne({ defaultPreset: true, user });
|
|
|
|
if (currentDefault && currentDefault.presetId !== presetId) {
|
|
await Preset.findByIdAndUpdate(currentDefault._id, {
|
|
$unset: { defaultPreset: '', order: '' },
|
|
});
|
|
}
|
|
} else if (defaultPreset === false) {
|
|
update.defaultPreset = undefined;
|
|
update.order = undefined;
|
|
setter['$unset'] = { defaultPreset: '', order: '' };
|
|
}
|
|
|
|
setter.$set = update;
|
|
return await Preset.findOneAndUpdate({ presetId, user }, setter, { new: true, upsert: true });
|
|
} catch (error) {
|
|
logger.error('[savePreset] Error saving preset', error);
|
|
return { message: 'Error saving preset' };
|
|
}
|
|
},
|
|
deletePresets: async (user, filter) => {
|
|
// let toRemove = await Preset.find({ ...filter, user }).select('presetId');
|
|
// const ids = toRemove.map((instance) => instance.presetId);
|
|
let deleteCount = await Preset.deleteMany({ ...filter, user });
|
|
return deleteCount;
|
|
},
|
|
};
|