From e999f3c8f468ff514e49fe1bdb300e9ee73e38de Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Sat, 14 Oct 2023 13:46:00 -0700 Subject: [PATCH] check error --- discollama.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/discollama.py b/discollama.py index 70961ad..24b85d5 100644 --- a/discollama.py +++ b/discollama.py @@ -45,6 +45,9 @@ async def generate_response(prompt, context=[]): async def buffered_generate_response(prompt, context=[]): buffer = '' async for part in generate_response(prompt, context): + if error := part.get('error'): + raise Exception(error) + if part['done']: yield buffer, part break