@mastra/rag
Version:
The Retrieval-Augmented Generation (RAG) module contains document processing and embedding utilities.
27 lines • 1.01 kB
TypeScript
import type { MastraLanguageModel } from '@mastra/core/agent';
import type { KeywordExtractPrompt, QuestionExtractPrompt, SummaryPrompt, TitleExtractorPrompt, TitleCombinePrompt } from '../prompts/index.js';
export type KeywordExtractArgs = {
llm?: MastraLanguageModel;
keywords?: number;
promptTemplate?: KeywordExtractPrompt['template'];
};
export type QuestionAnswerExtractArgs = {
llm?: MastraLanguageModel;
questions?: number;
promptTemplate?: QuestionExtractPrompt['template'];
embeddingOnly?: boolean;
};
export type SummaryExtractArgs = {
llm?: MastraLanguageModel;
summaries?: string[];
promptTemplate?: SummaryPrompt['template'];
};
export type TitleExtractorsArgs = {
llm?: MastraLanguageModel;
nodes?: number;
nodeTemplate?: TitleExtractorPrompt['template'];
combineTemplate?: TitleCombinePrompt['template'];
};
export declare const STRIP_REGEX: RegExp;
export declare const baseLLM: MastraLanguageModel;
//# sourceMappingURL=types.d.ts.map