@agnishc/edb-agent-steer 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,7 @@
+# Changelog
+
+## [Unreleased]
+
+### Added
+- Initial release: steer / queue / discard / edit prompt for mid-turn messages
+- `/steer <text>` command for direct steering without the prompt

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Agnish Chakraborty
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,38 @@
+# @agnishc/edb-agent-steer
+
+A Pi CLI extension that intercepts messages sent while the agent is already running and presents a compact single-keypress prompt:
+
+```
+  ↵  "your message here"
+
+  s steer   q queue   d discard   e edit
+```
+
+| Action | Behaviour |
+|--------|-----------|
+| `s` steer | Delivers the message before the next LLM call (same turn) |
+| `q` queue | Delivers the message after the agent fully finishes |
+| `d` discard | Throws the message away |
+| `e` edit / Esc | Restores the text to the editor |
+
+When the agent is **idle**, messages pass through normally — no prompt appears.
+
+Also registers a `/steer <text>` command that bypasses the prompt and steers directly.
+
+## Install
+
+```bash
+pi install npm:@agnishc/edb-agent-steer
+```
+
+## Usage
+
+Just type while the agent is running. The prompt appears automatically.
+
+```
+/steer reconsider the approach — use a queue instead
+```
+
+## License
+
+[MIT](LICENSE) © Agnish Chakraborty

package/package.json

@@ -0,0 +1,26 @@
+{
+	"name": "@agnishc/edb-agent-steer",
+	"version": "0.1.0",
+	"description": "Pi extension: intercepts mid-turn messages with a steer / queue / discard / edit prompt",
+	"keywords": ["pi-package", "pi-extension", "edb"],
+	"type": "module",
+	"license": "MIT",
+	"author": "Agnish Chakraborty",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/agnishcc/pi-extention-monorepo.git",
+		"directory": "packages/edb-agent-steer"
+	},
+	"homepage": "https://github.com/agnishcc/pi-extention-monorepo/tree/main/packages/edb-agent-steer#readme",
+	"bugs": { "url": "https://github.com/agnishcc/pi-extention-monorepo/issues" },
+	"publishConfig": { "access": "public" },
+	"scripts": { "test": "vitest run" },
+	"files": ["src", "README.md", "LICENSE", "CHANGELOG.md"],
+	"pi": {
+		"extensions": ["./src/index.ts"]
+	},
+	"peerDependencies": {
+		"@mariozechner/pi-coding-agent": "*",
+		"@mariozechner/pi-tui": "*"
+	}
+}

package/src/component.ts

@@ -0,0 +1,52 @@
+import { matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
+import type { SteerChoice } from "./types";
+
+// ── Component ──────────────────────────────────────────────────────────────────
+
+export class SteerPromptComponent {
+	private cachedWidth?: number;
+	private cachedLines?: string[];
+
+	constructor(
+		private readonly message: string,
+		private readonly theme: any,
+		private readonly done: (choice: SteerChoice) => void,
+	) {}
+
+	handleInput(data: string): void {
+		if (matchesKey(data, "s")) this.done("steer");
+		else if (matchesKey(data, "q")) this.done("queue");
+		else if (matchesKey(data, "d")) this.done("discard");
+		else if (matchesKey(data, "e") || matchesKey(data, "escape")) this.done("edit");
+	}
+
+	render(width: number): string[] {
+		if (this.cachedLines && this.cachedWidth === width) return this.cachedLines;
+
+		const th = this.theme;
+		const maxMsgW = Math.max(10, width - 8);
+		const preview = this.message.length > maxMsgW ? `${this.message.slice(0, maxMsgW - 1)}…` : this.message;
+
+		const key = (k: string, label: string) => `${th.fg("accent", k)}${th.fg("dim", ` ${label}`)}`;
+
+		const lines: string[] = [
+			"",
+			truncateToWidth(`  ${th.fg("muted", "↵")}  ${th.fg("dim", `"${preview}"`)}`, width),
+			"",
+			truncateToWidth(
+				`  ${key("s", "steer")}   ${key("q", "queue")}   ${key("d", "discard")}   ${key("e", "edit")}`,
+				width,
+			),
+			"",
+		];
+
+		this.cachedWidth = width;
+		this.cachedLines = lines;
+		return lines;
+	}
+
+	invalidate(): void {
+		this.cachedWidth = undefined;
+		this.cachedLines = undefined;
+	}
+}

package/src/index.ts

@@ -0,0 +1,93 @@
+/**
+ * pi-agent-steer
+ *
+ * When the user submits a message while the agent is already running, a
+ * compact inline prompt appears showing four single-key options:
+ *
+ *   s  steer   — deliver before the next LLM call (same session)
+ *   q  queue   — deliver after the agent fully finishes
+ *   d  discard — throw the message away
+ *   e  edit    — restore the text to the editor (also: Esc)
+ *
+ * A single keypress acts immediately — no Enter, no dialog, no Esc conflicts.
+ * When the agent is idle the message is passed through normally — no prompt.
+ * The /steer <text> slash command skips the prompt entirely and always steers directly.
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { SteerPromptComponent } from "./component";
+import type { SteerChoice } from "./types";
+
+// ── Extension ──────────────────────────────────────────────────────────────────
+
+export default function steerExtension(pi: ExtensionAPI): void {
+	// ── Input interceptor ─────────────────────────────────────────────────────
+	pi.on("input", async (event, ctx) => {
+		// Only intercept typed messages, not extension-injected ones.
+		if (event.source !== "interactive") return { action: "continue" };
+
+		// Don't intercept empty input — Esc in the editor fires input with
+		// event.text === "" and we must not block that.
+		if (!event.text?.trim()) return { action: "continue" };
+
+		// Agent is idle — pass through normally.
+		if (ctx.isIdle()) return { action: "continue" };
+
+		// No interactive UI (print / JSON mode) — pass through.
+		if (!ctx.hasUI) return { action: "continue" };
+
+		// Show the single-keypress prompt.
+		const choice = await ctx.ui.custom<SteerChoice>(
+			(_tui, theme, _kb, done) => new SteerPromptComponent(event.text, theme, done),
+		);
+
+		// edit / Esc / unexpected null → restore text to editor, do nothing else.
+		if (!choice || choice === "edit") {
+			ctx.ui.setEditorText(event.text);
+			return { action: "handled" };
+		}
+
+		if (choice === "discard") {
+			ctx.ui.notify("Message discarded", "info");
+			return { action: "handled" };
+		}
+
+		if (choice === "steer") {
+			pi.sendUserMessage(event.text, { deliverAs: "steer" });
+			ctx.ui.notify("◀ Steered — before the next LLM call", "info");
+			return { action: "handled" };
+		}
+
+		if (choice === "queue") {
+			pi.sendUserMessage(event.text, { deliverAs: "followUp" });
+			ctx.ui.notify("⏳ Queued — after agent finishes", "info");
+			return { action: "handled" };
+		}
+
+		return { action: "continue" };
+	});
+
+	// ── /steer slash command — explicit steer, bypasses the prompt ────────────
+	pi.registerCommand("steer", {
+		description: "Steer a message directly into the running agent turn. " + "Usage: /steer <text>",
+
+		handler: async (args, ctx) => {
+			const content = args?.trim();
+
+			if (!content) {
+				ctx.ui.notify("/steer <text>  — steer text before the next LLM call", "info");
+				return;
+			}
+
+			const running = !ctx.isIdle();
+
+			if (running) {
+				pi.sendUserMessage(content, { deliverAs: "steer" });
+			} else {
+				pi.sendUserMessage(content);
+			}
+
+			ctx.ui.notify(running ? "◀ Steered — before the next LLM call" : "◀ Steered — triggering response", "info");
+		},
+	});
+}

package/src/types.ts

@@ -0,0 +1,3 @@
+// ── Types ──────────────────────────────────────────────────────────────────────
+
+export type SteerChoice = "steer" | "queue" | "discard" | "edit";