From aa12716974f9ff4eeaed2c012e39e9a06c2f311c Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Thu, 16 Nov 2023 18:54:59 +0100 Subject: Update MyShell.py --- g4f/Provider/MyShell.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'g4f/Provider') diff --git a/g4f/Provider/MyShell.py b/g4f/Provider/MyShell.py index dc3934fb..ac3290b5 100644 --- a/g4f/Provider/MyShell.py +++ b/g4f/Provider/MyShell.py @@ -65,6 +65,7 @@ window.reader = response.body.getReader(); driver.execute_script(script.replace("{body}", json.dumps(data))) script = """ chunk = await window.reader.read(); +if (chunk['done']) return null; text = await (new Response(chunk['value']).text()); content = ''; text.split('\\n').forEach((line, index) => { @@ -79,8 +80,12 @@ text.split('\\n').forEach((line, index) => { }); return content; """ - while chunk := driver.execute_script(script): - yield chunk + while True: + chunk = driver.execute_script(script): + if chunk: + yield chunk + elif chunk != "": + break finally: driver.close() if not browser: -- cgit v1.2.3