Embabel Agent Documentation/com.embabel.agent.rag.lucene/LuceneRagFacetProvider

LuceneRagFacetProvider

class LuceneRagFacetProvider constructor(val name: String, embeddingModel: <Error class: unknown class>? = null, vectorWeight: Double = 0.5, chunkerConfig: ContentChunker.Config = ContentChunker.DefaultConfig()) : AbstractWritableContentElementRepository, RagFacetProvider, Closeable

LuceneRagService with optional vector search support via an EmbeddingModel. Works in memory. Be careful if loading excessive content!

Constructors

constructor(name: String, embeddingModel: <Error class: unknown class>? = null, vectorWeight: Double = 0.5, chunkerConfig: ContentChunker.Config = ContentChunker.DefaultConfig())

Properties

name

open val name: String

Functions

open fun accept(documents: List<<Error class: unknown class>>)

clear

fun clear()

Clear all stored content - useful for testing

open override fun close()

count

open override fun count(): Int

facets

open override fun facets(): List<RagFacet<out Retrievable>>

findAll

fun findAll(): List<Chunk>

findById

open override fun findById(id: String): ContentElement?

findChunksById

open override fun findChunksById(chunkIds: List<String>): List<Chunk>

getStatistics

fun getStatistics(): LuceneStatistics

Get statistics about the current state

infoString

open fun infoString(verbose: Boolean?, indent: Int): String

onNewRetrievables

open override fun onNewRetrievables(retrievables: List<Retrievable>)

The Retrievables have been saved to the store, but Retrievables are special, and we probably want to embed them

provision

open fun provision()

Provision this rag service if necessary

save

open override fun save(element: ContentElement): ContentElement

fun search(ragRequest: RagRequest): RagFacetResults<Chunk>

writeContent

override fun writeContent(root: MaterializedDocument): List<String>

Will call save on the root and all descendants. The database only needs to store each descendant and link by id, rather than otherwise consider the entire structure.