nat.experimental.test_time_compute.selection.threshold_selector#

Attributes#

Classes#

ThresholdSelector

Downselects only those TTCItems whose 'score' >= config.threshold.

Functions#

register_threshold_selector(config, builder)

Module Contents#

logger#

class ThresholdSelector( config: nat.data_models.ttc_strategy.TTCStrategyBaseConfig, )#

Bases: nat.experimental.test_time_compute.models.strategy_base.StrategyBase

Downselects only those TTCItems whose ‘score’ >= config.threshold.

async build_components(builder: nat.builder.builder.Builder) → None#: Build the components required for the selector.

supported_pipeline_types() → list[nat.experimental.test_time_compute.models.stage_enums.PipelineTypeEnum]#: Return the stage types supported by this selector.

stage_type() → nat.experimental.test_time_compute.models.stage_enums.StageTypeEnum#: Return the stage type of this strategy.

async ainvoke(

items: list[nat.experimental.test_time_compute.models.ttc_item.TTCItem],

original_prompt: str | None = None,

agent_context: str | None = None,

**kwargs,

) → list[nat.experimental.test_time_compute.models.ttc_item.TTCItem]#

async register_threshold_selector( config: nat.experimental.test_time_compute.models.selection_config.ThresholdSelectionConfig, builder: nat.builder.builder.Builder, )#