UNPKG

@mastra/core

Version:

Mastra is a framework for building AI-powered applications and agents with a modern TypeScript stack.

348 lines • 12.6 kB
import type { ToolSet } from 'ai-v5'; import z from 'zod'; import type { OutputSchema } from '../../../stream/base/schema.js'; import type { OuterLLMRun } from '../../types.js'; export declare function createLLMMappingStep<Tools extends ToolSet = ToolSet, OUTPUT extends OutputSchema = undefined>({ models, telemetry_settings, _internal, modelStreamSpan, ...rest }: OuterLLMRun<Tools, OUTPUT>, llmExecutionStep: any): import("../../..").Step<"llmExecutionMappingStep", z.ZodObject<any, z.UnknownKeysParam, z.ZodTypeAny, { [x: string]: any; }, { [x: string]: any; }>, z.ZodArray<z.ZodObject<{ toolCallId: z.ZodString; toolName: z.ZodString; args: z.ZodRecord<z.ZodString, z.ZodAny>; providerMetadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>; providerExecuted: z.ZodOptional<z.ZodBoolean>; output: z.ZodOptional<z.ZodAny>; } & { result: z.ZodAny; error: z.ZodOptional<z.ZodAny>; }, "strip", z.ZodTypeAny, { toolCallId: string; toolName: string; args: Record<string, any>; error?: any; providerMetadata?: Record<string, any> | undefined; providerExecuted?: boolean | undefined; output?: any; result?: any; }, { toolCallId: string; toolName: string; args: Record<string, any>; error?: any; providerMetadata?: Record<string, any> | undefined; providerExecuted?: boolean | undefined; output?: any; result?: any; }>, "many">, z.ZodObject<{ messageId: z.ZodString; messages: z.ZodObject<{ all: z.ZodArray<z.ZodAny, "many">; user: z.ZodArray<z.ZodAny, "many">; nonUser: z.ZodArray<z.ZodAny, "many">; }, "strip", z.ZodTypeAny, { user: any[]; all: any[]; nonUser: any[]; }, { user: any[]; all: any[]; nonUser: any[]; }>; output: z.ZodObject<{ text: z.ZodOptional<z.ZodString>; reasoning: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; reasoningText: z.ZodOptional<z.ZodString>; files: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; toolCalls: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; toolResults: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; sources: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; staticToolCalls: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; dynamicToolCalls: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; staticToolResults: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; dynamicToolResults: z.ZodOptional<z.ZodArray<z.ZodAny, "many">>; usage: z.ZodObject<{ inputTokens: z.ZodNumber; outputTokens: z.ZodNumber; totalTokens: z.ZodNumber; reasoningTokens: z.ZodOptional<z.ZodNumber>; cachedInputTokens: z.ZodOptional<z.ZodNumber>; }, "strip", z.ZodTypeAny, { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }, { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }>; steps: z.ZodArray<z.ZodAny, "many">; }, "strip", z.ZodTypeAny, { usage: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }; steps: any[]; text?: string | undefined; reasoning?: any[] | undefined; sources?: any[] | undefined; files?: any[] | undefined; toolCalls?: any[] | undefined; toolResults?: any[] | undefined; reasoningText?: string | undefined; staticToolCalls?: any[] | undefined; dynamicToolCalls?: any[] | undefined; staticToolResults?: any[] | undefined; dynamicToolResults?: any[] | undefined; }, { usage: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }; steps: any[]; text?: string | undefined; reasoning?: any[] | undefined; sources?: any[] | undefined; files?: any[] | undefined; toolCalls?: any[] | undefined; toolResults?: any[] | undefined; reasoningText?: string | undefined; staticToolCalls?: any[] | undefined; dynamicToolCalls?: any[] | undefined; staticToolResults?: any[] | undefined; dynamicToolResults?: any[] | undefined; }>; metadata: z.ZodObject<{ id: z.ZodOptional<z.ZodString>; model: z.ZodOptional<z.ZodString>; modelId: z.ZodOptional<z.ZodString>; modelMetadata: z.ZodOptional<z.ZodObject<{ modelId: z.ZodString; modelVersion: z.ZodString; modelProvider: z.ZodString; }, "strip", z.ZodTypeAny, { modelId: string; modelVersion: string; modelProvider: string; }, { modelId: string; modelVersion: string; modelProvider: string; }>>; timestamp: z.ZodOptional<z.ZodDate>; providerMetadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>; headers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>; request: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>; }, "strip", z.ZodTypeAny, { id?: string | undefined; providerMetadata?: Record<string, any> | undefined; headers?: Record<string, string> | undefined; model?: string | undefined; request?: Record<string, any> | undefined; timestamp?: Date | undefined; modelId?: string | undefined; modelMetadata?: { modelId: string; modelVersion: string; modelProvider: string; } | undefined; }, { id?: string | undefined; providerMetadata?: Record<string, any> | undefined; headers?: Record<string, string> | undefined; model?: string | undefined; request?: Record<string, any> | undefined; timestamp?: Date | undefined; modelId?: string | undefined; modelMetadata?: { modelId: string; modelVersion: string; modelProvider: string; } | undefined; }>; stepResult: z.ZodObject<{ reason: z.ZodString; warnings: z.ZodArray<z.ZodAny, "many">; isContinued: z.ZodBoolean; logprobs: z.ZodOptional<z.ZodAny>; totalUsage: z.ZodOptional<z.ZodObject<{ inputTokens: z.ZodNumber; outputTokens: z.ZodNumber; totalTokens: z.ZodNumber; reasoningTokens: z.ZodOptional<z.ZodNumber>; cachedInputTokens: z.ZodOptional<z.ZodNumber>; }, "strip", z.ZodTypeAny, { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }, { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }>>; headers: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>; messageId: z.ZodOptional<z.ZodString>; request: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>; }, "strip", z.ZodTypeAny, { reason: string; warnings: any[]; isContinued: boolean; messageId?: string | undefined; headers?: Record<string, string> | undefined; request?: Record<string, any> | undefined; logprobs?: any; totalUsage?: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; } | undefined; }, { reason: string; warnings: any[]; isContinued: boolean; messageId?: string | undefined; headers?: Record<string, string> | undefined; request?: Record<string, any> | undefined; logprobs?: any; totalUsage?: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; } | undefined; }>; }, "strip", z.ZodTypeAny, { output: { usage: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }; steps: any[]; text?: string | undefined; reasoning?: any[] | undefined; sources?: any[] | undefined; files?: any[] | undefined; toolCalls?: any[] | undefined; toolResults?: any[] | undefined; reasoningText?: string | undefined; staticToolCalls?: any[] | undefined; dynamicToolCalls?: any[] | undefined; staticToolResults?: any[] | undefined; dynamicToolResults?: any[] | undefined; }; messageId: string; metadata: { id?: string | undefined; providerMetadata?: Record<string, any> | undefined; headers?: Record<string, string> | undefined; model?: string | undefined; request?: Record<string, any> | undefined; timestamp?: Date | undefined; modelId?: string | undefined; modelMetadata?: { modelId: string; modelVersion: string; modelProvider: string; } | undefined; }; messages: { user: any[]; all: any[]; nonUser: any[]; }; stepResult: { reason: string; warnings: any[]; isContinued: boolean; messageId?: string | undefined; headers?: Record<string, string> | undefined; request?: Record<string, any> | undefined; logprobs?: any; totalUsage?: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; } | undefined; }; }, { output: { usage: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; }; steps: any[]; text?: string | undefined; reasoning?: any[] | undefined; sources?: any[] | undefined; files?: any[] | undefined; toolCalls?: any[] | undefined; toolResults?: any[] | undefined; reasoningText?: string | undefined; staticToolCalls?: any[] | undefined; dynamicToolCalls?: any[] | undefined; staticToolResults?: any[] | undefined; dynamicToolResults?: any[] | undefined; }; messageId: string; metadata: { id?: string | undefined; providerMetadata?: Record<string, any> | undefined; headers?: Record<string, string> | undefined; model?: string | undefined; request?: Record<string, any> | undefined; timestamp?: Date | undefined; modelId?: string | undefined; modelMetadata?: { modelId: string; modelVersion: string; modelProvider: string; } | undefined; }; messages: { user: any[]; all: any[]; nonUser: any[]; }; stepResult: { reason: string; warnings: any[]; isContinued: boolean; messageId?: string | undefined; headers?: Record<string, string> | undefined; request?: Record<string, any> | undefined; logprobs?: any; totalUsage?: { inputTokens: number; outputTokens: number; totalTokens: number; reasoningTokens?: number | undefined; cachedInputTokens?: number | undefined; } | undefined; }; }>, z.ZodType<any, z.ZodTypeDef, any>, z.ZodType<any, z.ZodTypeDef, any>, import("../../..").DefaultEngineType>; //# sourceMappingURL=llm-mapping-step.d.ts.map