From 2ed7efc80e0e5b9f1ee6725696b9eda177bd073c Mon Sep 17 00:00:00 2001 From: Sanjiv Das Date: Mon, 4 Mar 2024 16:39:33 -0800 Subject: [PATCH] update anthropic models --- packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index eda848b87..9cbf61b13 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -793,7 +793,7 @@ def __init__(self, *args, **kwargs): async def _acall(self, *args, **kwargs) -> Coroutine[Any, Any, str]: return await self._call_in_executor(*args, **kwargs) - +# See model ID list here: https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html class BedrockProvider(BaseProvider, Bedrock): id = "bedrock" name = "Amazon Bedrock" @@ -821,15 +821,15 @@ class BedrockProvider(BaseProvider, Bedrock): async def _acall(self, *args, **kwargs) -> Coroutine[Any, Any, str]: return await self._call_in_executor(*args, **kwargs) - +# See model ID list here: https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html class BedrockChatProvider(BaseProvider, BedrockChat): id = "bedrock-chat" name = "Amazon Bedrock Chat" models = [ - "anthropic.claude-v1", "anthropic.claude-v2", "anthropic.claude-v2:1", "anthropic.claude-instant-v1", + "anthropic.claude-3-sonnet-20240229-v1:0" ] model_id_key = "model_id" pypi_package_deps = ["boto3"]