From a20d2c08e468733303fbce5e1a96502ba0d8e77d Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Fri, 1 Dec 2023 16:11:41 -0800 Subject: [PATCH 1/4] Adds new parameters --- packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 4 ++++ 1 file changed, 4 insertions(+) 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" From 780e3e53315f65ffbd1d6e7d0f67a9f8bfdbb6b7 Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Wed, 6 Dec 2023 15:53:26 -0800 Subject: [PATCH 2/4] Updates LangChain to newest version --- packages/jupyter-ai-magics/pyproject.toml | 2 +- packages/jupyter-ai/pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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 0f8d4a58a..0bbbfbae1 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]", From f953a76d93a061a5b32fb1209d17e60fe48fb25a Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Fri, 8 Dec 2023 11:52:43 -0800 Subject: [PATCH 3/4] 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"] From 154fd2f6864ed9764539f4afcd67ccc97819433e Mon Sep 17 00:00:00 2001 From: Jason Weill Date: Fri, 8 Dec 2023 12:00:30 -0800 Subject: [PATCH 4/4] Moves llama2 chat models back to BedrockProvider --- 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 2329ee842..33d220f94 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -658,6 +658,8 @@ 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"] @@ -683,8 +685,6 @@ 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"]