codefuse-chatbot/examples/stop.py

36 lines
921 B
Python

import docker, sys, os
from loguru import logger
src_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
)
sys.path.append(src_dir)
from configs.server_config import (
SANDBOX_CONTRAINER_NAME, CONTRAINER_NAME, SANDBOX_SERVER, DOCKER_SERVICE
)
from start import check_docker, check_process
try:
client = docker.from_env()
except:
client = None
def stop_main():
#
check_docker(client, SANDBOX_CONTRAINER_NAME, do_stop=True, )
check_process(f"port={SANDBOX_SERVER['port']}", do_stop=True)
check_process(f"port=5050", do_stop=True)
#
check_docker(client, CONTRAINER_NAME, do_stop=True, )
check_process("api.py", do_stop=True)
check_process("sdfile_api.py", do_stop=True)
check_process("llm_api.py", do_stop=True)
check_process("webui.py", do_stop=True)
if __name__ == "__main__":
stop_main()