agents.reasoning_and_critique.tot.config

Configuration for the Tree of Thoughts agent.

This module defines the configuration schema for the ToT agent, including engine configurations, algorithm parameters, and state schema.

Classes

TOTAgentConfig

Configuration for the Tree of Thoughts agent.

Module Contents

class agents.reasoning_and_critique.tot.config.TOTAgentConfig

Bases: haive.core.engine.agent.agent.AgentConfig

Configuration for the Tree of Thoughts agent.

This configuration specifies the LLM engines used for generation and scoring, as well as the parameters for the search algorithm.

classmethod create_for_problem_type(content_type='string', **kwargs)

Create a TOT agent configuration specialized for a specific problem type.

Parameters:

  • content_type (str) – Type of content (‘string’, ‘equation’, etc.)

  • **kwargs – Additional configuration parameters

Returns:

Configured TOTAgentConfig

Return type:

TOTAgentConfig

get_engine(engine_key)

Get an engine by key from the engines dictionary.

Parameters:

engine_key (str) – Key of the engine to retrieve

Returns:

The requested engine configuration

Raises:

KeyError – If the engine key is not found

Return type:

haive.core.engine.aug_llm.AugLLMConfig