소스 검색

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"]