Bug fix: Some non-standard forms of error return are not caught (#1877)

2024-07-01 20:35:49 +08:00 · 2024-07-01 20:35:49 +08:00 · 6cd2d80dfd
parent 18d3245fc9
commit 6cd2d80dfd
1 changed files with 2 additions and 1 deletions
--- a/request_llms/oai_std_model_template.py
+++ b/request_llms/oai_std_model_template.py
@ -44,7 +44,8 @@ def decode_chunk(chunk):
    try:
        chunk = json.loads(chunk[6:])
    except:
-        finish_reason = "JSON_ERROR"
+        respose = "API_ERROR"
+        finish_reason = chunk
    # 错误处理部分
    if "error" in chunk:
        respose = "API_ERROR"