diff --git a/pyinfra/visitor/response_formatter/formatters/default.py b/pyinfra/visitor/response_formatter/formatters/default.py index 43bf489..c2c2f0e 100644 --- a/pyinfra/visitor/response_formatter/formatters/default.py +++ b/pyinfra/visitor/response_formatter/formatters/default.py @@ -1,4 +1,4 @@ -from funcy import first +from funcy import first, omit from pyinfra.visitor.response_formatter.formatter import ResponseFormatter @@ -10,4 +10,4 @@ class DefaultResponseFormatter(ResponseFormatter): """ def format(self, message): - return {**message, "responseFile": first(message["response_files"])} + return {**omit(message, ["response_files"]), "responseFile": first(message["response_files"])}