Merge branch 'main' into cachedclient

microsoft · Jan 9, 2025 · 97e8d66 · 97e8d66
2 parents 7addcf4 + 0446ce9
commit 97e8d66
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/python/packages/autogen-ext/src/autogen_ext/models/openai/_model_info.py b/python/packages/autogen-ext/src/autogen_ext/models/openai/_model_info.py
@@ -5,6 +5,7 @@
 # Based on: https://platform.openai.com/docs/models/continuous-model-upgrades
 # This is a moving target, so correctness is checked by the model value returned by openai against expected values at runtime``
 _MODEL_POINTERS = {
+    "o1": "o1-2024-12-17",
     "o1-preview": "o1-preview-2024-09-12",
     "o1-mini": "o1-mini-2024-09-12",
     "gpt-4o": "gpt-4o-2024-08-06",
@@ -18,6 +19,12 @@
 }
 
 _MODEL_INFO: Dict[str, ModelInfo] = {
+    "o1-2024-12-17": {
+        "vision": False,
+        "function_calling": False,
+        "json_output": False,
+        "family": ModelFamily.O1,
+    },
     "o1-preview-2024-09-12": {
         "vision": False,
         "function_calling": False,
@@ -117,6 +124,7 @@
 }
 
 _MODEL_TOKEN_LIMITS: Dict[str, int] = {
+    "o1-2024-12-17": 200000,
     "o1-preview-2024-09-12": 128000,
     "o1-mini-2024-09-12": 128000,
     "gpt-4o-2024-08-06": 128000,