nat.experimental.test_time_compute.selection.threshold_selector

Attributes

logger

Classes

ThresholdSelector

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

Functions

register_threshold_selector(config, builder)

Module Contents

logger

class ThresholdSelector

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

supported_pipeline_types() → list[nat.experimental.test_time_compute.models.stage_enums.PipelineTypeEnum]

stage_type() → nat.experimental.test_time_compute.models.stage_enums.StageTypeEnum

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,

)