Archives/gpt4free
2
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-19 03:25:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use asyncio subprocess in OpenaiChat

Browse Source
This commit is contained in:
Heiner Lohaus 2023-10-28 19:02:39 +02:00
parent dc04ca9306
commit cc301a3dd8
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
g4f/Provider/needs_auth/OpenaiChat.py
View File

@ -1,7 +1,7 @@
from __future__ import annotations from __future__ import annotations
import uuid, json, time, os import uuid, json, time, os
import tempfile, subprocess, shutil import tempfile, shutil, asyncio
from ..base_provider import AsyncGeneratorProvider from ..base_provider import AsyncGeneratorProvider
from ..helper import get_browser, get_cookies, format_prompt, get_event_loop from ..helper import get_browser, get_cookies, format_prompt, get_event_loop
@ -174,7 +174,14 @@ fun.getToken(config).then(token => {
tmp.write(source.encode()) tmp.write(source.encode())
tmp.close() tmp.close()
try: try:
p = subprocess.Popen([node, tmp.name], stdout=subprocess.PIPE) p = await asyncio.create_subprocess_exec(
return p.stdout.read().decode() node, tmp.name,
stderr=asyncio.subprocess.PIPE,
stdout=asyncio.subprocess.PIPE
)
stdout, stderr = await p.communicate()
if p.returncode == 0:
return stdout.decode()
raise RuntimeError(f"Exec Error: {stderr.decode()}")
finally: finally:
os.unlink(tmp.name) os.unlink(tmp.name)