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)