From 1d078eb5b660662c677a7d155c6aecd5d4d58418 Mon Sep 17 00:00:00 2001
From: Jason Weill <93281816+JasonWeill@users.noreply.github.com>
Date: Fri, 8 Dec 2023 12:13:23 -0800
Subject: [PATCH] Backport PR #499: Adds new models to Bedrock provider

---
 packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 4 ++++
 packages/jupyter-ai-magics/pyproject.toml                 | 2 +-
 packages/jupyter-ai/pyproject.toml                        | 2 +-
 3 files changed, 6 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 8155c2f2b..33d220f94 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -656,7 +656,10 @@ class BedrockProvider(BaseProvider, Bedrock):
         "amazon.titan-text-express-v1",
         "ai21.j2-ultra-v1",
         "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"]
@@ -680,6 +683,7 @@ class BedrockChatProvider(BaseProvider, BedrockChat):
     models = [
         "anthropic.claude-v1",
         "anthropic.claude-v2",
+        "anthropic.claude-v2:1",
         "anthropic.claude-instant-v1",
     ]
     model_id_key = "model_id"
diff --git a/packages/jupyter-ai-magics/pyproject.toml b/packages/jupyter-ai-magics/pyproject.toml
index bb7a6dc57..aa9e76916 100644
--- a/packages/jupyter-ai-magics/pyproject.toml
+++ b/packages/jupyter-ai-magics/pyproject.toml
@@ -23,7 +23,7 @@ dynamic = ["version", "description", "authors", "urls", "keywords"]
 dependencies = [
     "ipython",
     "importlib_metadata>=5.2.0",
-    "langchain==0.0.318",
+    "langchain==0.0.346",
     "typing_extensions>=4.5.0",
     "click~=8.0",
     "jsonpath-ng>=1.5.3,<2",
diff --git a/packages/jupyter-ai/pyproject.toml b/packages/jupyter-ai/pyproject.toml
index 6c150eba4..2492fd55c 100644
--- a/packages/jupyter-ai/pyproject.toml
+++ b/packages/jupyter-ai/pyproject.toml
@@ -27,7 +27,7 @@ dependencies = [
     "openai~=0.26",
     "aiosqlite>=0.18",
     "importlib_metadata>=5.2.0",
-    "langchain==0.0.318",
+    "langchain==0.0.346",
     "tiktoken", # required for OpenAIEmbeddings
     "jupyter_ai_magics",
     "dask[distributed]",