@aexhq/agentloop-pi 6.1.2 → 6.2.0

package/README.md

@@ -62,7 +62,7 @@ siblings. For hosted applications, publish bytes with Aex's attachment API and r
 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.
 
-Version 6.1 targets Brain SDK 0.24 and preserves the 0.23 WIT and URL-media contract. Existing sessions keep their
+Version 6.2 targets Brain SDK 0.25 and preserves the 0.23 WIT and URL-media contract. 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.
 
@@ -81,3 +81,17 @@ without rebuilding Wasm. Run `npm test -w packages/loop-pi` for the Component bu
 tests, then the compiled journeys described in the repository README. All release gates remain
 required. A rebuilt Component needs a new admission and session. Package source is not a public
 logic-library entry point, and existing sessions keep their admitted implementation.
+
+## Hosted structured output
+
+Set `output: { schema: jsonSchema, maxCorrections: 2 }` on the loop to validate JSON Schema
+2020-12 inside the hosted turn. `z.toJSONSchema(schema)` supplies a schema from Zod. Schemas
+compile strictly before model effects: unknown keywords, unresolved external references and
+unsupported patterns fail explicitly. Standard formats are validated without coercion.
+
+Corrections run without tools and retain previous results. They finish in the same turn when
+using `session.submit()`, without a client process driving extra turns. Only a valid final
+answer emits `assistant_message`; its parsed value is the turn result. Refusal, truncation,
+provider failure and cancellation do not trigger formatting retries. Corrections default to
+two, with a supported range of zero through ten. Application business rules still belong in
+its tools or persistence boundary.

package/dist/index.mjs

@@ -9,7 +9,11 @@ var options = z.object({
   // contextWindow - reserveTokens, keep ~keepRecentTokens of recent messages.
   reserveTokens: z.number().int().positive().default(16384),
   keepRecentTokens: z.number().int().positive().default(2e4),
-  compaction: z.boolean().default(true)
+  compaction: z.boolean().default(true),
+  output: z.strictObject({
+    schema: z.union([z.boolean(), z.record(z.string(), z.unknown())]),
+    maxCorrections: z.number().int().min(0).max(10).default(2)
+  }).optional()
 }).strict();
 var pi = agentloop({
   options,

package/index.d.ts

@@ -12,6 +12,8 @@ export interface PiOptions {
   readonly keepRecentTokens?: number;
   /** Disable automatic compaction entirely. Default true (enabled). */
   readonly compaction?: boolean;
+  /** Hosted JSON Schema 2020-12 validation. Corrections run without tools in the same turn. */
+  readonly output?: { readonly schema: boolean | Readonly<Record<string, unknown>>; readonly maxCorrections?: number };
 }
 
 export declare const pi: (options: PiOptions) => PlacedAgentloop;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexhq/agentloop-pi",
-  "version": "6.1.2",
+  "version": "6.2.0",
   "description": "Pi-style parallel-Tool Brain extension",
   "license": "MIT",
   "repository": {
@@ -19,7 +19,10 @@
     "tag": "next"
   },
   "exports": {
-    ".": { "types": "./index.d.ts", "import": "./dist/index.mjs" },
+    ".": {
+      "types": "./index.d.ts",
+      "import": "./dist/index.mjs"
+    },
     "./package.json": "./package.json"
   },
   "files": [
@@ -37,8 +40,9 @@
     "prepack": "npm run build"
   },
   "dependencies": {
-    "@aexhq/brain": "0.24.3",
+    "@aexhq/brain": "0.25.1",
+    "ajv": "8.20.0",
+    "ajv-formats": "3.0.1",
     "zod": "4.4.3"
-  },
-  "devDependencies": {}
+  }
 }

package/src/index.mjs

@@ -9,6 +9,8 @@ const options = z.object({
   reserveTokens: z.number().int().positive().default(16_384),
   keepRecentTokens: z.number().int().positive().default(20_000),
   compaction: z.boolean().default(true),
+  output: z.strictObject({ schema: z.union([z.boolean(), z.record(z.string(), z.unknown())]),
+    maxCorrections: z.number().int().min(0).max(10).default(2) }).optional(),
 }).strict();
 
 export const pi = agentloop({

package/src/logic.mjs

@@ -1,6 +1,7 @@
 import { toolPlacement } from "../../../shared/tool-placement.mjs";
 import { observeEvents } from "../../../shared/loop-events.mjs";
 import { toolResult } from "../../../shared/tool-output.mjs";
+import { structuredOutput } from "../../../shared/structured-output.mjs";
 
 const CHECKPOINT_PREFIX = "Context checkpoint from earlier in this conversation:\n\n";
 
@@ -64,6 +65,7 @@ export async function runPi(input, context) {
     ...input.configuration,
   };
   const placement = toolPlacement(input.tools, options);
+  const output = structuredOutput(options.output);
   const transcript = cloneJson(input.transcript);
   const observed_sequence = await observeEvents(context, transcript, (await context.kv.read("observed_sequence")) ?? 0);
   const saved = await context.kv.read("checkpoint");
@@ -111,7 +113,7 @@ export async function runPi(input, context) {
       await context.setTranscript(transcript);
       await context.kv.put("checkpoint", checkpoint);
     }
-    const { message, stop_reason } = await context.model({ messages: transcript, tools: placement.definitions });
+    const { message, stop_reason } = await context.model(output?.request(transcript, placement.definitions) ?? { messages: transcript, tools: placement.definitions });
     transcript.push(message);
     await context.setTranscript(transcript);
     await context.kv.put("checkpoint", checkpoint);
@@ -119,9 +121,16 @@ export async function runPi(input, context) {
       .filter((block) => block.type === "tool_use")
       .map((block) => ({ call_id: block.id, name: block.name, input: block.input }));
     if (calls.length === 0) {
+      const final = output?.accept(text(message), stop_reason);
+      if (final?.correction) {
+        transcript.push(final.correction);
+        await context.setTranscript(transcript);
+        continue;
+      }
       await context.emit("output_emitted", { type: "assistant_message", message: text(message) });
-      return {};
+      return final ? { result: final.value } : {};
     }
+    if (output?.correcting) output.rejectTools();
     if (stop_reason === "max_tokens") {
       transcript.push({
         role: "user",