@aexhq/agentloop-pi 2.1.0 → 3.0.0

package/README.md

@@ -17,10 +17,10 @@ pi's steering and follow-up queues and per-tool `executionMode` are host-app sea
 equivalent and are not ported.
 
 ```ts
-import { brainWasm } from "@aexhq/brain";
+import { brainEnv } from "@aexhq/brain";
 import { pi } from "@aexhq/agentloop-pi";
 
-const loopRuntime = brainWasm();
+const loopRuntime = brainEnv({ name: "brain" });
 const session = await brain.sessions.create({
   agentloop: pi({ env: loopRuntime, contextWindow: 200_000 }),
   model,
@@ -32,8 +32,14 @@ The component is built by this package's publisher. Brain consumes the resulting
 does not compile its JavaScript source.
 
 The loop reads paginated session Events before each turn and saves its observation cursor in
-slots. Interrupted turns and environment failures enter the transcript as runtime observations.
+kv. Interrupted turns and environment failures enter the transcript as runtime observations.
 A failed tool result goes back to the model with `is_error`; the loop does not retry it automatically.
-Brain can release execution between turns without losing this transcript or cursor. Physical
-browser and sandbox state follows the Environment provider's TTL and may be lost. The official
-`tool-env` inspection and lifecycle tool is planned after the MVP.
+Brain can release execution between turns without losing this transcript or cursor. The caller
+controls Environment lifetime; a browser closure or provider resource loss can still destroy its
+physical state.
+
+By default the model sees canonical Tool schemas. A Tool with multiple placements requires an
+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.

package/dist/index.mjs

@@ -2,6 +2,8 @@
 import { agentloop, component } from "@aexhq/brain";
 import { z } from "zod";
 var options = z.object({
+  environmentSelection: z.enum(["hidden", "model"]).default("hidden"),
+  placements: z.record(z.string(), z.string()).default({}),
   contextWindow: z.number().int().positive().default(2e5),
   // pi defaults (compaction.ts): compact when context exceeds
   // contextWindow - reserveTokens, keep ~keepRecentTokens of recent messages.

package/index.d.ts

@@ -1,7 +1,9 @@
-import type { AgentloopBinding, Environment } from "@aexhq/brain";
+import type { PlacedAgentloop, Environment } from "@aexhq/brain";
 
 export interface PiOptions {
   readonly env: Environment;
+  readonly environmentSelection?: "hidden" | "model";
+  readonly placements?: Readonly<Record<string, string>>;
   /** Model context window in tokens the compaction budget is measured against. Default 200000. */
   readonly contextWindow?: number;
   /** Compact when the estimated context exceeds contextWindow - reserveTokens. pi default 16384. */
@@ -12,4 +14,4 @@ export interface PiOptions {
   readonly compaction?: boolean;
 }
 
-export declare const pi: (options: PiOptions) => AgentloopBinding;
+export declare const pi: (options: PiOptions) => PlacedAgentloop;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexhq/agentloop-pi",
-  "version": "2.1.0",
+  "version": "3.0.0",
   "description": "Pi-style parallel-Tool Brain extension",
   "license": "MIT",
   "repository": {
@@ -34,7 +34,7 @@
     "prepack": "npm run build"
   },
   "dependencies": {
-    "@aexhq/brain": "0.17.0",
+    "@aexhq/brain": "0.18.0",
     "zod": "4.4.3"
   },
   "devDependencies": {}

package/src/component.mjs

@@ -19,7 +19,7 @@ export async function turn(input) {
     const output = await runPi({
       input: JSON.parse(input.inputJson),
       transcript: JSON.parse(input.transcriptJson),
-      slots: JSON.parse(input.slotsJson),
+      kv: JSON.parse(input.kvJson),
       events: JSON.parse(input.eventsJson),
       configuration: JSON.parse(input.configurationJson),
       system: input.system,
@@ -33,7 +33,7 @@ export async function turn(input) {
     });
     return {
       transcriptJson: JSON.stringify(output.transcript),
-      slotsJson: JSON.stringify(output.slots),
+      kvJson: JSON.stringify(output.kv),
       resultJson: output.result === undefined ? undefined : JSON.stringify(output.result),
     };
   } catch (error) {

package/src/index.mjs

@@ -1,6 +1,8 @@
 import { agentloop, component } from "@aexhq/brain";
 import { z } from "zod";
 const options = z.object({
+  environmentSelection: z.enum(["hidden", "model"]).default("hidden"),
+  placements: z.record(z.string(), z.string()).default({}),
   contextWindow: z.number().int().positive().default(200_000),
   // pi defaults (compaction.ts): compact when context exceeds
   // contextWindow - reserveTokens, keep ~keepRecentTokens of recent messages.

package/src/logic.mjs

@@ -1,3 +1,4 @@
+import { toolPlacement } from "../../../shared/tool-placement.mjs";
 import { observeEvents } from "../../../shared/loop-events.mjs";
 
 const CHECKPOINT_PREFIX = "Context checkpoint from earlier in this conversation:\n\n";
@@ -59,9 +60,10 @@ export async function runPi(input, context) {
     compaction: true,
     ...input.configuration,
   };
+  const placement = toolPlacement(input.tools, options);
   const transcript = cloneJson(input.transcript);
-  const observed_sequence = await observeEvents(context, transcript, input.slots.observed_sequence ?? 0);
-  const saved = input.slots.checkpoint;
+  const observed_sequence = await observeEvents(context, transcript, input.kv.observed_sequence ?? 0);
+  const saved = input.kv.checkpoint;
   const checkpoint = saved === undefined ? { summary: null } : cloneJson(saved);
   const body = () => checkpoint.summary === null ? transcript : transcript.slice(1);
   const shouldCompact = () =>
@@ -86,6 +88,7 @@ export async function runPi(input, context) {
     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({
+      tools: [],
       messages: [{ role: "user", content: [{ type: "text", text: `${previous}${serializeConversation(messages.slice(0, cut))}\n\n${prompt}` }] }],
     });
     checkpoint.summary = text(message);
@@ -97,14 +100,14 @@ export async function runPi(input, context) {
   transcript.push({ role: "user", content: [{ type: "text", text: input.input.message }] });
   for (;;) {
     if (shouldCompact()) await compact();
-    const { message, stop_reason } = await context.model({ messages: transcript });
+    const { message, stop_reason } = await context.model({ messages: transcript, tools: placement.definitions });
     transcript.push(message);
     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, slots: { checkpoint, observed_sequence } };
+      return { transcript, kv: { checkpoint, observed_sequence } };
     }
     if (stop_reason === "max_tokens") {
       transcript.push({
@@ -113,7 +116,7 @@ export async function runPi(input, context) {
       });
       continue;
     }
-    const results = await context.dispatch(calls);
+    const results = await context.dispatch(calls.map(placement.invocation));
     const byCall = new Map(results.map((result) => [result.call_id, result]));
     transcript.push({
       role: "user",