Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions verifiers/envs/experimental/mcp_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,13 +148,13 @@ def __init__(
self,
# MCPEnv is designed for global server processes, not per-rollout,
# stateful server instances with mutable task-specific data.
mcp_servers: List[MCPServerConfig | dict] = [],
mcp_servers: List[MCPServerConfig | dict] | None = None,
max_turns: int = 10,
error_formatter: Callable[[Exception], str] = lambda e: f"Error: {str(e)}",
**kwargs,
):
self.mcp_servers: List[MCPServerConfig] = []
if mcp_servers:
if mcp_servers is not None:
for server in mcp_servers:
if isinstance(server, MCPServerConfig):
self.mcp_servers.append(server)
Expand Down
4 changes: 3 additions & 1 deletion verifiers/envs/stateful_tool_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def __init__(
self.skipped_args: dict[str, list[str]] = {}
self.max_turns: int = max_turns

def add_tool(self, tool: Callable, args_to_skip: list[str] = []):
def add_tool(self, tool: Callable, args_to_skip: list[str] | None = None):
"""Add a tool, optionally hiding arguments from the agent's view.

Skipped args are removed from the schema shown to the agent but can be
Expand All @@ -75,6 +75,8 @@ def add_tool(self, tool: Callable, args_to_skip: list[str] = []):

Assumes all non-skipped args use standard JSON types (no remaining $ref/$defs).
"""
if args_to_skip is None:
args_to_skip = []
self.tools.append(tool)
tool_def = convert_func_to_tool_def(filter_signature(tool, args_to_skip))
params = tool_def.parameters
Expand Down