UNPKG

@mastra/core

Version:

The core foundation of the Mastra framework, providing essential components and interfaces for building AI-powered applications.

33 lines 3.17 kB
import type { CoreMessage, LanguageModel } from 'ai'; import type { JSONSchema7 } from 'json-schema'; import type { ZodSchema } from 'zod'; import type { MastraPrimitives } from '../../action'; import type { Mastra } from '../../mastra'; import { MastraLLMBase } from './base'; import type { GenerateObjectWithMessagesArgs, GenerateTextResult, GenerateObjectResult, GenerateTextWithMessagesArgs, ToolSet, GenerateReturn, StreamTextWithMessagesArgs, StreamTextResult, StreamObjectWithMessagesArgs, StreamObjectResult, StreamReturn } from './base.types'; export declare class MastraLLM extends MastraLLMBase { #private; constructor({ model, mastra }: { model: LanguageModel; mastra?: Mastra; }); __registerPrimitives(p: MastraPrimitives): void; __registerMastra(p: Mastra): void; getProvider(): string; getModelId(): string; getModel(): import("ai").LanguageModelV1; private _applySchemaCompat; __text<Tools extends ToolSet, Z extends ZodSchema | JSONSchema7 | undefined>({ runId, messages, maxSteps, tools, temperature, toolChoice, onStepFinish, experimental_output, telemetry, threadId, resourceId, runtimeContext, ...rest }: GenerateTextWithMessagesArgs<Tools, Z>): Promise<GenerateTextResult<Tools, Z>>; __textObject<Z extends ZodSchema | JSONSchema7>({ messages, structuredOutput, runId, telemetry, threadId, resourceId, runtimeContext, ...rest }: GenerateObjectWithMessagesArgs<Z>): Promise<GenerateObjectResult<Z>>; __stream<Tools extends ToolSet, Z extends ZodSchema | JSONSchema7 | undefined = undefined>({ messages, onStepFinish, onFinish, maxSteps, tools, runId, temperature, toolChoice, experimental_output, telemetry, threadId, resourceId, runtimeContext, ...rest }: StreamTextWithMessagesArgs<Tools, Z>): StreamTextResult<Tools, Z>; __streamObject<T extends ZodSchema | JSONSchema7>({ messages, runId, runtimeContext, threadId, resourceId, onFinish, structuredOutput, telemetry, ...rest }: StreamObjectWithMessagesArgs<T>): StreamObjectResult<T>; generate<Output extends ZodSchema | JSONSchema7 | undefined = undefined, StructuredOutput extends ZodSchema | JSONSchema7 | undefined = undefined, Tools extends ToolSet = ToolSet>(messages: string | string[] | CoreMessage[], { output, ...rest }: Omit<Output extends undefined ? GenerateTextWithMessagesArgs<Tools, StructuredOutput> : Omit<GenerateObjectWithMessagesArgs<NonNullable<Output>>, 'structuredOutput' | 'output'>, 'messages'> & { output?: Output; }): Promise<GenerateReturn<Tools, Output, StructuredOutput>>; stream<Output extends ZodSchema | JSONSchema7 | undefined = undefined, StructuredOutput extends ZodSchema | JSONSchema7 | undefined = undefined, Tools extends ToolSet = ToolSet>(messages: string | string[] | CoreMessage[], { maxSteps, output, onFinish, ...rest }: Omit<Output extends undefined ? StreamTextWithMessagesArgs<Tools, StructuredOutput> : Omit<StreamObjectWithMessagesArgs<NonNullable<Output>>, 'structuredOutput' | 'output'> & { maxSteps?: never; }, 'messages'> & { output?: Output; }): StreamReturn<Tools, Output, StructuredOutput>; } //# sourceMappingURL=model.d.ts.map