From 1a34927314b0135a938233f78f3eb037311c310e Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 21 Oct 2023 21:11:07 -0700 Subject: [PATCH] Make API URLs more visible --- extensions/api/blocking_api.py | 6 +++--- extensions/api/streaming_api.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/extensions/api/blocking_api.py b/extensions/api/blocking_api.py index 8d5850cf..ecc327aa 100644 --- a/extensions/api/blocking_api.py +++ b/extensions/api/blocking_api.py @@ -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() diff --git a/extensions/api/streaming_api.py b/extensions/api/streaming_api.py index 71113c2e..2968ed8d 100644 --- a/extensions/api/streaming_api.py +++ b/extensions/api/streaming_api.py @@ -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))