Enhance the robustness of the code (#1879)

### What problem does this PR solve?

Enhance the robustness of the code

### Type of change

- [x] Refactoring

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾
2024-08-09 10:18:08 +08:00
committed by GitHub
parent afccbc88e8
commit 411c645134
2 changed files with 2 additions and 2 deletions

View File

@ -72,7 +72,7 @@ class Base(ABC):
+ num_tokens_from_string(resp.choices[0].delta.content)
)
if not hasattr(resp, "usage") or not resp.usage
else resp.usage["total_tokens"]
else resp.usage.get("total_tokens",total_tokens)
)
if resp.choices[0].finish_reason == "length":
ans += "...\nFor the content length reason, it stopped, continue?" if is_english(