Merge pull request #81 from hardchor/add-gpt-4o-2024-11-20

Add support for gpt-4o model version 2024-11-20
ag2ai · Nov 24, 2024 · 1946d0d · 1946d0d
2 parents b225872 + 639ea33
commit 1946d0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/autogen/oai/openai_utils.py b/autogen/oai/openai_utils.py
@@ -35,6 +35,7 @@
     "gpt-4o": (0.005, 0.015),
     "gpt-4o-2024-05-13": (0.005, 0.015),
     "gpt-4o-2024-08-06": (0.0025, 0.01),
+    "gpt-4o-2024-11-20": (0.0025, 0.01),
     # gpt-4-turbo
     "gpt-4-turbo-2024-04-09": (0.01, 0.03),
     # gpt-4

diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
@@ -56,6 +56,7 @@ def get_max_token_limit(model: str = "gpt-3.5-turbo-0613") -> int:
         "gpt-4o": 128000,
         "gpt-4o-2024-05-13": 128000,
         "gpt-4o-2024-08-06": 128000,
+        "gpt-4o-2024-11-20": 128000,
         "gpt-4o-mini": 128000,
         "gpt-4o-mini-2024-07-18": 128000,
     }
@@ -131,6 +132,7 @@ def _num_token_from_messages(messages: Union[List, Dict], model="gpt-3.5-turbo-0
         "gpt-4o",
         "gpt-4o-2024-05-13",
         "gpt-4o-2024-08-06",
+        "gpt-4o-2024-11-20",
         "gpt-4o-mini",
         "gpt-4o-mini-2024-07-18",
     }: