rag-vector/ai.koog.rag.vector.storage

Package-level declarations

Types

EmbeddingStorage

open class EmbeddingStorage<Document>(embedder: DocumentEmbedder<Document>, storage: VectorStorageBackend<Document>) : VectorStorage<Document, SimilaritySearchRequest> , CapabilityAwareStorage

A VectorStorage naive implementation that composes a DocumentEmbedder with a VectorStorageBackend.

FileDocumentEmbeddingStorage

open class FileDocumentEmbeddingStorage<Document, Path>(embedder: DocumentEmbedder<Document>, documentProvider: DocumentProvider<Path, Document>, fs: FileSystemProvider.ReadWrite<Path>, root: Path) : EmbeddingStorage<Document>

A file-based VectorStorage implementation.

InMemoryDocumentEmbeddingStorage

open class InMemoryDocumentEmbeddingStorage<Document>(embedder: DocumentEmbedder<Document>) : EmbeddingStorage<Document>

An in-memory VectorStorage implementation.

InMemoryTextDocumentEmbeddingStorage

class InMemoryTextDocumentEmbeddingStorage<Document, Path>(embedder: Embedder, documentReader: DocumentProvider<Path, Document>) : InMemoryDocumentEmbeddingStorage<Document>

Implementation of an in-memory storage solution for text document embeddings.

JVMFileDocumentEmbeddingStorage

class JVMFileDocumentEmbeddingStorage(embedder: DocumentEmbedder<Path>, root: Path) : FileDocumentEmbeddingStorage<Path, Path>

A file-system-based VectorStorage implementation for managing and embedding documents represented by file paths.

JVMFileEmbeddingStorage

class JVMFileEmbeddingStorage(embedder: Embedder, root: Path) : TextFileDocumentEmbeddingStorage<Path, Path>

A JVM-specific implementation of TextFileDocumentEmbeddingStorage tailored for text document embedding and storage within a file system. This class utilizes a ai.koog.rag.base.files.JVMDocumentProvider to handle document reading and manages embeddings using a provided ai.koog.embeddings.base.Embedder.

TextFileDocumentEmbeddingStorage

open class TextFileDocumentEmbeddingStorage<Document, Path>(embedder: Embedder, documentProvider: DocumentProvider<Path, Document>, fs: FileSystemProvider.ReadWrite<Path>, root: Path) : FileDocumentEmbeddingStorage<Document, Path>

A file-based VectorStorage implementation for text documents.

interface VectorStorage<Document, in Request : SearchRequest> : WriteStorage<Document> , LookupStorage<Document> , SearchStorage<Document, Request> , DeletionStorage

Interface for a vector storage that combines document ingestion and retrieval.