nat.experimental.test_time_compute.selection.best_of_n_selector#

Attributes#

logger

Classes#

BestOfNSelector

Abstract base class for strategy implementations.

Functions#

register_best_of_n_selector(config, builder)

Register the BestOfNSelector strategy.

Module Contents#

logger#

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

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

Abstract base class for strategy implementations.

This class defines the interface for strategies that can be used in the TTC framework. Concrete strategy classes should implement the methods defined in this class.

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

supported_pipeline_types() → [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) → [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.