Merge pull request #54 from Fr4nc3/psl-postgresbugfix

fix: delete page issue on cosmos db selection
Fr4nc3 · Dec 6, 2024 · da13aa7 · da13aa7
2 parents 396b68b + d4fa94c
commit da13aa7
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/code/backend/pages/03_Delete_Data.py b/code/backend/pages/03_Delete_Data.py
@@ -47,10 +47,9 @@ def load_css(file_path):
     search_handler = Search.get_search_handler(env_helper)
     results = search_handler.get_files()
     if (
-        results is None
-        or (hasattr(results, "get_count") and results.get_count() == 0)
-        or len(results) == 0
-    ):
+        env_helper.DATABASE_TYPE == "CosmosDB"
+        and (results is None or results.get_count() == 0)
+    ) or (env_helper.DATABASE_TYPE == "PostgreSQL" and len(results) == 0):
         st.info("No files to delete")
         st.stop()
     else: