nat.experimental.test_time_compute.models.strategy_base#

Classes#

StrategyBase

Abstract base class for strategy implementations.

Module Contents#

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

Bases: abc.ABC

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.

config: nat.data_models.ttc_strategy.TTCStrategyBaseConfig#

pipeline_type: nat.experimental.test_time_compute.models.stage_enums.PipelineTypeEnum | None = None#

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

Async:

Build the components required for the selector.

abstractmethod 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:

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

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

set_pipeline_type( pipeline_type: nat.experimental.test_time_compute.models.stage_enums.PipelineTypeEnum, ) → None#: Set the pipeline type for this strategy.