1 files changed, 7 insertions, 2 deletions
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: