Fix return statement in request handler to ensure output is returned …

…on success
mobiusml · Nov 12, 2024 · f26671f · f26671f
1 parent 1e6054b
commit f26671f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aana/api/request_handler.py b/aana/api/request_handler.py
@@ -148,9 +148,10 @@ async def execute_task(self, task_id: str | UUID) -> Any:
                 TaskRepository(session).update_status(
                     task_id, TaskStatus.FAILED, 0, error
                 )
+        else:
+            return out
         finally:
             self.running_tasks.remove(task_id)
-        return out
 
     @app.get(
         "/tasks/get/{task_id}",