SuperAGI

1
import os
2
import sys
3
import subprocess
4
from time import sleep
5
import shutil
6
from sys import platform
7
from superagi.lib.logger import logger
8

9
def check_command(command, message):
10
    if not shutil.which(command):
11
        logger.info(message)
12
        sys.exit(1)
13

14

15
def run_npm_commands(shell=False):
16
    os.chdir("gui")
17
    try:
18
        subprocess.run(["npm", "install"], check=True,shell=shell)
19
    except subprocess.CalledProcessError:
20
        logger.error(f"Error during '{' '.join(sys.exc_info()[1].cmd)}'. Exiting.")
21
        sys.exit(1)
22
    os.chdir("..")
23

24

25
def run_server(shell=False):
26
    api_process = subprocess.Popen(["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"], shell=shell)
27
    # celery_process = None
28
    celery_process = subprocess.Popen(["celery", "-A", "superagi.worker", "worker", "--loglevel=info"], shell=shell)
29
    os.chdir("gui")
30
    ui_process = subprocess.Popen(["npm", "run", "dev"], shell=shell)
31
    os.chdir("..")
32
    return api_process, ui_process , celery_process
33

34

35
def cleanup(api_process, ui_process, celery_process):
36
    logger.info("Shutting down processes...")
37
    api_process.terminate()
38
    ui_process.terminate()
39
    celery_process.terminate()
40
    logger.info("Processes terminated. Exiting.")
41
    sys.exit(1)
42

43

44
if __name__ == "__main__":
45
    check_command("node", "Node.js is not installed. Please install it and try again.")
46
    check_command("npm", "npm is not installed. Please install npm to proceed.")
47
    check_command("uvicorn", "uvicorn is not installed. Please install uvicorn to proceed.")
48

49
    isWindows = False
50
    if platform == "win32" or platform == "cygwin":
51
        isWindows = True
52
    run_npm_commands(shell=isWindows)
53

54
    try:
55
        api_process, ui_process, celery_process = run_server(isWindows)
56
        while True:
57
            try:
58
                sleep(30)
59
            except KeyboardInterrupt:
60
                cleanup(api_process, ui_process, celery_process)
61
    except Exception as e:
62
        cleanup(api_process, ui_process, celery_process)