From 50a14718baeacfada5cf7008e2761a801adbd968 Mon Sep 17 00:00:00 2001 From: geekan Date: Wed, 7 Feb 2024 16:37:23 +0800 Subject: [PATCH 1/2] refine log --- metagpt/provider/openai_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 63e68c9bd..120748d15 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -236,7 +236,7 @@ class OpenAILLM(BaseLLM): usage.prompt_tokens = count_message_tokens(messages, self.model) usage.completion_tokens = count_string_tokens(rsp, self.model) except Exception as e: - logger.error(f"usage calculation failed: {e}") + logger.warning(f"usage calculation failed: {e}") return usage From ce63e455dfe1071a99ee421c1e17df07db20200d Mon Sep 17 00:00:00 2001 From: geekan Date: Wed, 7 Feb 2024 17:03:10 +0800 Subject: [PATCH 2/2] fix bug --- metagpt/provider/openai_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 120748d15..756f8c483 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -253,7 +253,7 @@ class OpenAILLM(BaseLLM): def _get_max_tokens(self, messages: list[dict]): if not self.auto_max_tokens: return self.config.max_token - return get_max_completion_tokens(messages, self.model, self.config.max_tokens) + return get_max_completion_tokens(messages, self.model, self.config.max_token) @handle_exception async def amoderation(self, content: Union[str, list[str]]):