UNPKG

@tanstack/ai

Version:

Type-safe TypeScript AI SDK for streaming chat, tool calling, agents, structured outputs, and multimodal generation.

97 lines (96 loc) 3.15 kB
import { AgentLoopStrategy } from '../../types.js'; /** * Creates a strategy that continues for a maximum number of **model turns** * (iterations), not tool calls. * * One iteration can still emit many parallel tool calls. Prefer * {@link maxToolCalls} (and optionally `maxToolCallsPerTurn` on `chat()`) * when you need a tool-call budget. * * @param max - Maximum number of model turns to allow * @returns AgentLoopStrategy that stops after max iterations * * @example * ```typescript * const stream = chat({ * adapter: openaiText(), * model: "gpt-4o", * messages: [...], * tools: [weatherTool], * agentLoopStrategy: maxIterations(3), // Max 3 model turns * }); * ``` */ export declare function maxIterations(max: number): AgentLoopStrategy; /** * Creates a strategy that continues while `toolCallCount < max`. * * Unlike {@link maxIterations} (which counts model turns), this bounds * **emitted** tool calls counted during the run (including ones skipped by * `maxToolCallsPerTurn`). Strategies only run between turns, so the turn that * crosses `max` is not truncated — the final count (and executions, unless * `maxToolCallsPerTurn` is set) may exceed `max`. Pair with * `chat({ maxToolCallsPerTurn })` to also cap parallel fan-out inside a single * turn. * * @param max - Maximum cumulative emitted tool calls before stopping further turns * @returns AgentLoopStrategy that returns true while `toolCallCount < max` * * @example * ```typescript * import { chat, combineStrategies, maxIterations, maxToolCalls } from '@tanstack/ai' * * const stream = chat({ * adapter: openaiText('gpt-4o'), * messages: [...], * tools: [weatherTool], * maxToolCallsPerTurn: 10, * agentLoopStrategy: combineStrategies([ * maxIterations(20), * maxToolCalls(20), * ]), * }) * ``` */ export declare function maxToolCalls(max: number): AgentLoopStrategy; /** * Creates a strategy that continues until a specific finish reason is encountered * * @param stopReasons - Finish reasons that should stop the loop * @returns AgentLoopStrategy that stops on specific finish reasons * * @example * ```typescript * const stream = chat({ * adapter: openaiText(), * model: "gpt-4o", * messages: [...], * tools: [weatherTool], * agentLoopStrategy: untilFinishReason(["stop", "length"]), * }); * ``` */ export declare function untilFinishReason(stopReasons: Array<string>): AgentLoopStrategy; /** * Creates a strategy that combines multiple strategies with AND logic * All strategies must return true to continue * * @param strategies - Array of strategies to combine * @returns AgentLoopStrategy that continues only if all strategies return true * * @example * ```typescript * const stream = chat({ * adapter: openaiText(), * model: "gpt-4o", * messages: [...], * tools: [weatherTool], * agentLoopStrategy: combineStrategies([ * maxIterations(10), * maxToolCalls(20), * ({ messages }) => messages.length < 100, * ]), * }); * ``` */ export declare function combineStrategies(strategies: Array<AgentLoopStrategy>): AgentLoopStrategy;