From 73a7ba5764d2c3b25247b19ec6339887eb853130 Mon Sep 17 00:00:00 2001 From: Rohan Thacker Date: Sat, 8 Feb 2025 04:57:14 +0530 Subject: [PATCH] Added the Claude family of models to ModelFamily (#5443) Added the Claude family of models to the `ModelFamily` class. Co-authored-by: Eric Zhu --- .../src/autogen_core/models/_model_client.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/packages/autogen-core/src/autogen_core/models/_model_client.py b/python/packages/autogen-core/src/autogen_core/models/_model_client.py index e3ff0403fb88..4a046a3cf5f9 100644 --- a/python/packages/autogen-core/src/autogen_core/models/_model_client.py +++ b/python/packages/autogen-core/src/autogen_core/models/_model_client.py @@ -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", ]