inspect-swe 0.2.6__py3-none-any.whl → 0.2.7__py3-none-any.whl

inspect_swe/_claude_code/claude_code.py

@@ -29,6 +29,8 @@ def claude_code(
     """),
     system_prompt: str | None = None,
     mcp_servers: Sequence[MCPServerConfig] | None = None,
+    allowed_tools: list[str] | None = None,
+    disallowed_tools: list[str] | None = None,
     attempts: int | AgentAttempts = 1,
     model: str | None = None,
     small_model: str | None = None,
@@ -43,6 +45,8 @@ def claude_code(
 
     The agent can either use a version of Claude Code installed in the sandbox, or can download a version and install it in the sandbox (see docs on `version` option below for details).
 
+    Use `allowed_tools` and `disallowed_tools` to control access to tools. See [Tools available to Claude](https://docs.anthropic.com/en/docs/claude-code/settings#tools-available-to-claude) for the list of built-in tools and [How to use Allowed Tools in Claude Code](https://www.instructa.ai/blog/claude-code/how-to-use-allowed-tools-in-claude-code) for details on the supported syntax. Note that `allowed_tools` enables you to filter allowed parameter values and `disallowed_tools` enables you to remove tools entirely. In other words, `allowed_tools` is not a complete list of what tools are available but rather just filters on tool parameters---to remove tools you need to explicitly set `disallowed_tools`.
+
     Use the `attempts` option to enable additional submissions if the initial
     submission(s) are incorrect (by default, no additional attempts are permitted).
 
@@ -51,6 +55,8 @@ def claude_code(
         description: Agent description (used in multi-agent systems with `as_tool()` and `handoff()`)
         system_prompt: Additional system prompt to append to default system prompt.
         mcp_servers: MCP servers to make available to the agent.
+        allowed_tools: Parameter filters for built-in tools.
+        disallowed_tools: List of tool names to disallow entirely.
         attempts: Configure agent to make multiple attempts.
         model: Model name to use for Opus and Sonnet calls (defaults to main model for task).
         small_model: Model to use for Haiku calls (defaults to main model for task).
@@ -99,8 +105,19 @@ def claude_code(
                 cmd.extend(["--append-system-prompt", "\n\n".join(system_messages)])
 
             # mcp servers
+            cmd_allowed_tools = allowed_tools or []
             if mcp_servers:
-                cmd.extend(mcp_server_args(mcp_servers))
+                mcp_server_args, mcp_allowed_tools = resolve_mcp_servers(mcp_servers)
+                cmd.extend(mcp_server_args)
+                cmd_allowed_tools.extend(mcp_allowed_tools)
+
+            # add allowed and disallowed tools
+            if len(cmd_allowed_tools) > 0:
+                cmd.append("--allowed-tools")
+                cmd.append(",".join(cmd_allowed_tools))
+            if disallowed_tools is not None and len(disallowed_tools) > 0:
+                cmd.append("--disallowed-tools")
+                cmd.append(",".join(disallowed_tools))
 
             # user prompt
             prompt = "\n\n".join(
@@ -174,7 +191,9 @@ def claude_code(
     return agent_with(execute, name=name, description=description)
 
 
-def mcp_server_args(mcp_servers: Sequence[MCPServerConfig]) -> list[str]:
+def resolve_mcp_servers(
+    mcp_servers: Sequence[MCPServerConfig],
+) -> tuple[list[str], list[str]]:
     # build servers and allowed tools
     mcp_servers_json: dict[str, dict[str, Any]] = {}
     allowed_tools: list[str] = []
@@ -194,14 +213,11 @@ def mcp_server_args(mcp_servers: Sequence[MCPServerConfig]) -> list[str]:
             )
 
     # map to cli args
-    cmds: list[str] = []
+    mcp_config_cmds: list[str] = []
     if len(mcp_servers_json) > 0:
-        cmds.append("--mcp-config")
-        cmds.append(
+        mcp_config_cmds.append("--mcp-config")
+        mcp_config_cmds.append(
             to_json({"mcpServers": mcp_servers_json}, exclude_none=True).decode()
         )
-    if len(allowed_tools):
-        cmds.append("--allowed-tools")
-        cmds.append(",".join(allowed_tools))
 
-    return cmds
+    return mcp_config_cmds, allowed_tools

inspect_swe/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.6'
-__version_tuple__ = version_tuple = (0, 2, 6)
+__version__ = version = '0.2.7'
+__version_tuple__ = version_tuple = (0, 2, 7)
 
 __commit_id__ = commit_id = None

{inspect_swe-0.2.6.dist-info → inspect_swe-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inspect_swe
-Version: 0.2.6
+Version: 0.2.7
 Summary: Software engineering agents for Inspect AI.
 Project-URL: Documentation, https://meridianlabs-ai.github.io/inspect_swe/
 Project-URL: Source Code, https://github.com/meridianlabs-ai/inspect_swe

{inspect_swe-0.2.6.dist-info → inspect_swe-0.2.7.dist-info}/RECORD

@@ -1,9 +1,9 @@
 inspect_swe/__init__.py,sha256=yJ9tBcF2Wy11mVmLh1fTYXgYcsSHv30GAW-tVwE-r3s,342
 inspect_swe/_registry.py,sha256=jM37ysrY39Ufd67GRKbiwfSViOLlm-82lm_JEaWKshw,97
-inspect_swe/_version.py,sha256=2Q6v117QPuRsVsIEaHT3nJJVx7xxa47FYOkmuhVbGAI,704
+inspect_swe/_version.py,sha256=yXzK2akXKIKUAfJk0WCQothqygqvndys6GBuXxo-wk0,704
 inspect_swe/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 inspect_swe/_claude_code/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-inspect_swe/_claude_code/claude_code.py,sha256=bRYPsSYK3G4J5mPbgcpZSSKqWtUnhAy4_3acWDyaORs,8359
+inspect_swe/_claude_code/claude_code.py,sha256=V1C79iWhVawTYy-JeRsUYM6VdiGWO_bjH_qRWf3r5lM,9825
 inspect_swe/_claude_code/install/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 inspect_swe/_claude_code/install/cache.py,sha256=k08bCxGq-iYVpO16LNQhPjxTM9p2iecpqMjqYd2WBss,1708
 inspect_swe/_claude_code/install/download.py,sha256=s1y4CDHVbJenfsR7OUwwxr5QFp-rDi4XnIxumDEvmws,3217
@@ -20,8 +20,8 @@ inspect_swe/_util/download.py,sha256=cCUau4ZBOKezpotJV5-v3JY_5CuYDZ-VcWlLf_EyNL0
 inspect_swe/_util/platform.py,sha256=wm4efIFfdyTeaV2oxOXVvYl1u22MHX3jQMERHJMgv7A,339
 inspect_swe/_util/sandbox.py,sha256=2wYmVz5EGUDBhqbN3NgLAOsyKeU-KRI161MZMJ54n4M,1769
 inspect_swe/_util/trace.py,sha256=mFHmBKn2F8iJP9PpTHaCseMHnTMz3ErRx6RCKV83rZk,139
-inspect_swe-0.2.6.dist-info/METADATA,sha256=SJwbJADMfKoOBkiQ0_RST4iR1hjoHNM0nrn-kN_kU_o,1724
-inspect_swe-0.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-inspect_swe-0.2.6.dist-info/entry_points.txt,sha256=OzpvUhd7M3T2Rog4MjwJAxIKeX5ljiR0mVYM9GefBKg,49
-inspect_swe-0.2.6.dist-info/licenses/LICENSE,sha256=Hi3UDcbD6yCKZ1mcgt7pprzSG0rDEnSrbrm3XinyiDA,1070
-inspect_swe-0.2.6.dist-info/RECORD,,
+inspect_swe-0.2.7.dist-info/METADATA,sha256=keVzMz6nRC72XluU9hqrPlgixueQtCj5bsNm98A6L0A,1724
+inspect_swe-0.2.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+inspect_swe-0.2.7.dist-info/entry_points.txt,sha256=OzpvUhd7M3T2Rog4MjwJAxIKeX5ljiR0mVYM9GefBKg,49
+inspect_swe-0.2.7.dist-info/licenses/LICENSE,sha256=Hi3UDcbD6yCKZ1mcgt7pprzSG0rDEnSrbrm3XinyiDA,1070
+inspect_swe-0.2.7.dist-info/RECORD,,