fix: add error handling for invalid total items in collection retrieval

ist356 · Nov 20, 2024 · 3c77c30 · 3c77c30
1 parent 7a1f599
commit 3c77c30
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/ia_collection_analyzer/iahelper.py b/src/ia_collection_analyzer/iahelper.py
@@ -67,6 +67,12 @@ def get_collection(collection_id, progress_hook=None) -> list:
         )
         collection = []
         total_items = search.num_found
+        try:
+            total_items = int(search.num_found)
+        except ValueError:
+            total_items = 0
+            logger.error(f"Failed to get total items for {collection_id}: search.num_found={search.num_found}")
+            return []
         if progress_hook:
             progress_hook(0, total_items)
         for result in tqdm(