From befe79775b1d1c63b8764e56745e868a5d91adb5 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 23 Sep 2023 23:28:37 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(celery=5Fapp.py):=20add=20re?= =?UTF-8?q?turn=20type=20annotation=20to=20make=5Fcelery=20function=20to?= =?UTF-8?q?=20improve=20code=20readability=20and=20maintainability=20?= =?UTF-8?q?=F0=9F=94=A7=20chore(celeryconfig.py):=20update=20broker=5Furl?= =?UTF-8?q?=20default=20value=20to=20use=20RabbitMQ=20instead=20of=20Redis?= =?UTF-8?q?=20for=20better=20performance=20and=20scalability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/core/celery_app.py | 2 +- src/backend/langflow/core/celeryconfig.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) 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"