Skip to content

Commit 73a7ba5

Browse files
rohanthackerekzhu
andauthoredFeb 7, 2025··
Added the Claude family of models to ModelFamily (#5443)
Added the Claude family of models to the `ModelFamily` class. Co-authored-by: Eric Zhu <[email protected]>
1 parent 9f1c4c9 commit 73a7ba5

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed
 

‎python/packages/autogen-core/src/autogen_core/models/_model_client.py

+10
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,11 @@ class ModelFamily:
2727
GEMINI_1_5_FLASH = "gemini-1.5-flash"
2828
GEMINI_1_5_PRO = "gemini-1.5-pro"
2929
GEMINI_2_0_FLASH = "gemini-2.0-flash"
30+
CLAUDE_3_HAIKU = "claude-3-haiku"
31+
CLAUDE_3_SONNET = "claude-3-sonnet"
32+
CLAUDE_3_OPUS = "claude-3-opus"
33+
CLAUDE_3_5_HAIKU = "claude-3.5-haiku"
34+
CLAUDE_3_5_SONNET = "claude-3.5-sonnet"
3035
UNKNOWN = "unknown"
3136

3237
ANY: TypeAlias = Literal[
@@ -39,6 +44,11 @@ class ModelFamily:
3944
"gemini-1.5-flash",
4045
"gemini-1.5-pro",
4146
"gemini-2.0-flash",
47+
"claude-3-haiku",
48+
"claude-3-sonnet",
49+
"claude-3-opus",
50+
"claude-3.5-haiku",
51+
"claude-3.5-sonnet",
4252
"unknown",
4353
]
4454

0 commit comments

Comments
 (0)
Please sign in to comment.