Add Claude-3-Haiku

Added the Claude 3 Haiku model to Anthropic Chat providers and also to Bedrock Claude Chat providers
jupyterlab · Mar 20, 2024 · aff0425 · aff0425
1 parent 4814904
commit aff0425
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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"]