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

binary-husky · Jul 1, 2024 · 6cd2d80 · 6cd2d80
1 parent 18d3245
commit 6cd2d80
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"