JVMFileEmbeddingStorage

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.

embedder

The embedding implementation used to generate and compare vector embeddings.

root

The root directory where the document storage system is initialized.

Constructors

constructor(embedder: Embedder, root: Path)

Creates an instance of JVMFileEmbeddingStorage.

open suspend override fun add(documents: List<Path>, namespace: String?): List<String>

Retrieves a flow of all documents stored in the system.

open suspend override fun delete(ids: List<String>, namespace: String?): List<String>

open suspend override fun get(ids: List<String>, namespace: String?): List<Path>

open suspend override fun search(request: SimilaritySearchRequest, namespace: String?): List<SearchResult<Path>>

open suspend override fun update(documents: Map<String, Path>, namespace: String?): List<String>