koog-agents-optimization

0.1.0-SNAPSHOT

koog-agents-optimization/ai.koog.agents.optimization.training.metrics.impl/ConsumptionMetric

ConsumptionMetric

@Serializable(with = ConsumptionMetricSerializer::class)

@SerialName(value = "consumption")

data class ConsumptionMetric(val key: MetricKey<ConsumptionMetric> = KEY, val totalConsumption: LLMConsumptionOrNA = LLMConsumptionOrNA.NA, val minPerSubstage: Double? = null, val maxPerSubstage: Double? = null, val avgPerSubstage: Double? = null, val withRetries: ConsumptionWithRetries? = null) : Metric

Total / min / max / avg LLM consumption across substages. Canonical totals are final-attempt only; withRetries, when non-null, exposes the retries-aware counterparts.

Constructors

ConsumptionMetric

constructor(key: MetricKey<ConsumptionMetric> = KEY, totalConsumption: LLMConsumptionOrNA = LLMConsumptionOrNA.NA, minPerSubstage: Double? = null, maxPerSubstage: Double? = null, avgPerSubstage: Double? = null, withRetries: ConsumptionWithRetries? = null)

Types

object Companion

Key holder for ConsumptionMetric.

Properties

val avgPerSubstage: Double? = null

Mean per-substage total cost across finished substages, or null if none.

@Transient

open override val key: MetricKey<ConsumptionMetric>

The key that identifies this metric in a ai.koog.agents.optimization.training.metrics.MetricsMap.

val maxPerSubstage: Double? = null

Largest per-substage total cost across finished substages, or null if none.

val minPerSubstage: Double? = null

Smallest per-substage total cost across finished substages, or null if none.

totalConsumption

val totalConsumption: LLMConsumptionOrNA

Running sum of LLM consumption across substages (final-attempt only), or NA if none observed.

val withRetries: ConsumptionWithRetries? = null

Retries-aware totals; null when no retries fired below this stage.

Functions

open override fun recompute(currentStage: StageRecord, substages: List<TrainingRecord>, isStageUpdate: Boolean): ConsumptionMetric

Recompute this metric from the full list of substages records.

fun totalOrNull(): LLMConsumption?

Get the total as an LLMConsumption or null.