From 2e344365896dce4e6c694456ca77ff91b5c19fa8 Mon Sep 17 00:00:00 2001 From: tacascer Date: Mon, 22 Apr 2024 20:25:32 -0400 Subject: [PATCH] chore: update GPT3.5 models --- pr_agent/algo/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 484ed0458..cab47378c 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -1,8 +1,9 @@ MAX_TOKENS = { 'text-embedding-ada-002': 8000, - 'gpt-3.5-turbo': 4000, + 'gpt-3.5-turbo': 16000, + 'gpt-3.5-turbo-0125': 16000, 'gpt-3.5-turbo-0613': 4000, - 'gpt-3.5-turbo-0301': 4000, + 'gpt-3.5-turbo-1106': 16000, 'gpt-3.5-turbo-16k': 16000, 'gpt-3.5-turbo-16k-0613': 16000, 'gpt-4': 8000,