Fix get task results missing response

CybercentreCanada · Sep 28, 2021 · a3918ce · a3918ce
1 parent 34cddbd
commit a3918ce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)