feat(BA-1008): Make action processor work with async functions (#3999)

Author: seedspirit

changes/3999.feature.md

@@ -0,0 +1 @@
+Make action processor work with async functions

src/ai/backend/manager/actions/processor.py

@@ -1,6 +1,6 @@
 import asyncio
 from datetime import datetime
-from typing import Callable, Generic, Optional
+from typing import Awaitable, Callable, Generic, Optional
 
 from .action import (
     BaseActionResultMeta,
@@ -13,11 +13,11 @@
 
 class ActionProcessor(Generic[TAction, TActionResult]):
     _monitors: list[ActionMonitor]
-    _func: Callable[[TAction], TActionResult]
+    _func: Callable[[TAction], Awaitable[TActionResult]]
 
     def __init__(
         self,
-        func: Callable[[TAction], TActionResult],
+        func: Callable[[TAction], Awaitable[TActionResult]],
         monitors: Optional[list[ActionMonitor]] = None,
     ) -> None:
         self._func = func
@@ -27,7 +27,7 @@ async def _run(self, action: TAction) -> ProcessResult[TActionResult]:
         started_at = datetime.now()
         status: str
         try:
-            result = self._func(action)
+            result = await self._func(action)
             status = "success"
             description = "Success"
         except Exception as e: