correct error capture output

arcalot · Dec 20, 2023 · 88a84be · 88a84be
1 parent 2f683c3
commit 88a84be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arcaflow_plugin_opensearch/opensearch_plugin.py b/arcaflow_plugin_opensearch/opensearch_plugin.py
@@ -76,7 +76,10 @@ def process_bulk_list_generator():
         )
 
         if resp["errors"]:
-            raise Exception(f"Document status: {resp['_shards']}")
+            shards = {}
+            for i in resp["items"]:
+                shards[list(i.values())[0]["_id"]] = list(i.values())[0]["_shards"]
+            raise Exception(f"Document status: {str(shards)}")
 
         # # # # TODO -- Adapting the below from
         # # # # https://github.com/cloud-bulldozer/benchmark-wrapper/blob/