nat.experimental.test_time_compute.selection.llm_based_output_merging_selector#

Attributes#

logger

Classes#

LLMBasedOutputMergingSelector

Abstract base class for strategy implementations.

Functions#

register_llm_based_output_merging_selector(config, builder)

Module Contents#

logger#

class LLMBasedOutputMergingSelector( 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.

llm_bound = None#

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]#

Merge the outputs of multiple planning items into a single output

Args:: original_prompt (str): The prompt the user provided the agent. agent_context (str): The context of the agent, if applicable. items (list[TTCItem]): The list of planning items to select from.
Returns:: TTCItem: The selected planning item.

async register_llm_based_output_merging_selector( config: nat.experimental.test_time_compute.models.selection_config.LLMBasedOutputMergingConfig, builder: nat.builder.builder.Builder, )#: Register the LLMBasedOutputMergingSelector with the builder.