refactor: original changes

This commit is contained in:
Danny Avila 2025-05-30 04:28:22 -04:00
parent fa9177180f
commit f9c0e9853f
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
83 changed files with 413 additions and 505 deletions

View file

@ -1,7 +1,7 @@
const mongoose = require('mongoose');
const { File } = require('@librechat/data-schemas');
const { EToolResources } = require('librechat-data-provider');
const { logger } = require('~/config');
const db = require('~/lib/db/connectDb');
/**
* Finds a file by its file_id with additional query options.
@ -10,7 +10,7 @@ const db = require('~/lib/db/connectDb');
* @returns {Promise<MongoFile>} A promise that resolves to the file document or null.
*/
const findFileById = async (file_id, options = {}) => {
return await db.models.File.findOne({ file_id, ...options }).lean();
return await File.findOne({ file_id, ...options }).lean();
};
/**
@ -23,7 +23,7 @@ const findFileById = async (file_id, options = {}) => {
*/
const getFiles = async (filter, _sortOptions, selectFields = { text: 0 }) => {
const sortOptions = { updatedAt: -1, ..._sortOptions };
return await db.models.File.find(filter).select(selectFields).sort(sortOptions).lean();
return await File.find(filter).select(selectFields).sort(sortOptions).lean();
};
/**
@ -79,7 +79,7 @@ const createFile = async (data, disableTTL) => {
delete fileData.expiresAt;
}
return await db.models.File.findOneAndUpdate({ file_id: data.file_id }, fileData, {
return await File.findOneAndUpdate({ file_id: data.file_id }, fileData, {
new: true,
upsert: true,
}).lean();
@ -96,7 +96,7 @@ const updateFile = async (data) => {
$set: update,
$unset: { expiresAt: '' }, // Remove the expiresAt field to prevent TTL
};
return await db.models.File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
return await File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
};
/**
@ -110,7 +110,7 @@ const updateFileUsage = async (data) => {
$inc: { usage: inc },
$unset: { expiresAt: '', temp_file_id: '' },
};
return await db.models.File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
return await File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
};
/**
@ -119,7 +119,7 @@ const updateFileUsage = async (data) => {
* @returns {Promise<MongoFile>} A promise that resolves to the deleted file document or null.
*/
const deleteFile = async (file_id) => {
return await db.models.File.findOneAndDelete({ file_id }).lean();
return await File.findOneAndDelete({ file_id }).lean();
};
/**
@ -128,7 +128,7 @@ const deleteFile = async (file_id) => {
* @returns {Promise<MongoFile>} A promise that resolves to the deleted file document or null.
*/
const deleteFileByFilter = async (filter) => {
return await db.models.File.findOneAndDelete(filter).lean();
return await File.findOneAndDelete(filter).lean();
};
/**
@ -141,7 +141,7 @@ const deleteFiles = async (file_ids, user) => {
if (user) {
deleteQuery = { user: user };
}
return await db.models.File.deleteMany(deleteQuery);
return await File.deleteMany(deleteQuery);
};
/**