Update token_count_utils.py (microsoft#2531)

* Update token_count_utils.py Update the token counts of new gpt models ref: https://platform.openai.com/docs/models/gpt-4-turbo-and-gpt-4 * format code with pre-commit --------- Co-authored-by: Chi Wang <[email protected]> Co-authored-by: Eric Zhu <[email protected]>
jayralencar · Apr 30, 2024 · 17a102d · 17a102d
1 parent 5b5bd78
commit 17a102d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
@@ -14,14 +14,17 @@ def get_max_token_limit(model: str = "gpt-3.5-turbo-0613") -> int:
     model = re.sub(r"^gpt4", "gpt-4", model)
 
     max_token_limit = {
-        "gpt-3.5-turbo": 4096,
+        "gpt-3.5-turbo": 16385,
+        "gpt-3.5-turbo-0125": 16385,
         "gpt-3.5-turbo-0301": 4096,
         "gpt-3.5-turbo-0613": 4096,
         "gpt-3.5-turbo-instruct": 4096,
         "gpt-3.5-turbo-16k": 16385,
         "gpt-3.5-turbo-16k-0613": 16385,
         "gpt-3.5-turbo-1106": 16385,
         "gpt-4": 8192,
+        "gpt-4-turbo": 128000,
+        "gpt-4-turbo-2024-04-09": 128000,
         "gpt-4-32k": 32768,
         "gpt-4-32k-0314": 32768,  # deprecate in Sep
         "gpt-4-0314": 8192,  # deprecate in Sep