Embabel Agent Documentation/com.embabel.agent.rag/RagRequest

RagRequest

data class RagRequest(val query: String, val similarityThreshold: <Error class: unknown class> = 0.8, val topK: Int = 8, val hyDE: HyDE? = null, val desiredMaxLatency: Duration = Duration.ofMillis(5000), val compressionConfig: CompressionConfig = CompressionConfig(), val contentElementSearch: ContentElementSearch = ContentElementSearch.CHUNKS_ONLY, val entitySearch: EntitySearch? = null, val timestamp: Instant = Instant.now()) : RagRequestRefinement<RagRequest>

RAG request. Contains a query and parameters for similarity search.

Parameters

query

the query string to search for

similarityThreshold

the minimum similarity score for results (default is 0.8)

topK

the maximum number of results to return (default is 8) If set, only the given entities will be searched for.

Constructors

RagRequest

constructor(query: String, similarityThreshold: <Error class: unknown class> = 0.8, topK: Int = 8, hyDE: HyDE? = null, desiredMaxLatency: Duration = Duration.ofMillis(5000), compressionConfig: CompressionConfig = CompressionConfig(), contentElementSearch: ContentElementSearch = ContentElementSearch.CHUNKS_ONLY, entitySearch: EntitySearch? = null, timestamp: Instant = Instant.now())