(env_name, tasks_data, available_tools, env_num=1, seed=1, history_length=2, max_steps=30)
| 22 | |
| 23 | |
| 24 | def build_env(env_name, tasks_data, available_tools, env_num=1, seed=1, history_length=2, max_steps=30): |
| 25 | group_n = 1 |
| 26 | if env_name == "gaia": |
| 27 | # Build GAIA/Tool Use environments |
| 28 | from openmanus_rl.environments.env_package.tool_use.projection import tool_use_projection |
| 29 | from openmanus_rl.environments.env_package.tool_use.envs import build_tool_use_envs |
| 30 | from openmanus_rl.environments.env_package.tool_use.manager import ToolUseEnvironmentManager |
| 31 | |
| 32 | envs = build_tool_use_envs( |
| 33 | tasks_data=tasks_data, |
| 34 | available_tools=available_tools, |
| 35 | seed=seed, |
| 36 | env_num=env_num, |
| 37 | group_n=group_n, |
| 38 | is_train=True |
| 39 | ) |
| 40 | |
| 41 | # Minimal config object with required fields |
| 42 | cfg = SimpleNamespace( |
| 43 | env=SimpleNamespace( |
| 44 | env_name="tool_use", |
| 45 | history_length=history_length, |
| 46 | max_steps=max_steps # Controlled by CLI |
| 47 | ) |
| 48 | ) |
| 49 | env_manager = ToolUseEnvironmentManager(envs, tool_use_projection, cfg) |
| 50 | else: |
| 51 | raise ValueError(f"Unsupported environment name: {env_name}") |
| 52 | |
| 53 | return env_manager |
| 54 | |
| 55 | |
| 56 | class Agent: |
no test coverage detected