import pip

required_deps = {
    "requests": "requests",
    "openai": "openai",
    "anthropic": "anthropic",
    "colorlog": "colorlog",
    "mirai": "yiri-mirai-rc",
    "aiocqhttp": "aiocqhttp",
    "botpy": "qq-botpy",
    "PIL": "pillow",
    "nakuru": "nakuru-project-idk",
    "tiktoken": "tiktoken",
    "yaml": "pyyaml",
    "aiohttp": "aiohttp",
    "psutil": "psutil",
    "async_lru": "async-lru",
}


async def check_deps() -> list[str]:
    global required_deps

    missing_deps = []
    for dep in required_deps:
        try:
            __import__(dep)
        except ImportError:
            missing_deps.append(dep)
    return missing_deps

async def install_deps(deps: list[str]):
    global required_deps
    
    for dep in deps:
        pip.main(["install", required_deps[dep]])