Files
LangBot/pkg/core/bootutils/deps.py
2024-02-19 18:59:52 +08:00

33 lines
697 B
Python

import pip
required_deps = {
"requests": "requests",
"openai": "openai",
"colorlog": "colorlog",
"mirai": "yiri-mirai-rc",
"PIL": "pillow",
"nakuru": "nakuru-project-idk",
"CallingGPT": "CallingGPT",
"tiktoken": "tiktoken",
"yaml": "pyyaml",
"aiohttp": "aiohttp",
}
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]])