From bcadd173f66d40f8a101d5835e360de0c509bccb Mon Sep 17 00:00:00 2001
From: HRUSHIKESH DOKALA <96101829+Hk669@users.noreply.github.com>
Date: Sun, 19 May 2024 19:02:58 +0530
Subject: [PATCH] gpt40 tokens update (#2717)

---
 autogen/token_count_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
index 589d7b404a7d..b71dbc428a13 100644
--- 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]