fix: error raise status code not exist (#888)

2023-08-17 15:33:35 +08:00
parent cd78adb0ab
commit 0cc0b6e052
1 changed files with 4 additions and 1 deletions
--- a/api/core/third_party/spark/spark_llm.py
+++ b/api/core/third_party/spark/spark_llm.py
@@ -102,7 +102,10 @@ class SparkLLMClient:
        data = json.loads(message)
        code = data['header']['code']
        if code != 0:
-            self.queue.put({'error': f"Code: {code}, Error: {data['header']['message']}"})
+            self.queue.put({
+                'status_code': 400,
+                'error': f"Code: {code}, Error: {data['header']['message']}"
+            })
            ws.close()
        else:
            choices = data["payload"]["choices"]