QChatGPT/pkg/utils/pkgmgr.py

from pip._internal import main as pipmain

# from . import log


def install(package):
    pipmain(['install', package])
    # log.reset_logging()

def install_upgrade(package):
    pipmain(['install', '--upgrade', package, "-i", "https://pypi.tuna.tsinghua.edu.cn/simple",
                    "--trusted-host", "pypi.tuna.tsinghua.edu.cn"])
    # log.reset_logging()


def run_pip(params: list):
    pipmain(params)
    # log.reset_logging()


def install_requirements(file):
    pipmain(['install', '-r', file, "-i", "https://pypi.tuna.tsinghua.edu.cn/simple",
                    "--trusted-host", "pypi.tuna.tsinghua.edu.cn"])
    # log.reset_logging()


if __name__ == "__main__":
    try:
        install("openai11")
    except Exception as e:
        print(111)
        print(e)

    print(222)