diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index 9cbf61b13..5dd66c6d7 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -793,6 +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" @@ -821,6 +822,7 @@ 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" @@ -829,7 +831,7 @@ class BedrockChatProvider(BaseProvider, BedrockChat): "anthropic.claude-v2", "anthropic.claude-v2:1", "anthropic.claude-instant-v1", - "anthropic.claude-3-sonnet-20240229-v1:0" + "anthropic.claude-3-sonnet-20240229-v1:0", ] model_id_key = "model_id" pypi_package_deps = ["boto3"]