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