瀏覽代碼

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

takatost 1 年之前
父節點
當前提交
0cc0b6e052
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      api/core/third_party/spark/spark_llm.py

+ 4 - 1
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"]