Pārlūkot izejas kodu

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

takatost 1 gadu atpakaļ
vecāks
revīzija
0cc0b6e052
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  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"]