@ax-llm/ax 16.1.8 → 16.1.9

package/index.d.cts

@@ -341,6 +341,7 @@ type AxFunctionHandler = (args?: any, extra?: Readonly<{
     debug?: boolean;
     ai?: AxAIService;
     step?: AxStepContext;
+    abortSignal?: AbortSignal;
 }>) => unknown;
 type AxFunctionJSONSchema = {
     type: string;
@@ -2812,7 +2813,14 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     private thoughtFieldName;
     private signatureToolCallingManager?;
     private structuredOutputFunctionFallback;
+    private abortController?;
+    private _stopRequested;
     constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]> | AxSignature<any, any>, options?: Readonly<AxProgramForwardOptions<any>>);
+    /**
+     * Stops an in-flight generation. Causes `forward()` / `streamingForward()`
+     * to throw `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     setInstruction(instruction: string): void;
     getInstruction(): string | undefined;
     private getSignatureName;
@@ -8845,10 +8853,17 @@ TState extends AxFlowState = IN> implements AxFlowable<IN, OUT> {
     private readonly flowLogger?;
     private readonly timingLogger?;
     private readonly defaultAIOptions?;
+    private abortController?;
+    private _stopRequested;
     /**
      * Converts a string to camelCase for valid field names
      */
     private toCamelCase;
+    /**
+     * Stops an in-flight `forward()` call. Causes it to throw
+     * `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     /**
      * Executes a list of steps with comprehensive logging
      */
@@ -10037,7 +10052,9 @@ interface AxCodeInterpreter {
  * A persistent code execution session. Variables persist across `execute()` calls.
  */
 interface AxCodeSession {
-    execute(code: string): Promise<unknown>;
+    execute(code: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<unknown>;
     close(): void;
 }
 /**
@@ -10120,6 +10137,8 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
     private rlmConfig?;
     private rlmContextFields?;
     private rlmProgram?;
+    private abortController?;
+    private _stopRequested;
     private name;
     private func;
     constructor({ ai, name, description, definition, signature, agents, functions, }: Readonly<{
@@ -10131,6 +10150,11 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
         agents?: AxAgentic<IN, OUT>[];
         functions?: AxInputFunctionType;
     }>, options?: Readonly<AxAgentOptions>);
+    /**
+     * Stops an in-flight forward/streamingForward call. Causes the call
+     * to throw `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     /**
      * Creates a new AxAgent instance with type-safe signature parsing.
      * This is the recommended way to create agents with string-based signatures.

package/index.d.ts

@@ -341,6 +341,7 @@ type AxFunctionHandler = (args?: any, extra?: Readonly<{
     debug?: boolean;
     ai?: AxAIService;
     step?: AxStepContext;
+    abortSignal?: AbortSignal;
 }>) => unknown;
 type AxFunctionJSONSchema = {
     type: string;
@@ -2812,7 +2813,14 @@ declare class AxGen<IN = any, OUT extends AxGenOut = any> extends AxProgram<IN,
     private thoughtFieldName;
     private signatureToolCallingManager?;
     private structuredOutputFunctionFallback;
+    private abortController?;
+    private _stopRequested;
     constructor(signature: NonNullable<ConstructorParameters<typeof AxSignature>[0]> | AxSignature<any, any>, options?: Readonly<AxProgramForwardOptions<any>>);
+    /**
+     * Stops an in-flight generation. Causes `forward()` / `streamingForward()`
+     * to throw `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     setInstruction(instruction: string): void;
     getInstruction(): string | undefined;
     private getSignatureName;
@@ -8845,10 +8853,17 @@ TState extends AxFlowState = IN> implements AxFlowable<IN, OUT> {
     private readonly flowLogger?;
     private readonly timingLogger?;
     private readonly defaultAIOptions?;
+    private abortController?;
+    private _stopRequested;
     /**
      * Converts a string to camelCase for valid field names
      */
     private toCamelCase;
+    /**
+     * Stops an in-flight `forward()` call. Causes it to throw
+     * `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     /**
      * Executes a list of steps with comprehensive logging
      */
@@ -10037,7 +10052,9 @@ interface AxCodeInterpreter {
  * A persistent code execution session. Variables persist across `execute()` calls.
  */
 interface AxCodeSession {
-    execute(code: string): Promise<unknown>;
+    execute(code: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<unknown>;
     close(): void;
 }
 /**
@@ -10120,6 +10137,8 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
     private rlmConfig?;
     private rlmContextFields?;
     private rlmProgram?;
+    private abortController?;
+    private _stopRequested;
     private name;
     private func;
     constructor({ ai, name, description, definition, signature, agents, functions, }: Readonly<{
@@ -10131,6 +10150,11 @@ declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAge
         agents?: AxAgentic<IN, OUT>[];
         functions?: AxInputFunctionType;
     }>, options?: Readonly<AxAgentOptions>);
+    /**
+     * Stops an in-flight forward/streamingForward call. Causes the call
+     * to throw `AxAIServiceAbortedError`.
+     */
+    stop(): void;
     /**
      * Creates a new AxAgent instance with type-safe signature parsing.
      * This is the recommended way to create agents with string-based signatures.