@mastra/rag/dist/document/extractors/types.d.ts

The Retrieval-Augmented Generation (RAG) module contains document processing and embedding utilities.

import type { MastraLanguageModel, MastraLegacyLanguageModel } from '@mastra/core/agent';
import type { z } from 'zod';
import type { KeywordExtractPrompt, QuestionExtractPrompt, SummaryPrompt, TitleExtractorPrompt, TitleCombinePrompt } from '../prompts/index.js';
export type KeywordExtractArgs = {
    llm?: MastraLegacyLanguageModel | MastraLanguageModel;
    keywords?: number;
    promptTemplate?: KeywordExtractPrompt['template'];
};
export type QuestionAnswerExtractArgs = {
    llm?: MastraLegacyLanguageModel | MastraLanguageModel;
    questions?: number;
    promptTemplate?: QuestionExtractPrompt['template'];
    embeddingOnly?: boolean;
};
export type SummaryExtractArgs = {
    llm?: MastraLegacyLanguageModel | MastraLanguageModel;
    summaries?: string[];
    promptTemplate?: SummaryPrompt['template'];
};
export type TitleExtractorsArgs = {
    llm?: MastraLegacyLanguageModel | MastraLanguageModel;
    nodes?: number;
    nodeTemplate?: TitleExtractorPrompt['template'];
    combineTemplate?: TitleCombinePrompt['template'];
};
export type SchemaExtractArgs<T extends z.ZodType = z.ZodType> = {
    schema: T;
    llm?: MastraLegacyLanguageModel | MastraLanguageModel;
    instructions?: string;
    metadataKey?: string;
};
export declare const STRIP_REGEX: RegExp;
export declare const baseLLM: MastraLegacyLanguageModel | MastraLanguageModel;
//# sourceMappingURL=types.d.ts.map