Ails throw the contents of an error response

pull/834/head
zengrr 1 year ago
parent 6a4463889e
commit 1ed448e584

@ -72,6 +72,8 @@ class Ails(BaseProvider):
if b"content" in token: if b"content" in token:
completion_chunk = json.loads(token.decode().replace("data: ", "")) completion_chunk = json.loads(token.decode().replace("data: ", ""))
token = completion_chunk["choices"][0]["delta"].get("content") token = completion_chunk["choices"][0]["delta"].get("content")
if "ai.ls".lower() in token.lower():
raise Exception("Response Error: " + token)
if token != None: if token != None:
yield token yield token

Loading…
Cancel
Save