@meetsmore-oss/use-ai-server 1.9.4 → 1.10.0

package/dist/agents/AISDKAgent.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModel } from 'ai';
+import { LanguageModel, type JSONValue } from 'ai';
 import type { Agent, AgentInput, EventEmitter, AgentResult } from './types';
 import type { ToolDefinition } from '../types';
 import { type CacheBreakpointFn } from './anthropicCache';
@@ -147,6 +147,33 @@ export interface AISDKAgentConfig {
      * ```
      */
     cacheBreakpoint?: CacheBreakpointFn;
+    /**
+     * Provider-specific options passed directly to `streamText`.
+     * Can be used for AI Gateway features like model fallbacks, provider routing, etc.
+     *
+     * @example
+     * ```typescript
+     * // Model fallbacks via AI Gateway
+     * {
+     *   providerOptions: {
+     *     gateway: {
+     *       models: ['anthropic/claude-opus-4.6', 'google/gemini-3.1-pro-preview'],
+     *     },
+     *   },
+     * }
+     *
+     * // Model fallbacks + provider routing
+     * {
+     *   providerOptions: {
+     *     gateway: {
+     *       models: ['openai/gpt-5-nano', 'anthropic/claude-opus-4.6'],
+     *       order: ['azure', 'openai'],
+     *     },
+     *   },
+     * }
+     * ```
+     */
+    providerOptions?: Record<string, Record<string, JSONValue>>;
     /**
      * Maximum number of tokens the model can output per response.
      * @default 4096
@@ -213,6 +240,7 @@ export interface AISDKAgentConfig {
  */
 export declare class AISDKAgent implements Agent {
     private model;
+    private providerOptions?;
     private name;
     private annotation?;
     private toolFilter?;
@@ -252,6 +280,16 @@ export declare class AISDKAgent implements Agent {
      * Each step runs ONE model invocation, then checks for updated tools/state.
      */
     private executeStepLoop;
+    /**
+     * Detects incomplete tool calls caused by maxOutputTokens truncation and injects
+     * synthetic error tool_results into ctx so the model can retry with shorter arguments.
+     *
+     * When the stream is cut mid-tool-input, tool-input-start fires but tool-call never fires.
+     *
+     * Mutates ctx.allResponseMessages and ctx.currentMessages as a side effect.
+     * @returns true if recovery messages were injected (caller should continue to next step)
+     */
+    private handleIncompleteToolCalls;
     /**
      * Processes a single stream chunk, emitting the appropriate AG-UI events.
      * Mutates stepCtx.stepHadToolCalls when a tool call is detected.
@@ -291,6 +329,12 @@ export declare class AISDKAgent implements Agent {
      * @returns A SystemModelMessage with the state, or undefined if no state
      */
     private buildStateMessage;
+    /**
+     * When maxSteps is exhausted mid-tool-call chain, overrides step config
+     * to strip tools and inject a summary prompt so the model can summarize progress.
+     * No-ops when isGracefulSummaryStep is false.
+     */
+    private applyGracefulSummaryOverrides;
     /**
      * Filters tools using the configured filter function.
      * If no filter is configured, returns all tools.

package/dist/agents/AISDKAgent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AISDKAgent.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0B,aAAa,EAA2D,MAAM,IAAI,CAAC;AAKpH,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAiB,MAAM,SAAS,CAAC;AAC3F,OAAO,KAAK,EAAE,cAAc,EAAuB,MAAM,UAAU,CAAC;AA0BpE,OAAO,EAAyB,KAAK,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AA8FjF;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;;;;;;;;OAUG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,YAAY,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzD;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,OAAO,CAAC;IAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6CG;IACH,eAAe,CAAC,EAAE,iBAAiB,CAAC;IAEpC;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAKD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,qBAAa,UAAW,YAAW,KAAK;IACtC,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAC,CAAoC;IACvD,OAAO,CAAC,YAAY,CAAC,CAA4C;IACjE,OAAO,CAAC,eAAe,CAAC,CAAoB;IAC5C,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,gBAAgB;IAYpC,OAAO,IAAI,MAAM;IAIjB,aAAa,IAAI,MAAM,GAAG,SAAS;IAInC;;;OAGG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBxE;;;OAGG;YACW,gBAAgB;IAsC9B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAuBnB;;;OAGG;YACW,eAAe;IAwH7B;;;OAGG;IAEH,OAAO,CAAC,kBAAkB;IAqL1B;;;OAGG;IACH,OAAO,CAAC,WAAW;IA2DnB;;;OAGG;IACH,OAAO,CAAC,cAAc;IA+CtB;;;;;OAKG;YACW,mBAAmB;IAcjC;;;;;;;OAOG;IACH,OAAO,CAAC,yBAAyB;IAgBjC;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAQzB;;;OAGG;IACH,OAAO,CAAC,WAAW;IAsBnB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAyD7B,OAAO,CAAC,mBAAmB;IAyC3B;;;;;OAKG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAMpC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAKlC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAG9B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAG/B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAI9B;IAEX;;;;OAIG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAWhD;IAEJ,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAOtC;IAEH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAM1B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAqC;IAEhF;;;;;;;;OAQG;IACH,OAAO,CAAC,gBAAgB;CAgBzB"}
+{"version":3,"file":"AISDKAgent.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0B,aAAa,EAA8G,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAMvL,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAiB,MAAM,SAAS,CAAC;AAC3F,OAAO,KAAK,EAAE,cAAc,EAAuB,MAAM,UAAU,CAAC;AA0BpE,OAAO,EAAyB,KAAK,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAoGjF;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;;;;;;;;OAUG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,YAAY,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzD;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,OAAO,CAAC;IAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6CG;IACH,eAAe,CAAC,EAAE,iBAAiB,CAAC;IAEpC;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;IAE5D;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAKD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,qBAAa,UAAW,YAAW,KAAK;IACtC,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,eAAe,CAAC,CAA4C;IACpE,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAC,CAAoC;IACvD,OAAO,CAAC,YAAY,CAAC,CAA4C;IACjE,OAAO,CAAC,eAAe,CAAC,CAAoB;IAC5C,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,gBAAgB;IAapC,OAAO,IAAI,MAAM;IAIjB,aAAa,IAAI,MAAM,GAAG,SAAS;IAInC;;;OAGG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBxE;;;OAGG;YACW,gBAAgB;IAqC9B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAuBnB;;;OAGG;YACW,eAAe;IA4H7B;;;;;;;;OAQG;IACH,OAAO,CAAC,yBAAyB;IAsBjC;;;OAGG;IAEH,OAAO,CAAC,kBAAkB;IAmN1B;;;OAGG;IACH,OAAO,CAAC,WAAW;IAqDnB;;;OAGG;IACH,OAAO,CAAC,cAAc;IA+CtB;;;;;OAKG;YACW,mBAAmB;IAcjC;;;;;;;OAOG;IACH,OAAO,CAAC,yBAAyB;IAgBjC;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAQzB;;;;OAIG;IACH,OAAO,CAAC,6BAA6B;IAUrC;;;OAGG;IACH,OAAO,CAAC,WAAW;IAsBnB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAyD7B,OAAO,CAAC,mBAAmB;IAyC3B;;;;;OAKG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAMpC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAKlC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAG9B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAG/B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAI9B;IAEX;;;;OAIG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAWhD;IAEJ,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAOtC;IAEH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAM1B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAqC;IAEhF;;;;;;;;OAQG;IACH,OAAO,CAAC,gBAAgB;CAgBzB"}

package/dist/agents/AISDKAgent.perStepEvents.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=AISDKAgent.perStepEvents.test.d.ts.map

package/dist/agents/AISDKAgent.perStepEvents.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AISDKAgent.perStepEvents.test.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.perStepEvents.test.ts"],"names":[],"mappings":""}

package/dist/index.d.ts

@@ -3,7 +3,7 @@ export type { UseAIServerConfig, McpEndpointConfig, ToolDefinition, CorsOptions
 export type { ClientSession } from './server';
 export type { Agent, AgentInput, EventEmitter, AgentResult } from './agents';
 export { AISDKAgent, type AISDKAgentConfig, type MessageWithCacheContext, type CacheTtl, type CacheBreakpointResult, type CacheBreakpointFn } from './agents';
-export type { UseAIServerPlugin, MessageHandler } from './plugins';
+export type { UseAIServerPlugin, MessageHandler, BeforeRunAgentResult } from './plugins';
 export { FeedbackPlugin } from './plugins';
 export type { SpanProcessor } from './instrumentation';
 export { logger } from './logger';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACjG,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAG9C,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,KAAK,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,KAAK,QAAQ,EAAE,KAAK,qBAAqB,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG9J,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGvD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAGlC,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,KAAK,OAAO,EAAE,MAAM,aAAa,CAAC;AAGzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGzF,OAAO,EACL,uBAAuB,EACvB,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,EAC1B,yBAAyB,EACzB,KAAK,uBAAuB,GAC7B,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,wBAAwB,EACxB,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,EACH,EAAE,EACF,GAAG,GACJ,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACjG,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAG9C,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,KAAK,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,KAAK,QAAQ,EAAE,KAAK,qBAAqB,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG9J,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACzF,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGvD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAGlC,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,KAAK,OAAO,EAAE,MAAM,aAAa,CAAC;AAGzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGzF,OAAO,EACL,uBAAuB,EACvB,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,EAC1B,yBAAyB,EACzB,KAAK,uBAAuB,GAC7B,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,wBAAwB,EACxB,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,EACH,EAAE,EACF,GAAG,GACJ,MAAM,SAAS,CAAC"}

package/dist/index.js

@@ -90464,16 +90464,30 @@ ${block.text}`
         this.sendEvent(session.socket, event);
       }
     };
+    const agentInput = {
+      session,
+      runId,
+      messages: incomingMessages,
+      tools: session.tools,
+      state,
+      systemPrompt,
+      originalInput: message.data
+    };
+    for (const plugin of this.plugins) {
+      if (plugin.beforeRunAgent) {
+        const result = await plugin.beforeRunAgent(agentInput);
+        if (result?.abort) {
+          this.sendEvent(session.socket, {
+            type: export_EventType.RUN_ERROR,
+            message: result.message,
+            timestamp: Date.now()
+          });
+          return;
+        }
+      }
+    }
     try {
-      await selectedAgent.run({
-        session,
-        runId,
-        messages: incomingMessages,
-        tools: session.tools,
-        state,
-        systemPrompt,
-        originalInput: message.data
-      }, eventEmitter);
+      await selectedAgent.run(agentInput, eventEmitter);
     } finally {
       delete session.currentMcpHeaders;
     }
@@ -114134,6 +114148,7 @@ var DEFAULT_MAX_STEPS = 10;
 
 class AISDKAgent {
   model;
+  providerOptions;
   name;
   annotation;
   toolFilter;
@@ -114144,6 +114159,7 @@ class AISDKAgent {
   maxSteps;
   constructor(config2) {
     this.model = config2.model;
+    this.providerOptions = config2.providerOptions;
     this.name = config2.name || "ai-sdk";
     this.annotation = config2.annotation;
     this.toolFilter = config2.toolFilter;
@@ -114191,14 +114207,13 @@ class AISDKAgent {
       originalInput,
       staticSystemMessages,
       streamTextStarted: false,
-      messageId: null,
-      hasEmittedTextStart: false,
       finalText: "",
       currentStepNumber: 0,
       hasAnyContent: false,
       currentMessages: [...sanitizedInputMessages],
       allResponseMessages: [],
-      response: null
+      response: null,
+      lastStepHadToolCalls: false
     };
   }
   emitRunStartEvents(ctx, events) {
@@ -114245,46 +114260,52 @@ class AISDKAgent {
     });
   }
   async executeStepLoop(ctx, events, span) {
-    for (let stepIteration = 0;stepIteration < this.maxSteps; stepIteration++) {
+    for (let stepIteration = 0;stepIteration <= this.maxSteps; stepIteration++) {
+      const isGracefulSummaryStep = stepIteration === this.maxSteps;
+      if (isGracefulSummaryStep && !ctx.lastStepHadToolCalls)
+        break;
       const stepCtx = {
         currentTools: ctx.session.tools,
         activeToolCalls: new Map,
-        stepHadToolCalls: false
+        completedToolCalls: new Set,
+        stepHadToolCalls: false,
+        messageId: null,
+        hasEmittedTextStart: false,
+        stepFinishReason: undefined
       };
-      logger2.debug("Starting step iteration", {
-        stepIteration,
-        toolCount: stepCtx.currentTools.length,
-        toolNames: stepCtx.currentTools.map((t) => t.name)
-      });
       const stateMessage = this.buildStateMessage(ctx.session.state);
       const messagesForStep = [
         ...ctx.staticSystemMessages || [],
         ...stateMessage ? [stateMessage] : [],
         ...ctx.currentMessages
       ];
-      const messagesWithCache = applyCacheBreakpoints(messagesForStep, this.cacheBreakpoint, this.model);
+      const stepConfig = {
+        messages: messagesForStep,
+        tools: stepCtx.currentTools.length > 0 ? this.sanitizeToolsForAPI(this.filterTools(stepCtx.currentTools), ctx.session, events) : undefined,
+        metadata: {
+          sessionId: ctx.session.clientId,
+          threadId: ctx.session.threadId,
+          runId: ctx.runId,
+          ipAddress: ctx.session.ipAddress,
+          toolCount: stepCtx.currentTools.length,
+          stepIteration,
+          ...ctx.originalInput.forwardedProps?.telemetryMetadata || {}
+        }
+      };
+      this.applyGracefulSummaryOverrides(isGracefulSummaryStep, stepConfig);
+      logger2.debug("Starting step iteration", { stepIteration, ...stepConfig.metadata });
+      const messagesWithCache = applyCacheBreakpoints(stepConfig.messages, this.cacheBreakpoint, this.model);
       ctx.streamTextStarted = true;
       const createStream = () => streamText({
         model: this.model,
         messages: messagesWithCache,
-        tools: stepCtx.currentTools.length > 0 ? this.sanitizeToolsForAPI(this.filterTools(stepCtx.currentTools), ctx.session, events) : undefined,
+        tools: stepConfig.tools,
         stopWhen: stepCountIs(1),
         maxOutputTokens: this.maxOutputTokens,
         temperature: this.temperature,
         abortSignal: ctx.session.abortController?.signal,
-        experimental_telemetry: span.active ? {
-          isEnabled: true,
-          functionId: "use-ai",
-          metadata: {
-            sessionId: ctx.session.clientId,
-            threadId: ctx.session.threadId,
-            runId: ctx.runId,
-            ipAddress: ctx.session.ipAddress,
-            toolCount: stepCtx.currentTools.length,
-            stepIteration,
-            ...ctx.originalInput.forwardedProps?.telemetryMetadata || {}
-          }
-        } : undefined,
+        providerOptions: this.providerOptions,
+        experimental_telemetry: span.active ? { isEnabled: true, functionId: "use-ai", metadata: stepConfig.metadata } : undefined,
         onStepFinish: ({ usage, finishReason }) => {
           logger2.debug("Step finished", { usage, finishReason, stepIteration });
         }
@@ -114306,11 +114327,15 @@ class AISDKAgent {
       ctx.response = response;
       const stepMessages = this.sanitizeMessages(response.messages);
       ctx.allResponseMessages.push(...stepMessages);
+      ctx.currentMessages = [...ctx.currentMessages, ...stepMessages];
+      if (this.handleIncompleteToolCalls(ctx, stepCtx)) {
+        continue;
+      }
+      ctx.lastStepHadToolCalls = stepCtx.stepHadToolCalls;
       if (!stepCtx.stepHadToolCalls) {
         logger2.debug("Step had no tool calls, finishing run", { stepIteration });
         break;
       }
-      ctx.currentMessages = [...ctx.currentMessages, ...stepMessages];
       logger2.debug("Continuing to next step after tool calls", {
         stepIteration,
         newMessageCount: ctx.currentMessages.length,
@@ -114318,6 +114343,20 @@ class AISDKAgent {
       });
     }
   }
+  handleIncompleteToolCalls(ctx, stepCtx) {
+    const incompleteToolCalls = [...stepCtx.activeToolCalls.entries()].filter(([id]) => !stepCtx.completedToolCalls.has(id)).map(([id, call]) => ({ id, ...call }));
+    const recoveryMessages = buildRecoveryToolResults(incompleteToolCalls, stepCtx.stepFinishReason, this.maxOutputTokens);
+    if (recoveryMessages.length === 0) {
+      return false;
+    }
+    const sanitized = this.sanitizeMessages(recoveryMessages);
+    ctx.allResponseMessages.push(...sanitized);
+    ctx.currentMessages = [...ctx.currentMessages, ...sanitized];
+    logger2.warn("Incomplete tool calls detected (likely maxOutputTokens exceeded)", {
+      incompleteCount: incompleteToolCalls.length
+    });
+    return true;
+  }
   processStreamChunk(chunk, ctx, stepCtx, events) {
     switch (chunk.type) {
       case "start-step": {
@@ -114330,19 +114369,19 @@ class AISDKAgent {
       }
       case "text-delta": {
         ctx.hasAnyContent = true;
-        if (!ctx.hasEmittedTextStart) {
-          ctx.messageId = v4_default();
+        if (!stepCtx.hasEmittedTextStart) {
+          stepCtx.messageId = v4_default();
           events.emit({
             type: export_EventType.TEXT_MESSAGE_START,
-            messageId: ctx.messageId,
+            messageId: stepCtx.messageId,
             role: "assistant",
             timestamp: Date.now()
           });
-          ctx.hasEmittedTextStart = true;
+          stepCtx.hasEmittedTextStart = true;
         }
         events.emit({
           type: export_EventType.TEXT_MESSAGE_CONTENT,
-          messageId: ctx.messageId,
+          messageId: stepCtx.messageId,
           delta: chunk.text,
           timestamp: Date.now()
         });
@@ -114356,13 +114395,22 @@ class AISDKAgent {
       case "tool-input-start": {
         ctx.hasAnyContent = true;
         stepCtx.stepHadToolCalls = true;
+        if (stepCtx.messageId) {
+          events.emit({
+            type: export_EventType.TEXT_MESSAGE_END,
+            messageId: stepCtx.messageId,
+            timestamp: Date.now()
+          });
+        }
         const toolDef = stepCtx.currentTools.find((t) => t.name === chunk.toolName);
         const annotations = getToolAnnotations(toolDef);
+        const parentId = stepCtx.messageId ?? v4_default();
+        stepCtx.messageId = null;
         const toolCallStartEvent = {
           type: export_EventType.TOOL_CALL_START,
           toolCallId: chunk.id,
           toolCallName: chunk.toolName,
-          parentMessageId: ctx.messageId ?? v4_default(),
+          parentMessageId: parentId,
           timestamp: Date.now()
         };
         if (annotations) {
@@ -114387,6 +114435,7 @@ class AISDKAgent {
       }
       case "tool-call": {
         stepCtx.stepHadToolCalls = true;
+        stepCtx.completedToolCalls.add(chunk.toolCallId);
         const toolCall = stepCtx.activeToolCalls.get(chunk.toolCallId);
         const finalArgs = JSON.stringify(chunk.input);
         if (toolCall && toolCall.args.length === 0) {
@@ -114417,7 +114466,19 @@ class AISDKAgent {
         });
         return;
       }
+      case "finish": {
+        stepCtx.stepFinishReason = chunk.finishReason;
+        return;
+      }
       case "finish-step": {
+        if (stepCtx.messageId) {
+          events.emit({
+            type: export_EventType.TEXT_MESSAGE_END,
+            messageId: stepCtx.messageId,
+            timestamp: Date.now()
+          });
+          stepCtx.messageId = null;
+        }
         events.emit({
           type: export_EventType.STEP_FINISHED,
           stepName: `step-${ctx.currentStepNumber - 1}`,
@@ -114444,13 +114505,6 @@ class AISDKAgent {
     }
   }
   finalizeRun(ctx, events, span) {
-    if (ctx.hasEmittedTextStart && ctx.messageId) {
-      events.emit({
-        type: export_EventType.TEXT_MESSAGE_END,
-        messageId: ctx.messageId,
-        timestamp: Date.now()
-      });
-    }
     if (!ctx.hasAnyContent) {
       span.endWithError("Empty response from AI");
       events.emit({
@@ -114548,6 +114602,13 @@ class AISDKAgent {
 ${JSON.stringify(state, null, 2)}`
     };
   }
+  applyGracefulSummaryOverrides(isGracefulSummaryStep, stepConfig) {
+    if (!isGracefulSummaryStep)
+      return;
+    stepConfig.messages.push({ role: "user", content: "max steps reached, summarize progress" });
+    stepConfig.tools = undefined;
+    Object.assign(stepConfig.metadata, { toolCount: 0, gracefulSummary: true });
+  }
   filterTools(tools) {
     if (!this.toolFilter) {
       return tools;
@@ -114690,6 +114751,35 @@ ${val.text}`
     }
   }
 }
+function buildRecoveryToolResults(incompleteToolCalls, stepFinishReason, maxOutputTokens) {
+  if (incompleteToolCalls.length === 0 || stepFinishReason !== "length") {
+    return [];
+  }
+  const recoveryAssistantMessage = {
+    role: "assistant",
+    content: incompleteToolCalls.map((toolCall) => ({
+      type: "tool-call",
+      toolCallId: toolCall.id,
+      toolName: toolCall.name,
+      input: {}
+    }))
+  };
+  const recoveryToolResults = incompleteToolCalls.map((toolCall) => ({
+    role: "tool",
+    content: [
+      {
+        type: "tool-result",
+        toolCallId: toolCall.id,
+        toolName: toolCall.name,
+        output: {
+          type: "text",
+          value: `Error: Tool call "${toolCall.name}" failed because its arguments were cut off mid-stream by the output token limit (maxOutputTokens: ${maxOutputTokens}). This call was recorded with args={} as a placeholder — retry with shorter arguments. Truncated args (first 200 chars): ${toolCall.args.substring(0, 200)}`
+        }
+      }
+    ]
+  }));
+  return [recoveryAssistantMessage, ...recoveryToolResults];
+}
 // ../../node_modules/.bun/zod-to-json-schema@3.25.1+27912429049419a2/node_modules/zod-to-json-schema/dist/esm/Options.js
 var ignoreOverride2 = Symbol("Let zodToJsonSchema decide on which parser to use");
 var defaultOptions2 = {

package/dist/plugins/index.d.ts

@@ -1,3 +1,3 @@
-export type { UseAIServerPlugin, MessageHandler } from './types';
+export type { UseAIServerPlugin, MessageHandler, BeforeRunAgentResult } from './types';
 export { FeedbackPlugin } from './FeedbackPlugin';
 //# sourceMappingURL=index.d.ts.map

package/dist/plugins/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/plugins/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/plugins/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AACvF,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}

package/dist/plugins/types.d.ts

@@ -1,4 +1,4 @@
-import type { ClientSession } from '../agents/types';
+import type { ClientSession, AgentInput } from '../agents/types';
 import type { UseAIClientMessage } from '../types';
 /**
  * Message handler function type for plugin-registered message handlers.
@@ -7,6 +7,14 @@ import type { UseAIClientMessage } from '../types';
  * @param message - The incoming message from the client
  */
 export type MessageHandler = (session: ClientSession, message: UseAIClientMessage) => Promise<void> | void;
+/**
+ * Result from a beforeRunAgent hook indicating the run should be aborted.
+ * Return this from beforeRunAgent to block the run and send the message to the client as RUN_ERROR.
+ */
+export interface BeforeRunAgentResult {
+    abort: true;
+    message: string;
+}
 /**
  * Plugin interface for extending UseAIServer functionality.
  *
@@ -65,6 +73,17 @@ export interface UseAIServerPlugin {
      * @param session - The disconnecting client session
      */
     onClientDisconnect?(session: ClientSession): void;
+    /**
+     * Optional hook called before an agent run starts.
+     * Use this to perform pre-run checks such as authentication, quota enforcement, etc.
+     *
+     * Return `{ abort: true, message: '...' }` to block the run — the message is sent
+     * to the client as a RUN_ERROR event. Return `void` (or `undefined`) to allow the run.
+     *
+     * @param input - The agent input containing session, messages, tools, and state
+     * @returns Promise resolving to an abort result or void
+     */
+    beforeRunAgent?(input: AgentInput): Promise<BeforeRunAgentResult | void>;
     /**
      * Optional cleanup hook called when the server is shutting down.
      * Use this to flush pending data, close connections, etc.

package/dist/plugins/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/plugins/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAEnD;;;;;GAKG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE3G;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,WAAW,iBAAiB;IAChC;;;;;OAKG;IACH,OAAO,IAAI,MAAM,CAAC;IAElB;;;;;OAKG;IACH,gBAAgB,CAAC,MAAM,EAAE;QACvB,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI,CAAC;KACrE,GAAG,IAAI,CAAC;IAET;;;;OAIG;IACH,eAAe,CAAC,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CAAC;IAE/C;;;;OAIG;IACH,kBAAkB,CAAC,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CAAC;IAElD;;;;;OAKG;IACH,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/plugins/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAEnD;;;;;GAKG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE3G;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,IAAI,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,WAAW,iBAAiB;IAChC;;;;;OAKG;IACH,OAAO,IAAI,MAAM,CAAC;IAElB;;;;;OAKG;IACH,gBAAgB,CAAC,MAAM,EAAE;QACvB,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI,CAAC;KACrE,GAAG,IAAI,CAAC;IAET;;;;OAIG;IACH,eAAe,CAAC,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CAAC;IAE/C;;;;OAIG;IACH,kBAAkB,CAAC,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CAAC;IAElD;;;;;;;;;OASG;IACH,cAAc,CAAC,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC,CAAC;IAEzE;;;;;OAKG;IACH,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB"}

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,iBAAiB,EAYlB,MAAM,SAAS,CAAC;AAOjB,OAAO,KAAK,EAAqB,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAezE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAEpD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8CG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAiB;IAC3B,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,YAAY,CAAoC;IACxD,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,OAAO,CAAyC;IACxD,OAAO,CAAC,MAAM,CAIZ;IACF,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,eAAe,CAAiB;IACxC,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,eAAe,CAA0C;IACjE,OAAO,CAAC,YAAY,CAAgC;IACpD,OAAO,CAAC,WAAW,CAA8B;IAEjD,OAAO,CAAC,eAAe,CAAkB;IAEzC;;;;;OAKG;gBACS,MAAM,EAAE,iBAAiB;IAoGrC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBjC;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAiBzB;;;;;;OAMG;IACI,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI;IAS1E,OAAO,CAAC,mBAAmB;YAqGb,mBAAmB;YA8BnB,cAAc;IAmT5B,OAAO,CAAC,iBAAiB;IAKzB,OAAO,CAAC,gBAAgB;IAiDxB,OAAO,CAAC,0BAA0B;IAqBlC,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,SAAS;IAMjB;;;;;;;;;;;OAWG;YACW,qBAAqB;IA+CnC;;;;;;;OAOG;IACH,OAAO,CAAC,oBAAoB;IAwC5B;;;;;OAKG;IACH,OAAO,CAAC,cAAc;IAoBtB;;;;;;OAMG;IACH,OAAO,CAAC,yBAAyB;IAYjC;;;OAGG;IACU,KAAK;CAsBnB"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,iBAAiB,EAYlB,MAAM,SAAS,CAAC;AAOjB,OAAO,KAAK,EAAqB,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAezE,YAAY,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAEpD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8CG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAiB;IAC3B,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,YAAY,CAAoC;IACxD,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,OAAO,CAAyC;IACxD,OAAO,CAAC,MAAM,CAIZ;IACF,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,eAAe,CAAiB;IACxC,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,eAAe,CAA0C;IACjE,OAAO,CAAC,YAAY,CAAgC;IACpD,OAAO,CAAC,WAAW,CAA8B;IAEjD,OAAO,CAAC,eAAe,CAAkB;IAEzC;;;;;OAKG;gBACS,MAAM,EAAE,iBAAiB;IAoGrC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBjC;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAiBzB;;;;;;OAMG;IACI,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI;IAS1E,OAAO,CAAC,mBAAmB;YAqGb,mBAAmB;YA8BnB,cAAc;IAkU5B,OAAO,CAAC,iBAAiB;IAKzB,OAAO,CAAC,gBAAgB;IAiDxB,OAAO,CAAC,0BAA0B;IAqBlC,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,SAAS;IAMjB;;;;;;;;;;;OAWG;YACW,qBAAqB;IA+CnC;;;;;;;OAOG;IACH,OAAO,CAAC,oBAAoB;IAwC5B;;;;;OAKG;IACH,OAAO,CAAC,cAAc;IAoBtB;;;;;;OAMG;IACH,OAAO,CAAC,yBAAyB;IAYjC;;;OAGG;IACU,KAAK;CAsBnB"}

package/dist/src/agents/AISDKAgent.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModel } from 'ai';
+import { LanguageModel, type JSONValue } from 'ai';
 import type { Agent, AgentInput, EventEmitter, AgentResult } from './types';
 import type { ToolDefinition } from '../types';
 import { type CacheBreakpointFn } from './anthropicCache';
@@ -147,6 +147,33 @@ export interface AISDKAgentConfig {
      * ```
      */
     cacheBreakpoint?: CacheBreakpointFn;
+    /**
+     * Provider-specific options passed directly to `streamText`.
+     * Can be used for AI Gateway features like model fallbacks, provider routing, etc.
+     *
+     * @example
+     * ```typescript
+     * // Model fallbacks via AI Gateway
+     * {
+     *   providerOptions: {
+     *     gateway: {
+     *       models: ['anthropic/claude-opus-4.6', 'google/gemini-3.1-pro-preview'],
+     *     },
+     *   },
+     * }
+     *
+     * // Model fallbacks + provider routing
+     * {
+     *   providerOptions: {
+     *     gateway: {
+     *       models: ['openai/gpt-5-nano', 'anthropic/claude-opus-4.6'],
+     *       order: ['azure', 'openai'],
+     *     },
+     *   },
+     * }
+     * ```
+     */
+    providerOptions?: Record<string, Record<string, JSONValue>>;
     /**
      * Maximum number of tokens the model can output per response.
      * @default 4096
@@ -213,6 +240,7 @@ export interface AISDKAgentConfig {
  */
 export declare class AISDKAgent implements Agent {
     private model;
+    private providerOptions?;
     private name;
     private annotation?;
     private toolFilter?;
@@ -252,6 +280,16 @@ export declare class AISDKAgent implements Agent {
      * Each step runs ONE model invocation, then checks for updated tools/state.
      */
     private executeStepLoop;
+    /**
+     * Detects incomplete tool calls caused by maxOutputTokens truncation and injects
+     * synthetic error tool_results into ctx so the model can retry with shorter arguments.
+     *
+     * When the stream is cut mid-tool-input, tool-input-start fires but tool-call never fires.
+     *
+     * Mutates ctx.allResponseMessages and ctx.currentMessages as a side effect.
+     * @returns true if recovery messages were injected (caller should continue to next step)
+     */
+    private handleIncompleteToolCalls;
     /**
      * Processes a single stream chunk, emitting the appropriate AG-UI events.
      * Mutates stepCtx.stepHadToolCalls when a tool call is detected.
@@ -291,6 +329,12 @@ export declare class AISDKAgent implements Agent {
      * @returns A SystemModelMessage with the state, or undefined if no state
      */
     private buildStateMessage;
+    /**
+     * When maxSteps is exhausted mid-tool-call chain, overrides step config
+     * to strip tools and inject a summary prompt so the model can summarize progress.
+     * No-ops when isGracefulSummaryStep is false.
+     */
+    private applyGracefulSummaryOverrides;
     /**
      * Filters tools using the configured filter function.
      * If no filter is configured, returns all tools.

package/dist/src/agents/AISDKAgent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AISDKAgent.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0B,aAAa,EAA2D,MAAM,IAAI,CAAC;AAKpH,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAiB,MAAM,SAAS,CAAC;AAC3F,OAAO,KAAK,EAAE,cAAc,EAAuB,MAAM,UAAU,CAAC;AA0BpE,OAAO,EAAyB,KAAK,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AA8FjF;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;;;;;;;;OAUG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,YAAY,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzD;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,OAAO,CAAC;IAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6CG;IACH,eAAe,CAAC,EAAE,iBAAiB,CAAC;IAEpC;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAKD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,qBAAa,UAAW,YAAW,KAAK;IACtC,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAC,CAAoC;IACvD,OAAO,CAAC,YAAY,CAAC,CAA4C;IACjE,OAAO,CAAC,eAAe,CAAC,CAAoB;IAC5C,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,gBAAgB;IAYpC,OAAO,IAAI,MAAM;IAIjB,aAAa,IAAI,MAAM,GAAG,SAAS;IAInC;;;OAGG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBxE;;;OAGG;YACW,gBAAgB;IAsC9B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAuBnB;;;OAGG;YACW,eAAe;IAwH7B;;;OAGG;IAEH,OAAO,CAAC,kBAAkB;IAqL1B;;;OAGG;IACH,OAAO,CAAC,WAAW;IA2DnB;;;OAGG;IACH,OAAO,CAAC,cAAc;IA+CtB;;;;;OAKG;YACW,mBAAmB;IAcjC;;;;;;;OAOG;IACH,OAAO,CAAC,yBAAyB;IAgBjC;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAQzB;;;OAGG;IACH,OAAO,CAAC,WAAW;IAsBnB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAyD7B,OAAO,CAAC,mBAAmB;IAyC3B;;;;;OAKG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAMpC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAKlC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAG9B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAG/B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAI9B;IAEX;;;;OAIG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAWhD;IAEJ,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAOtC;IAEH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAM1B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAqC;IAEhF;;;;;;;;OAQG;IACH,OAAO,CAAC,gBAAgB;CAgBzB"}
+{"version":3,"file":"AISDKAgent.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0B,aAAa,EAA8G,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAMvL,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAiB,MAAM,SAAS,CAAC;AAC3F,OAAO,KAAK,EAAE,cAAc,EAAuB,MAAM,UAAU,CAAC;AA0BpE,OAAO,EAAyB,KAAK,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAoGjF;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B;;;;;;;;;;;;;;;;;;OAkBG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd;;;;;;;;;;OAUG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,YAAY,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzD;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,OAAO,CAAC;IAE/C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6CG;IACH,eAAe,CAAC,EAAE,iBAAiB,CAAC;IAEpC;;;;;;;;;;;;;;;;;;;;;;;;;OAyBG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;IAE5D;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAKD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2CG;AACH,qBAAa,UAAW,YAAW,KAAK;IACtC,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,eAAe,CAAC,CAA4C;IACpE,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,UAAU,CAAC,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAC,CAAoC;IACvD,OAAO,CAAC,YAAY,CAAC,CAA4C;IACjE,OAAO,CAAC,eAAe,CAAC,CAAoB;IAC5C,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAS;gBAEb,MAAM,EAAE,gBAAgB;IAapC,OAAO,IAAI,MAAM;IAIjB,aAAa,IAAI,MAAM,GAAG,SAAS;IAInC;;;OAGG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAgBxE;;;OAGG;YACW,gBAAgB;IAqC9B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAuBnB;;;OAGG;YACW,eAAe;IA4H7B;;;;;;;;OAQG;IACH,OAAO,CAAC,yBAAyB;IAsBjC;;;OAGG;IAEH,OAAO,CAAC,kBAAkB;IAmN1B;;;OAGG;IACH,OAAO,CAAC,WAAW;IAqDnB;;;OAGG;IACH,OAAO,CAAC,cAAc;IA+CtB;;;;;OAKG;YACW,mBAAmB;IAcjC;;;;;;;OAOG;IACH,OAAO,CAAC,yBAAyB;IAgBjC;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAQzB;;;;OAIG;IACH,OAAO,CAAC,6BAA6B;IAUrC;;;OAGG;IACH,OAAO,CAAC,WAAW;IAsBnB;;;;OAIG;IACH,OAAO,CAAC,qBAAqB;IAyD7B,OAAO,CAAC,mBAAmB;IAyC3B;;;;;OAKG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAMpC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAKlC;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAG9B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAG/B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAI9B;IAEX;;;;OAIG;IACH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,4BAA4B,CAWhD;IAEJ,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAOtC;IAEH,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAM1B;IAEX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAqC;IAEhF;;;;;;;;OAQG;IACH,OAAO,CAAC,gBAAgB;CAgBzB"}

package/dist/src/agents/AISDKAgent.perStepEvents.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=AISDKAgent.perStepEvents.test.d.ts.map

package/dist/src/agents/AISDKAgent.perStepEvents.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AISDKAgent.perStepEvents.test.d.ts","sourceRoot":"","sources":["../../../src/agents/AISDKAgent.perStepEvents.test.ts"],"names":[],"mappings":""}

package/dist/src/index.d.ts

@@ -3,7 +3,7 @@ export type { UseAIServerConfig, McpEndpointConfig, ToolDefinition, CorsOptions
 export type { ClientSession } from './server';
 export type { Agent, AgentInput, EventEmitter, AgentResult } from './agents';
 export { AISDKAgent, type AISDKAgentConfig, type MessageWithCacheContext, type CacheTtl, type CacheBreakpointResult, type CacheBreakpointFn } from './agents';
-export type { UseAIServerPlugin, MessageHandler } from './plugins';
+export type { UseAIServerPlugin, MessageHandler, BeforeRunAgentResult } from './plugins';
 export { FeedbackPlugin } from './plugins';
 export type { SpanProcessor } from './instrumentation';
 export { logger } from './logger';

package/dist/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACjG,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAG9C,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,KAAK,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,KAAK,QAAQ,EAAE,KAAK,qBAAqB,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG9J,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGvD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAGlC,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,KAAK,OAAO,EAAE,MAAM,aAAa,CAAC;AAGzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGzF,OAAO,EACL,uBAAuB,EACvB,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,EAC1B,yBAAyB,EACzB,KAAK,uBAAuB,GAC7B,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,wBAAwB,EACxB,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,EACH,EAAE,EACF,GAAG,GACJ,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACjG,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAG9C,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,UAAU,EAAE,KAAK,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,KAAK,QAAQ,EAAE,KAAK,qBAAqB,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG9J,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACzF,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGvD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAGlC,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,KAAK,OAAO,EAAE,MAAM,aAAa,CAAC;AAGzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAGzF,OAAO,EACL,uBAAuB,EACvB,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,EAC1B,yBAAyB,EACzB,KAAK,uBAAuB,GAC7B,MAAM,OAAO,CAAC;AAGf,OAAO,EACL,wBAAwB,EACxB,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,EACH,EAAE,EACF,GAAG,GACJ,MAAM,SAAS,CAAC"}

package/dist/src/plugins/index.d.ts

@@ -1,3 +1,3 @@
-export type { UseAIServerPlugin, MessageHandler } from './types';
+export type { UseAIServerPlugin, MessageHandler, BeforeRunAgentResult } from './types';
 export { FeedbackPlugin } from './FeedbackPlugin';
 //# sourceMappingURL=index.d.ts.map

package/dist/src/plugins/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/plugins/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/plugins/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,iBAAiB,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AACvF,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC"}