Przeglądaj źródła

fix(storage): ensure `storage_runner` initialization within app context (#9627)

-LAN- 6 miesięcy temu
rodzic
commit
e8abbe0623
1 zmienionych plików z 2 dodań i 1 usunięć
  1. 2 1
      api/extensions/ext_storage.py

+ 2 - 1
api/extensions/ext_storage.py

@@ -15,7 +15,8 @@ class Storage:
 
 
     def init_app(self, app: Flask):
     def init_app(self, app: Flask):
         storage_factory = self.get_storage_factory(dify_config.STORAGE_TYPE)
         storage_factory = self.get_storage_factory(dify_config.STORAGE_TYPE)
-        self.storage_runner = storage_factory()
+        with app.app_context():
+            self.storage_runner = storage_factory()
 
 
     @staticmethod
     @staticmethod
     def get_storage_factory(storage_type: str) -> type[BaseStorage]:
     def get_storage_factory(storage_type: str) -> type[BaseStorage]: