shariq-pi-extensions 0.2.11 → 0.2.13

package/extensions/factory-provider/factory/websocket.ts

@@ -63,13 +63,23 @@ export function createFactoryResponsesWebSocketFetch(options: FactoryWebSocketFe
 			});
 
 			const abort = () => {
-				socket.close();
-				void writer.abort(new Error("Factory request was aborted"));
+				init?.signal?.removeEventListener("abort", abort);
+				try {
+					socket.close();
+				} catch {
+					// Ignore socket close errors during abort.
+				}
+				void writer.abort(init?.signal?.reason || new Error("Factory request was aborted")).catch(() => undefined);
 				if (!settled) {
 					settled = true;
-					reject(new Error("Factory request was aborted"));
+					reject(init?.signal?.reason || new Error("Factory request was aborted"));
 				}
 			};
+
+			if (init?.signal?.aborted) {
+				abort();
+				return;
+			}
 			init?.signal?.addEventListener("abort", abort, { once: true });
 
 			socket.once("open", () => {
@@ -86,11 +96,11 @@ export function createFactoryResponsesWebSocketFetch(options: FactoryWebSocketFe
 
 			socket.on("message", (data: RawData) => {
 				const text = data.toString();
-				void writer.write(encoder.encode(`data: ${text}\n\n`));
+				void writer.write(encoder.encode(`data: ${text}\n\n`)).catch(() => undefined);
 				try {
 					const event = JSON.parse(text) as { type?: string };
 					if (event.type === "response.completed" || event.type === "response.failed" || event.type === "error") {
-						void writer.close();
+						void writer.close().catch(() => undefined);
 						socket.close();
 					}
 				} catch {
@@ -99,6 +109,7 @@ export function createFactoryResponsesWebSocketFetch(options: FactoryWebSocketFe
 			});
 
 			socket.once("unexpected-response", (_request, response) => {
+				init?.signal?.removeEventListener("abort", abort);
 				const chunks: Buffer[] = [];
 				response.on("data", (chunk: Buffer) => chunks.push(chunk));
 				response.on("end", () => {
@@ -114,11 +125,12 @@ export function createFactoryResponsesWebSocketFetch(options: FactoryWebSocketFe
 			});
 
 			socket.once("error", (error) => {
+				init?.signal?.removeEventListener("abort", abort);
 				if (!settled) {
 					settled = true;
 					reject(error);
 				} else if (opened) {
-					void writer.abort(error);
+					void writer.abort(error).catch(() => undefined);
 				}
 			});
 			socket.once("close", () => {

package/extensions/smart-compaction/README.md

@@ -40,7 +40,10 @@ Settings are persisted in `~/.pi/agent/smart-compaction.json`:
   "version": 1,
   "enabled": true,
   "model": "inherit",
-  "thinkingLevel": "medium",
-  "maxSummaryTokens": 16384
+  "thinkingLevel": "inherit"
 }
 ```
+
+- `model`: `"inherit"` (uses current active session model) or explicit `"provider/model-id"`.
+- `thinkingLevel`: `"inherit"` (uses current session's thinking level) or `"off" | "low" | "medium" | "high" | "max"`.
+- `maxSummaryTokens`: optional override integer; if omitted, dynamically defaults to the model's full native output capacity (65,536–128,000+ tokens) so summaries are never artificially truncated.

package/extensions/smart-compaction/config.ts

@@ -6,16 +6,15 @@ export interface SmartCompactionConfig {
   version: 1;
   enabled: boolean;
   model: string; // "inherit" or "provider/model-id"
-  thinkingLevel?: "off" | "low" | "medium" | "high";
-  maxSummaryTokens?: number;
+  thinkingLevel?: "inherit" | "off" | "low" | "medium" | "high" | "max";
+  maxSummaryTokens?: number; // optional override; defaults to model's full capacity
 }
 
 export const DEFAULT_SMART_COMPACTION_CONFIG: SmartCompactionConfig = {
   version: 1,
   enabled: true,
   model: "inherit",
-  thinkingLevel: "medium",
-  maxSummaryTokens: 16384,
+  thinkingLevel: "inherit",
 };
 
 export function smartCompactionConfigPath(): string {
@@ -30,12 +29,12 @@ export function loadSmartCompactionConfig(file = smartCompactionConfigPath()): S
       version: 1,
       enabled: typeof raw.enabled === "boolean" ? raw.enabled : DEFAULT_SMART_COMPACTION_CONFIG.enabled,
       model: typeof raw.model === "string" && raw.model.trim() ? raw.model.trim() : DEFAULT_SMART_COMPACTION_CONFIG.model,
-      thinkingLevel: raw.thinkingLevel && ["off", "low", "medium", "high"].includes(raw.thinkingLevel)
-        ? raw.thinkingLevel
+      thinkingLevel: raw.thinkingLevel && ["inherit", "off", "low", "medium", "high", "max"].includes(raw.thinkingLevel)
+        ? (raw.thinkingLevel as SmartCompactionConfig["thinkingLevel"])
         : DEFAULT_SMART_COMPACTION_CONFIG.thinkingLevel,
       maxSummaryTokens: typeof raw.maxSummaryTokens === "number" && raw.maxSummaryTokens > 0
         ? raw.maxSummaryTokens
-        : DEFAULT_SMART_COMPACTION_CONFIG.maxSummaryTokens,
+        : undefined,
     };
   } catch {
     return { ...DEFAULT_SMART_COMPACTION_CONFIG };
@@ -50,8 +49,8 @@ export function saveSmartCompactionConfig(config: SmartCompactionConfig, file =
     version: 1,
     enabled: config.enabled,
     model: config.model || "inherit",
-    thinkingLevel: config.thinkingLevel ?? "medium",
-    maxSummaryTokens: config.maxSummaryTokens ?? 16384,
+    thinkingLevel: config.thinkingLevel ?? "inherit",
+    maxSummaryTokens: config.maxSummaryTokens,
   };
   try {
     fs.writeFileSync(temporary, `${JSON.stringify(document, null, 2)}\n`, { mode: 0o600 });

package/extensions/smart-compaction/engine.ts

@@ -44,7 +44,7 @@ export function resolveCompactionModel(
 
 export interface RunSmartCompactionOptions {
   event: SessionBeforeCompactEvent;
-  ctx: Pick<ExtensionContext, "model" | "modelRegistry">;
+  ctx: Pick<ExtensionContext, "model" | "modelRegistry" | "thinkingLevel">;
   config: SmartCompactionConfig;
 }
 
@@ -97,18 +97,27 @@ export async function runSmartCompaction(
     ],
   };
 
-  const requestedMaxTokens = config.maxSummaryTokens ?? 16384;
-  const maxTokens = model.maxTokens > 0 ? Math.min(requestedMaxTokens, model.maxTokens) : requestedMaxTokens;
-
   const completeOptions: Record<string, unknown> = {
-    maxTokens,
     signal,
     cacheRetention: "none",
     sessionId: uuidv7(),
   };
 
-  if (model.reasoning && config.thinkingLevel && config.thinkingLevel !== "off") {
-    completeOptions.reasoning = config.thinkingLevel;
+  // If user explicitly configured a maxSummaryTokens override, pass it.
+  // Otherwise, omit maxTokens so the provider uses the model's full native maximum output capacity (e.g. 128k, 65k).
+  if (typeof config.maxSummaryTokens === "number" && config.maxSummaryTokens > 0) {
+    completeOptions.maxTokens = config.maxSummaryTokens;
+  }
+
+  // Resolve reasoning effort / thinking level
+  if (model.reasoning) {
+    const desiredThinking = config.thinkingLevel === "inherit" || !config.thinkingLevel
+      ? ctx.thinkingLevel
+      : config.thinkingLevel;
+
+    if (desiredThinking && desiredThinking !== "off") {
+      completeOptions.reasoning = desiredThinking;
+    }
   }
 
   const response = await ctx.modelRegistry.complete(model, context, completeOptions as any);

package/extensions/smart-compaction/index.ts

@@ -164,11 +164,19 @@ export function createSmartCompactionExtension(options: SmartCompactionExtension
         }
 
         // Default status
+        const currentModelDesc = config.model === "inherit"
+          ? `inherit (${cmdCtx.model ? `${cmdCtx.model.provider}/${cmdCtx.model.id}` : "active session model"})`
+          : config.model;
+        const currentThinkingDesc = config.thinkingLevel === "inherit"
+          ? `inherit (${cmdCtx.thinkingLevel ?? "session default"})`
+          : (config.thinkingLevel ?? "inherit");
+        const maxTokensDesc = config.maxSummaryTokens ? `${config.maxSummaryTokens}` : "unlimited (full model output capacity)";
+
         const status = [
           `Smart Compaction: ${config.enabled ? "ENABLED" : "DISABLED"}`,
-          `Model: ${config.model} (${config.model === "inherit" ? (cmdCtx.model ? `${cmdCtx.model.provider}/${cmdCtx.model.id}` : "inherit") : config.model})`,
-          `Thinking Level: ${config.thinkingLevel ?? "medium"}`,
-          `Max Tokens: ${config.maxSummaryTokens ?? 4096}`,
+          `Model: ${currentModelDesc}`,
+          `Thinking Level: ${currentThinkingDesc}`,
+          `Max Output Tokens: ${maxTokensDesc}`,
           "",
           "Commands:",
           "  /smart-compaction enable | disable",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shariq-pi-extensions",
-  "version": "0.2.11",
+  "version": "0.2.13",
   "description": "Cross-platform extension suite for the Pi coding agent.",
   "license": "MIT",
   "author": "Shariq Riaz",