UNPKG

ai-utils.js

Version:

Build AI applications, chatbots, and agents with JavaScript and TypeScript.

36 lines (35 loc) 1.53 kB
import { TextEmbeddingModel, TextEmbeddingModelSettings } from "../model-function/embed-text/TextEmbeddingModel.js"; import { Run } from "../run/Run.js"; import { TextChunk } from "../text-chunk/TextChunk.js"; import { TextChunkRetrieverSettings } from "../text-chunk/retrieve-text-chunks/TextChunkRetriever.js"; import { VectorIndex } from "./VectorIndex.js"; import { FunctionOptions } from "../model-function/FunctionOptions.js"; export declare class VectorIndexTextChunkStore<CHUNK extends TextChunk, INDEX, MODEL extends TextEmbeddingModel<unknown, TextEmbeddingModelSettings>> { private readonly _index; private readonly generateId; private readonly embeddingModel; private readonly queryFunctionId?; private readonly upsertFunctionId?; constructor({ index, generateId, embeddingModel, queryFunctionId, upsertFunctionId, }: { index: VectorIndex<CHUNK, INDEX>; generateId?: () => string; embeddingModel: MODEL; queryFunctionId?: string; upsertFunctionId?: string; }); upsertChunk({ id, chunk, }: { id?: string; keyText: string; chunk: CHUNK; }, options?: { run?: Run; }): Promise<void>; upsertManyChunks({ ids, chunks, }: { ids?: Array<string | undefined>; chunks: CHUNK[]; }, options?: { run?: Run; }): Promise<void>; retrieveSimilarTextChunks(queryText: string, options?: FunctionOptions<TextChunkRetrieverSettings> | undefined): Promise<CHUNK[]>; get index(): INDEX; }