omp-auto-loop 0.2.2 → 0.2.4

package/index.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI, ExtensionContext } from "@oh-my-pi/pi-coding-agent";
 import { buildPrompt, emptyState, getSystemPromptAddition, type LoopState, updateWidget } from "./state.ts";
+
 import { getLoopControlToolDefinition, handleLoopControlTool, renderLoopControlCall, renderLoopControlResult } from "./tool.ts";
 
 export default function (pi: ExtensionAPI) {
@@ -61,19 +62,21 @@ export default function (pi: ExtensionAPI) {
 
 		if (state.nextScheduled) {
 			state.nextScheduled = false;
+			pi.sendMessage(
+				{ customType: "loop-iteration", content: buildPrompt(state), display: false },
+				{ triggerTurn: true, deliverAs: "nextTurn" }
+			);
 			return;
 		}
 
 		// Fallback: LLM stopped without calling loop_control or confirming done
 		state.currentStep++;
 		updateWidget(state, ctx);
-		setTimeout(() => {
-			pi.sendMessage({
-				customType: "loop-fallback",
-				content: "You stopped without calling `loop_control`. If the task is incomplete, continue working. If done, call `loop_control` with status 'done'.",
-				display: false
-			}, { triggerTurn: true, deliverAs: "steer" });
-		}, 100);
+		pi.sendMessage({
+			customType: "loop-fallback",
+			content: "You stopped without calling `loop_control`. If the task is incomplete, continue working. If done, call `loop_control` with status 'done'.",
+			display: false
+		}, { triggerTurn: true, deliverAs: "nextTurn" });
 	});
 
 	pi.registerTool({
@@ -84,15 +87,15 @@ export default function (pi: ExtensionAPI) {
 			updateWidget(state, ctx);
 			return { content: result.content, details: result.details };
 		},
-		renderCall: renderLoopControlCall as any,
-		renderResult: renderLoopControlResult as any,
-	});
-
-	const stopLoop = (ctx: ExtensionContext, reason: string) => {
-		if (!state.active) {
-			ctx.ui.notify("No active loop", "info");
-			return;
-		}
+                renderCall: renderLoopControlCall as any,
+                renderResult: renderLoopControlResult as any,
+        });
+
+        const stopLoop = (ctx: ExtensionContext, reason: string) => {
+                if (!state.active) {
+                        ctx.ui.notify("No active loop", "info");
+                        return;
+                }
 		state.active = false;
 		state.done = true;
 		state.reasonDone = reason;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "omp-auto-loop",
-	"version": "0.2.2",
+	"version": "0.2.4",
 	"description": "General-purpose auto agent loop plugin for omp.",
 	"type": "module",
 	"main": "index.ts",

package/state.ts

@@ -25,15 +25,13 @@ export function buildPrompt(state: LoopState): string {
 
 export function updateWidget(state: LoopState, ctx: ExtensionContext) {
         if (!state.active) {
-                ctx.ui.setStatus("loop", undefined);
                 ctx.ui.setWidget("loop", undefined);
                 return;
         }
         const label = `iteration ${state.currentStep + 1}`;
-        ctx.ui.setStatus("loop", `🔄 ${label}`);
         ctx.ui.setWidget("loop", [
                 `┌─ Loop ──────────`,
-                `│ ${label}`,
+                `│ 🔄 ${label}`,
                 `└─ Ctrl+Shift+X to stop ─`,
         ]);
 }

package/tool.ts

@@ -33,9 +33,6 @@ export function handleLoopControlTool(opts: {
         }
 
         const newState = { ...state, currentStep: state.currentStep + 1, confirmingDone: false, nextScheduled: true };
-        setTimeout(() => {
-                pi.sendMessage({ customType: "loop-iteration", content: buildPrompt(newState), display: false }, { triggerTurn: true, deliverAs: "steer" });
-        }, 100);
 
         return {
                 content: [{ type: "text", text: `→ Advancing to step ${newState.currentStep + 1}. Summary: ${params.summary}` }],
@@ -57,8 +54,15 @@ export function getLoopControlToolDefinition() {
         };
 }
 
-export function renderLoopControlCall(args: { status: string }, theme: any) {
-        return new Text(theme.fg("toolTitle", theme.bold("loop_control ")) + theme.fg(args.status === "done" ? "success" : "accent", args.status), 0, 0);
+export function renderLoopControlCall(args: { status: string; summary?: string; reason?: string }, theme: any) {
+        let text = theme.fg("toolTitle", theme.bold("loop_control ")) + theme.fg(args.status === "done" ? "success" : "accent", args.status);
+        if (args.summary) {
+                text += `\n${theme.dim("Summary: ")}${args.summary}`;
+        }
+        if (args.reason) {
+                text += `\n${theme.dim("Reason: ")}${args.reason}`;
+        }
+        return new Text(text, 0, 0);
 }
 
 export function renderLoopControlResult(result: { details?: LoopState }, _opts: unknown, theme: any) {

package/test-run.ts

@@ -1,2 +0,0 @@
-import { loadExtensions } from "@oh-my-pi/pi-coding-agent/src/extensibility/extensions/loader";
-console.log("loader available");