File size: 3,228 Bytes
f0743f4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | import { EModelEndpoint, getEndpointField } from 'librechat-data-provider';
import type { TEndpointsConfig, TConfig } from 'librechat-data-provider';
import { getAvailableEndpoints, getEndpointsFilter, mapEndpoints } from './endpoints';
const mockEndpointsConfig: TEndpointsConfig = {
[EModelEndpoint.openAI]: { type: undefined, iconURL: 'openAI_icon.png', order: 0 },
[EModelEndpoint.google]: { type: undefined, iconURL: 'google_icon.png', order: 1 },
Mistral: { type: EModelEndpoint.custom, iconURL: 'custom_icon.png', order: 2 },
};
describe('getEndpointField', () => {
it('returns undefined if endpointsConfig is undefined', () => {
expect(getEndpointField(undefined, EModelEndpoint.openAI, 'type')).toBeUndefined();
});
it('returns undefined if endpoint is null', () => {
expect(getEndpointField(mockEndpointsConfig, null, 'type')).toBeUndefined();
});
it('returns undefined if endpoint is undefined', () => {
expect(getEndpointField(mockEndpointsConfig, undefined, 'type')).toBeUndefined();
});
it('returns the correct value for a valid endpoint and property', () => {
expect(getEndpointField(mockEndpointsConfig, EModelEndpoint.openAI, 'order')).toEqual(0);
expect(getEndpointField(mockEndpointsConfig, EModelEndpoint.google, 'iconURL')).toEqual(
'google_icon.png',
);
});
it('returns undefined for a valid endpoint but an invalid property', () => {
/* Type assertion as 'nonexistentProperty' is intentionally not a valid property of TConfig */
expect(
getEndpointField(
mockEndpointsConfig,
EModelEndpoint.openAI,
'nonexistentProperty' as keyof TConfig,
),
).toBeUndefined();
});
it('returns the correct value for a non-enum endpoint and valid property', () => {
expect(getEndpointField(mockEndpointsConfig, 'Mistral', 'type')).toEqual(EModelEndpoint.custom);
});
it('returns undefined for a non-enum endpoint with an invalid property', () => {
expect(
getEndpointField(mockEndpointsConfig, 'Mistral', 'nonexistentProperty' as keyof TConfig),
).toBeUndefined();
});
});
describe('getEndpointsFilter', () => {
it('returns an empty object if endpointsConfig is undefined', () => {
expect(getEndpointsFilter(undefined)).toEqual({});
});
it('returns a filter object based on endpointsConfig', () => {
const expectedFilter = {
[EModelEndpoint.openAI]: true,
[EModelEndpoint.google]: true,
Mistral: true,
};
expect(getEndpointsFilter(mockEndpointsConfig)).toEqual(expectedFilter);
});
});
describe('getAvailableEndpoints', () => {
it('returns available endpoints based on filter and config', () => {
const filter = {
[EModelEndpoint.openAI]: true,
[EModelEndpoint.google]: false,
Mistral: true,
};
const expectedEndpoints = [EModelEndpoint.openAI, 'Mistral'];
expect(getAvailableEndpoints(filter, mockEndpointsConfig)).toEqual(expectedEndpoints);
});
});
describe('mapEndpoints', () => {
it('returns sorted available endpoints', () => {
const expectedOrder = [EModelEndpoint.openAI, EModelEndpoint.google, 'Mistral'];
expect(mapEndpoints(mockEndpointsConfig)).toEqual(expectedOrder);
});
});
|