gpt40 tokens update (#2717)

microsoft · Jul 30, 2024 · bcadd17 · bcadd17
1 parent 49b02c5
commit bcadd17
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
@@ -34,6 +34,8 @@ def get_max_token_limit(model: str = "gpt-3.5-turbo-0613") -> int:
         "gpt-4-0125-preview": 128000,
         "gpt-4-turbo-preview": 128000,
         "gpt-4-vision-preview": 128000,
+        "gpt-4o": 128000,
+        "gpt-4o-2024-05-13": 128000,
     }
     return max_token_limit[model]