openNAMU__openNAMU/route/go_api_func_llm.py

from .tool.func import *

def api_func_llm(db_set):
    with get_db_connect() as conn:
        if flask.request.method == 'POST':
            other_set = {}
            other_set["prompt"] = flask.request.form.get('prompt', '')
            other_set["ip"] = ip_check()
            other_set = json.dumps(other_set)

            if platform.system() == 'Linux':
                if platform.machine() in ["AMD64", "x86_64"]:
                    data = subprocess.Popen([os.path.join(".", "route_go", "bin", "main.amd64.bin"), sys._getframe().f_code.co_name, db_set, other_set], stdout = subprocess.PIPE).communicate()[0]
                else:
                    data = subprocess.Popen([os.path.join(".", "route_go", "bin", "main.arm64.bin"), sys._getframe().f_code.co_name, db_set, other_set], stdout = subprocess.PIPE).communicate()[0]
            else:
                if platform.machine() in ["AMD64", "x86_64"]:
                    data = subprocess.Popen([os.path.join(".", "route_go", "bin", "main.amd64.exe"), sys._getframe().f_code.co_name, db_set, other_set], stdout = subprocess.PIPE).communicate()[0]
                else:
                    data = subprocess.Popen([os.path.join(".", "route_go", "bin", "main.arm64.exe"), sys._getframe().f_code.co_name, db_set, other_set], stdout = subprocess.PIPE).communicate()[0]

            data = data.decode('utf8')

            return flask.Response(response = data, status = 200, mimetype = 'application/json')
        else:
            return flask.jsonify({})