include request body into VespaResult

1 files changed, 9 insertions, 4 deletions

diff --git a/python/vespa/vespa/query.py b/python/vespa/vespa/query.py
index f78c8a23380..39d8d6d5c2b 100644
--- a/python/vespa/vespa/query.py
+++ b/python/vespa/vespa/query.py
@@ -204,13 +204,18 @@ class Query(object):
 
 
 class VespaResult(object):
-    def __init__(self, vespa_result):
-        self.vespa_result = vespa_result
+    def __init__(self, vespa_result, request_body=None):
+        self._vespa_result = vespa_result
+        self._request_body = request_body
+
+    @property
+    def request_body(self) -> Optional[Dict]:
+        return self._request_body
 
     @property
     def json(self) -> Dict:
-        return self.vespa_result
+        return self._vespa_result
 
     @property
     def hits(self) -> List:
-        return self.vespa_result.get("root", {}).get("children", [])
+        return self._vespa_result.get("root", {}).get("children", [])