Add helper function to raise first encountered exception in `TasksSum…

…mary` (#1396)

cognite/client/_api/datapoints.py

@@ -1637,6 +1637,6 @@ def fetch_datapoints(self) -> list[dict[str, Any]]:
             for chunk in split_into_chunks(self._all_identifiers, self.dps_client._RETRIEVE_LATEST_LIMIT)
         ]
         tasks_summary = execute_tasks(self.dps_client._post, tasks, max_workers=self.dps_client._config.max_workers)
-        if tasks_summary.exceptions:
-            raise tasks_summary.exceptions[0]
+        tasks_summary.raise_first_encountered_exception()
+
         return tasks_summary.joined_results(lambda res: res.json()["items"])

cognite/client/_api/raw.py

@@ -614,6 +614,6 @@ def list(
             for cursor in cursors
         ]
         summary = utils._concurrency.execute_tasks(self._list, tasks, max_workers=self._config.max_workers)
-        if summary.exceptions:
-            raise summary.exceptions[0]
+        summary.raise_first_encountered_exception()
+
         return RowList(summary.joined_results())

cognite/client/_api/relationships.py

@@ -311,8 +311,8 @@ def list(
                 tasks,
                 max_workers=self._config.max_workers,
             )
-            if tasks_summary.exceptions:
-                raise tasks_summary.exceptions[0]
+            tasks_summary.raise_first_encountered_exception()
+
             return RelationshipList(tasks_summary.joined_results())
         return self._list(
             list_cls=RelationshipList,

cognite/client/_api/sequences.py

@@ -1058,8 +1058,7 @@ def _fetch_sequence(post_obj: dict[str, Any]) -> SequenceData:
         tasks_summary = utils._concurrency.execute_tasks(
             _fetch_sequence, [(x,) for x in post_objs], max_workers=self._config.max_workers
         )
-        if tasks_summary.exceptions:
-            raise tasks_summary.exceptions[0]
+        tasks_summary.raise_first_encountered_exception()
         results = tasks_summary.joined_results()
         if len(post_objs) == 1:
             return results[0]

cognite/client/_api/synthetic_time_series.py

@@ -92,9 +92,7 @@ def query(
         datapoints_summary = utils._concurrency.execute_tasks(
             self._fetch_datapoints, tasks, max_workers=self._config.max_workers
         )
-
-        if datapoints_summary.exceptions:
-            raise datapoints_summary.exceptions[0]
+        datapoints_summary.raise_first_encountered_exception()
 
         return (
             DatapointsList(datapoints_summary.results, cognite_client=self._cognite_client)

cognite/client/_api_client.py

@@ -506,8 +506,7 @@ def get_partition(partition_num: int) -> list[dict[str, Any]]:
             tasks_summary = utils._concurrency.execute_tasks(
                 get_partition, [(partition,) for partition in next_cursors], max_workers=partitions
             )
-            if tasks_summary.exceptions:
-                raise tasks_summary.exceptions[0]
+            tasks_summary.raise_first_encountered_exception()
 
             for item in tasks_summary.joined_results():
                 yield resource_cls._load(item, cognite_client=self._cognite_client)
@@ -619,8 +618,8 @@ def get_partition(partition: int) -> list[dict[str, Any]]:
 
         tasks = [(f"{i + 1}/{partitions}",) for i in range(partitions)]
         tasks_summary = utils._concurrency.execute_tasks(get_partition, tasks, max_workers=partitions)
-        if tasks_summary.exceptions:
-            raise tasks_summary.exceptions[0]
+        tasks_summary.raise_first_encountered_exception()
+
         return list_cls._load(tasks_summary.joined_results(), cognite_client=self._cognite_client)
 
     def _aggregate(

cognite/client/utils/_concurrency.py

@@ -73,6 +73,10 @@ def raise_compound_exception_if_failed_tasks(
             self.exceptions, successful=successful, failed=failed, unknown=unknown, unwrap_fn=str_format_element_fn
         )
 
+    def raise_first_encountered_exception(self) -> None:
+        if self.exceptions:
+            raise self.exceptions[0]
+
 
 def collect_exc_info_and_raise(
     exceptions: list[Exception],