koog-agents-optimization/ai.koog.agents.optimization.training/CapturingPromptExecutor

CapturingPromptExecutor

abstract class CapturingPromptExecutor : PromptExecutor

A PromptExecutor that accumulates the LLMConsumption of the calls it intercepts and hands it back via collectAndClear. The training infrastructure creates one per leaf prompt call (via the session's capturingExecutorFactory) and reads its consumption between retry attempts; the concrete provider-aware extraction lives in the app.

Inheritors

ConsumptionCapturingPromptExecutor

Constructors

CapturingPromptExecutor

constructor()

Functions

abstract fun close()

collectAndClear

abstract fun collectAndClear(): LLMConsumption?

Returns the consumption accumulated since the previous call (or since construction), then resets the accumulator. Returns null when nothing was captured.

execute

fun execute(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, executorService: ExecutorService?): List<Message.Response>

abstract suspend fun execute(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): List<Message.Response>

executeMultipleChoices

fun executeMultipleChoices(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, executorService: ExecutorService?): List<LLMChoice>

open suspend fun executeMultipleChoices(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): List<LLMChoice>

executeStreaming

abstract fun executeStreaming(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): Flow<StreamFrame>

executeStreamingWithPublisher

fun executeStreamingWithPublisher(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): Flow.Publisher<StreamFrame>

executeStructuredOrThrow

inline suspend fun <OutputStructT> PromptExecutor.executeStructuredOrThrow(prompt: Prompt, model: LLModel): OutputStructT

Executes prompt against model, requesting a structured response of type OutputStructT, and returns the parsed data.

getBasicJsonSchemaGenerator

open fun getBasicJsonSchemaGenerator(model: LLModel): BasicJsonSchemaGenerator

getStandardJsonSchemaGenerator

open fun getStandardJsonSchemaGenerator(model: LLModel): StandardJsonSchemaGenerator

models

fun models(executorService: ExecutorService?): List<LLModel>

open suspend fun models(): List<LLModel>

moderate

fun moderate(prompt: Prompt, model: LLModel, executorService: ExecutorService?): ModerationResult

abstract suspend fun moderate(prompt: Prompt, model: LLModel): ModerationResult