From 83c8ccb6b9733f61b799f3104c7d504c3f51fe9e Mon Sep 17 00:00:00 2001 From: usamimeri_renko <1710269958@qq.com> Date: Fri, 26 Apr 2024 17:15:24 +0800 Subject: [PATCH] update --- metagpt/provider/bedrock/amazon_bedrock_api.py | 5 ----- tests/metagpt/provider/test_amazon_bedrock_api.py | 8 ++------ 2 files changed, 2 insertions(+), 11 deletions(-) diff --git a/metagpt/provider/bedrock/amazon_bedrock_api.py b/metagpt/provider/bedrock/amazon_bedrock_api.py index 3d1b08f47..31862e2d5 100644 --- a/metagpt/provider/bedrock/amazon_bedrock_api.py +++ b/metagpt/provider/bedrock/amazon_bedrock_api.py @@ -17,11 +17,6 @@ except ImportError: @register_provider([LLMType.AMAZON_BEDROCK]) class AmazonBedrockLLM(BaseLLM): - """ - check out: - https://docs.aws.amazon.com/code-library/latest/ug/python_3_bedrock-runtime_code_examples.html - """ - def __init__(self, config: LLMConfig): self.config = config self.__client = self.__init_client("bedrock-runtime") diff --git a/tests/metagpt/provider/test_amazon_bedrock_api.py b/tests/metagpt/provider/test_amazon_bedrock_api.py index bf8e467e7..46d44e81b 100644 --- a/tests/metagpt/provider/test_amazon_bedrock_api.py +++ b/tests/metagpt/provider/test_amazon_bedrock_api.py @@ -31,7 +31,7 @@ def get_bedrock_request_body(model_id) -> dict: return BEDROCK_PROVIDER_REQUEST_BODY[provider] -def is_subset(subset, superset): +def is_subset(subset, superset) -> bool: """Ensure all fields in request body are allowed. ```python @@ -71,9 +71,6 @@ class TestAPI: provider = bedrock_api._get_provider() request_body = json.loads(provider.get_request_body( messages, **bedrock_api._generate_kwargs)) - print(get_bedrock_request_body( - bedrock_api.config.model)) - print(request_body) assert is_subset(request_body, get_bedrock_request_body( bedrock_api.config.model)) @@ -88,5 +85,4 @@ class TestAPI: mock_bedrock_provider_response) mocker.patch("metagpt.provider.bedrock.amazon_bedrock_api.AmazonBedrockLLM.invoke_model_with_response_stream", mock_bedrock_provider_stream_response) - assert bedrock_api._chat_completion_stream( - messages) == "Hello World" + assert bedrock_api._chat_completion_stream(messages) == "Hello World"