Backport PR #1069: Merge Anthropic language model providers (#1076)

Co-authored-by: Sanjiv Das <[email protected]>
jupyterlab · Oct 31, 2024 · 44b3071 · 44b3071
1 parent dee8cea
commit 44b3071
Showing 1 changed file with 1 addition and 4 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
@@ -10,8 +10,6 @@ class AnthropicProvider(BaseProvider, AnthropicLLM):
         "claude-v1",
         "claude-v1.0",
         "claude-v1.2",
-        "claude-2",
-        "claude-2.0",
         "claude-instant-v1",
         "claude-instant-v1.0",
         "claude-instant-v1.2",
@@ -40,11 +38,10 @@ class ChatAnthropicProvider(
     BaseProvider, ChatAnthropic
 ):  # https://docs.anthropic.com/en/docs/about-claude/models
     id = "anthropic-chat"
-    name = "ChatAnthropic"
+    name = "Anthropic"
     models = [
         "claude-2.0",
         "claude-2.1",
-        "claude-instant-1.2",
         "claude-3-opus-20240229",
         "claude-3-sonnet-20240229",
         "claude-3-haiku-20240307",