CAMEL - Build Multi-Agent AI Systems

MultiStepEnv

class MultiStepEnv(ABC):

A multi-step environment for reinforcement learning with LLMs.

def __init__(
    self,
    extractor: BaseExtractor,
    max_steps: Optional[int] = None,
    **kwargs
):

Initialize the environment. Parameters:

extractor: Extractor to process LLM responses.
max_steps: Maximum steps per episode. **kwargs: Additional environment parameters.

def _get_initial_state(self):

def _get_next_observation(self):

def _get_terminal_observation(self):

def is_done(self):

Returns: bool: A boolean flag.

def _is_done(self):

def metadata(self):

Returns: Dict[str, Any]: A copy of the environment’s metadata.

def current_step(self):

Returns: int: The number of the step we are currently in.