@aexhq/agentloop-pi 4.0.0 → 5.0.0

package/README.md

@@ -44,11 +44,14 @@ 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.
 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.
+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": "4.0.0",
+  "version": "5.0.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.20.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

@@ -64,9 +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;
-  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 +108,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 }));