Compare commits

...

1 Commits

Author SHA1 Message Date
John Wang
c0e6f30ee6 fix: error raise status code not exist 2023-08-17 15:33:11 +08:00

View File

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