@output.ai/llm 0.2.12 → 0.2.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/llm",
-  "version": "0.2.12",
+  "version": "0.2.13",
   "description": "Framework abstraction to interact with LLM models",
   "type": "module",
   "main": "src/index.js",

package/src/index.d.ts

@@ -4,7 +4,10 @@ import type {
   GenerateObjectResult as AIGenerateObjectResult,
   CallSettings,
   ToolSet,
-  ToolChoice
+  ToolChoice,
+  StopCondition,
+  GenerateTextOnStepFinishCallback,
+  PrepareStepFunction
 } from 'ai';
 
 /**
@@ -100,12 +103,20 @@ export type {
   CallSettings,
   ToolSet,
   ToolChoice,
-  Tool
+  Tool,
+  StopCondition,
+  StepResult,
+  GenerateTextOnStepFinishCallback,
+  PrepareStepFunction,
+  PrepareStepResult
 } from 'ai';
 
 // Re-export the tool helper function for creating tools
 export { tool } from 'ai';
 
+// Re-export stop condition helpers for multi-step workflows
+export { stepCountIs, hasToolCall } from 'ai';
+
 /**
  * Common AI SDK options that can be passed through to all generate functions.
  * These options are passed directly to the underlying AI SDK call.
@@ -113,7 +124,7 @@ export { tool } from 'ai';
 type AiSdkOptions = Partial<Omit<CallSettings, 'maxOutputTokens'>>;
 
 /**
- * AI SDK options specific to generateText, including tool calling support.
+ * AI SDK options specific to generateText, including tool calling and multi-step support.
  * @typeParam TOOLS - The tools available for the model to call
  */
 type GenerateTextAiSdkOptions<TOOLS extends ToolSet = ToolSet> = AiSdkOptions & {
@@ -123,14 +134,23 @@ type GenerateTextAiSdkOptions<TOOLS extends ToolSet = ToolSet> = AiSdkOptions &
   toolChoice?: ToolChoice<TOOLS>;
   /** Limit which tools are active without changing types */
   activeTools?: Array<keyof TOOLS>;
+  /** Maximum number of automatic tool execution rounds (multi-step) */
+  maxSteps?: number;
+  /** Custom stop conditions for multi-step execution */
+  stopWhen?: StopCondition<TOOLS> | StopCondition<TOOLS>[];
+  /** Callback after each step completes */
+  onStepFinish?: GenerateTextOnStepFinishCallback<TOOLS>;
+  /** Customize each step before execution */
+  prepareStep?: PrepareStepFunction<TOOLS>;
 };
 
 /**
  * Result from generateText including full AI SDK response metadata.
  * Extends AI SDK's GenerateTextResult with a unified `result` field.
+ * @typeParam TOOLS - The tools available for the model to call (preserves typing on steps)
  */
-export type GenerateTextResult =
-  AIGenerateTextResult<Record<string, never>, unknown> & {
+export type GenerateTextResult<TOOLS extends ToolSet = ToolSet> =
+  AIGenerateTextResult<TOOLS, unknown> & {
     /** Unified field name alias for 'text' - provides consistency across all generate* functions */
     result: string;
   };
@@ -177,7 +197,7 @@ export function generateText<TOOLS extends ToolSet = ToolSet>(
     prompt: string,
     variables?: Record<string, string | number | boolean>
   } & GenerateTextAiSdkOptions<TOOLS>
-): Promise<GenerateTextResult>;
+): Promise<GenerateTextResult<TOOLS>>;
 
 /**
  * Use an LLM model to generate an object with a fixed schema.