update error messages and fix local search bug

backend/src/api/common.py

@@ -118,7 +118,7 @@ def sanitize_name(name: str | None) -> str | None:
     Sanitize a human-readable name by converting it to a SHA256 hash, then truncate
     to 128 bit length to ensure it is within the 63 character limit imposed by Azure Storage.
 
-    The sanitized name will be used to identify container names for Azure Storage and CosmosDB.
+    The sanitized name will be used to identify container names in both Azure Storage and CosmosDB.
 
     Args:
     -----
@@ -138,7 +138,7 @@ def sanitize_name(name: str | None) -> str | None:
 
 def retrieve_original_blob_container_name(sanitized_name: str) -> str | None:
     """
-    Retrieve the original human-readable container name of a sanitized name.
+    Retrieve the original human-readable container name of a sanitized blob name.
 
     Args:
     -----
@@ -166,7 +166,7 @@ def retrieve_original_blob_container_name(sanitized_name: str) -> str | None:
 
 def retrieve_original_entity_config_name(sanitized_name: str) -> str | None:
     """
-    Retrieve the original human-readable entity config name of a sanitized name.
+    Retrieve the original human-readable entity config name of a sanitized entity config name.
 
     Args:
     -----

backend/src/api/data.py

@@ -144,7 +144,7 @@ async def upload_files(
     except ValueError:
         raise HTTPException(
             status_code=500,
-            detail=f"Invalid container name: '{storage_name}'. Please try a different name.",
+            detail=f"Invalid blob container name: '{storage_name}'. Please try a different name.",
         )
     try:
         blob_service_client = BlobServiceClientSingletonAsync.get_instance()

backend/src/api/query.py

@@ -235,11 +235,11 @@ async def local_query(request: GraphRequest):
         text_unit_df.document_ids = text_unit_df.document_ids.apply(
             lambda x: [i + add_on for i in x]
         )
-        text_unit_df.entity_ids = text_unit_df.entity_ids.apply(
-            lambda x: [i + add_on for i in x]
+        text_unit_df.entity_ids = text_unit_df.entity_ids.map(
+            lambda x: [i + add_on for i in x], na_action="ignore"
         )
-        text_unit_df.relationship_ids = text_unit_df.relationship_ids.apply(
-            lambda x: [i + add_on for i in x]
+        text_unit_df.relationship_ids = text_unit_df.relationship_ids.map(
+            lambda x: [i + add_on for i in x], na_action="ignore"
         )
         text_unit_dfs.append(text_unit_df)