diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index 563c00c4f..5fb15ee57 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -437,7 +437,7 @@ def is_api_key_exc(cls, e: Exception): return False -class ChatAnthropicProvider(BaseProvider, ChatAnthropic): +class ChatAnthropicProvider(BaseProvider, ChatAnthropic): # https://docs.anthropic.com/claude/docs/models-overview id = "anthropic-chat" name = "ChatAnthropic" models = [ @@ -446,6 +446,7 @@ class ChatAnthropicProvider(BaseProvider, ChatAnthropic): "claude-instant-1.2", "claude-3-opus-20240229", "claude-3-sonnet-20240229", + "claude-3-haiku-20240307", ] model_id_key = "model" pypi_package_deps = ["anthropic"] @@ -829,6 +830,7 @@ class BedrockChatProvider(BaseProvider, BedrockChat): "anthropic.claude-v2:1", "anthropic.claude-instant-v1", "anthropic.claude-3-sonnet-20240229-v1:0", + "anthropic.claude-3-haiku-20240307-v1:0", ] model_id_key = "model_id" pypi_package_deps = ["boto3"]