Prechádzať zdrojové kódy

chore: Update some type hints in config. (#5833)

-LAN- 9 mesiacov pred
rodič
commit
32d85fb896

+ 3 - 0
api/configs/app_config.py

@@ -8,6 +8,9 @@ from configs.middleware import MiddlewareConfig
 from configs.packaging import PackagingInfo
 
 
+# TODO: Both `BaseModel` and `BaseSettings` has `model_config` attribute but they are in different types.
+# This inheritance is depends on the order of the classes.
+# It is better to use `BaseSettings` as the base class.
 class DifyConfig(
     # based on pydantic-settings
     BaseSettings,

+ 1 - 1
api/configs/feature/__init__.py

@@ -136,7 +136,7 @@ class HttpConfig(BaseModel):
     def CONSOLE_CORS_ALLOW_ORIGINS(self) -> list[str]:
         return self.inner_CONSOLE_CORS_ALLOW_ORIGINS.split(',')
 
-    inner_WEB_API_CORS_ALLOW_ORIGINS: Optional[str] = Field(
+    inner_WEB_API_CORS_ALLOW_ORIGINS: str = Field(
         description='',
         validation_alias=AliasChoices('WEB_API_CORS_ALLOW_ORIGINS'),
         default='*',

+ 1 - 1
api/configs/middleware/__init__.py

@@ -144,7 +144,7 @@ class CeleryConfig(DatabaseConfig):
 
     @computed_field
     @property
-    def CELERY_RESULT_BACKEND(self) -> str:
+    def CELERY_RESULT_BACKEND(self) -> str | None:
         return 'db+{}'.format(self.SQLALCHEMY_DATABASE_URI) \
             if self.CELERY_BACKEND == 'database' else self.CELERY_BROKER_URL