apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-02 12:22:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

openai_api: refine logic

Browse source
This commit is contained in:
geekan 2023-12-19 11:52:23 +08:00
parent ad8f7ebfb9
commit 602818e2ba
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/provider/openai_api.py
View file

@ -329,7 +329,8 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
usage["completion_tokens"] = completion_tokens
return usage
except Exception as e:
logger.error("usage calculation failed!", e)
logger.error(f"{self.model} usage calculation failed!", e)
return {}
else:
return usage
@ -360,7 +361,7 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
return results
def _update_costs(self, usage: dict):
if CONFIG.calc_usage:
if CONFIG.calc_usage and usage:
try:
prompt_tokens = int(usage["prompt_tokens"])
completion_tokens = int(usage["completion_tokens"])