fix: async task response deserialization crash (#4865)

bentoml · Jul 17, 2024 · 4c41872 · 4c41872
1 parent 2df8f44
commit 4c41872
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/_bentoml_impl/tasks/result.py b/src/_bentoml_impl/tasks/result.py
@@ -163,7 +163,7 @@ async def get(self, task_id: str) -> CompletedResultRow[Request, Response]:
             row[4],
             row[5],
             await self.serializer.deserialize_request(row[1]),
-            await self.serializer.deserialize_request(row[1]),
+            await self.serializer.deserialize_response(row[3]),
             row[6],
         )
 

diff --git a/src/_bentoml_impl/tasks/serde.py b/src/_bentoml_impl/tasks/serde.py
@@ -18,7 +18,7 @@ async def _consume_response(response: Response) -> bytes:
     # This will collect the response body into a buffer
     # and run until the background task is done
     buffer = io.BytesIO()
-    scope: dict[str, t.Any] = {}
+    scope: dict[str, t.Any] = {"type": "http"}
 
     def receive() -> t.Awaitable[Message]:
         # A fake receive that never returns