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

Package-level declarations

Types

class MappedChunk(val id: String, val uri: String? = null, val text: String, val parentId: String, var metadata: Map<String, Any?> = emptyMap())

NeoOgmConfig

class NeoOgmConfig(properties: NeoRagServiceProperties)

NeoRagServiceProperties

class NeoRagServiceProperties

OgmCypherSearch

class OgmCypherSearch(sessionFactory: <Error class: unknown class>, queryResolver: LogicalQueryResolver) : CypherSearch

OgmMappedEntity

open class OgmMappedEntity(val id: String, val uri: String? = null, val chunks: List<MappedChunk> = emptyList())

Superclass for all entities that are mapped using Neo4j OGM.

OgmMappedNamedAndDescribedEntity

abstract class OgmMappedNamedAndDescribedEntity(val name: String, val id: String, val uri: String? = null) : OgmMappedEntity

OgmMetadataSchemaResolver

class OgmMetadataSchemaResolver(sessionFactory: <Error class: unknown class>, ogmCypherSearch: OgmCypherSearch, properties: NeoRagServiceProperties) : SchemaResolver

Infers schema from OGM metadata

OgmRagFacetProvider

class OgmRagFacetProvider(modelProvider: <Error class: unknown class>, ogmCypherSearch: OgmCypherSearch, schemaResolver: SchemaResolver, sessionFactory: <Error class: unknown class>, platformTransactionManager: <Error class: unknown class>, properties: NeoRagServiceProperties, val enhancers: List<<Error class: unknown class>> = emptyList())

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

SchemaDrivenCypherRagQueryGenerator

class SchemaDrivenCypherRagQueryGenerator(modelProvider: <Error class: unknown class>, schema: DataDictionary) : CypherRagQueryGenerator

Generate RAG queries from a given schema

Functions

cleanLlmCypher

fun cleanLlmCypher(rawResponse: String): String

Try to extract Cypher from the response