koog-spring-ai-starter-model-chat/ai.koog.spring.ai.chat

Package-level declarations

Types

Extension point for provider-specific org.springframework.ai.chat.prompt.ChatOptions customization.

@ConfigurationProperties(prefix = "koog.spring.ai.chat")

data class KoogSpringAiChatProperties(val enabled: Boolean = true, val chatModelBeanName: String? = null, val moderationModelBeanName: String? = null, val provider: String? = null, val dispatcher: KoogSpringAiChatProperties.DispatcherProperties = DispatcherProperties())

Configuration properties for the Koog Spring AI Chat Model adapter.

SpringAiChatAutoConfiguration

@AutoConfiguration(afterName = ["org.springframework.ai.model.anthropic.autoconfigure.AnthropicChatAutoConfiguration", "org.springframework.ai.model.azure.openai.autoconfigure.AzureOpenAiChatAutoConfiguration", "org.springframework.ai.model.bedrock.converse.autoconfigure.BedrockConverseProxyChatAutoConfiguration", "org.springframework.ai.model.deepseek.autoconfigure.DeepSeekChatAutoConfiguration", "org.springframework.ai.model.google.genai.autoconfigure.chat.GoogleGenAiChatAutoConfiguration", "org.springframework.ai.model.huggingface.autoconfigure.HuggingfaceChatAutoConfiguration", "org.springframework.ai.model.minimax.autoconfigure.MiniMaxChatAutoConfiguration", "org.springframework.ai.model.mistralai.autoconfigure.MistralAiChatAutoConfiguration", "org.springframework.ai.model.oci.genai.autoconfigure.OCIGenAiChatAutoConfiguration", "org.springframework.ai.model.ollama.autoconfigure.OllamaChatAutoConfiguration", "org.springframework.ai.model.openai.autoconfigure.OpenAiChatAutoConfiguration", "org.springframework.ai.model.openaisdk.autoconfigure.OpenAiSdkChatAutoConfiguration", "org.springframework.ai.model.vertexai.autoconfigure.gemini.VertexAiGeminiChatAutoConfiguration", "org.springframework.ai.model.zhipuai.autoconfigure.ZhiPuAiChatAutoConfiguration"])

@EnableConfigurationProperties(value = [KoogSpringAiChatProperties::class])

@ConditionalOnClass(value = [ChatModel::class])

@ConditionalOnProperty(prefix = "koog.spring.ai.chat", name = ["enabled"], havingValue = "true", matchIfMissing = true)

open class SpringAiChatAutoConfiguration

Auto-configuration for the Koog Spring AI Chat Model adapter.

SpringAiLLMClient

class SpringAiLLMClient(chatModel: ChatModel, provider: LLMProvider, clock: Clock, dispatcher: CoroutineDispatcher, chatOptionsCustomizer: ChatOptionsCustomizer, moderationModel: ModerationModel?) : LLMClient

An LLMClient implementation that delegates to a Spring AI ChatModel.

SpringAiLLMProvider

object SpringAiLLMProvider : LLMProvider

An ai.koog.prompt.llm.LLMProvider representing a Spring AI-backed provider.

Functions

koogMessageToSpringMessage

fun koogMessageToSpringMessage(message: Message): Message

Converts a Koog Message to a Spring AI SpringMessage.

koogToolDescriptorToToolCallback

fun koogToolDescriptorToToolCallback(descriptor: ToolDescriptor): ToolCallback

Converts a Koog ToolDescriptor to a Spring AI ToolCallback.

springGenerationToKoogResponses

fun springGenerationToKoogResponses(generation: Generation, clock: Clock = kotlin.time.Clock.System, usage: Usage? = null): List<Message.Response>

Converts a Spring AI Generation (from a org.springframework.ai.chat.model.ChatResponse) to a list of Koog Message.Response.

springModerationResultToKoogModerationResult

fun springModerationResultToKoogModerationResult(springResult: Moderation): ModerationResult

Converts a Spring AI SpringModeration to a Koog ModerationResult.