From e337e3accb0568f78c0d65cf66c4e2b0677fe3b6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 5 Mar 2024 00:56:49 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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"]