From aa12716974f9ff4eeaed2c012e39e9a06c2f311c Mon Sep 17 00:00:00 2001
From: H Lohaus <hlohaus@users.noreply.github.com>
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