add back anthropic models removed in rebase

srdas · Oct 28, 2024 · 63dff52 · 63dff52
1 parent c5fdaa1
commit 63dff52
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/anthropic.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/anthropic.py
@@ -49,6 +49,7 @@ class ChatAnthropicProvider(
         "claude-3-sonnet-20240229",
         "claude-3-haiku-20240307",
         "claude-3-5-sonnet-20240620",
+        "claude-3-5-sonnet-20241022",
     ]
     model_id_key = "model"
     pypi_package_deps = ["anthropic"]

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py
@@ -69,6 +69,7 @@ class BedrockChatProvider(BaseProvider, ChatBedrock):
         "anthropic.claude-3-haiku-20240307-v1:0",
         "anthropic.claude-3-opus-20240229-v1:0",
         "anthropic.claude-3-5-sonnet-20240620-v1:0",
+        "anthropic.claude-3-5-sonnet-20241022-v2:0",
         "meta.llama2-13b-chat-v1",
         "meta.llama2-70b-chat-v1",
         "meta.llama3-8b-instruct-v1:0",