Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-02-24 03:14:08 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

🚀 feat: Support for GPT-3.5 Turbo/0125 Model (#1704)

Browse source 
* 🚀 feat: Support for GPT-3.5 Turbo/0125 Model

* ci: fix tx test
This commit is contained in:
Danny Avila 2024-02-02 01:01:11 -05:00 • committed by GitHub
parent 30e143e96d
commit 8479ac7293
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 15 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/models/tx.js
View file

@ -12,6 +12,7 @@ const tokenValues = {
'16k': { prompt: 3, completion: 4 },
'gpt-3.5-turbo-1106': { prompt: 1, completion: 2 },
'gpt-4-1106': { prompt: 10, completion: 30 },
'gpt-3.5-turbo-0125': { prompt: 0.5, completion: 1.5 },
};
/**
@ -29,6 +30,8 @@ const getValueKey = (model, endpoint) => {
if (modelName.includes('gpt-3.5-turbo-16k')) {
return '16k';
} else if (modelName.includes('gpt-3.5-turbo-0125')) {
return 'gpt-3.5-turbo-0125';
} else if (modelName.includes('gpt-3.5-turbo-1106')) {
return 'gpt-3.5-turbo-1106';
} else if (modelName.includes('gpt-3.5')) {

3
api/models/tx.spec.js
View file

@ -90,6 +90,9 @@ describe('getMultiplier', () => {
expect(getMultiplier({ tokenType: 'completion', model: 'gpt-4-turbo-vision-preview' })).toBe(
tokenValues['gpt-4-1106'].completion,
);
expect(getMultiplier({ tokenType: 'completion', model: 'gpt-3.5-turbo-0125' })).toBe(
tokenValues['gpt-3.5-turbo-0125'].completion,
);
});
it('should return defaultRate if derived valueKey does not match any known patterns', () => {

1
api/utils/tokens.js
View file

@ -55,6 +55,7 @@ const openAIModels = {
'gpt-3.5-turbo-16k': 16375, // -10 from max
'gpt-3.5-turbo-16k-0613': 16375, // -10 from max
'gpt-3.5-turbo-1106': 16375, // -10 from max
'gpt-3.5-turbo-0125': 16375, // -10 from max
'mistral-': 31990, // -10 from max
};

3
api/utils/tokens.spec.js
View file

@ -92,6 +92,9 @@ describe('getModelMaxTokens', () => {
expect(getModelMaxTokens('gpt-4-0125-preview')).toBe(
maxTokensMap[EModelEndpoint.openAI]['gpt-4-0125'],
);
expect(getModelMaxTokens('gpt-3.5-turbo-0125')).toBe(
maxTokensMap[EModelEndpoint.openAI]['gpt-3.5-turbo-0125'],
);
});
test('should return correct tokens for Anthropic models', () => {