Parcourir la source

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

tags/0.3.15
takatost il y a 2 ans
Parent
révision
0cc0b6e052
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 4 ajouts et 1 suppressions
  1. 4
    1
      api/core/third_party/spark/spark_llm.py

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

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

Chargement…
Annuler
Enregistrer