فهرست منبع

refactor: Create a `dify_config` with Pydantic. (#5938)

-LAN- 9 ماه پیش
والد
کامیت
cb8feb732f
3فایلهای تغییر یافته به همراه8 افزوده شده و 3 حذف شده
  1. 2 2
      api/app.py
  2. 2 1
      api/commands.py
  3. 4 0
      api/configs/__init__.py

+ 2 - 2
api/app.py

@@ -1,6 +1,6 @@
 import os
 
-from configs.app_config import DifyConfig
+from configs import dify_config
 
 if not os.environ.get("DEBUG") or os.environ.get("DEBUG", "false").lower() != 'true':
     from gevent import monkey
@@ -81,7 +81,7 @@ def create_flask_app_with_configs() -> Flask:
     with configs loaded from .env file
     """
     dify_app = DifyApp(__name__)
-    dify_app.config.from_mapping(DifyConfig().model_dump())
+    dify_app.config.from_mapping(dify_config.model_dump())
 
     # populate configs into system environment variables
     for key, value in dify_app.config.items():

+ 2 - 1
api/commands.py

@@ -8,6 +8,7 @@ import click
 from flask import current_app
 from werkzeug.exceptions import NotFound
 
+from configs import dify_config
 from constants.languages import languages
 from core.rag.datasource.vdb.vector_factory import Vector
 from core.rag.datasource.vdb.vector_type import VectorType
@@ -112,7 +113,7 @@ def reset_encrypt_key_pair():
     After the reset, all LLM credentials will become invalid, requiring re-entry.
     Only support SELF_HOSTED mode.
     """
-    if current_app.config['EDITION'] != 'SELF_HOSTED':
+    if dify_config.EDITION != 'SELF_HOSTED':
         click.echo(click.style('Sorry, only support SELF_HOSTED mode.', fg='red'))
         return
 

+ 4 - 0
api/configs/__init__.py

@@ -0,0 +1,4 @@
+
+from .app_config import DifyConfig
+
+dify_config = DifyConfig()