From e7c7b44a0b45e40fd78c644576bbe23689ba3605 Mon Sep 17 00:00:00 2001 From: Wei-Jianan Date: Fri, 17 May 2024 18:44:05 +0800 Subject: [PATCH] [format] --- metagpt/provider/bedrock/bedrock_provider.py | 9 ++++++++- metagpt/provider/bedrock_api.py | 6 +++--- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/metagpt/provider/bedrock/bedrock_provider.py b/metagpt/provider/bedrock/bedrock_provider.py index 06a0029bd..1236bf56b 100644 --- a/metagpt/provider/bedrock/bedrock_provider.py +++ b/metagpt/provider/bedrock/bedrock_provider.py @@ -33,7 +33,14 @@ class AnthropicProvider(BaseBedrockProvider): def get_request_body(self, messages: list[dict], generate_kwargs, *args, **kwargs) -> str: system_message, user_messages = self._split_system_user_messages(messages) - body = json.dumps({"messages": user_messages, "anthropic_version": "bedrock-2023-05-31", "system": system_message, **generate_kwargs}) + body = json.dumps( + { + "messages": user_messages, + "anthropic_version": "bedrock-2023-05-31", + "system": system_message, + **generate_kwargs, + } + ) return body def _get_completion_from_dict(self, rsp_dict: dict) -> str: diff --git a/metagpt/provider/bedrock_api.py b/metagpt/provider/bedrock_api.py index b4a5f267f..f30d4701e 100644 --- a/metagpt/provider/bedrock_api.py +++ b/metagpt/provider/bedrock_api.py @@ -132,7 +132,7 @@ class BedrockLLM(BaseLLM): prompt_tokens = int(headers.get("x-amzn-bedrock-input-token-count", 0)) completion_tokens = int(headers.get("x-amzn-bedrock-output-token-count", 0)) usage = { - "prompt_tokens": prompt_tokens, - "completion_tokens": completion_tokens, - } + "prompt_tokens": prompt_tokens, + "completion_tokens": completion_tokens, + } return usage