@pithos-kit/squiggle 0.0.0 → 0.4.1

package/README.md

@@ -39,7 +39,7 @@ pi -e ./pithos.squiggle
 ```yaml
 pi:
   extensions:
-    "@pithos-kit/squiggle": "npm:0.4.0"
+    "@pithos-kit/squiggle": "npm:0.4.1"
 ```
 
 ## Configuration

package/extensions/squiggle.ts

@@ -17,7 +17,10 @@ Show whether Squiggle is enabled and which correction model it uses.
 Options:
   --help, -h  Show this help`;
 
-export default function squiggle(pi: ExtensionAPI) {
+export function registerSquiggle(
+	pi: ExtensionAPI,
+	correctPrompt: (input: string, ctx: ExtensionContext, config: SquiggleConfig) => Promise<string | null> = correctWithModel,
+) {
 	let runtimeMode: SquiggleConfig["mode"] | undefined;
 
 	pi.on("session_start", async (_event, ctx) => {
@@ -58,25 +61,18 @@ export default function squiggle(pi: ExtensionAPI) {
 		if (!event.text.trim()) return { action: "continue" };
 
 		const stopIndicator = startSquiggleIndicator(ctx);
-		const corrected = await correctWithModel(event.text, ctx, config).finally(stopIndicator);
+		const corrected = await correctPrompt(event.text, ctx, config).finally(stopIndicator);
 		if (!corrected || corrected === event.text) return { action: "continue" };
 
 		if (ctx.hasUI) ctx.ui.notify(formatColoredDiff(event.text, corrected), "info");
-
-		// In interactive mode, `transform` changes what the agent receives, but the
-		// already-submitted prompt may still be rendered as originally typed. To make
-		// the visible user message corrected too, swallow the original input and
-		// resubmit the corrected text as an extension-originated user message. The
-		// source guard above prevents a correction loop.
-		if (event.source === "interactive") {
-			pi.sendUserMessage(corrected);
-			return { action: "handled" };
-		}
-
 		return { action: "transform", text: corrected };
 	});
 }
 
+export default function squiggle(pi: ExtensionAPI): void {
+	registerSquiggle(pi);
+}
+
 function isHelpRequest(args: string): boolean {
 	const normalized = args.trim();
 	return normalized === "--help" || normalized === "-h";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pithos-kit/squiggle",
-  "version": "0.0.0",
+  "version": "0.4.1",
   "description": "Quietly polish grammar and spelling in your Pi prompts.",
   "keywords": [
     "pi-package",
@@ -31,16 +31,8 @@
     "summary": "Quietly polish grammar and spelling in user prompts.",
     "minimumPi": ">=0.83.0",
     "commands": [
-      {
-        "name": "squiggle",
-        "usage": "/squiggle [toggle|--help]",
-        "summary": "Toggle prompt correction."
-      },
-      {
-        "name": "squiggle-status",
-        "usage": "/squiggle-status [--help]",
-        "summary": "Show correction status and configuration."
-      }
+      { "name": "squiggle", "usage": "/squiggle [toggle|--help]", "summary": "Toggle prompt correction." },
+      { "name": "squiggle-status", "usage": "/squiggle-status [--help]", "summary": "Show correction status and configuration." }
     ],
     "tools": [],
     "prompts": [],
@@ -48,26 +40,10 @@
     "themes": [],
     "agents": [],
     "configuration": [
-      {
-        "kind": "file",
-        "key": ".pi/squiggle.json",
-        "summary": "Project correction mode, model, and input limit."
-      },
-      {
-        "kind": "environment",
-        "key": "SQUIGGLE_MODE",
-        "summary": "Override correction mode."
-      },
-      {
-        "kind": "environment",
-        "key": "SQUIGGLE_MODEL",
-        "summary": "Override the correction model."
-      },
-      {
-        "kind": "environment",
-        "key": "SQUIGGLE_MAX_CHARS",
-        "summary": "Override the maximum corrected input length."
-      }
+      { "kind": "file", "key": ".pi/squiggle.json", "summary": "Project correction mode, model, and input limit." },
+      { "kind": "environment", "key": "SQUIGGLE_MODE", "summary": "Override correction mode." },
+      { "kind": "environment", "key": "SQUIGGLE_MODEL", "summary": "Override the correction model." },
+      { "kind": "environment", "key": "SQUIGGLE_MAX_CHARS", "summary": "Override the maximum corrected input length." }
     ]
   },
   "pi": {