UNPKG

@mastra/rag

Version:

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

31 lines 1.08 kB
import type { MastraLanguageModel } from '@mastra/core/agent'; import type { TitleCombinePrompt, TitleExtractorPrompt } from '../prompts'; import type { BaseNode } from '../schema'; import { BaseExtractor } from './base'; import type { TitleExtractorsArgs } from './types'; type ExtractTitle = { documentTitle: string; }; /** * Extract title from a list of nodes. */ export declare class TitleExtractor extends BaseExtractor { llm: MastraLanguageModel; isTextNodeOnly: boolean; nodes: number; nodeTemplate: TitleExtractorPrompt; combineTemplate: TitleCombinePrompt; constructor(options?: TitleExtractorsArgs); /** * Extract titles from a list of nodes. * @param {BaseNode[]} nodes Nodes to extract titles from. * @returns {Promise<BaseNode<ExtractTitle>[]>} Titles extracted from the nodes. */ extract(nodes: BaseNode[]): Promise<Array<ExtractTitle>>; private filterNodes; private separateNodesByDocument; private extractTitles; private getTitlesCandidates; } export {}; //# sourceMappingURL=title.d.ts.map