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

0cc0b6e0 · takatost · GitHub · cd78adb0 · 0cc0b6e0
Unverified Commit 0cc0b6e0 authored Aug 17, 2023 by takatost Committed by GitHub Aug 17, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

spark_llm.py api/core/third_party/spark/spark_llm.py +4 -1

No files found.
--- 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"]