UNPKG

@mastra/rag

Version:

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

27 lines 1.01 kB
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