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

Package-level declarations

Types

data class DualShotConfig(val summaryWords: Int = 100)

class DualShotRagServiceSearchTools(val ragService: RagService, val options: RagOptions, val summarizer: RagResponseSummarizer)

Expose a RagService as tools. Once the tools instance is created, options such as similarity cutoff are immutable and will be used consistently in all calls. The LLM needs to provide only the search query.

RagOptions

data class RagOptions constructor(val similarityThreshold: <Error class: unknown class> = 0.7, val topK: Int = 8, val hyDE: HyDE? = null, val desiredMaxLatency: Duration = Duration.ofMillis(5000), val compressionConfig: CompressionConfig = CompressionConfig(), val llm: <Error class: unknown class> = LlmOptions.withAutoLlm(), val contentElementSearch: ContentElementSearch = ContentElementSearch.CHUNKS_ONLY, val entitySearch: EntitySearch? = null, val ragResponseFormatter: RagResponseFormatter = SimpleRagResponseFormatter, val service: String? = null, val listener: RagEventListener = RagEventListener.NOOP, val dualShot: DualShotConfig? = null) : RagRequestRefinement<RagOptions>

Operations for RAG use as an LLM tool. Options are immutable and stable.

SingleShotRagServiceSearchTools

class SingleShotRagServiceSearchTools(val ragService: RagService, val options: RagOptions)