diff --git a/src/_bentoml_impl/server/app.py b/src/_bentoml_impl/server/app.py
index e0e229a4786..571a08978d2 100644
--- a/src/_bentoml_impl/server/app.py
+++ b/src/_bentoml_impl/server/app.py
@@ -440,6 +440,7 @@ async def cancel_task(self, request: Request) -> Response:
 
     async def _run_task(self, task_id: str, name: str, request: Request) -> None:
         try:
+            self.service.context.request.state.task_id = task_id
             resp = await self.api_endpoint_wrapper(name, request)
             await self._result_store.set_result(
                 task_id,