update qianfan pypi version

This commit is contained in:
betterwang 2024-02-23 12:45:06 +08:00
parent 41d6d2be1d
commit 655ba5647b
2 changed files with 3 additions and 3 deletions

View file

@ -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