apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

reformat

Browse source
This commit is contained in:
cotran 2024-10-24 15:29:22 -07:00
parent 2a4b5ec9fb
commit 4b012ffd8c
2 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
model_server/app/cli.py
View file

@ -98,12 +98,11 @@ def check_and_install_lsof():
print(f"Failed to install lsof: {install_error}")
def stop_server(port=51000, wait=True, timeout=10):
def kill_process(port=51000, wait=True, timeout=10):
"""Stop the running Uvicorn server."""
log.info("Stopping model server")
try:
# Run the function to check and install lsof if necessary
check_and_install_lsof()
# Step 1: Run lsof command to get the process using the port
lsof_command = f"lsof -n | grep {port} | grep -i LISTEN"
result = subprocess.run(
@ -156,6 +155,11 @@ def stop_server(port=51000, wait=True, timeout=10):
print(f"Error occurred: {e}")
def stop_server(port=51000, wait=True, timeout=10):
check_and_install_lsof()
kill_process(port, wait, timeout)
def restart_server(port=51000):
"""Restart the Uvicorn server."""
stop_server(port)