Support for additional mata data in api call response

This commit is contained in:
allegroai 2022-02-13 19:30:36 +02:00
parent 604a38035b
commit e149af58b1

View File

@ -584,6 +584,10 @@ class APICall(DataContainer):
def json_flags(self): def json_flags(self):
return self._json_flags return self._json_flags
@property
def extra_meta_fields(self):
return {}
def mark_end(self): def mark_end(self):
self._end_ts = time.time() self._end_ts = time.time()
self._duration = int((self._end_ts - self._start_ts) * 1000) self._duration = int((self._end_ts - self._start_ts) * 1000)
@ -634,6 +638,7 @@ class APICall(DataContainer):
"result_msg": self.result.msg, "result_msg": self.result.msg,
"error_stack": self.result.traceback if include_stack else None, "error_stack": self.result.traceback if include_stack else None,
"error_data": self.result.error_data, "error_data": self.result.error_data,
**self.extra_meta_fields,
}, },
"data": self.result.data, "data": self.result.data,
} }