mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 08:50:15 +01:00
🦙 refactor: Normalize Ollama Config Names (#4657)
This commit is contained in:
parent
c27b26cc31
commit
d60a0af878
3 changed files with 17 additions and 14 deletions
|
|
@ -1,6 +1,6 @@
|
||||||
const { EModelEndpoint, extractEnvVariable } = require('librechat-data-provider');
|
const { EModelEndpoint, extractEnvVariable } = require('librechat-data-provider');
|
||||||
|
const { isUserProvided, normalizeEndpointName } = require('~/server/utils');
|
||||||
const { getCustomConfig } = require('./getCustomConfig');
|
const { getCustomConfig } = require('./getCustomConfig');
|
||||||
const { isUserProvided } = require('~/server/utils');
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Load config endpoints from the cached configuration object
|
* Load config endpoints from the cached configuration object
|
||||||
|
|
@ -29,7 +29,8 @@ async function loadConfigEndpoints(req) {
|
||||||
|
|
||||||
for (let i = 0; i < customEndpoints.length; i++) {
|
for (let i = 0; i < customEndpoints.length; i++) {
|
||||||
const endpoint = customEndpoints[i];
|
const endpoint = customEndpoints[i];
|
||||||
const { baseURL, apiKey, name, iconURL, modelDisplayLabel } = endpoint;
|
const { baseURL, apiKey, name: configName, iconURL, modelDisplayLabel } = endpoint;
|
||||||
|
const name = normalizeEndpointName(configName);
|
||||||
|
|
||||||
const resolvedApiKey = extractEnvVariable(apiKey);
|
const resolvedApiKey = extractEnvVariable(apiKey);
|
||||||
const resolvedBaseURL = extractEnvVariable(baseURL);
|
const resolvedBaseURL = extractEnvVariable(baseURL);
|
||||||
|
|
|
||||||
|
|
@ -1,16 +1,7 @@
|
||||||
const { Providers } = require('@librechat/agents');
|
|
||||||
const { EModelEndpoint, extractEnvVariable } = require('librechat-data-provider');
|
const { EModelEndpoint, extractEnvVariable } = require('librechat-data-provider');
|
||||||
|
const { isUserProvided, normalizeEndpointName } = require('~/server/utils');
|
||||||
const { fetchModels } = require('~/server/services/ModelService');
|
const { fetchModels } = require('~/server/services/ModelService');
|
||||||
const { getCustomConfig } = require('./getCustomConfig');
|
const { getCustomConfig } = require('./getCustomConfig');
|
||||||
const { isUserProvided } = require('~/server/utils');
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param {string} name
|
|
||||||
* @returns {string}
|
|
||||||
*/
|
|
||||||
function normalizeEndpointName(name = '') {
|
|
||||||
return name.toLowerCase() === Providers.OLLAMA ? Providers.OLLAMA : name;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Load config endpoints from the cached configuration object
|
* Load config endpoints from the cached configuration object
|
||||||
|
|
|
||||||
|
|
@ -7,6 +7,7 @@ const {
|
||||||
defaultRetrievalModels,
|
defaultRetrievalModels,
|
||||||
defaultAssistantsVersion,
|
defaultAssistantsVersion,
|
||||||
} = require('librechat-data-provider');
|
} = require('librechat-data-provider');
|
||||||
|
const { Providers } = require('@librechat/agents');
|
||||||
const { getCitations, citeText } = require('./citations');
|
const { getCitations, citeText } = require('./citations');
|
||||||
const partialRight = require('lodash/partialRight');
|
const partialRight = require('lodash/partialRight');
|
||||||
const { sendMessage } = require('./streamResponse');
|
const { sendMessage } = require('./streamResponse');
|
||||||
|
|
@ -212,13 +213,23 @@ function generateConfig(key, baseURL, endpoint) {
|
||||||
return config;
|
return config;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Normalize the endpoint name to system-expected value.
|
||||||
|
* @param {string} name
|
||||||
|
* @returns {string}
|
||||||
|
*/
|
||||||
|
function normalizeEndpointName(name = '') {
|
||||||
|
return name.toLowerCase() === Providers.OLLAMA ? Providers.OLLAMA : name;
|
||||||
|
}
|
||||||
|
|
||||||
module.exports = {
|
module.exports = {
|
||||||
createOnProgress,
|
|
||||||
isEnabled,
|
isEnabled,
|
||||||
handleText,
|
handleText,
|
||||||
formatSteps,
|
formatSteps,
|
||||||
formatAction,
|
formatAction,
|
||||||
addSpaceIfNeeded,
|
|
||||||
isUserProvided,
|
isUserProvided,
|
||||||
generateConfig,
|
generateConfig,
|
||||||
|
addSpaceIfNeeded,
|
||||||
|
createOnProgress,
|
||||||
|
normalizeEndpointName,
|
||||||
};
|
};
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue