Embabel Agent Documentation/com.embabel.agent.rag.neo.ogm/OgmRagFacetProvider

OgmRagFacetProvider

class OgmRagFacetProvider(modelProvider: <Error class: unknown class>, ogmCypherSearch: OgmCypherSearch, schemaResolver: SchemaResolver, sessionFactory: <Error class: unknown class>, platformTransactionManager: <Error class: unknown class>, properties: NeoRagServiceProperties) : AbstractWritableContentElementRepository, RagFacetProvider

Performs RAG queries in readonly transactions using Neo4j OGM. Requires a Neo4j OGM PlatformTransactionManager to be configured in the Spring context.

Constructors

OgmRagFacetProvider

constructor(modelProvider: <Error class: unknown class>, ogmCypherSearch: OgmCypherSearch, schemaResolver: SchemaResolver, sessionFactory: <Error class: unknown class>, platformTransactionManager: <Error class: unknown class>, properties: NeoRagServiceProperties)

Properties

name

open val name: String

Functions

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

count

open override fun count(): Int

Return the total number of content elements in the repository

embeddingFor

fun embeddingFor(text: String): Embedding

entityVectorSearch

fun entityVectorSearch(request: <Error class: unknown class>, embedding: FloatArray, labels: Set<String>): List<<Error class: unknown class><EntityData>>

facets

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

findAll

fun findAll(): List<ContentElement>

findById

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

findChunksById

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

findChunksForEntity

open override fun findChunksForEntity(entityId: String): List<Chunk>

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 override fun provision()

Provision this rag service if necessary

save

open override fun save(element: ContentElement): ContentElement

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

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.