yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-02 10:50:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add gemini 2.5 fash compatibility (#6574)

Browse Source 
<!-- Thank you for your contribution! Please review
https://microsoft.github.io/autogen/docs/Contribute before opening a
pull request. -->

<!-- Please add a reviewer to the assignee section when you create a PR.
If you don't have the access to it, we will shortly find a reviewer and
assign them to your PR. -->

## Why are these changes needed?

<!-- Please give a short summary of the change and the problem this
solves. -->

## Related issue number

<!-- For example: "Closes #1234" -->

## Checks

- [ ] I've included any doc changes needed for
<https://microsoft.github.io/autogen/>. See
<https://github.com/microsoft/autogen/blob/main/CONTRIBUTING.md> to
build and test documentation locally.
- [ ] I've added tests (if relevant) corresponding to the changes
introduced in this PR.
- [ ] I've made sure all auto checks have passed.
This commit is contained in:
Damien Menigaux 2025-05-21 21:38:03 +02:00 committed by GitHub
parent 1578cd955f
commit c6c8693b2c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 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

@ -31,6 +31,7 @@ class ModelFamily:
GEMINI_1_5_PRO = "gemini-1.5-pro"
GEMINI_2_0_FLASH = "gemini-2.0-flash"
GEMINI_2_5_PRO = "gemini-2.5-pro"
GEMINI_2_5_FLASH = "gemini-2.5-flash"
CLAUDE_3_HAIKU = "claude-3-haiku"
CLAUDE_3_SONNET = "claude-3-sonnet"
CLAUDE_3_OPUS = "claude-3-opus"
@ -64,6 +65,7 @@ class ModelFamily:
"gemini-1.5-pro",
"gemini-2.0-flash",
"gemini-2.5-pro",
"gemini-2.5-flash"
# anthropic_models
"claude-3-haiku",
"claude-3-sonnet",
@ -107,6 +109,7 @@ class ModelFamily:
ModelFamily.GEMINI_1_5_PRO,
ModelFamily.GEMINI_2_0_FLASH,
ModelFamily.GEMINI_2_5_PRO,
ModelFamily.GEMINI_2_5_FLASH,
)
@staticmethod

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

@ -308,6 +308,14 @@ _MODEL_INFO: Dict[str, ModelInfo] = {
"structured_output": True,
"multiple_system_messages": False,
},
"gemini-2.5-flash-preview-05-20": {
"vision": True,
"function_calling": True,
"json_output": True,
"family": ModelFamily.GEMINI_2_5_FLASH,
"structured_output": True,
"multiple_system_messages": False,
},
"claude-3-haiku-20240307": {
"vision": True,
"function_calling": True,
@ -422,6 +430,7 @@ _MODEL_TOKEN_LIMITS: Dict[str, int] = {
"gemini-2.0-flash": 1048576,
"gemini-2.0-flash-lite-preview-02-05": 1048576,
"gemini-2.5-pro-preview-03-25": 2097152,
"gemini-2.5-flash-preview-05-20": 1048576,
"claude-3-haiku-20240307": 50000,
"claude-3-sonnet-20240229": 40000,
"claude-3-opus-20240229": 20000,