pi-freeflow 1.0.8 → 1.1.0

package/extensions/index.ts

@@ -948,15 +948,20 @@ function normalizeRequestBody(
 	}
 
 	// 3. Max & Min tokens clamping (OpenCode Zen requires min 16; Vercel relay caps at 64k)
-	const mt = (body.max_tokens ?? body.maxTokens ?? body.max_output_tokens) as number | undefined;
-	if (typeof mt === "number") {
-		let clamped = mt;
-		if (clamped < 16) clamped = 16;
-		else if (isRelay && clamped > RELAY_MAX_TOKENS) clamped = RELAY_MAX_TOKENS;
-
-		if (body.max_tokens !== undefined) body.max_tokens = clamped;
-		if (body.maxTokens !== undefined) body.maxTokens = clamped;
-		if (body.max_output_tokens !== undefined) body.max_output_tokens = clamped;
+	const clampTokens = (val: number): number => {
+		if (val < 16) return 16;
+		if (isRelay && val > RELAY_MAX_TOKENS) return RELAY_MAX_TOKENS;
+		return val;
+	};
+
+	if (typeof body.max_tokens === "number") {
+		body.max_tokens = clampTokens(body.max_tokens);
+	}
+	if (typeof body.maxTokens === "number") {
+		body.maxTokens = clampTokens(body.maxTokens);
+	}
+	if (typeof body.max_output_tokens === "number") {
+		body.max_output_tokens = clampTokens(body.max_output_tokens);
 	}
 
 	return body;
@@ -1659,10 +1664,12 @@ export default async function (pi: ExtensionAPI) {
 		);
 	});
 
-	pi.on("session_shutdown", () => {
-		log("info", "shutting down proxy...");
-		server.close();
-		rateLimitMap.clear();
-		log("info", "shutdown complete");
+	pi.on?.("session_shutdown", () => {
+		if (server) {
+			log("info", "shutting down proxy...");
+			server.close();
+			rateLimitMap.clear();
+			log("info", "shutdown complete");
+		}
 	});
 }

package/package.json

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