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

Package-level declarations

Types

class AgentRunFailureException(val failedRecord: AgentRunRecord.Failed, cause: Throwable) : Exception

Thrown by runAgent().getOrThrow() / runAgentOrThrow() to propagate a recorded agent run failure. The failedRecord is already added to the stage's substages before this exception is constructed.

DatasetMaxFailureRateExceededException

class DatasetMaxFailureRateExceededException(val failedRatio: Ratio, val threshold: Double) : RuntimeException

Thrown synchronously by StageScope.iterateDataset after the loop completes, when the observed per-item failure rate exceeds the configured threshold. Carries the structured cause (rate + threshold) so the recognizer can describe the abort without parsing a string.

PromptExecutionFailureException

class PromptExecutionFailureException(val failedRecord: PromptExecutionRecord.Failed, cause: Throwable) : Exception

Thrown when a prompt execution fails. Wraps the already-recorded PromptExecutionRecord.Failed so that the enclosing resolveStageFailure can extract the analyzed failure directly.

StageFailedException

class StageFailedException(val failedRecord: StageRecord, cause: Throwable) : Exception

Thrown by runStage(...).getOrThrow() / runStageOrThrow() to propagate a recorded stage failure. The failedRecord is already added to the parent's substages before this exception is constructed.

Functions

resolveStageFailure

fun resolveStageFailure(exception: Exception, analyzer: FailureAnalyzer): AnalyzedFailure

Returns the AnalyzedFailure for an exception caught inside a stage block.