koog-spring-ai-starter-model-embedding/ai.koog.spring.ai.embedding/SpringAiLLMEmbeddingProvider

SpringAiLLMEmbeddingProvider

class SpringAiLLMEmbeddingProvider(embeddingModel: EmbeddingModel, dispatcher: CoroutineDispatcher) : LLMEmbeddingProvider(source)

An LLMEmbeddingProvider implementation that delegates to a Spring AI EmbeddingModel.

When multiple embedding models are registered in the Spring context, use the koog.spring.ai.embedding.embedding-model-bean-name property to select the desired bean. The LLModel.id is forwarded to the underlying Spring AI model via EmbeddingOptions so that backends which support runtime model selection (e.g. OpenAI-compatible endpoints) can honour it; backends that ignore the option will simply use their pre-configured model.

Parameters

embeddingModel

the Spring AI embedding model to delegate to

dispatcher

the CoroutineDispatcher used for blocking embedding calls

Constructors

SpringAiLLMEmbeddingProvider

constructor(embeddingModel: EmbeddingModel, dispatcher: CoroutineDispatcher)

Types

A Java-friendly builder for SpringAiLLMEmbeddingProvider.

object Companion

Java-friendly builder access.

Functions

open suspend override fun embed(text: String, model: LLModel): List<Double>

Embeds the given text using the configured Spring AI EmbeddingModel.

open suspend override fun embed(inputs: List<String>, model: LLModel): List<List<Double>>

Embeds the given inputs using the configured Spring AI EmbeddingModel.

@JvmName(name = "embed")

fun embedBlocking(text: String, model: LLModel, executorService: ExecutorService?): List<Double>

@JvmName(name = "embed")

fun embedBlocking(inputs: List<String>, model: LLModel, executorService: ExecutorService?): List<List<Double>>