koog-agents-optimization

0.1.0-SNAPSHOT

koog-agents-optimization/ai.koog.agents.optimization.core/OptimizationArtifact

OptimizationArtifact

@Serializable

data class OptimizationArtifact(val strategyInstruction: String? = null, val strategyDemonstrations: List<Demonstration> = emptyList(), val subgraphInstructions: Map<String, String> = emptyMap(), val subgraphDemonstrations: Map<String, List<Demonstration>> = emptyMap())

The serializable output of an optimization run, holding optimized instructions and demonstrations.

Produced by optimizers, consumed by optimizable subgraphs at runtime, and persisted to the filesystem between training and inference.

Immutability simplifies parallel evaluation: each optimization trial can safely hold its own artifact instance without synchronization.

Subgraph entries use the subgraph name as key. The strategy-level instruction and demonstrations are stored separately from subgraph entries.

Constructors

OptimizationArtifact

constructor(strategyInstruction: String? = null, strategyDemonstrations: List<Demonstration> = emptyList(), subgraphInstructions: Map<String, String> = emptyMap(), subgraphDemonstrations: Map<String, List<Demonstration>> = emptyMap())

Types

object Companion

Holds the agent-storage key under which the artifact is published at runtime.

Properties

strategyDemonstrations

val strategyDemonstrations: List<Demonstration>

Demonstrations for the strategy level.

strategyInstruction

val strategyInstruction: String? = null

Optimized instruction for the strategy's system prompt, or null to keep the original.

subgraphDemonstrations

val subgraphDemonstrations: Map<String, List<Demonstration>>

Map from subgraph name to demonstration list.

subgraphInstructions

val subgraphInstructions: Map<String, String>

Map from subgraph name to optimized instruction text.

Functions

getDemonstrations

fun getDemonstrations(subgraphName: String): List<Demonstration>

Returns the demonstrations for the given subgraph, or an empty list if not present.

fun getInstruction(subgraphName: String): String?

Returns the optimized instruction for the given subgraph, or null if not present.

fun mergeWith(other: OptimizationArtifact): OptimizationArtifact

Merges this artifact with another, with the other artifact's entries taking precedence.

withStrategyDemonstrations

fun withStrategyDemonstrations(demonstrations: List<Demonstration>): OptimizationArtifact

Creates a new artifact with updated strategy-level demonstrations.

withStrategyInstruction

fun withStrategyInstruction(instruction: String): OptimizationArtifact

Creates a new artifact with an updated strategy-level instruction.

withSubgraphDemonstrations

fun withSubgraphDemonstrations(subgraphName: String, demonstrations: List<Demonstration>): OptimizationArtifact

Creates a new artifact with additional or updated subgraph demonstrations.

withSubgraphInstruction

fun withSubgraphInstruction(subgraphName: String, instruction: String): OptimizationArtifact

Creates a new artifact with an additional or updated subgraph instruction.