UNPKG

@genkit-ai/dev-local-vectorstore

Version:

Genkit AI framework plugin for temporary local vector database.

34 lines (30 loc) 1.25 kB
import * as _genkit_ai_ai_retriever from '@genkit-ai/ai/retriever'; import { z } from 'genkit'; import { EmbedderArgument } from 'genkit/embedder'; import { GenkitPlugin } from 'genkit/plugin'; interface Params<EmbedderCustomOptions extends z.ZodTypeAny> { indexName: string; embedder: EmbedderArgument<EmbedderCustomOptions>; embedderOptions?: z.infer<EmbedderCustomOptions>; } /** * Local file-based vectorstore plugin that provides retriever and indexer. * * NOT INTENDED FOR USE IN PRODUCTION */ declare function devLocalVectorstore<EmbedderCustomOptions extends z.ZodTypeAny>(params: Params<EmbedderCustomOptions>[]): GenkitPlugin; /** * Local file-based vectorstore retriever reference */ declare function devLocalRetrieverRef(indexName: string): _genkit_ai_ai_retriever.RetrieverReference<z.ZodOptional<z.ZodObject<{ k: z.ZodOptional<z.ZodNumber>; }, "strip", z.ZodTypeAny, { k?: number | undefined; }, { k?: number | undefined; }>>>; /** * Local file-based indexer reference */ declare function devLocalIndexerRef(indexName: string): _genkit_ai_ai_retriever.IndexerReference<z.ZodOptional<z.ZodNull>>; export { devLocalVectorstore as default, devLocalIndexerRef, devLocalRetrieverRef, devLocalVectorstore };