@aexhq/agentloop-pi 4.0.0 → 5.1.0

package/README.md

@@ -44,11 +44,17 @@ explicit `placements: { toolName: "environmentName" }` option. Set
 schema. The loop unwraps that choice before dispatch; Brain validates the selected pair and the
 canonical input. Both modes use the same session Tool placements.
 
-Transcript and kv changes are saved inline through Brain's state services. A later model or
+Transcript changes and `ctx.kv.read/put/delete` use Brain's state services. KV mutations are
+committed inline; missing keys remain distinct from stored JSON null. A later model or
 Tool failure preserves acknowledged writes; turn output contains only the result. Retained
 native model blocks pass through unchanged, and user images enter model context as media.
+Version 5.1 also presents successful Tool outputs shaped as
+`{ type: "aex_tool_output", version: 1, content, media: [{ type: "image", url }] }`
+as Tool-result content and image media. Browser and MCP extensions use this presentation
+convention; ordinary JSON results keep their existing behavior.
 Compaction explicitly resets the response format and installs a summary only after `end_turn`;
 a truncated, refused, or unknown summary leaves the original saved context intact.
 
-Rebuild Components against the matching Brain WIT when upgrading. Existing sessions keep their
-immutable loop implementation and must be recreated for this interface change.
+Version 5.1 targets Brain 0.22 and preserves the existing WIT. Existing sessions keep their
+immutable loop implementation; create new sessions to adopt the updated loop. Keep matching
+server/artifacts for recovery. An upgrade does not migrate or delete session data.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexhq/agentloop-pi",
-  "version": "4.0.0",
+  "version": "5.1.0",
   "description": "Pi-style parallel-Tool Brain extension",
   "license": "MIT",
   "repository": {
@@ -30,11 +30,12 @@
   ],
   "scripts": {
     "build": "node ../../tools/build-agentloop.mjs src/component.mjs src/index.mjs dist",
-    "test": "npm run build && node --test",
+    "test": "npm run build && node --test *.test.mjs",
+    "test:journeys": "node --test test/journeys.mjs",
     "prepack": "npm run build"
   },
   "dependencies": {
-    "@aexhq/brain": "0.19.0",
+    "@aexhq/brain": "0.22.0",
     "zod": "4.4.3"
   },
   "devDependencies": {}

package/src/component.mjs

@@ -19,14 +19,20 @@ export async function turn(input) {
     const output = await runPi({
       input: JSON.parse(input.inputJson),
       transcript: JSON.parse(input.transcriptJson),
-      kv: JSON.parse(input.kvJson),
       events: JSON.parse(input.eventsJson),
       configuration: JSON.parse(input.configurationJson),
       system: input.system,
       tools: JSON.parse(input.toolsJson),
     }, {
       setTranscript: (messages) => hosted(() => host.setTranscript(JSON.stringify(messages))),
-      setKv: (key, value) => hosted(() => host.setKv(key, JSON.stringify(value))),
+      kv: {
+        read: (key) => {
+          const value = hosted(() => host.kvRead(key));
+          return value === undefined ? undefined : JSON.parse(value);
+        },
+        put: (key, value) => hosted(() => host.kvPut(key, JSON.stringify(value))),
+        delete: (key) => hosted(() => host.kvDelete(key)),
+      },
       events: (after) => JSON.parse(hosted(() => host.events(BigInt(after)))),
       model: (request) => JSON.parse(hosted(() => host.model(JSON.stringify(request)))),
       dispatch: (calls) => JSON.parse(hosted(() => host.dispatch(JSON.stringify(calls)))),

package/src/logic.mjs

@@ -1,5 +1,6 @@
 import { toolPlacement } from "../../../shared/tool-placement.mjs";
 import { observeEvents } from "../../../shared/loop-events.mjs";
+import { toolResult } from "../../../shared/tool-output.mjs";
 
 const CHECKPOINT_PREFIX = "Context checkpoint from earlier in this conversation:\n\n";
 
@@ -64,9 +65,9 @@ export async function runPi(input, context) {
   };
   const placement = toolPlacement(input.tools, options);
   const transcript = cloneJson(input.transcript);
-  const observed_sequence = await observeEvents(context, transcript, input.kv.observed_sequence ?? 0);
-  const saved = input.kv.checkpoint;
-  await context.setKv("observed_sequence", observed_sequence);
+  const observed_sequence = await observeEvents(context, transcript, (await context.kv.read("observed_sequence")) ?? 0);
+  const saved = await context.kv.read("checkpoint");
+  await context.kv.put("observed_sequence", observed_sequence);
   const checkpoint = saved === undefined ? { summary: null } : cloneJson(saved);
   const body = () => checkpoint.summary === null ? transcript : transcript.slice(1);
   const shouldCompact = () =>
@@ -108,12 +109,12 @@ export async function runPi(input, context) {
     if (shouldCompact()) {
       await compact();
       await context.setTranscript(transcript);
-      await context.setKv("checkpoint", checkpoint);
+      await context.kv.put("checkpoint", checkpoint);
     }
     const { message, stop_reason } = await context.model({ messages: transcript, tools: placement.definitions });
     transcript.push(message);
     await context.setTranscript(transcript);
-    await context.setKv("checkpoint", checkpoint);
+    await context.kv.put("checkpoint", checkpoint);
     const calls = message.content
       .filter((block) => block.type === "tool_use")
       .map((block) => ({ call_id: block.id, name: block.name, input: block.input }));
@@ -132,15 +133,7 @@ export async function runPi(input, context) {
     const byCall = new Map(results.map((result) => [result.call_id, result]));
     transcript.push({
       role: "user",
-      content: calls.map(({ call_id }) => {
-        const result = byCall.get(call_id);
-        return {
-          type: "tool_result",
-          tool_use_id: call_id,
-          content: result === undefined ? "Tool produced no result." : result.output,
-          is_error: result === undefined ? true : result.is_error,
-        };
-      }),
+      content: calls.map(({ call_id }) => toolResult(call_id, byCall.get(call_id))),
     });
   }
 }