diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index 563c00c4f..f27831e87 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -437,7 +437,9 @@ 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 +448,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 +832,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"]