mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 17:00:15 +01:00
5d40d0a37a
* chore: bump data-provider and add MESSAGES CacheKey * refactor: avoid saving messages while streaming, save partial text to cache instead * fix(ci): processChunks * chore: logging aborted request to debug * feat: set stream rate for token processing * chore: specify default stream rate * fix(ci): Update AppService.js to use optional chaining for endpointLocals assignment * refactor: abstract the error handler * feat: streamRate for assistants; refactor: update default rate for token * refactor: update error handling in assistants/errors.js * refactor: update error handling in assistants/errors.js
58 lines
2.4 KiB
JavaScript
58 lines
2.4 KiB
JavaScript
const {
|
|
Capabilities,
|
|
assistantEndpointSchema,
|
|
defaultAssistantsVersion,
|
|
} = require('librechat-data-provider');
|
|
const { logger } = require('~/config');
|
|
|
|
/**
|
|
* Sets up the minimum, default Assistants configuration if Azure OpenAI Assistants option is enabled.
|
|
* @returns {Partial<TAssistantEndpoint>} The Assistants endpoint configuration.
|
|
*/
|
|
function azureAssistantsDefaults() {
|
|
return {
|
|
capabilities: [Capabilities.tools, Capabilities.actions, Capabilities.code_interpreter],
|
|
version: defaultAssistantsVersion.azureAssistants,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Sets up the Assistants configuration from the config (`librechat.yaml`) file.
|
|
* @param {TCustomConfig} config - The loaded custom configuration.
|
|
* @param {EModelEndpoint.assistants|EModelEndpoint.azureAssistants} assistantsEndpoint - The Assistants endpoint name.
|
|
* - The previously loaded assistants configuration from Azure OpenAI Assistants option.
|
|
* @param {Partial<TAssistantEndpoint>} [prevConfig]
|
|
* @returns {Partial<TAssistantEndpoint>} The Assistants endpoint configuration.
|
|
*/
|
|
function assistantsConfigSetup(config, assistantsEndpoint, prevConfig = {}) {
|
|
const assistantsConfig = config.endpoints[assistantsEndpoint];
|
|
const parsedConfig = assistantEndpointSchema.parse(assistantsConfig);
|
|
if (assistantsConfig.supportedIds?.length && assistantsConfig.excludedIds?.length) {
|
|
logger.warn(
|
|
`Configuration conflict: The '${assistantsEndpoint}' endpoint has both 'supportedIds' and 'excludedIds' defined. The 'excludedIds' will be ignored.`,
|
|
);
|
|
}
|
|
if (
|
|
assistantsConfig.privateAssistants &&
|
|
(assistantsConfig.supportedIds?.length || assistantsConfig.excludedIds?.length)
|
|
) {
|
|
logger.warn(
|
|
`Configuration conflict: The '${assistantsEndpoint}' endpoint has both 'privateAssistants' and 'supportedIds' or 'excludedIds' defined. The 'supportedIds' and 'excludedIds' will be ignored.`,
|
|
);
|
|
}
|
|
|
|
return {
|
|
...prevConfig,
|
|
retrievalModels: parsedConfig.retrievalModels,
|
|
disableBuilder: parsedConfig.disableBuilder,
|
|
pollIntervalMs: parsedConfig.pollIntervalMs,
|
|
supportedIds: parsedConfig.supportedIds,
|
|
capabilities: parsedConfig.capabilities,
|
|
excludedIds: parsedConfig.excludedIds,
|
|
privateAssistants: parsedConfig.privateAssistants,
|
|
timeoutMs: parsedConfig.timeoutMs,
|
|
streamRate: parsedConfig.streamRate,
|
|
};
|
|
}
|
|
|
|
module.exports = { azureAssistantsDefaults, assistantsConfigSetup };
|