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

Package-level declarations

Types

@Serializable

sealed class AgentRunRecord : TrainingRecord, LeafRecordWithRetries

Records the outcome of a single agent run (possibly across retries). Leaf record — no substages.

Leaf record that may carry retry history. The elapsedTime and consumption fields are the final attempt's values; the extension helpers below combine them with previousAttempts.

PromptExecutionRecord

@Serializable

sealed class PromptExecutionRecord : TrainingRecord, LeafRecordWithRetries

Records the outcome of a single prompt execution (possibly across retries). Leaf record — no substages.

StageRecord

@Serializable

data class StageRecord(val name: String, var additionalData: JsonElement? = null, var failure: AnalyzedFailure? = null, var actionLog: JsonElement? = null, val metrics: MetricsMap = MetricsMap(), var isFinished: Boolean = false, var substagesTotal: Int? = null, var realElapsed: Duration = Duration.ZERO, val substages: MutableList<TrainingRecord> = mutableListOf()) : TrainingRecord

Record of a stage -- the universal node in the training records tree.

TrainingRecord

@Serializable

sealed class TrainingRecord

Base class for all records in the training records tree.

TrainingResult

@Serializable

data class TrainingResult(val rootStage: StageRecord, val executionMetadata: ExecutionMetadata)

Top-level persisted result of a training session: the tree of StageRecords plus the ExecutionMetadata captured by the runner (pod name, start/completion timestamps).

Functions

aggregatedConsumption

fun LeafRecordWithRetries.aggregatedConsumption(): LLMConsumptionOrNA

Final attempt + retry attempts. Returns NA only if every contributing value is NA.

aggregatedElapsedTime

fun LeafRecordWithRetries.aggregatedElapsedTime(): Duration

Final attempt + retry attempts + retry wait delays.

consumptionInRetries

fun LeafRecordWithRetries.consumptionInRetries(): LLMConsumptionOrNA

Retry attempts only (excludes the final attempt). NA if all retries had NA consumption.

elapsedTimeInRetries

fun LeafRecordWithRetries.elapsedTimeInRetries(): Duration

Retry-attempt elapsed + retry wait delays (excludes the final attempt).