mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 06:00:56 +02:00
bf0a84e45a
* chore: bump @librechat/agents to v1.9.8 for rscript support * chore: fix @langchain/google-genai dep., match agents * chore: fix @langchain/google-vertexai to v0.1.5, match with agents * chore: bump @librechat/agents to v1.9.9 * chore: update @librechat/agents to v1.9.91 and @langchain/google-vertexai to v0.1.6 * chore: increase MAX_FILE_SIZE to 150MB for file uploads * chore: bump @librechat/agents to v1.9.92 * feat: support `recursionLimit` for agents * chore: update configuration version to 1.2.1 in librechat.yaml and config.ts * feat: add R language SVG icon to the assets and include it in ApiKeyDialog * feat: add support for new vision model 'o1' and exclude 'o1-mini'
85 lines
3 KiB
JavaScript
85 lines
3 KiB
JavaScript
// Code Files
|
|
const axios = require('axios');
|
|
const FormData = require('form-data');
|
|
const { getCodeBaseURL } = require('@librechat/agents');
|
|
|
|
const MAX_FILE_SIZE = 150 * 1024 * 1024;
|
|
|
|
/**
|
|
* Retrieves a download stream for a specified file.
|
|
* @param {string} fileIdentifier - The identifier for the file (e.g., "session_id/fileId").
|
|
* @param {string} apiKey - The API key for authentication.
|
|
* @returns {Promise<AxiosResponse>} A promise that resolves to a readable stream of the file content.
|
|
* @throws {Error} If there's an error during the download process.
|
|
*/
|
|
async function getCodeOutputDownloadStream(fileIdentifier, apiKey) {
|
|
try {
|
|
const baseURL = getCodeBaseURL();
|
|
const response = await axios({
|
|
method: 'get',
|
|
url: `${baseURL}/download/${fileIdentifier}`,
|
|
responseType: 'stream',
|
|
headers: {
|
|
'User-Agent': 'LibreChat/1.0',
|
|
'X-API-Key': apiKey,
|
|
},
|
|
timeout: 15000,
|
|
});
|
|
|
|
return response;
|
|
} catch (error) {
|
|
throw new Error(`Error downloading file: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Uploads a file to the Code Environment server.
|
|
* @param {Object} params - The params object.
|
|
* @param {ServerRequest} params.req - The request object from Express. It should have a `user` property with an `id`
|
|
* representing the user, and an `app.locals.paths` object with an `uploads` path.
|
|
* @param {import('fs').ReadStream | import('stream').Readable} params.stream - The read stream for the file.
|
|
* @param {string} params.filename - The name of the file.
|
|
* @param {string} params.apiKey - The API key for authentication.
|
|
* @param {string} [params.entity_id] - Optional entity ID for the file.
|
|
* @returns {Promise<string>}
|
|
* @throws {Error} If there's an error during the upload process.
|
|
*/
|
|
async function uploadCodeEnvFile({ req, stream, filename, apiKey, entity_id = '' }) {
|
|
try {
|
|
const form = new FormData();
|
|
if (entity_id.length > 0) {
|
|
form.append('entity_id', entity_id);
|
|
}
|
|
form.append('file', stream, filename);
|
|
|
|
const baseURL = getCodeBaseURL();
|
|
const response = await axios.post(`${baseURL}/upload`, form, {
|
|
headers: {
|
|
...form.getHeaders(),
|
|
'Content-Type': 'multipart/form-data',
|
|
'User-Agent': 'LibreChat/1.0',
|
|
'User-Id': req.user.id,
|
|
'X-API-Key': apiKey,
|
|
},
|
|
maxContentLength: MAX_FILE_SIZE,
|
|
maxBodyLength: MAX_FILE_SIZE,
|
|
});
|
|
|
|
/** @type {{ message: string; session_id: string; files: Array<{ fileId: string; filename: string }> }} */
|
|
const result = response.data;
|
|
if (result.message !== 'success') {
|
|
throw new Error(`Error uploading file: ${result.message}`);
|
|
}
|
|
|
|
const fileIdentifier = `${result.session_id}/${result.files[0].fileId}`;
|
|
if (entity_id.length === 0) {
|
|
return fileIdentifier;
|
|
}
|
|
|
|
return `${fileIdentifier}?entity_id=${entity_id}`;
|
|
} catch (error) {
|
|
throw new Error(`Error uploading file: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
module.exports = { getCodeOutputDownloadStream, uploadCodeEnvFile };
|