wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

added: gemini 2.5 pro preview (#6226)

This commit is contained in:
Ardent Illumina 2025-04-06 03:27:56 +03:00 committed by GitHub
parent 0a314c17c7
commit b1ae4ac79e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/packages/autogen-core/src/autogen_core/models/_model_client.py
View File

@ -28,6 +28,7 @@ class ModelFamily:
GEMINI_1_5_FLASH = "gemini-1.5-flash"
GEMINI_1_5_PRO = "gemini-1.5-pro"
GEMINI_2_0_FLASH = "gemini-2.0-flash"
GEMINI_2_5_PRO = "gemini-2.5-pro"
CLAUDE_3_HAIKU = "claude-3-haiku"
CLAUDE_3_SONNET = "claude-3-sonnet"
CLAUDE_3_OPUS = "claude-3-opus"
@ -47,6 +48,7 @@ class ModelFamily:
"gemini-1.5-flash",
"gemini-1.5-pro",
"gemini-2.0-flash",
"gemini-2.5-pro",
"claude-3-haiku",
"claude-3-sonnet",
"claude-3-opus",
@ -76,6 +78,7 @@ class ModelFamily:
ModelFamily.GEMINI_1_5_FLASH,
ModelFamily.GEMINI_1_5_PRO,
ModelFamily.GEMINI_2_0_FLASH,
ModelFamily.GEMINI_2_5_PRO,
)
@staticmethod

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

@ -209,6 +209,13 @@ _MODEL_INFO: Dict[str, ModelInfo] = {
"family": ModelFamily.GEMINI_2_0_FLASH,
"structured_output": True,
},
"gemini-2.5-pro-preview-03-25": {
"vision": True,
"function_calling": True,
"json_output": True,
"family": ModelFamily.GEMINI_2_5_PRO,
"structured_output": True,
},
"claude-3-haiku-20240307": {
"vision": True,
"function_calling": True,
@ -279,6 +286,7 @@ _MODEL_TOKEN_LIMITS: Dict[str, int] = {
"gemini-1.5-pro": 2097152,
"gemini-2.0-flash": 1048576,
"gemini-2.0-flash-lite-preview-02-05": 1048576,
"gemini-2.5-pro-preview-03-25": 2097152,
"claude-3-haiku-20240307": 50000,
"claude-3-sonnet-20240229": 40000,
"claude-3-opus-20240229": 20000,