mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-18 09:20:15 +01:00
9288e84454
- Added support for Entra ID authentication in OpenAIClient and related services. - Updated header management to conditionally use Entra ID access tokens or API keys based on environment configuration. - Introduced utility functions for Entra ID token retrieval and credential management. - Enhanced tests to verify Entra ID authentication flow and its integration with Azure configurations.
58 lines
2.1 KiB
JavaScript
58 lines
2.1 KiB
JavaScript
const { isUserProvided, shouldUseEntraId } = require('@librechat/api');
|
|
const { EModelEndpoint } = require('librechat-data-provider');
|
|
const { generateConfig } = require('~/server/utils/handleText');
|
|
|
|
const {
|
|
OPENAI_API_KEY: openAIApiKey,
|
|
AZURE_ASSISTANTS_API_KEY: azureAssistantsApiKey,
|
|
ASSISTANTS_API_KEY: assistantsApiKey,
|
|
AZURE_API_KEY: azureOpenAIApiKey,
|
|
ANTHROPIC_API_KEY: anthropicApiKey,
|
|
CHATGPT_TOKEN: chatGPTToken,
|
|
PLUGINS_USE_AZURE,
|
|
GOOGLE_KEY: googleKey,
|
|
OPENAI_REVERSE_PROXY,
|
|
AZURE_OPENAI_BASEURL,
|
|
ASSISTANTS_BASE_URL,
|
|
AZURE_ASSISTANTS_BASE_URL,
|
|
} = process.env ?? {};
|
|
|
|
// Note: For Entra ID, we can't determine the actual token here since this is synchronous
|
|
// The actual token will be obtained in the initialize functions.
|
|
// Still we need to set a placeholder token to avoid errors.
|
|
const finalAzureOpenAIApiKey = shouldUseEntraId() ? 'entra-id-placeholder' : azureOpenAIApiKey;
|
|
|
|
const useAzurePlugins = !!PLUGINS_USE_AZURE;
|
|
|
|
const userProvidedOpenAI = useAzurePlugins
|
|
? isUserProvided(finalAzureOpenAIApiKey)
|
|
: isUserProvided(openAIApiKey);
|
|
|
|
module.exports = {
|
|
config: {
|
|
openAIApiKey,
|
|
azureOpenAIApiKey: finalAzureOpenAIApiKey,
|
|
useAzurePlugins,
|
|
userProvidedOpenAI,
|
|
googleKey,
|
|
[EModelEndpoint.anthropic]: generateConfig(anthropicApiKey),
|
|
[EModelEndpoint.chatGPTBrowser]: generateConfig(chatGPTToken),
|
|
[EModelEndpoint.openAI]: generateConfig(openAIApiKey, OPENAI_REVERSE_PROXY),
|
|
[EModelEndpoint.azureOpenAI]: generateConfig(finalAzureOpenAIApiKey, AZURE_OPENAI_BASEURL),
|
|
[EModelEndpoint.assistants]: generateConfig(
|
|
assistantsApiKey,
|
|
ASSISTANTS_BASE_URL,
|
|
EModelEndpoint.assistants,
|
|
),
|
|
[EModelEndpoint.azureAssistants]: generateConfig(
|
|
azureAssistantsApiKey,
|
|
AZURE_ASSISTANTS_BASE_URL,
|
|
EModelEndpoint.azureAssistants,
|
|
),
|
|
[EModelEndpoint.bedrock]: generateConfig(
|
|
process.env.BEDROCK_AWS_SECRET_ACCESS_KEY ?? process.env.BEDROCK_AWS_DEFAULT_REGION,
|
|
),
|
|
/* key will be part of separate config */
|
|
[EModelEndpoint.agents]: generateConfig('true', undefined, EModelEndpoint.agents),
|
|
},
|
|
};
|