From 5ca1696da6e39dea9b9d4d465f733ff0064b3c0c Mon Sep 17 00:00:00 2001 From: "David L. Qiu" Date: Mon, 25 Nov 2024 10:32:42 -0800 Subject: [PATCH] pre-commit --- .../jupyter_ai_magics/partner_providers/aws.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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]: