From c5a6003ef5dc4b06728a695be45867860cb99530 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 23 Sep 2023 23:29:20 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(manager.py):=20replace=20log?= =?UTF-8?q?ging=20module=20with=20loguru=20logger=20for=20consistent=20log?= =?UTF-8?q?ging=20across=20the=20project=20=E2=9C=A8=20feat(manager.py):?= =?UTF-8?q?=20add=20debug=20logs=20to=20check=20Celery=20availability=20an?= =?UTF-8?q?d=20backend=20being=20used=20=F0=9F=94=A7=20fix(manager.py):=20?= =?UTF-8?q?move=20check=5Fcelery=5Favailability=20function=20definition=20?= =?UTF-8?q?above=20its=20usage=20to=20improve=20code=20readability=20?= =?UTF-8?q?=E2=9C=A8=20feat(manager.py):=20add=20debug=20logs=20to=20show?= =?UTF-8?q?=20task=20launch=20details=20and=20backend=20being=20used?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/task/manager.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/services/task/manager.py b/src/backend/langflow/services/task/manager.py index 4083978df..ab74d916e 100644 --- a/src/backend/langflow/services/task/manager.py +++ b/src/backend/langflow/services/task/manager.py @@ -1,19 +1,26 @@ from typing import Any, Callable, Coroutine, Union -import logging - +from loguru import logger from langflow.services.base import Service from langflow.services.task.backends.anyio import AnyIOBackend from langflow.services.task.backends.base import TaskBackend from langflow.services.task.utils import get_celery_worker_status -try: + +def check_celery_availability(): from langflow.worker import celery_app try: status = get_celery_worker_status(celery_app) + logger.debug(f"Celery status: {status}") except Exception as e: - logging.error(f"An error occurred: {e}") + logger.error(f"An error occurred: {e}") status = {"availability": None} + return status + + +try: + status = check_celery_availability() + USE_CELERY = status.get("availability") is not None except ImportError: USE_CELERY = False @@ -30,7 +37,9 @@ class TaskService(Service): if USE_CELERY: from langflow.services.task.backends.celery import CeleryBackend + logger.debug("Using Celery backend") return CeleryBackend() + logger.debug("Using AnyIO backend") return AnyIOBackend() # In your TaskService class @@ -51,6 +60,8 @@ class TaskService(Service): async def launch_task( self, task_func: Callable[..., Any], *args: Any, **kwargs: Any ) -> Any: + logger.debug(f"Launching task {task_func} with args {args} and kwargs {kwargs}") + logger.debug(f"Using backend {self.backend}") task = self.backend.launch_task(task_func, *args, **kwargs) return await task if isinstance(task, Coroutine) else task