nat.experimental.test_time_compute.selection.best_of_n_selector#

Attributes#

Classes#

BestOfNSelector

Functions#

register_best_of_n_selector(config, builder)

Register the BestOfNSelector strategy.

Module Contents#

logger#

class BestOfNSelector#

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

async build_components(builder: nat.builder.builder.Builder) → None#

supported_pipeline_types() → [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) → [nat.experimental.test_time_compute.models.ttc_item.TTCItem]#

async register_best_of_n_selector( config: nat.experimental.test_time_compute.models.selection_config.BestOfNSelectionConfig, builder: nat.builder.builder.Builder, )#: Register the BestOfNSelector strategy.