UNPKG

@llamaindex/core

Version:
33 lines (29 loc) 1.27 kB
import { MessageContent } from '../../llms/dist/index.js'; import { BaseRetriever } from '../../retriever/dist/index.js'; import { TextNode, BaseNode } from '../../schema/dist/index.js'; /** * @todo refactor this module, most of the part is broken * reference * - https://github.com/run-llama/LlamaIndexTS/pull/531 * - https://github.com/run-llama/LlamaIndexTS/pull/416 */ declare abstract class BaseObjectNodeMapping { abstract fromObjects<OT>(objs: OT[], ...args: any[]): BaseObjectNodeMapping; abstract objNodeMapping(): Record<any, any>; abstract toNode(obj: any): TextNode; validateObject(obj: any): void; addObj(obj: any): void; abstract _addObj(obj: any): void; toNodes(objs: any[]): TextNode[]; abstract _fromNode(node: BaseNode): any; fromNode(node: BaseNode): any; abstract persist(persistDir: string, objNodeMappingFilename: string): void; } declare class ObjectRetriever<T = unknown> { _retriever: BaseRetriever; _objectNodeMapping: BaseObjectNodeMapping; constructor(retriever: BaseRetriever, objectNodeMapping: BaseObjectNodeMapping); get retriever(): BaseRetriever; retrieve(strOrQueryBundle: MessageContent): Promise<T[]>; } export { BaseObjectNodeMapping, ObjectRetriever };