mirror of
https://github.com/danny-avila/LibreChat.git
synced 2026-06-13 20:00:27 +00:00
* feat: add agent file retention exemption * refactor: centralize agent file retention policy
43 lines
1.4 KiB
JavaScript
43 lines
1.4 KiB
JavaScript
const {
|
|
getRetentionExpiry: getRetentionExpiryWithDeps,
|
|
getAgentFileRetentionExpiry: getAgentFileRetentionExpiryWithDeps,
|
|
} = require('@librechat/api');
|
|
const { logger, createTempChatExpirationDate } = require('@librechat/data-schemas');
|
|
const db = require('~/models');
|
|
|
|
const getRetentionDependencies = () => ({
|
|
getConvo: db.getConvoRetention ?? db.getConvo,
|
|
createExpirationDate: createTempChatExpirationDate,
|
|
logger,
|
|
});
|
|
|
|
/**
|
|
* Returns `{ expiredAt }` when the request indicates data retention applies, otherwise `{}`.
|
|
* Spread into file data objects before calling createFile.
|
|
* @param {ServerRequest} req
|
|
* @returns {Promise<{ expiredAt?: Date | null }>}
|
|
*/
|
|
async function getRetentionExpiry(req) {
|
|
return getRetentionExpiryWithDeps(req, getRetentionDependencies());
|
|
}
|
|
|
|
/**
|
|
* Returns `{ expiredAt }` for agent file uploads when retention applies, otherwise `{}`.
|
|
* @param {object} params
|
|
* @param {ServerRequest} params.req
|
|
* @param {boolean} [params.messageAttachment]
|
|
* @param {string} [params.tool_resource]
|
|
* @param {string} [params.toolResource]
|
|
* @returns {Promise<{ expiredAt?: Date | null }>}
|
|
*/
|
|
async function getAgentFileRetentionExpiry({ tool_resource, toolResource, ...params }) {
|
|
return getAgentFileRetentionExpiryWithDeps(
|
|
{ ...params, toolResource: tool_resource ?? toolResource },
|
|
getRetentionDependencies(),
|
|
);
|
|
}
|
|
|
|
module.exports = {
|
|
getRetentionExpiry,
|
|
getAgentFileRetentionExpiry,
|
|
};
|