From 9721e9d6cf38ae3d8e8776bbcbbbd6aa68ed77de Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:33:43 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(manager.py):=20add=20validat?= =?UTF-8?q?ion=20to=20check=20if=20task=20function=20has=20an=20apply=20me?= =?UTF-8?q?thod=20before=20calling=20it=20=E2=9C=A8=20feat(manager.py):=20?= =?UTF-8?q?change=20return=20type=20of=20launch=5Ftask=20method=20to=20Any?= =?UTF-8?q?=20to=20allow=20for=20more=20flexibility=20in=20return=20values?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/task/manager.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/services/task/manager.py b/src/backend/langflow/services/task/manager.py index c3e5e6444..cdf731917 100644 --- a/src/backend/langflow/services/task/manager.py +++ b/src/backend/langflow/services/task/manager.py @@ -42,13 +42,15 @@ class TaskManager(Service): ) -> Any: if not self.use_celery: return None, await task_func(*args, **kwargs) + if not hasattr(task_func, "apply"): + raise ValueError(f"Task function {task_func} does not have an apply method") task = task_func.apply(args=args, kwargs=kwargs) result = task.get() return task.id, result async def launch_task( self, task_func: Callable[..., Any], *args: Any, **kwargs: Any - ) -> Union[str, str]: + ) -> Any: return await self.backend.launch_task(task_func, *args, **kwargs) def get_task(self, task_id: Union[int, str]) -> Any: