feat: 热更新和热重载功能

pkg/openai/modelmgr.py

@@ -0,0 +1,34 @@
+# 提供与模型交互的抽象接口
+
+COMPLETION_MODELS = {
+    'text-davinci-003'
+}
+
+EDIT_MODELS = {
+
+}
+
+IMAGE_MODELS = {
+
+}
+
+
+# ModelManager
+# 由session包含
+class ModelMgr(object):
+
+    using_completion_model = ""
+    using_edit_model = ""
+    using_image_model = ""
+
+    def __init__(self):
+        pass
+
+    def get_using_completion_model(self):
+        return self.using_completion_model
+
+    def get_using_edit_model(self):
+        return self.using_edit_model
+
+    def get_using_image_model(self):
+        return self.using_image_model

pkg/utils/reloader.py

@@ -0,0 +1,19 @@
+import logging
+
+import pkg
+import importlib
+import pkgutil
+
+
+def walk(module, prefix=''):
+    for item in pkgutil.iter_modules(module.__path__):
+        if item.ispkg:
+            walk(__import__(module.__name__ + '.' + item.name, fromlist=['']), prefix + item.name + '.')
+        else:
+            logging.info('reload module: {}'.format(prefix + item.name))
+            importlib.reload(__import__(module.__name__ + '.' + item.name, fromlist=['']))
+
+
+def reload_all():
+    walk(pkg)
+    importlib.reload(__import__('config'))

pkg/utils/updater.py

@@ -0,0 +1,7 @@
+def update_all():
+    try:
+        from dulwich import porcelain
+        repo = porcelain.open_repo('.')
+        porcelain.pull(repo)
+    except ModuleNotFoundError:
+        raise Exception("dulwich模块未安装,请查看 https://github.com/RockChinQ/QChatGPT/issues/77")