diff --git a/runpod/serverless/modules/rp_fastapi.py b/runpod/serverless/modules/rp_fastapi.py index f4aff225c..b7458af77 100644 --- a/runpod/serverless/modules/rp_fastapi.py +++ b/runpod/serverless/modules/rp_fastapi.py @@ -339,7 +339,7 @@ async def _sim_runsync(self, job_request: DefaultRequest) -> JobOutput: thread.start() return jsonable_encoder( - {"id": job.id, "status": "COMPLETED", "output": job_output["output"]} + {"id": job.id, "status": "COMPLETED", "output": job_output.get("output")} ) # ---------------------------------- stream ---------------------------------- # @@ -416,5 +416,5 @@ async def _sim_status(self, job_id: str) -> JobOutput: thread.start() return jsonable_encoder( - {"id": job_id, "status": "COMPLETED", "output": job_output["output"]} + {"id": job.id, "status": "COMPLETED", "output": job_output.get("output")} )