@caupulican/pi-agent-core 0.90.6 → 0.90.11

package/dist/agent-loop.d.ts

@@ -3,9 +3,16 @@
  * Transforms to Message[] only at the LLM call boundary.
  */
 import { EventStream } from "@caupulican/pi-ai/event-stream";
+import { ToolFailureRecoveryGate } from "./tool-failure-recovery-gate.ts";
 import type { AgentContext, AgentEvent, AgentLoopConfig, AgentMessage, AgentToolCall, StreamFn, ToolCallRepairInfo } from "./types.ts";
-export { resolveRequestPreflightMaxTokens } from "./provider-request-planner.ts";
+export { narrowRequestMaxTokens, resolveRequestPreflightMaxTokens } from "./provider-request-planner.ts";
 export type AgentEventSink = (event: AgentEvent) => Promise<void> | void;
+/** Bounded no-progress state retained across host-owned continuations of one logical prompt. */
+export interface AgentLoopContinuationState {
+    stallWindow: string[];
+    toolFailureRecoveryGate: ToolFailureRecoveryGate;
+}
+export declare function createAgentLoopContinuationState(): AgentLoopContinuationState;
 /**
  * Start an agent loop with a new prompt message.
  * The prompt is added to the context and events are emitted for it.
@@ -20,8 +27,8 @@ export declare function agentLoop(prompts: AgentMessage[], context: AgentContext
  * This cannot be validated here since `convertToLlm` is only called once per turn.
  */
 export declare function agentLoopContinue(context: AgentContext, config: AgentLoopConfig, signal?: AbortSignal, streamFn?: StreamFn): EventStream<AgentEvent, AgentMessage[]>;
-export declare function runAgentLoop(prompts: AgentMessage[], context: AgentContext, config: AgentLoopConfig, emit: AgentEventSink, signal?: AbortSignal, streamFn?: StreamFn): Promise<AgentMessage[]>;
-export declare function runAgentLoopContinue(context: AgentContext, config: AgentLoopConfig, emit: AgentEventSink, signal?: AbortSignal, streamFn?: StreamFn): Promise<AgentMessage[]>;
+export declare function runAgentLoop(prompts: AgentMessage[], context: AgentContext, config: AgentLoopConfig, emit: AgentEventSink, signal?: AbortSignal, streamFn?: StreamFn, continuationState?: AgentLoopContinuationState): Promise<AgentMessage[]>;
+export declare function runAgentLoopContinue(context: AgentContext, config: AgentLoopConfig, emit: AgentEventSink, signal?: AbortSignal, streamFn?: StreamFn, continuationState?: AgentLoopContinuationState): Promise<AgentMessage[]>;
 /**
  * Start one provider request through the canonical agent-loop boundary.
  *

package/dist/agent-loop.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../src/agent-loop.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAoC7D,OAAO,KAAK,EACX,YAAY,EACZ,UAAU,EACV,eAAe,EACf,YAAY,EAEZ,aAAa,EAEb,QAAQ,EACR,kBAAkB,EAClB,MAAM,YAAY,CAAC;AAIpB,OAAO,EAAE,gCAAgC,EAAE,MAAM,+BAA+B,CAAC;AAEjF,MAAM,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAEzE;;;GAGG;AACH,wBAAgB,SAAS,CACxB,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,WAAW,CAAC,UAAU,EAAE,YAAY,EAAE,CAAC,CAEzC;AAED;;;;;;;GAOG;AACH,wBAAgB,iBAAiB,CAChC,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,WAAW,CAAC,UAAU,EAAE,YAAY,EAAE,CAAC,CAIzC;AAuBD,wBAAsB,YAAY,CACjC,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,IAAI,EAAE,cAAc,EACpB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,OAAO,CAAC,YAAY,EAAE,CAAC,CAgBzB;AAED,wBAAsB,oBAAoB,CACzC,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,IAAI,EAAE,cAAc,EACpB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,OAAO,CAAC,YAAY,EAAE,CAAC,CAWzB;AAmVD;;;;;;GAMG;AACH,wBAAsB,yBAAyB,CAC9C,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,QAAQ,CAAC,EAAE,QAAQ,GACjB,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAExC;AAwiCD,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,aAAa,GAAG,kBAAkB,GAAG,SAAS,CAO7F"}
+{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../src/agent-loop.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AA4B7D,OAAO,EAGN,uBAAuB,EAGvB,MAAM,iCAAiC,CAAC;AAGzC,OAAO,KAAK,EACX,YAAY,EACZ,UAAU,EACV,eAAe,EACf,YAAY,EAEZ,aAAa,EAEb,QAAQ,EACR,kBAAkB,EAClB,MAAM,YAAY,CAAC;AAIpB,OAAO,EAAE,sBAAsB,EAAE,gCAAgC,EAAE,MAAM,+BAA+B,CAAC;AAEzG,MAAM,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAEzE,gGAAgG;AAChG,MAAM,WAAW,0BAA0B;IAC1C,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,uBAAuB,EAAE,uBAAuB,CAAC;CACjD;AAED,wBAAgB,gCAAgC,IAAI,0BAA0B,CAE7E;AAED;;;GAGG;AACH,wBAAgB,SAAS,CACxB,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,WAAW,CAAC,UAAU,EAAE,YAAY,EAAE,CAAC,CAEzC;AAED;;;;;;;GAOG;AACH,wBAAgB,iBAAiB,CAChC,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,GACjB,WAAW,CAAC,UAAU,EAAE,YAAY,EAAE,CAAC,CAIzC;AAuBD,wBAAsB,YAAY,CACjC,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,IAAI,EAAE,cAAc,EACpB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,EACnB,iBAAiB,GAAE,0BAA+D,GAChF,OAAO,CAAC,YAAY,EAAE,CAAC,CAgBzB;AAED,wBAAsB,oBAAoB,CACzC,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,IAAI,EAAE,cAAc,EACpB,MAAM,CAAC,EAAE,WAAW,EACpB,QAAQ,CAAC,EAAE,QAAQ,EACnB,iBAAiB,GAAE,0BAA+D,GAChF,OAAO,CAAC,YAAY,EAAE,CAAC,CAWzB;AAoVD;;;;;;GAMG;AACH,wBAAsB,yBAAyB,CAC9C,OAAO,EAAE,YAAY,EACrB,MAAM,EAAE,eAAe,EACvB,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,QAAQ,CAAC,EAAE,QAAQ,GACjB,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAExC;AAiiCD,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,aAAa,GAAG,kBAAkB,GAAG,SAAS,CAO7F"}

package/dist/agent-loop.js

@@ -9,9 +9,13 @@ import { startPlannedAgentProviderRequest } from "./provider-request-planner.js"
 import { assessToolFailure, clearToolFailure, createRepeatedToolFailureResult, createToolFailureMemoryTracker, createToolFailureOperationExhaustedResult, createToolFailureRecoveryExhaustedResult, createToolFailureResult, getUnresolvedToolFailure, normalizeToolSignature, rememberToolFailure, toolFailureCorrection, } from "./tool-failure-memory.js";
 import { TOOL_FAILURE_RECOVERY_ACCOUNTING_WAVE_SIZE, ToolFailureRecoveryGate, } from "./tool-failure-recovery-gate.js";
 import { appendMandatoryToolFailureDeliveryPrompt } from "./tool-failure-recovery-protocol.js";
+import { rejectNativeToolProtocolResidue } from "./tool-protocol-residue.js";
 import { DEFAULT_MAX_STALL_TURNS } from "./types.js";
 import { createEmptyUsage } from "./usage.js";
-export { resolveRequestPreflightMaxTokens } from "./provider-request-planner.js";
+export { narrowRequestMaxTokens, resolveRequestPreflightMaxTokens } from "./provider-request-planner.js";
+export function createAgentLoopContinuationState() {
+    return { stallWindow: [], toolFailureRecoveryGate: new ToolFailureRecoveryGate() };
+}
 /**
  * Start an agent loop with a new prompt message.
  * The prompt is added to the context and events are emitted for it.
@@ -46,7 +50,7 @@ function streamAgentLoop(config, signal, run) {
     });
     return stream;
 }
-export async function runAgentLoop(prompts, context, config, emit, signal, streamFn) {
+export async function runAgentLoop(prompts, context, config, emit, signal, streamFn, continuationState = createAgentLoopContinuationState()) {
     const newMessages = [...prompts];
     const currentContext = {
         ...context,
@@ -58,16 +62,16 @@ export async function runAgentLoop(prompts, context, config, emit, signal, strea
         await emit({ type: "message_start", message: prompt });
         await emit({ type: "message_end", message: prompt });
     }
-    await runLoop(currentContext, newMessages, config, signal, emit, streamFn);
+    await runLoop(currentContext, newMessages, config, signal, emit, streamFn, continuationState);
     return newMessages;
 }
-export async function runAgentLoopContinue(context, config, emit, signal, streamFn) {
+export async function runAgentLoopContinue(context, config, emit, signal, streamFn, continuationState = createAgentLoopContinuationState()) {
     assertContinuableContext(context);
     const newMessages = [];
     const currentContext = { ...context, messages: [...context.messages] };
     await emit({ type: "agent_start" });
     await emit({ type: "turn_start" });
-    await runLoop(currentContext, newMessages, config, signal, emit, streamFn);
+    await runLoop(currentContext, newMessages, config, signal, emit, streamFn, continuationState);
     return newMessages;
 }
 function assertContinuableContext(context) {
@@ -167,7 +171,7 @@ function repeatsSuccessfulTextProtocolBatch(incomingSignature, toolCalls, previo
 /**
  * Main loop logic shared by agentLoop and agentLoopContinue.
  */
-async function runLoop(initialContext, newMessages, initialConfig, signal, emit, streamFn) {
+async function runLoop(initialContext, newMessages, initialConfig, signal, emit, streamFn, continuationState = createAgentLoopContinuationState()) {
     let currentContext = initialContext;
     let config = initialConfig;
     let firstTurn = true;
@@ -181,11 +185,11 @@ async function runLoop(initialContext, newMessages, initialConfig, signal, emit,
     // window slides past it. Counts only turns that issued tool calls, so varied/long work never trips
     // it. `0` disables.
     const stallLimit = config.maxStallTurns ?? DEFAULT_MAX_STALL_TURNS;
-    const stallWindow = [];
+    const stallWindow = continuationState.stallWindow;
     const validationFailureTracker = { repeats: 0 };
     const repairTeachTracker = new Map();
     let toolFailureMemory = createToolFailureMemoryTracker(currentContext.messages);
-    const toolFailureRecoveryGate = new ToolFailureRecoveryGate();
+    const toolFailureRecoveryGate = continuationState.toolFailureRecoveryGate;
     let mandatoryRecoveryDeliveryPending = false;
     let lastSuccessfulTextProtocolBatch;
     // Check for steering messages at start (user may have typed while waiting)
@@ -376,7 +380,7 @@ async function streamAssistantResponse(context, config, signal, emit, streamFn)
     const response = await startAgentProviderRequest(context, config, signal, streamFn);
     let partialMessage = null;
     let addedPartial = false;
-    for await (const event of response) {
+    responseEvents: for await (const event of response) {
         switch (event.type) {
             case "start":
                 partialMessage = event.partial;
@@ -404,23 +408,11 @@ async function streamAssistantResponse(context, config, signal, emit, streamFn)
                 }
                 break;
             case "done":
-            case "error": {
-                const finalMessage = await response.result();
-                if (addedPartial) {
-                    context.messages[context.messages.length - 1] = finalMessage;
-                }
-                else {
-                    context.messages.push(finalMessage);
-                }
-                if (!addedPartial) {
-                    await emit({ type: "message_start", message: { ...finalMessage } });
-                }
-                await emit({ type: "message_end", message: finalMessage });
-                return finalMessage;
-            }
+            case "error":
+                break responseEvents;
         }
     }
-    const finalMessage = await response.result();
+    const finalMessage = rejectNativeToolProtocolResidue(await response.result(), context.tools ?? [], Boolean(config.textToolCallProtocol));
     if (addedPartial) {
         context.messages[context.messages.length - 1] = finalMessage;
     }