diff --git a/metagpt/utils/cost_manager.py b/metagpt/utils/cost_manager.py index 4e6b65b2c..929d0b2e5 100644 --- a/metagpt/utils/cost_manager.py +++ b/metagpt/utils/cost_manager.py @@ -42,7 +42,7 @@ class CostManager(BaseModel): """ self.total_prompt_tokens += prompt_tokens self.total_completion_tokens += completion_tokens - if model not in TOKEN_COSTS: + if model not in self.token_costs: logger.warning(f"Model {model} not found in TOKEN_COSTS.") return diff --git a/requirements.txt b/requirements.txt index 9759248bb..4651cb258 100644 --- a/requirements.txt +++ b/requirements.txt @@ -67,5 +67,5 @@ playwright>=1.26 # used at metagpt/tools/libs/web_scraping.py anytree ipywidgets==8.1.1 Pillow -qianfan==0.3.1 -dashscope==1.14.1 \ No newline at end of file +qianfan==0.3.2 +dashscope==1.14.1