xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-12-26 20:54:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add exec_js helper def

Browse Source
This commit is contained in:
Heiner Lohaus 2023-11-03 22:24:35 +01:00
parent 65b5227500
commit dfefd22aa1
1 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -145,11 +145,6 @@ class OpenaiChat(AsyncGeneratorProvider):
return f"g4f.provider.{cls.__name__} supports: ({param})" return f"g4f.provider.{cls.__name__} supports: ({param})"
async def get_arkose_token(proxy: str = None) -> str: async def get_arkose_token(proxy: str = None) -> str:
node = shutil.which("node")
if not node:
if debug.logging:
print('OpenaiChat: "node" not found')
return
dir = os.path.dirname(os.path.dirname(__file__)) dir = os.path.dirname(os.path.dirname(__file__))
include = f'{dir}/npm/node_modules/funcaptcha' include = f'{dir}/npm/node_modules/funcaptcha'
config = { config = {
@ -175,23 +170,32 @@ fun.getToken(config).then(token => {
tmp.write(source.encode()) tmp.write(source.encode())
tmp.close() tmp.close()
try: try:
if sys.platform == 'win32': return await exec_js(tmp.name)
p = subprocess.Popen( finally:
[node, tmp.name], os.unlink(tmp.name)
stdout=subprocess.PIPE,
stderr=subprocess.PIPE async def exec_js(file: str) -> str:
) node = shutil.which("node")
if p.returncode == 0: if not node:
return p.stdout.read().decode() if debug.logging:
raise RuntimeError(f"Exec Error: {p.stderr.read().decode()}") print('OpenaiChat: "node" not found')
p = await asyncio.create_subprocess_exec( return
node, tmp.name, if sys.platform == 'win32':
stderr=asyncio.subprocess.PIPE, p = subprocess.Popen(
stdout=asyncio.subprocess.PIPE [node, file],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
) )
stdout, stderr = await p.communicate() stdout, stderr = p.communicate()
if p.returncode == 0: if p.returncode == 0:
return stdout.decode() return stdout.decode()
raise RuntimeError(f"Exec Error: {stderr.decode()}") raise RuntimeError(f"Exec Error: {stderr.decode()}")
finally: p = await asyncio.create_subprocess_exec(
os.unlink(tmp.name) node, file,
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()}")