From e50704a49f60b4d20efcc69aa90b24488f907f92 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:32:54 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(celery.py):=20fix=20type=20h?= =?UTF-8?q?inting=20error=20for=20AsyncResult=20import=20=F0=9F=94=92=20ch?= =?UTF-8?q?ore(celery.py):=20add=20check=20for=20delay=20method=20existenc?= =?UTF-8?q?e=20in=20launch=5Ftask=20to=20prevent=20errors=20=F0=9F=94=92?= =?UTF-8?q?=20chore(celery.py):=20change=20get=5Ftask=20return=20type=20hi?= =?UTF-8?q?nt=20to=20Any=20to=20match=20actual=20return=20type?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/task/backends/celery.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/task/backends/celery.py b/src/backend/langflow/services/task/backends/celery.py index 04e26159c..3a6e1f450 100644 --- a/src/backend/langflow/services/task/backends/celery.py +++ b/src/backend/langflow/services/task/backends/celery.py @@ -1,5 +1,5 @@ from typing import Any, Callable -from celery.result import AsyncResult +from celery.result import AsyncResult # type: ignore from langflow.services.task.backends.base import TaskBackend from langflow.worker import celery_app @@ -11,8 +11,10 @@ class CeleryBackend(TaskBackend): def launch_task( self, task_func: Callable[..., Any], *args: Any, **kwargs: Any ) -> str: - task = task_func.apply_async(args=args, kwargs=kwargs) + if not hasattr(task_func, "delay"): + raise ValueError(f"Task function {task_func} does not have a delay method") + task = task_func.delay(*args, **kwargs) return task.id - def get_task(self, task_id: str) -> AsyncResult: + def get_task(self, task_id: str) -> Any: return AsyncResult(task_id, app=self.celery_app)