diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py index 66a6838c5..646f65dbf 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/aws.py @@ -51,7 +51,7 @@ async def _acall(self, *args, **kwargs) -> Coroutine[Any, Any, str]: class BedrockChatProvider(BaseProvider, ChatBedrock): id = "bedrock-chat" name = "Amazon Bedrock Chat" - + cri_models = [ # Anthropic models "anthropic.claude-3-haiku-20240307-v1:0", @@ -119,7 +119,7 @@ def __init__(self, *args, **kwargs): model_id = kwargs.pop("model_id") if model_id in self.cri_models: model_id = "us." + model_id - + super().__init__(*args, **kwargs, model_id=model_id) async def _acall(self, *args, **kwargs) -> Coroutine[Any, Any, str]: