diff --git a/src/backend/langflow/core/celery_app.py b/src/backend/langflow/core/celery_app.py index 85e2bc2d2..ef3fc6545 100644 --- a/src/backend/langflow/core/celery_app.py +++ b/src/backend/langflow/core/celery_app.py @@ -1,7 +1,7 @@ from celery import Celery # type: ignore -def make_celery(app_name: str, config: str): +def make_celery(app_name: str, config: str) -> Celery: celery_app = Celery(app_name) celery_app.config_from_object(config) celery_app.conf.task_routes = {"langflow.worker.tasks.*": {"queue": "langflow"}} diff --git a/src/backend/langflow/core/celeryconfig.py b/src/backend/langflow/core/celeryconfig.py index 6747135ae..35d51bba0 100644 --- a/src/backend/langflow/core/celeryconfig.py +++ b/src/backend/langflow/core/celeryconfig.py @@ -4,7 +4,8 @@ import os langflow_redis_host = os.environ.get("LANGFLOW_REDIS_HOST") langflow_redis_port = os.environ.get("LANGFLOW_REDIS_PORT") if "BROKER_URL" in os.environ and "RESULT_BACKEND" in os.environ: - broker_url = os.environ.get("BROKER_URL", "redis://localhost:6379/0") + # RabbitMQ + broker_url = os.environ.get("BROKER_URL", "amqp://localhost") result_backend = os.environ.get("RESULT_BACKEND", "redis://localhost:6379/0") elif langflow_redis_host and langflow_redis_port: broker_url = f"redis://{langflow_redis_host}:{langflow_redis_port}/0"