Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-23 20:00:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add conversation ID support to custom endpoint headers

Browse source 
- Add LIBRECHAT_CONVERSATION_ID to customUserVars when provided
- Pass conversation ID to header resolution for dynamic headers
- Add comprehensive test coverage

Enables custom endpoints to access conversation context using {{LIBRECHAT_CONVERSATION_ID}} placeholder.
This commit is contained in:
Gopal Sharma 2025-07-22 20:32:05 +05:30 • committed by Gopal Sharma
parent da3730b7d6
commit a8babbcebf
2 changed files with 18 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/server/services/Endpoints/custom/initialize.js
View file

@ -28,7 +28,10 @@ const initializeClient = async ({ req, res, endpointOption, optionsOnly, overrid
const CUSTOM_API_KEY = extractEnvVariable(endpointConfig.apiKey);
const CUSTOM_BASE_URL = extractEnvVariable(endpointConfig.baseURL);
let resolvedHeaders = resolveHeaders(endpointConfig.headers, req.user);
const customUserVars = {};
customUserVars.LIBRECHAT_CONVERSATION_ID = req.body.conversationId;
let resolvedHeaders = resolveHeaders(endpointConfig.headers, req.user, customUserVars);
if (CUSTOM_API_KEY.match(envVarRegex)) {
throw new Error(`Missing API Key for ${endpoint}.`);

14
api/server/services/Endpoints/custom/initialize.spec.js
View file

@ -64,6 +64,20 @@ describe('custom/initializeClient', () => {
jest.clearAllMocks();
});
it('calls resolveHeaders with conversation ID when provided', async () => {
const { resolveHeaders } = require('@librechat/api');
const requestWithConversationId = {
...mockRequest,
body: { ...mockRequest.body, conversationId: 'existing-conversation-123' },
};
await initializeClient({ req: requestWithConversationId, res: mockResponse, optionsOnly: true });
expect(resolveHeaders).toHaveBeenCalledWith(
{ 'x-user': '{{LIBRECHAT_USER_ID}}', 'x-email': '{{LIBRECHAT_USER_EMAIL}}' },
{ id: 'user-123', email: 'test@example.com' },
{ LIBRECHAT_CONVERSATION_ID: 'existing-conversation-123' },
);
});
it('calls resolveHeaders with headers and user', async () => {
const { resolveHeaders } = require('@librechat/api');
await initializeClient({ req: mockRequest, res: mockResponse, optionsOnly: true });