Added new models to token_count_utils (#1511)

* Added new models to token_count_utils * Added token costs for new models.
microsoft · Feb 2, 2024 · b0c1f8d · b0c1f8d
1 parent 9cd458e
commit b0c1f8d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/autogen/oai/openai_utils.py b/autogen/oai/openai_utils.py
@@ -46,6 +46,8 @@
     "gpt-3.5-turbo-1106": (0.001, 0.002),
     "gpt-35-turbo-1106": (0.001, 0.002),
     "gpt-4-1106-preview": (0.01, 0.03),
+    "gpt-4-0125-preview": (0.01, 0.03),
+    "gpt-4-turbo-preview": (0.01, 0.03),
     "gpt-4-1106-vision-preview": (0.01, 0.03),  # TODO: support vision pricing of images
 }
 

diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
@@ -28,6 +28,8 @@ def get_max_token_limit(model="gpt-3.5-turbo-0613"):
         "gpt-4-0613": 8192,
         "gpt-4-32k-0613": 32768,
         "gpt-4-1106-preview": 128000,
+        "gpt-4-0125-preview": 128000,
+        "gpt-4-turbo-preview": 128000,
         "gpt-4-vision-preview": 128000,
     }
     return max_token_limit[model]