From aad0857b39e9ad70ec8fd360c11caf96e9f15d2a Mon Sep 17 00:00:00 2001 From: Sanjiv Das Date: Mon, 29 Apr 2024 14:54:13 -0700 Subject: [PATCH] Backport PR #764: Updated LLM providers to Add New Bedrock Models IDs --- .../jupyter-ai-magics/jupyter_ai_magics/providers.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index 694007ded..3759000e9 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -627,12 +627,20 @@ class BedrockProvider(BaseProvider, Bedrock): name = "Amazon Bedrock" models = [ "amazon.titan-text-express-v1", + "amazon.titan-text-lite-v1", "ai21.j2-ultra-v1", "ai21.j2-mid-v1", "cohere.command-light-text-v14", "cohere.command-text-v14", + "cohere.command-r-v1:0", + "cohere.command-r-plus-v1:0", "meta.llama2-13b-chat-v1", "meta.llama2-70b-chat-v1", + "meta.llama3-8b-instruct-v1:0", + "meta.llama3-70b-instruct-v1:0", + "mistral.mistral-7b-instruct-v0:2", + "mistral.mixtral-8x7b-instruct-v0:1", + "mistral.mistral-large-2402-v1:0", ] model_id_key = "model_id" pypi_package_deps = ["boto3"] @@ -660,6 +668,7 @@ class BedrockChatProvider(BaseProvider, BedrockChat): "anthropic.claude-instant-v1", "anthropic.claude-3-sonnet-20240229-v1:0", "anthropic.claude-3-haiku-20240307-v1:0", + "anthropic.claude-3-opus-20240229-v1:0", ] model_id_key = "model_id" pypi_package_deps = ["boto3"]