mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 17:00:15 +01:00
8c22bb1d3d
* chore: replace violation cache accessors with enum * chore: fix test * chore(fileSchema): index timestamps * fix(ActionService): use encoding/caching strategy for handling assistant function character length limit * refactor(actions): async `domainParser` also resolve retrieved model (which is deployment name) to user-defined model * style(AssistantAction): add `whitespace-nowrap` for ellipsis * refactor(ActionService): if domain is less than or equal to encoded domain fixed length, return domain with replacement of separator * refactor(actions): use sessions/transactions for updating Assistant Action database records * chore: remove TTL from ENCODED_DOMAINS cache * refactor(domainParser): minor optimization and add tests * fix(spendTokens): use txData.user for token usage logging * refactor(actions): add helper function `withSession` for database operations with sessions/transactions * fix(PluginsClient): logger debug `message` field edge case
69 lines
2.7 KiB
JavaScript
69 lines
2.7 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const actionSchema = require('./schema/action');
|
|
|
|
const Action = mongoose.model('action', actionSchema);
|
|
|
|
/**
|
|
* Update an action with new data without overwriting existing properties,
|
|
* or create a new action if it doesn't exist, within a transaction session if provided.
|
|
*
|
|
* @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.
|
|
* @param {mongoose.ClientSession} [session] - The transaction session to use.
|
|
* @returns {Promise<Object>} The updated or newly created action document as a plain object.
|
|
*/
|
|
const updateAction = async (searchParams, updateData, session = null) => {
|
|
const options = { new: true, upsert: true, session };
|
|
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<Object>>} 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, within a transaction session if provided.
|
|
*
|
|
* @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.
|
|
* @param {mongoose.ClientSession} [session] - The transaction session to use (optional).
|
|
* @returns {Promise<Object>} A promise that resolves to the deleted action document as a plain object, or null if no document was found.
|
|
*/
|
|
const deleteAction = async (searchParams, session = null) => {
|
|
const options = session ? { session } : {};
|
|
return await Action.findOneAndDelete(searchParams, options).lean();
|
|
};
|
|
|
|
module.exports = {
|
|
updateAction,
|
|
getActions,
|
|
deleteAction,
|
|
};
|