mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 00:40:14 +01:00
🌐 feat: librechat.yaml from URL (#2064)
* feat: librechat.yaml from URL * doc update: librechat.yaml from URL * update dotenv.md - typo * Update loadCustomConfig.js * ci: specs for loadCustomConfig * fix(processFileURL): safe destructuring of saveURL result --------- Co-authored-by: fuegovic <fueg@live.ca> Co-authored-by: Fuegovic <32828263+fuegovic@users.noreply.github.com>
This commit is contained in:
parent
f5a754c8be
commit
ebcca16b94
5 changed files with 201 additions and 17 deletions
153
api/server/services/Config/loadCustomConfig.spec.js
Normal file
153
api/server/services/Config/loadCustomConfig.spec.js
Normal file
|
|
@ -0,0 +1,153 @@
|
|||
jest.mock('axios');
|
||||
jest.mock('~/cache/getLogStores');
|
||||
jest.mock('~/utils/loadYaml');
|
||||
|
||||
const axios = require('axios');
|
||||
const loadCustomConfig = require('./loadCustomConfig');
|
||||
const getLogStores = require('~/cache/getLogStores');
|
||||
const loadYaml = require('~/utils/loadYaml');
|
||||
const { logger } = require('~/config');
|
||||
|
||||
describe('loadCustomConfig', () => {
|
||||
const mockSet = jest.fn();
|
||||
const mockCache = { set: mockSet };
|
||||
|
||||
beforeEach(() => {
|
||||
jest.resetAllMocks();
|
||||
delete process.env.CONFIG_PATH;
|
||||
getLogStores.mockReturnValue(mockCache);
|
||||
});
|
||||
|
||||
it('should return null and log error if remote config fetch fails', async () => {
|
||||
process.env.CONFIG_PATH = 'http://example.com/config.yaml';
|
||||
axios.get.mockRejectedValue(new Error('Network error'));
|
||||
const result = await loadCustomConfig();
|
||||
expect(logger.error).toHaveBeenCalledTimes(1);
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should return null for an invalid local config file', async () => {
|
||||
process.env.CONFIG_PATH = 'localConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(null);
|
||||
const result = await loadCustomConfig();
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should parse, validate, and cache a valid local configuration', async () => {
|
||||
const mockConfig = {
|
||||
version: '1.0',
|
||||
cache: true,
|
||||
endpoints: {
|
||||
custom: [
|
||||
{
|
||||
name: 'mistral',
|
||||
apiKey: 'user_provided',
|
||||
baseURL: 'https://api.mistral.ai/v1',
|
||||
},
|
||||
],
|
||||
},
|
||||
};
|
||||
process.env.CONFIG_PATH = 'validConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(mockConfig);
|
||||
const result = await loadCustomConfig();
|
||||
|
||||
expect(result).toEqual(mockConfig);
|
||||
expect(mockSet).toHaveBeenCalledWith(expect.anything(), mockConfig);
|
||||
});
|
||||
|
||||
it('should return null and log if config schema validation fails', async () => {
|
||||
const invalidConfig = { invalidField: true };
|
||||
process.env.CONFIG_PATH = 'invalidConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(invalidConfig);
|
||||
|
||||
const result = await loadCustomConfig();
|
||||
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should handle and return null on YAML parse error for a string response from remote', async () => {
|
||||
process.env.CONFIG_PATH = 'http://example.com/config.yaml';
|
||||
axios.get.mockResolvedValue({ data: 'invalidYAMLContent' });
|
||||
|
||||
const result = await loadCustomConfig();
|
||||
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should return the custom config object for a valid remote config file', async () => {
|
||||
const mockConfig = {
|
||||
version: '1.0',
|
||||
cache: true,
|
||||
endpoints: {
|
||||
custom: [
|
||||
{
|
||||
name: 'mistral',
|
||||
apiKey: 'user_provided',
|
||||
baseURL: 'https://api.mistral.ai/v1',
|
||||
},
|
||||
],
|
||||
},
|
||||
};
|
||||
process.env.CONFIG_PATH = 'http://example.com/config.yaml';
|
||||
axios.get.mockResolvedValue({ data: mockConfig });
|
||||
const result = await loadCustomConfig();
|
||||
expect(result).toEqual(mockConfig);
|
||||
expect(mockSet).toHaveBeenCalledWith(expect.anything(), mockConfig);
|
||||
});
|
||||
|
||||
it('should return null if the remote config file is not found', async () => {
|
||||
process.env.CONFIG_PATH = 'http://example.com/config.yaml';
|
||||
axios.get.mockRejectedValue({ response: { status: 404 } });
|
||||
const result = await loadCustomConfig();
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should return null if the local config file is not found', async () => {
|
||||
process.env.CONFIG_PATH = 'nonExistentConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(null);
|
||||
const result = await loadCustomConfig();
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
|
||||
it('should not cache the config if cache is set to false', async () => {
|
||||
const mockConfig = {
|
||||
version: '1.0',
|
||||
cache: false,
|
||||
endpoints: {
|
||||
custom: [
|
||||
{
|
||||
name: 'mistral',
|
||||
apiKey: 'user_provided',
|
||||
baseURL: 'https://api.mistral.ai/v1',
|
||||
},
|
||||
],
|
||||
},
|
||||
};
|
||||
process.env.CONFIG_PATH = 'validConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(mockConfig);
|
||||
await loadCustomConfig();
|
||||
expect(mockSet).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('should log the loaded custom config', async () => {
|
||||
const mockConfig = {
|
||||
version: '1.0',
|
||||
cache: true,
|
||||
endpoints: {
|
||||
custom: [
|
||||
{
|
||||
name: 'mistral',
|
||||
apiKey: 'user_provided',
|
||||
baseURL: 'https://api.mistral.ai/v1',
|
||||
},
|
||||
],
|
||||
},
|
||||
};
|
||||
process.env.CONFIG_PATH = 'validConfig.yaml';
|
||||
loadYaml.mockReturnValueOnce(mockConfig);
|
||||
await loadCustomConfig();
|
||||
expect(logger.info).toHaveBeenCalledWith('Custom config file loaded:');
|
||||
expect(logger.info).toHaveBeenCalledWith(JSON.stringify(mockConfig, null, 2));
|
||||
expect(logger.debug).toHaveBeenCalledWith('Custom config:', mockConfig);
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue