koog-agents-optimization/ai.koog.agents.optimization.features

Package-level declarations

Types

Link copied to clipboard
class CollectedSubgraphTraces(maxPerSubgraph: Int)

Thread-safe container for collected subgraph execution traces and whole-agent trajectory.

Link copied to clipboard
object PromptOptimizationFeature : AIAgentGraphFeature<PromptPromptOptimizationFeatureConfig, OptimizationArtifact>

Feature that applies an OptimizationArtifact to an agent at strategy start.

Link copied to clipboard
class PromptPromptOptimizationFeatureConfig : FeatureConfig

Configuration for the PromptOptimizationFeature.

Link copied to clipboard
class SubgraphTraceCollectionConfig : FeatureConfig

Configuration for SubgraphTraceCollectionFeature.

Link copied to clipboard
object SubgraphTraceCollectionFeature : AIAgentGraphFeature<SubgraphTraceCollectionConfig, CollectedSubgraphTraces>

Optimizer-internal feature that captures both per-subgraph traces and whole-agent trajectory.

Functions

Link copied to clipboard
fun GraphAIAgent.FeatureContext.collectSubgraphTraces(configure: SubgraphTraceCollectionConfig.() -> Unit = {})

Installs the SubgraphTraceCollectionFeature to collect subgraph execution traces and whole-agent trajectory.

Link copied to clipboard
fun GraphAIAgent.FeatureContext.installPromptOptimization(configure: PromptPromptOptimizationFeatureConfig.() -> Unit = {})

Installs the PromptOptimizationFeature with the given configuration.