修复知识库的依赖问题

2023-09-12 11:35:31 +08:00
parent d495b73456
commit 4b5f13bff2
3 changed files with 25 additions and 23 deletions
--- a/crazy_functions/crazy_utils.py
+++ b/crazy_functions/crazy_utils.py
@@ -706,11 +706,14 @@ class knowledge_archive_interface():
        self.threadLock.release()
        return resp, prompt

-def try_install_deps(deps):
+def try_install_deps(deps, reload_m=[]):
+    import subprocess, sys, importlib
    for dep in deps:
-        import subprocess, sys
        subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--user', dep])
-
+    import site
+    importlib.reload(site)
+    for m in reload_m:
+        importlib.reload(__import__(m))

 class construct_html():
    def __init__(self) -> None: