pi-freeflow 1.1.9 → 1.2.0

package/extensions/index.ts

@@ -1453,8 +1453,8 @@ function startProxy(
 			} catch {}
 
 			const upstream: Upstream = isKilo ? "kilo" : "opencode";
+			const isStream = parsedBody?.stream === true;
 			if (!checkRateLimit(clientIP, upstream)) {
-				log("warn", `rate limit hit for ${upstream} from ${clientIP}`, { ip: clientIP, upstream }, reqId);
 				res.writeHead(429, { "content-type": "application/json" });
 				res.end(JSON.stringify({ error: "rate limit exceeded" }));
 				return;
@@ -1508,52 +1508,49 @@ function startProxy(
 							activeHost,
 						);
 						try {
-							if (relayBody && parsedBody) {
+							if (parsedBody) {
 								const relayBodyObj = structuredClone(parsedBody);
 								normalizeRequestBody(relayBodyObj, true, isKilo, reqId);
-								relayBody = Buffer.from(JSON.stringify(relayBodyObj));
+								const relayBody = Buffer.from(JSON.stringify(relayBodyObj));
+								const response = await relayFetch(fullUrl, {
+									method: req.method || "POST",
+									headers: relayHeaders,
+									body: relayBody,
+									signal: AbortSignal.timeout(300_000),
+								}, reqId);
+								if (isStream && response.ok && response.body) {
+									const ct =
+										response.headers.get("content-type") || "text/event-stream";
+									res.writeHead(response.status, {
+										"content-type": ct,
+										"cache-control": "no-cache, no-transform",
+										"connection": "keep-alive",
+										"x-accel-buffering": "no",
+									});
+									pipeUpstreamStream(
+										Readable.fromWeb(
+											response.body as unknown as WebReadableStream,
+										),
+										res,
+										req,
+										reqId,
+									);
+								} else {
+									const data = await response.text();
+									const ct =
+										response.headers.get("content-type") || "application/json";
+									res.writeHead(response.status, { "content-type": ct });
+									res.end(data);
+								}
+								return; // relay handled the response
 							}
-							const response = await relayFetch(fullUrl, {
-								method: req.method || "POST",
-								headers: relayHeaders,
-								body: relayBody,
-								signal: AbortSignal.timeout(300_000),
-							}, reqId);
-							const ct =
-								response.headers.get("content-type") || "application/json";
-							if (response.ok && response.body) {
-								const ct =
-									response.headers.get("content-type") || "text/event-stream";
-								res.writeHead(response.status, {
-									"content-type": ct,
-									"cache-control": "no-cache, no-transform",
-									"connection": "keep-alive",
-									"x-accel-buffering": "no",
-								});
-								pipeUpstreamStream(
-									Readable.fromWeb(
-										response.body as unknown as WebReadableStream,
-									),
-									res,
-									req,
-									reqId,
-								);
-							} else {
-								const data = await response.text();
-								const ct =
-									response.headers.get("content-type") || "application/json";
-								res.writeHead(response.status, { "content-type": ct });
-								res.end(data);
-							}
-							return; // relay handled the response
 						} catch (e) {
 							log("warn", "opencode relay failed, falling back to direct", {
 								error: String(e),
 							}, reqId);
- 							if (res.headersSent) return; // can't recover mid-stream
- 						}
+							if (res.headersSent) return; // can't recover mid-stream
+						}
  					}
- 					// direct path (existing, untouched)
 					// direct path — with debug trace and thinking-aware normalize
 					let directBody = Buffer.concat(bodyChunks);
 					if (parsedBody) {

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-freeflow",
 	"type": "module",
-	"version": "1.1.9",
+	"version": "1.2.0",
 	"description": "Personal multi-cloud rolling fallback relay for OpenCode Zen and KiloCode models in OMP/Pi",
 	"keywords": [
 		"pi-package",