mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 08:50:15 +01:00
0e719592c6
* feat: Add 'Run Code' and 'Temporary Chat' permissions to role management * feat: Add NextFunction typedef to api/typedefs.js * feat: Add temporary chat and run code permissions to role schema * refactor: Enhance access check middleware with logging for permission errors and better typing * refactor: Set default value of USE permission to true in multiConvoPermissionsSchema * refactor: Implement checkAccess function for separation of permission validation logic from middleware * feat: Integrate permission checks for tool execution and enhance Markdown code block with execution capability * fix: Convert REDIS_MAX_LISTENERS to a number, closes #5979
86 lines
2.2 KiB
JavaScript
86 lines
2.2 KiB
JavaScript
const fs = require('fs');
|
|
const ioredis = require('ioredis');
|
|
const KeyvRedis = require('@keyv/redis');
|
|
const { isEnabled } = require('~/server/utils');
|
|
const logger = require('~/config/winston');
|
|
|
|
const { REDIS_URI, USE_REDIS, USE_REDIS_CLUSTER, REDIS_CA, REDIS_KEY_PREFIX, REDIS_MAX_LISTENERS } =
|
|
process.env;
|
|
|
|
let keyvRedis;
|
|
const redis_prefix = REDIS_KEY_PREFIX || '';
|
|
const redis_max_listeners = Number(REDIS_MAX_LISTENERS) || 10;
|
|
|
|
function mapURI(uri) {
|
|
const regex =
|
|
/^(?:(?<scheme>\w+):\/\/)?(?:(?<user>[^:@]+)(?::(?<password>[^@]+))?@)?(?<host>[\w.-]+)(?::(?<port>\d{1,5}))?$/;
|
|
const match = uri.match(regex);
|
|
|
|
if (match) {
|
|
const { scheme, user, password, host, port } = match.groups;
|
|
|
|
return {
|
|
scheme: scheme || 'none',
|
|
user: user || null,
|
|
password: password || null,
|
|
host: host || null,
|
|
port: port || null,
|
|
};
|
|
} else {
|
|
const parts = uri.split(':');
|
|
if (parts.length === 2) {
|
|
return {
|
|
scheme: 'none',
|
|
user: null,
|
|
password: null,
|
|
host: parts[0],
|
|
port: parts[1],
|
|
};
|
|
}
|
|
|
|
return {
|
|
scheme: 'none',
|
|
user: null,
|
|
password: null,
|
|
host: uri,
|
|
port: null,
|
|
};
|
|
}
|
|
}
|
|
|
|
if (REDIS_URI && isEnabled(USE_REDIS)) {
|
|
let redisOptions = null;
|
|
let keyvOpts = {
|
|
useRedisSets: false,
|
|
keyPrefix: redis_prefix,
|
|
};
|
|
|
|
if (REDIS_CA) {
|
|
const ca = fs.readFileSync(REDIS_CA);
|
|
redisOptions = { tls: { ca } };
|
|
}
|
|
|
|
if (isEnabled(USE_REDIS_CLUSTER)) {
|
|
const hosts = REDIS_URI.split(',').map((item) => {
|
|
var value = mapURI(item);
|
|
|
|
return {
|
|
host: value.host,
|
|
port: value.port,
|
|
};
|
|
});
|
|
const cluster = new ioredis.Cluster(hosts, { redisOptions });
|
|
keyvRedis = new KeyvRedis(cluster, keyvOpts);
|
|
} else {
|
|
keyvRedis = new KeyvRedis(REDIS_URI, keyvOpts);
|
|
}
|
|
keyvRedis.on('error', (err) => logger.error('KeyvRedis connection error:', err));
|
|
keyvRedis.setMaxListeners(redis_max_listeners);
|
|
logger.info(
|
|
'[Optional] Redis initialized. Note: Redis support is experimental. If you have issues, disable it. Cache needs to be flushed for values to refresh.',
|
|
);
|
|
} else {
|
|
logger.info('[Optional] Redis not initialized. Note: Redis support is experimental.');
|
|
}
|
|
|
|
module.exports = keyvRedis;
|