@aexhq/agentloop-pi 3.0.0 → 5.0.0

package/README.md

@@ -43,3 +43,15 @@ explicit `placements: { toolName: "environmentName" }` option. Set
 `environmentSelection: "model"` to expose each Tool's authorized Environment names in its input
 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 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.
+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. Keep their matching server/artifacts for recovery and create
+new sessions explicitly when adopting Brain 0.20 / version 5 extensions; do not delete old data
+as an implicit upgrade step.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexhq/agentloop-pi",
-  "version": "3.0.0",
+  "version": "5.0.0",
   "description": "Pi-style parallel-Tool Brain extension",
   "license": "MIT",
   "repository": {
@@ -15,7 +15,7 @@
   },
   "publishConfig": {
     "access": "public",
-    "provenance": false,
+    "provenance": true,
     "tag": "next"
   },
   "exports": {
@@ -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.18.0",
+    "@aexhq/brain": "0.20.0",
     "zod": "4.4.3"
   },
   "devDependencies": {}

package/src/component.mjs

@@ -19,12 +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))),
+      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)))),
@@ -32,8 +40,6 @@ export async function turn(input) {
       telemetry: (record) => host.telemetry(JSON.stringify(record ?? null)),
     });
     return {
-      transcriptJson: JSON.stringify(output.transcript),
-      kvJson: JSON.stringify(output.kv),
       resultJson: output.result === undefined ? undefined : JSON.stringify(output.result),
     };
   } catch (error) {

package/src/logic.mjs

@@ -45,6 +45,8 @@ const text = (message) => message.content.filter((block) => block.type === "text
 const blockText = (block) => {
   if (block.type === "text") return block.text;
   if (block.type === "tool_use") return `[tool_use ${block.name}] ${JSON.stringify(block.input)}`;
+  if (block.type === "native") return `[provider continuation state: ${block.format}]`;
+  if (block.type === "image") return block.url.startsWith("https://") ? `[image ${block.url}]` : "[embedded image]";
   const output = typeof block.content === "string" ? block.content : JSON.stringify(block.content);
   return `[tool_result${block.is_error ? " (error)" : ""}] ${output.length > 2000 ? `${output.slice(0, 2000)}…` : output}`;
 };
@@ -62,8 +64,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;
+  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 = () =>
@@ -87,27 +90,36 @@ export async function runPi(input, context) {
     if (cut === 0) return;
     const previous = checkpoint.summary === null ? "" : `Previous summary:\n\n${checkpoint.summary}\n\n`;
     const prompt = checkpoint.summary === null ? SUMMARIZATION_PROMPT : `${UPDATE_RULES}${SUMMARIZATION_PROMPT}`;
-    const { message } = await context.model({
+    const { message, stop_reason } = await context.model({
+      response_format: null,
       tools: [],
       messages: [{ role: "user", content: [{ type: "text", text: `${previous}${serializeConversation(messages.slice(0, cut))}\n\n${prompt}` }] }],
     });
+    if (stop_reason !== "end_turn") throw new Error(`Compaction did not complete: ${stop_reason}`);
     checkpoint.summary = text(message);
     transcript.splice(0, transcript.length,
       { role: "user", content: [{ type: "text", text: `${CHECKPOINT_PREFIX}${checkpoint.summary}` }] },
       ...messages.slice(cut));
   };
 
-  transcript.push({ role: "user", content: [{ type: "text", text: input.input.message }] });
+  transcript.push({ role: "user", content: [{ type: "text", text: input.input.message }, ...(input.input.media ?? [])] });
   for (;;) {
-    if (shouldCompact()) await compact();
+    await context.setTranscript(transcript);
+    if (shouldCompact()) {
+      await compact();
+      await context.setTranscript(transcript);
+      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.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 }));
     if (calls.length === 0) {
       await context.emit("output_emitted", { type: "assistant_message", message: text(message) });
-      return { transcript, kv: { checkpoint, observed_sequence } };
+      return {};
     }
     if (stop_reason === "max_tokens") {
       transcript.push({