Skip to content

Commit

Permalink
pre-commit
Browse files Browse the repository at this point in the history
  • Loading branch information
dlqqq committed Nov 25, 2024
1 parent 7b9c7cf commit 5ca1696
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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]:
Expand Down

0 comments on commit 5ca1696

Please sign in to comment.