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

* 🚀 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

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
};

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', () => {