agent_studio.agent.base_agent

Contents

`agent_studio.agent.base_agent`#

Module Contents#

Classes#

Base class for agents.

Attributes#

`config`
`logger`

agent_studio.agent.base_agent.config[source]#

agent_studio.agent.base_agent.logger[source]#

class agent_studio.agent.base_agent.BaseAgent(model: agent_studio.llm.base_model.BaseModel)[source]#

Base class for agents.

name: str = 'base'[source]#

reset(instruction: str) → None[source]#

get_token_count() → int[source]#

generate_action(obs: numpy.ndarray | None) → tuple[str, str][source]#

step_action(confirmed: bool, **kwargs) → tuple[dict, bool][source]#: Executes the code and record the result.

abstract eval(final_obs: numpy.ndarray | None = None) → dict[str, Any][source]#

close() → None[source]#

abstract trajectory2intermediate_msg() → list[dict[str, Any]][source]#