yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-27 23:18:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added the Claude family of models to ModelFamily (#5443)

Browse Source 
Added the Claude family of models to the `ModelFamily` class. 

Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
This commit is contained in:
Rohan Thacker 2025-02-08 04:57:14 +05:30 committed by GitHub
parent 9f1c4c924f
commit 73a7ba5764
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,6 +27,11 @@ 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"
CLAUDE_3_HAIKU = "claude-3-haiku"
CLAUDE_3_SONNET = "claude-3-sonnet"
CLAUDE_3_OPUS = "claude-3-opus"
CLAUDE_3_5_HAIKU = "claude-3.5-haiku"
CLAUDE_3_5_SONNET = "claude-3.5-sonnet"
UNKNOWN = "unknown"
ANY: TypeAlias = Literal[
@ -39,6 +44,11 @@ class ModelFamily:
"gemini-1.5-flash",
"gemini-1.5-pro",
"gemini-2.0-flash",
"claude-3-haiku",
"claude-3-sonnet",
"claude-3-opus",
"claude-3.5-haiku",
"claude-3.5-sonnet",
"unknown",
]