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
77 lines
2.7 KiB
JavaScript
77 lines
2.7 KiB
JavaScript
const { Action } = require('~/db/models');
|
|
|
|
/**
|
|
* Update an action with new data without overwriting existing properties,
|
|
* or create a new action if it doesn't exist.
|
|
*
|
|
* @param {Object} searchParams - The search parameters to find the action to update.
|
|
* @param {string} searchParams.action_id - The ID of the action to update.
|
|
* @param {string} searchParams.user - The user ID of the action's author.
|
|
* @param {Object} updateData - An object containing the properties to update.
|
|
* @returns {Promise<Action>} The updated or newly created action document as a plain object.
|
|
*/
|
|
const updateAction = async (searchParams, updateData) => {
|
|
const options = { new: true, upsert: true };
|
|
return await Action.findOneAndUpdate(searchParams, updateData, options).lean();
|
|
};
|
|
|
|
/**
|
|
* Retrieves all actions that match the given search parameters.
|
|
*
|
|
* @param {Object} searchParams - The search parameters to find matching actions.
|
|
* @param {boolean} includeSensitive - Flag to include sensitive data in the metadata.
|
|
* @returns {Promise<Array<Action>>} A promise that resolves to an array of action documents as plain objects.
|
|
*/
|
|
const getActions = async (searchParams, includeSensitive = false) => {
|
|
const actions = await Action.find(searchParams).lean();
|
|
|
|
if (!includeSensitive) {
|
|
for (let i = 0; i < actions.length; i++) {
|
|
const metadata = actions[i].metadata;
|
|
if (!metadata) {
|
|
continue;
|
|
}
|
|
|
|
const sensitiveFields = ['api_key', 'oauth_client_id', 'oauth_client_secret'];
|
|
for (let field of sensitiveFields) {
|
|
if (metadata[field]) {
|
|
delete metadata[field];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return actions;
|
|
};
|
|
|
|
/**
|
|
* Deletes an action by params.
|
|
*
|
|
* @param {Object} searchParams - The search parameters to find the action to delete.
|
|
* @param {string} searchParams.action_id - The ID of the action to delete.
|
|
* @param {string} searchParams.user - The user ID of the action's author.
|
|
* @returns {Promise<Action>} A promise that resolves to the deleted action document as a plain object, or null if no document was found.
|
|
*/
|
|
const deleteAction = async (searchParams) => {
|
|
return await Action.findOneAndDelete(searchParams).lean();
|
|
};
|
|
|
|
/**
|
|
* Deletes actions by params.
|
|
*
|
|
* @param {Object} searchParams - The search parameters to find the actions to delete.
|
|
* @param {string} searchParams.action_id - The ID of the action(s) to delete.
|
|
* @param {string} searchParams.user - The user ID of the action's author.
|
|
* @returns {Promise<Number>} A promise that resolves to the number of deleted action documents.
|
|
*/
|
|
const deleteActions = async (searchParams) => {
|
|
const result = await Action.deleteMany(searchParams);
|
|
return result.deletedCount;
|
|
};
|
|
|
|
module.exports = {
|
|
getActions,
|
|
updateAction,
|
|
deleteAction,
|
|
deleteActions,
|
|
};
|