Merge pull request #264 from runpod/stream-test-method

justinmerrell · web-flow · commit 1b3ba536f2f4 · 2024-01-11T10:45:10.000-05:00
fix: added GET to stream method for fastAPI
diff --git a/runpod/serverless/modules/rp_fastapi.py b/runpod/serverless/modules/rp_fastapi.py
@@ -131,7 +131,7 @@ def __init__(self, config: Dict[str, Any]):
             description="Returns job output directly when called."
         )
         api_router.add_api_route(
-            "/stream/{job_id}", self._sim_stream, methods=["POST"],
+            "/stream/{job_id}", self._sim_stream, methods=["POST", "GET"],
             response_model_exclude_none=True, summary="Simulate stream behavior.",
             description="Aggregates the output of the job and returns it when the job is complete."
         )

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def __init__(self, config: Dict[str, Any]):`
`131`	`131`	`description="Returns job output directly when called."`
`132`	`132`	`)`
`133`	`133`	`api_router.add_api_route(`
`134`		`- "/stream/{job_id}", self._sim_stream, methods=["POST"],`
	`134`	`+ "/stream/{job_id}", self._sim_stream, methods=["POST", "GET"],`
`135`	`135`	`response_model_exclude_none=True, summary="Simulate stream behavior.",`
`136`	`136`	`description="Aggregates the output of the job and returns it when the job is complete."`
`137`	`137`	`)`