mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 00:40:14 +01:00
e16a6190a5
* feat: Update MCP package version and dependencies; refactor ToolContentPart type * refactor: Change module type to commonjs and update rollup configuration, remove unused dev dependency * refactor: Change async calls to synchronous for MCP and FlowStateManager retrieval * chore: Add eslint disable comment for i18next rule in DropdownPopup component * fix: improve statefulness of mcp servers selected if some were removed since last session * feat: implement conversation storage cleanup functions and integrate them into mutation success handlers * feat: enhance storage condition logic in useLocalStorageAlt to prevent unnecessary local storage writes * refactor: streamline local storage update logic in useLocalStorageAlt
92 lines
2.4 KiB
JavaScript
92 lines
2.4 KiB
JavaScript
const axios = require('axios');
|
|
const { EventSource } = require('eventsource');
|
|
const { Time, CacheKeys } = require('librechat-data-provider');
|
|
const { MCPManager, FlowStateManager } = require('librechat-mcp');
|
|
const logger = require('./winston');
|
|
|
|
global.EventSource = EventSource;
|
|
|
|
let mcpManager = null;
|
|
let flowManager = null;
|
|
|
|
/**
|
|
* @returns {MCPManager}
|
|
*/
|
|
function getMCPManager() {
|
|
if (!mcpManager) {
|
|
mcpManager = MCPManager.getInstance(logger);
|
|
}
|
|
return mcpManager;
|
|
}
|
|
|
|
/**
|
|
* @param {(key: string) => Keyv} getLogStores
|
|
* @returns {FlowStateManager}
|
|
*/
|
|
function getFlowStateManager(getLogStores) {
|
|
if (!flowManager) {
|
|
flowManager = new FlowStateManager(getLogStores(CacheKeys.FLOWS), {
|
|
ttl: Time.ONE_MINUTE * 3,
|
|
logger,
|
|
});
|
|
}
|
|
return flowManager;
|
|
}
|
|
|
|
/**
|
|
* Sends message data in Server Sent Events format.
|
|
* @param {ServerResponse} res - The server response.
|
|
* @param {{ data: string | Record<string, unknown>, event?: string }} event - The message event.
|
|
* @param {string} event.event - The type of event.
|
|
* @param {string} event.data - The message to be sent.
|
|
*/
|
|
const sendEvent = (res, event) => {
|
|
if (typeof event.data === 'string' && event.data.length === 0) {
|
|
return;
|
|
}
|
|
res.write(`event: message\ndata: ${JSON.stringify(event)}\n\n`);
|
|
};
|
|
|
|
/**
|
|
* Creates and configures an Axios instance with optional proxy settings.
|
|
*
|
|
* @typedef {import('axios').AxiosInstance} AxiosInstance
|
|
* @typedef {import('axios').AxiosProxyConfig} AxiosProxyConfig
|
|
*
|
|
* @returns {AxiosInstance} A configured Axios instance
|
|
* @throws {Error} If there's an issue creating the Axios instance or parsing the proxy URL
|
|
*/
|
|
function createAxiosInstance() {
|
|
const instance = axios.create();
|
|
|
|
if (process.env.proxy) {
|
|
try {
|
|
const url = new URL(process.env.proxy);
|
|
|
|
/** @type {AxiosProxyConfig} */
|
|
const proxyConfig = {
|
|
host: url.hostname.replace(/^\[|\]$/g, ''),
|
|
protocol: url.protocol.replace(':', ''),
|
|
};
|
|
|
|
if (url.port) {
|
|
proxyConfig.port = parseInt(url.port, 10);
|
|
}
|
|
|
|
instance.defaults.proxy = proxyConfig;
|
|
} catch (error) {
|
|
console.error('Error parsing proxy URL:', error);
|
|
throw new Error(`Invalid proxy URL: ${process.env.proxy}`);
|
|
}
|
|
}
|
|
|
|
return instance;
|
|
}
|
|
|
|
module.exports = {
|
|
logger,
|
|
sendEvent,
|
|
getMCPManager,
|
|
createAxiosInstance,
|
|
getFlowStateManager,
|
|
};
|