mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 06:00:56 +02:00
0c326797dd
* feat: add GOOGLE_MODELS env var * feat: add gemini vision support * refactor(GoogleClient): adjust clientOptions handling depending on model * fix(logger): fix redact logic and redact errors only * fix(GoogleClient): do not allow non-multiModal messages when gemini-pro-vision is selected * refactor(OpenAIClient): use `isVisionModel` client property to avoid calling validateVisionModel multiple times * refactor: better debug logging by correctly traversing, redacting sensitive info, and logging condensed versions of long values * refactor(GoogleClient): allow response errors to be thrown/caught above client handling so user receives meaningful error message debug orderedMessages, parentMessageId, and buildMessages result * refactor(AskController): use model from client.modelOptions.model when saving intermediate messages, which requires for the progress callback to be initialized after the client is initialized * feat(useSSE): revert to previous model if the model was auto-switched by backend due to message attachments * docs: update with google updates, notes about Gemini Pro Vision * fix: redis should not be initialized without USE_REDIS and increase max listeners to 20
96 lines
3.1 KiB
JavaScript
96 lines
3.1 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const fileSchema = require('./schema/fileSchema');
|
|
|
|
const File = mongoose.model('File', fileSchema);
|
|
|
|
/**
|
|
* Finds a file by its file_id with additional query options.
|
|
* @param {string} file_id - The unique identifier of the file.
|
|
* @param {object} options - Query options for filtering, projection, etc.
|
|
* @returns {Promise<MongoFile>} A promise that resolves to the file document or null.
|
|
*/
|
|
const findFileById = async (file_id, options = {}) => {
|
|
return await File.findOne({ file_id, ...options }).lean();
|
|
};
|
|
|
|
/**
|
|
* Retrieves files matching a given filter.
|
|
* @param {Object} filter - The filter criteria to apply.
|
|
* @returns {Promise<Array<MongoFile>>} A promise that resolves to an array of file documents.
|
|
*/
|
|
const getFiles = async (filter) => {
|
|
return await File.find(filter).lean();
|
|
};
|
|
|
|
/**
|
|
* Creates a new file with a TTL of 1 hour.
|
|
* @param {MongoFile} data - The file data to be created, must contain file_id.
|
|
* @returns {Promise<MongoFile>} A promise that resolves to the created file document.
|
|
*/
|
|
const createFile = async (data) => {
|
|
const fileData = {
|
|
...data,
|
|
expiresAt: new Date(Date.now() + 3600 * 1000),
|
|
};
|
|
return await File.findOneAndUpdate({ file_id: data.file_id }, fileData, {
|
|
new: true,
|
|
upsert: true,
|
|
}).lean();
|
|
};
|
|
|
|
/**
|
|
* Updates a file identified by file_id with new data and removes the TTL.
|
|
* @param {MongoFile} data - The data to update, must contain file_id.
|
|
* @returns {Promise<MongoFile>} A promise that resolves to the updated file document.
|
|
*/
|
|
const updateFile = async (data) => {
|
|
const { file_id, ...update } = data;
|
|
const updateOperation = {
|
|
$set: update,
|
|
$unset: { expiresAt: '' }, // Remove the expiresAt field to prevent TTL
|
|
};
|
|
return await File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
|
|
};
|
|
|
|
/**
|
|
* Increments the usage of a file identified by file_id.
|
|
* @param {MongoFile} data - The data to update, must contain file_id and the increment value for usage.
|
|
* @returns {Promise<MongoFile>} A promise that resolves to the updated file document.
|
|
*/
|
|
const updateFileUsage = async (data) => {
|
|
const { file_id, inc = 1 } = data;
|
|
const updateOperation = {
|
|
$inc: { usage: inc },
|
|
$unset: { expiresAt: '' },
|
|
};
|
|
return await File.findOneAndUpdate({ file_id }, updateOperation, { new: true }).lean();
|
|
};
|
|
|
|
/**
|
|
* Deletes a file identified by file_id.
|
|
* @param {string} file_id - The unique identifier of the file to delete.
|
|
* @returns {Promise<MongoFile>} A promise that resolves to the deleted file document or null.
|
|
*/
|
|
const deleteFile = async (file_id) => {
|
|
return await File.findOneAndDelete({ file_id }).lean();
|
|
};
|
|
|
|
/**
|
|
* Deletes multiple files identified by an array of file_ids.
|
|
* @param {Array<string>} file_ids - The unique identifiers of the files to delete.
|
|
* @returns {Promise<Object>} A promise that resolves to the result of the deletion operation.
|
|
*/
|
|
const deleteFiles = async (file_ids) => {
|
|
return await File.deleteMany({ file_id: { $in: file_ids } });
|
|
};
|
|
|
|
module.exports = {
|
|
File,
|
|
findFileById,
|
|
getFiles,
|
|
createFile,
|
|
updateFile,
|
|
updateFileUsage,
|
|
deleteFile,
|
|
deleteFiles,
|
|
};
|