jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 fix(manager.py): add validation to check if task function has an apply method before calling it

Browse source 
✨ feat(manager.py): change return type of launch_task method to Any to allow for more flexibility in return values
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-05 15:33:43 -03:00
commit 9721e9d6cf
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/services/task/manager.py
View file

@ -42,13 +42,15 @@ class TaskManager(Service):
) -> Any: ) -> Any:
if not self.use_celery: if not self.use_celery:
return None, await task_func(*args, **kwargs) 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) task = task_func.apply(args=args, kwargs=kwargs)
result = task.get() result = task.get()
return task.id, result return task.id, result
async def launch_task( async def launch_task(
self, task_func: Callable[..., Any], *args: Any, **kwargs: Any self, task_func: Callable[..., Any], *args: Any, **kwargs: Any
) -> Union[str, str]: ) -> Any:
return await self.backend.launch_task(task_func, *args, **kwargs) return await self.backend.launch_task(task_func, *args, **kwargs)
def get_task(self, task_id: Union[int, str]) -> Any: def get_task(self, task_id: Union[int, str]) -> Any: