diff --git a/api/.env.example b/api/.env.example index b8976e5b17..6ead14e9b0 100644 --- a/api/.env.example +++ b/api/.env.example @@ -54,7 +54,7 @@ REDIS_CLUSTERS_PASSWORD= # celery configuration CELERY_BROKER_URL=redis://:difyai123456@localhost:${REDIS_PORT}/1 - +CELERY_BACKEND=redis # PostgreSQL database configuration DB_USERNAME=postgres DB_PASSWORD=difyai123456 diff --git a/api/configs/middleware/__init__.py b/api/configs/middleware/__init__.py index 0c0c06dd46..ab45469219 100644 --- a/api/configs/middleware/__init__.py +++ b/api/configs/middleware/__init__.py @@ -209,9 +209,10 @@ class DatabaseConfig(BaseSettings): class CeleryConfig(DatabaseConfig): + CELERY_BACKEND: str = Field( description="Backend for Celery task results. Options: 'database', 'redis'.", - default="database", + default="redis", ) CELERY_BROKER_URL: Optional[str] = Field( @@ -240,6 +241,7 @@ class CeleryConfig(DatabaseConfig): @computed_field def CELERY_RESULT_BACKEND(self) -> str | None: + return ( "db+{}".format(self.SQLALCHEMY_DATABASE_URI) if self.CELERY_BACKEND == "database"