From b58bd9353f7e6778b06442fe69ed960ae5c0b590 Mon Sep 17 00:00:00 2001 From: Burkhard Reffeling <307162+hardchor@users.noreply.github.com> Date: Sun, 24 Nov 2024 12:54:46 +0000 Subject: [PATCH] Add support for gpt-4o model version 2024-11-20 in token limits --- autogen/oai/openai_utils.py | 1 + autogen/token_count_utils.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/autogen/oai/openai_utils.py b/autogen/oai/openai_utils.py index 8dfaa90517..e26096199f 100644 --- 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 index 82fb4e08fd..56975a279b 100644 --- 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", }: