Merge pull request #381 from CybercentreCanada/hotfix/async_task

Hotfix/async task
CybercentreCanada · Sep 28, 2021 · 73581c2 · 73581c2
2 parents aadc3ec + a3918ce
commit 73581c2
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)