diff --git a/src/google/adk/code_executors/agent_engine_sandbox_code_executor.py b/src/google/adk/code_executors/agent_engine_sandbox_code_executor.py index f601d045..69d1778a 100644 --- a/src/google/adk/code_executors/agent_engine_sandbox_code_executor.py +++ b/src/google/adk/code_executors/agent_engine_sandbox_code_executor.py @@ -134,8 +134,8 @@ class AgentEngineSandboxCodeExecutor(BaseCodeExecutor): or 'file_name' not in output.metadata.attributes ): json_output_data = json.loads(output.data.decode('utf-8')) - stdout = json_output_data.get('stdout', '') - stderr = json_output_data.get('stderr', '') + stdout = json_output_data.get('msg_out', '') + stderr = json_output_data.get('msg_err', '') else: file_name = '' if ( diff --git a/tests/unittests/code_executors/test_agent_engine_sandbox_code_executor.py b/tests/unittests/code_executors/test_agent_engine_sandbox_code_executor.py index c9480601..6022527f 100644 --- a/tests/unittests/code_executors/test_agent_engine_sandbox_code_executor.py +++ b/tests/unittests/code_executors/test_agent_engine_sandbox_code_executor.py @@ -71,7 +71,7 @@ class TestAgentEngineSandboxCodeExecutor: mock_json_output = MagicMock() mock_json_output.mime_type = "application/json" mock_json_output.data = json.dumps( - {"stdout": "hello world", "stderr": ""} + {"msg_out": "hello world", "msg_err": ""} ).encode("utf-8") mock_json_output.metadata = None