@mastra/rag
Version:
The Retrieval-Augmented Generation (RAG) module contains document processing and embedding utilities.
31 lines • 1.08 kB
TypeScript
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