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 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]",