refactor: 基本启动流程

2026-06-27 07:54:19 +00:00 · 2024-01-23 20:55:20 +08:00
parent e63c6ac723
commit 8d35ecd711
14 changed files with 345 additions and 5 deletions
@@ -0,0 +1,34 @@
+import pip
+
+required_deps = {
+    "requests": "requests",
+    "openai": "openai",
+    "dulwich": "dulwich",
+    "colorlog": "colorlog",
+    "mirai": "yiri-mirai-rc",
+    "func_timeout": "func_timeout",
+    "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]])