Proxy: forward method/query/body/headers, stream SSE both ways

node_probe.py

@@ -81,19 +81,43 @@ def register_routes(fa_app):
 
     @fa_app.get("/respite/v2/node/proxy/{path:path}")
     @fa_app.post("/respite/v2/node/proxy/{path:path}")
-    def _proxy(path: str):
-        """Forward to the Node backend on 127.0.0.1:3210 (streaming)."""
+    async def _proxy(request, path: str):
+        """Forward method, query string, body and headers to the Node backend
+        on 127.0.0.1:3210. Streaming both ways (SSE included)."""
         if not node_runtime.node_ready():
             start_ok = node_runtime.start_node_backend()
             if not start_ok:
                 return JSONResponse({"error": "node backend not running"}, status_code=503)
         url = f"http://127.0.0.1:{node_runtime.NODE_PORT}/{path}"
+        if request.url.query:
+            url += "?" + request.url.query
+        headers = {
+            k: v for k, v in request.headers.items()
+            if k.lower() not in ("host", "connection", "content-length", "accept-encoding")
+        }
+        body = await request.body()
         try:
-            with httpx.stream("GET", url, timeout=30.0) as r:
-                headers = dict(r.headers)
-                body = b""
-                for chunk in r.iter_bytes():
-                    body += chunk
-            return Response(content=body, status_code=r.status_code, headers=headers)
+            client = httpx.AsyncClient(timeout=httpx.Timeout(300.0, connect=10.0))
+            req = client.build_request(
+                request.method, url, headers=headers, content=body or None
+            )
+            upstream = await client.send(req, stream=True)
+            resp_headers = {
+                k: v for k, v in upstream.headers.items()
+                if k.lower() not in ("content-length", "transfer-encoding", "connection")
+            }
+            from starlette.responses import StreamingResponse
+
+            async def stream_gen():
+                try:
+                    async for chunk in upstream.aiter_bytes():
+                        yield chunk
+                finally:
+                    await upstream.aclose()
+                    await client.aclose()
+
+            return StreamingResponse(
+                stream_gen(), status_code=upstream.status_code, headers=resp_headers
+            )
         except Exception as e:
             return JSONResponse({"error": str(e)}, status_code=502)