RWKV-Runner/backend-python/utils/ngrok.py

14 lines
375 B
Python
Raw Permalink Normal View History

2023-05-07 17:27:54 +08:00
import os
2023-11-04 20:22:28 +08:00
import global_var
2023-05-07 17:27:54 +08:00
def ngrok_connect():
from pyngrok import ngrok, conf
2023-05-17 11:39:00 +08:00
2023-11-04 20:22:28 +08:00
conf.set_default(
conf.PyngrokConfig(ngrok_path="./ngrok.exe" if os.name == "nt" else "./ngrok")
)
2023-05-07 17:27:54 +08:00
ngrok.set_auth_token(os.environ["ngrok_token"])
2023-11-04 20:22:28 +08:00
http_tunnel = ngrok.connect(global_var.get(global_var.Args).port)
print(f"ngrok url: {http_tunnel.public_url}")