inspect-eval-utils 1.2.0__tar.gz → 1.2.1__tar.gz

{inspect_eval_utils-1.2.0 → inspect_eval_utils-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inspect-eval-utils
-Version: 1.2.0
+Version: 1.2.1
 Summary: Shared utilities for METR Inspect AI eval repos: task scaffolder + common runtime helpers.
 Project-URL: Repository, https://github.com/METR/inspect-eval-utils
 Project-URL: Issues, https://github.com/METR/inspect-eval-utils/issues

{inspect_eval_utils-1.2.0 → inspect_eval_utils-1.2.1}/src/inspect_eval_utils/tool_cli/_mechanism.py

@@ -277,15 +277,20 @@ def _required_bool_names(tool):
 
 
 def _call_rpc(method, *args, **kwargs):
+    # The RPC client is keyword-only after `method`; pass args by parameter name.
     try:
         if method == "list_tools":
             return call_{service_name}('list_tools')
         if method == "describe_tool":
-            return call_{service_name}('describe_tool', *args, **kwargs)
+            return call_{service_name}('describe_tool', tool_name=args[0])
         if method == "describe_tool_for_call":
-            return call_{service_name}('describe_tool_for_call', *args, **kwargs)
+            return call_{service_name}('describe_tool_for_call', tool_name=args[0])
         if method == "call_tool":
-            return call_{service_name}('call_tool', *args, **kwargs)
+            if len(args) > 2:
+                return call_{service_name}(
+                    'call_tool', tool_name=args[0], arguments=args[1], snapshot_token=args[2]
+                )
+            return call_{service_name}('call_tool', tool_name=args[0], arguments=args[1])
         return call_{service_name}(method, *args, **kwargs)
     except Exception as exc:
         print(str(exc), file=sys.stderr)