From f953a76d93a061a5b32fb1209d17e60fe48fb25a Mon Sep 17 00:00:00 2001
From: Jason Weill <jweill@amazon.com>
Date: Fri, 8 Dec 2023 11:52:43 -0800
Subject: [PATCH] Moves chat models to BedrockChat provider

---
 packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 33d220f94..2329ee842 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -658,8 +658,6 @@ class BedrockProvider(BaseProvider, Bedrock):
         "ai21.j2-mid-v1",
         "cohere.command-light-text-v14",
         "cohere.command-text-v14",
-        "meta.llama2-13b-chat-v1",
-        "meta.llama2-70b-chat-v1",
     ]
     model_id_key = "model_id"
     pypi_package_deps = ["boto3"]
@@ -685,6 +683,8 @@ class BedrockChatProvider(BaseProvider, BedrockChat):
         "anthropic.claude-v2",
         "anthropic.claude-v2:1",
         "anthropic.claude-instant-v1",
+        "meta.llama2-13b-chat-v1",
+        "meta.llama2-70b-chat-v1",
     ]
     model_id_key = "model_id"
     pypi_package_deps = ["boto3"]