@frockbot/plugin-shell 0.3.20 → 0.3.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@frockbot/plugin-shell",
-  "version": "0.3.20",
+  "version": "0.3.21",
   "private": false,
   "type": "module",
   "exports": {
@@ -52,42 +52,42 @@
   },
   "dependencies": {
     "@cordisjs/plugin-webui": "0.8.2",
-    "@frockbot/agent-runtime": "0.3.20",
-    "@frockbot/application-foundation": "0.3.20",
-    "@frockbot/catalog-core": "0.3.20",
-    "@frockbot/client-core": "0.3.20",
-    "@frockbot/client-ui": "0.3.20",
-    "@frockbot/computer-core": "0.3.20",
-    "@frockbot/computer-host-protocol": "0.3.20",
-    "@frockbot/configuration-core": "0.3.20",
-    "@frockbot/connection-core": "0.3.20",
-    "@frockbot/kernel-agent-loop": "0.3.20",
-    "@frockbot/kernel-composition": "0.3.20",
-    "@frockbot/kernel-contracts": "0.3.20",
-    "@frockbot/kernel-do": "0.3.20",
-    "@frockbot/machine-protocol": "0.3.20",
-    "@frockbot/plugin-applets": "0.3.20",
-    "@frockbot/plugin-authoring": "0.3.20",
-    "@frockbot/plugin-bot-template": "0.3.20",
-    "@frockbot/plugin-computer": "0.3.20",
-    "@frockbot/plugin-flock": "0.3.20",
-    "@frockbot/plugin-image": "0.3.20",
-    "@frockbot/plugin-machine-messages": "0.3.20",
-    "@frockbot/plugin-mcp": "0.3.20",
-    "@frockbot/plugin-memory": "0.3.20",
-    "@frockbot/plugin-package-catalog": "0.3.20",
-    "@frockbot/plugin-routines": "0.3.20",
-    "@frockbot/plugin-skills": "0.3.20",
-    "@frockbot/plugin-subagents": "0.3.20",
-    "@frockbot/plugin-user-machine": "0.3.20",
-    "@frockbot/protocol": "0.3.20",
+    "@frockbot/agent-runtime": "0.3.21",
+    "@frockbot/application-foundation": "0.3.21",
+    "@frockbot/catalog-core": "0.3.21",
+    "@frockbot/client-core": "0.3.21",
+    "@frockbot/client-ui": "0.3.21",
+    "@frockbot/computer-core": "0.3.21",
+    "@frockbot/computer-host-protocol": "0.3.21",
+    "@frockbot/configuration-core": "0.3.21",
+    "@frockbot/connection-core": "0.3.21",
+    "@frockbot/kernel-agent-loop": "0.3.21",
+    "@frockbot/kernel-composition": "0.3.21",
+    "@frockbot/kernel-contracts": "0.3.21",
+    "@frockbot/kernel-do": "0.3.21",
+    "@frockbot/machine-protocol": "0.3.21",
+    "@frockbot/plugin-applets": "0.3.21",
+    "@frockbot/plugin-authoring": "0.3.21",
+    "@frockbot/plugin-bot-template": "0.3.21",
+    "@frockbot/plugin-computer": "0.3.21",
+    "@frockbot/plugin-flock": "0.3.21",
+    "@frockbot/plugin-image": "0.3.21",
+    "@frockbot/plugin-machine-messages": "0.3.21",
+    "@frockbot/plugin-mcp": "0.3.21",
+    "@frockbot/plugin-memory": "0.3.21",
+    "@frockbot/plugin-package-catalog": "0.3.21",
+    "@frockbot/plugin-routines": "0.3.21",
+    "@frockbot/plugin-skills": "0.3.21",
+    "@frockbot/plugin-subagents": "0.3.21",
+    "@frockbot/plugin-user-machine": "0.3.21",
+    "@frockbot/protocol": "0.3.21",
     "cordis": "4.0.0-rc.8",
     "vue": "3.5.41"
   },
   "devDependencies": {
     "@cloudflare/workers-types": "latest",
-    "@frockbot/plugin-prompt": "0.3.20",
-    "@frockbot/plugin-tools": "0.3.20",
+    "@frockbot/plugin-prompt": "0.3.21",
+    "@frockbot/plugin-tools": "0.3.21",
     "@types/bun": "1.3.6",
     "@types/node": "26.2.0",
     "@vitejs/plugin-vue": "6.0.8",

package/src/agent.test.ts

@@ -7,6 +7,7 @@ import {
   type ToolCall,
   type ToolExecutionContext,
   type TurnTypeV1,
+  TURN_DEADLINE_MS_V1,
 } from "@frockbot/kernel-contracts";
 import { SystemPromptRegistry } from "@frockbot/plugin-prompt";
 import { ToolRegistry } from "@frockbot/plugin-tools";
@@ -19,6 +20,7 @@ import {
   PARENT_HANDOFF_CAPABILITY_V1,
   SEND_MESSAGE_ALIAS_V1,
   SEND_TO_USER_TOOL_V1,
+  TIME_BUDGET_WARNING_MS_V1,
   USER_VOICE_CAPABILITY_V1,
   WAKE_PARENT_TOOL_V1,
 } from "./agent.ts";
@@ -575,6 +577,36 @@ describe("the acknowledgement reaches the user", () => {
     }
   });
 
+  test("the last two minutes of a Turn tell the model to send a status", async () => {
+    const mounted = await mount();
+    try {
+      const assemble = (remainingMs: number) =>
+        mounted.root.systemPrompt.assemble({
+          sessionId: "session-1",
+          provider: "provider-1",
+          model: "model-1",
+          turnType: "chat",
+          step: { current: 61, max: 64 },
+          deadline: {
+            at: TURN_DEADLINE_MS_V1,
+            now: TURN_DEADLINE_MS_V1 - remainingMs,
+          },
+        });
+      const boundary = await assemble(TIME_BUDGET_WARNING_MS_V1);
+      expect(boundary.text).not.toContain("<time_budget>");
+      const warning = await assemble(TIME_BUDGET_WARNING_MS_V1 - 1);
+      expect(warning.text).toContain("<time_budget>");
+      expect(warning.text).toContain("fewer than 2 minutes left");
+      expect(warning.text).toContain("Do not start new work");
+      expect(warning.text).toContain(SEND_TO_USER_TOOL_V1);
+      expect(warning.text.indexOf("<time_budget>")).toBeGreaterThan(
+        warning.text.indexOf("<step_budget>"),
+      );
+    } finally {
+      await mounted.dispose();
+    }
+  });
+
   test("a step that already spoke is left alone, and a replay adds nothing", async () => {
     const mounted = await mount();
     try {

package/src/agent.ts

@@ -234,6 +234,11 @@ export const STEP_BUDGET_PROMPT_SECTION_V1 = "step-budget";
 export const STEP_BUDGET_PROMPT_ORDER_V1 = 90;
 /** The section appears when this many steps or fewer remain after this one. */
 export const STEP_BUDGET_WARNING_STEPS_V1 = 3;
+export const TIME_BUDGET_PROMPT_SECTION_V1 = "time-budget";
+/** Immediately after the step warning, so the time warning is read last. */
+export const TIME_BUDGET_PROMPT_ORDER_V1 = 91;
+/** The section appears only once strictly fewer than two minutes remain. */
+export const TIME_BUDGET_WARNING_MS_V1 = 2 * 60_000;
 
 /**
  * What the model is told when its reply is about to be stopped.
@@ -265,6 +270,23 @@ export function stepBudgetPromptTextV1(context: {
   ].join("\n");
 }
 
+/** Warns by wall-clock budget even when the model has used few steps. */
+export function timeBudgetPromptTextV1(context: {
+  deadline?: { at: number; now: number };
+}): string {
+  const deadline = context.deadline;
+  if (!deadline) return "";
+  const remaining = deadline.at - deadline.now;
+  if (!Number.isFinite(remaining) || remaining >= TIME_BUDGET_WARNING_MS_V1) {
+    return "";
+  }
+  return [
+    "<time_budget>",
+    `This Turn has fewer than 2 minutes left before it is stopped automatically. Do not start new work. Call \`${SEND_TO_USER_TOOL_V1}\` now with a short status for the person: what is finished, what is not, and what they can do next.`,
+    "</time_budget>",
+  ].join("\n");
+}
+
 export const CONVERSATION_PROMPT_TEXT_V1 = [
   "## Talking to the user",
   "",
@@ -469,6 +491,11 @@ export const shellAgentPlugin: Plugin.Function = (ctx) => {
       order: STEP_BUDGET_PROMPT_ORDER_V1,
       render: (context) => stepBudgetPromptTextV1(context),
     }),
+    ctx.systemPrompt.register({
+      id: TIME_BUDGET_PROMPT_SECTION_V1,
+      order: TIME_BUDGET_PROMPT_ORDER_V1,
+      render: (context) => timeBudgetPromptTextV1(context),
+    }),
     ctx.tools.register(
       createSendToUserTool(SEND_TO_USER_TOOL_V1, ctx.sessions),
       userVoice ? { admissionCeiling: userVoice } : undefined,