Skip to main content

research_town.envs package

Submodules

research_town.envs.env_base module

class research_town.envs.env_base.BaseEnv(name: str, config: Config)

Bases: ABC

abstract on_enter(**context: Any) → None

abstract on_exit() → Tuple[str, Dict[str, Any]]

abstract run() → Generator[Tuple[Progress, Agent], None, None]

research_town.envs.env_end module

class research_town.envs.env_end.EndEnv(name: str, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

research_town.envs.env_proposal_writing module

class research_town.envs.env_proposal_writing.ProposalWritingwithRAGEnv(name: str, log_db: LogDB, progress_db: ProgressDB, paper_db: PaperDB, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

research_town.envs.env_review_writing module

class research_town.envs.env_review_writing.ReviewWritingEnv(name: str, log_db: LogDB, progress_db: ProgressDB, paper_db: PaperDB, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

research_town.envs.env_start module

class research_town.envs.env_start.StartEnv(name: str, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

Module contents

class research_town.envs.BaseEnv(name: str, config: Config)

Bases: ABC

abstract on_enter(**context: Any) → None

abstract on_exit() → Tuple[str, Dict[str, Any]]

abstract run() → Generator[Tuple[Progress, Agent], None, None]

class research_town.envs.EndEnv(name: str, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

class research_town.envs.ProposalWritingwithRAGEnv(name: str, log_db: LogDB, progress_db: ProgressDB, paper_db: PaperDB, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

class research_town.envs.ReviewWritingEnv(name: str, log_db: LogDB, progress_db: ProgressDB, paper_db: PaperDB, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]

class research_town.envs.StartEnv(name: str, config: Config, agent_manager: AgentManager)

Bases: BaseEnv

on_enter(**context: Any) → None

on_exit() → tuple[str, dict[str, Any]]

run() → Generator[tuple[Progress, Agent], None, None]