github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add o1-2024-12-17 model (#4965)

Browse Source 
Co-authored-by: Jack Gerrits <jackgerrits@users.noreply.github.com>
This commit is contained in:
Griffin Bassman
2025-01-09 14:05:20 -05:00
committed by GitHub
parent f3ed7ae147
commit 0446ce924f
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/packages/autogen-ext/src/autogen_ext/models/openai/_model_info.py
View File

@@ -5,6 +5,7 @@ from autogen_core.models import ModelFamily, ModelInfo
# 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_POINTERS = {
}
_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_INFO: Dict[str, ModelInfo] = {
}
_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,