fix: better error message for stream (#11635)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
pull/11669/head
yihong 1 year ago committed by GitHub
parent b90f1581be
commit 7e154a467b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -478,6 +478,10 @@ class OAIAPICompatLargeLanguageModel(_CommonOaiApiCompat, LargeLanguageModel):
usage=usage,
)
break
# handle the error here. for issue #11629
if chunk_json.get("error") and chunk_json.get("choices") is None:
raise ValueError(chunk_json.get("error"))
if chunk_json:
if u := chunk_json.get("usage"):
usage = u

Loading…
Cancel
Save