diff --git a/src/strands_tools/batch.py b/src/strands_tools/batch.py index ac2046f3..1c19d258 100644 --- a/src/strands_tools/batch.py +++ b/src/strands_tools/batch.py @@ -109,7 +109,7 @@ def batch(tool: ToolUse, **kwargs) -> ToolResult: result = tool_fn(**arguments) if result["status"] == "success": - results.append({"name": tool_name, "status": "success", "result": result}) + results.append({"json": {"name": tool_name, "status": "success", "result": result}}) else: results.append( {"toolUseId": tool_use_id, "status": "error", "content": [{"text": "Tool missing"}]} diff --git a/tests/test_batch.py b/tests/test_batch.py index 590b75ba..be44c659 100644 --- a/tests/test_batch.py +++ b/tests/test_batch.py @@ -27,12 +27,12 @@ def test_batch_success(mock_agent): assert result["toolUseId"] == "mock_tool_id" assert result["status"] == "success" assert len(result["content"]) == 2 - assert result["content"][0]["name"] == "http_request" - assert result["content"][0]["status"] == "success" - assert result["content"][0]["result"]["result"]["ip"] == "127.0.0.1" - assert result["content"][1]["name"] == "use_aws" - assert result["content"][1]["status"] == "success" - assert result["content"][1]["result"]["result"]["buckets"] == ["bucket1", "bucket2"] + assert result["content"][0]["json"]["name"] == "http_request" + assert result["content"][0]["json"]["status"] == "success" + assert result["content"][0]["json"]["result"]["result"]["ip"] == "127.0.0.1" + assert result["content"][1]["json"]["name"] == "use_aws" + assert result["content"][1]["json"]["status"] == "success" + assert result["content"][1]["json"]["result"]["result"]["buckets"] == ["bucket1", "bucket2"] def test_batch_missing_tool(mock_agent):