nat.experimental.test_time_compute.search.single_shot_multi_plan_planner

Attributes

logger

Classes

SingleShotMultiPlanPlanner

Implementation of the Single Shot Multi Plan Planner.

Functions

register_single_shot_multi_plan_planner(config, builder)

Register the SingleShotMultiPlanPlanner strategy with the provided configuration.

Module Contents

logger

class SingleShotMultiPlanPlanner(

config: nat.data_models.ttc_strategy.TTCStrategyBaseConfig,

)

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

Implementation of the Single Shot Multi Plan Planner. This planner generates multiple plans in a single shot.

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

Generate a TTCItem based on the provided prompt.

async register_single_shot_multi_plan_planner(

config: nat.experimental.test_time_compute.models.search_config.SingleShotMultiPlanConfig,

builder: nat.builder.builder.Builder,

)

Register the SingleShotMultiPlanPlanner strategy with the provided configuration.