Update providers.py (jupyterlab#314)

Marchlak · Oct 28, 2024 · 635ddc3 · 635ddc3
1 parent e932fe3
commit 635ddc3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -172,6 +172,7 @@ class AnthropicProvider(BaseProvider, Anthropic):
         "claude-v1",
         "claude-v1.0",
         "claude-v1.2",
+        "claude-2",
         "claude-instant-v1",
         "claude-instant-v1.0",
     ]
@@ -435,6 +436,7 @@ class BedrockProvider(BaseProvider, Bedrock):
         "amazon.titan-tg1-large",
         "anthropic.claude-v1",
         "anthropic.claude-instant-v1",
+        "anthropic.claude-v2",
         "ai21.j2-jumbo-instruct",
         "ai21.j2-grande-instruct",
     ]