From a3918ce4ad5b2ba48099bd5a021ec5f5bf1c240a Mon Sep 17 00:00:00 2001 From: Steve Garon Date: Tue, 28 Sep 2021 14:46:12 +0000 Subject: [PATCH] Fix get task results missing response --- assemblyline/datastore/stores/es_store.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/assemblyline/datastore/stores/es_store.py b/assemblyline/datastore/stores/es_store.py index a6aaa806a..d0dd09e65 100644 --- a/assemblyline/datastore/stores/es_store.py +++ b/assemblyline/datastore/stores/es_store.py @@ -333,7 +333,10 @@ def _get_task_results(self, task): else: raise - return res['response'] + try: + return res['response'] + except KeyError: + return res['task']['status'] def _safe_index_copy(self, copy_function, src, target, body=None, min_status='yellow'): ret = copy_function(src, target, body=body, request_timeout=60)