AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Make API URLs more visible

Browse Source
This commit is contained in:
oobabooga 2023-10-21 21:11:07 -07:00
parent 09f807af83
commit 1a34927314
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
extensions/api/blocking_api.py
View File

@ -212,7 +212,7 @@ def _run_server(port: int, share: bool = False, tunnel_id=str):
server.socket = context.wrap_socket(server.socket, server_side=True)
def on_start(public_url: str):
logger.info(f'Starting non-streaming server at public url {public_url}/api')
logger.info(f'Blocking API URL: \n\n{public_url}/api\n')
if share:
try:
@ -221,9 +221,9 @@ def _run_server(port: int, share: bool = False, tunnel_id=str):
pass
else:
if ssl_verify:
logger.info(f'Starting API at https://{address}:{port}/api')
logger.info(f'Blocking API URL: \n\nhttps://{address}:{port}/api\n')
else:
logger.info(f'Starting API at http://{address}:{port}/api')
logger.info(f'Blocking API URL: \n\nhttp://{address}:{port}/api\n')
server.serve_forever()

6
extensions/api/streaming_api.py
View File

@ -122,7 +122,7 @@ def _run_server(port: int, share: bool = False, tunnel_id=str):
def on_start(public_url: str):
public_url = public_url.replace('https://', 'wss://')
logger.info(f'Starting streaming server at public url {public_url}{PATH}')
logger.info(f'Streaming API URL: \n\n{public_url}{PATH}\n')
if share:
try:
@ -131,9 +131,9 @@ def _run_server(port: int, share: bool = False, tunnel_id=str):
print(e)
else:
if ssl_verify:
logger.info(f'Starting streaming server at wss://{address}:{port}{PATH}')
logger.info(f'Streaming API URL: \n\nwss://{address}:{port}{PATH}\n')
else:
logger.info(f'Starting streaming server at ws://{address}:{port}{PATH}')
logger.info(f'Streaming API URL: \n\nws://{address}:{port}{PATH}\n')
asyncio.run(_run(host=address, port=port))