pt
/
gpt4free
mirror of https://github.com/xtekky/gpt4free.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update MyShell.py

This commit is contained in:
H Lohaus 2023-11-16 18:54:59 +01:00 committed by GitHub
parent 5c5235671e
commit aa12716974
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
g4f/Provider/MyShell.py
View File

@ -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: