Skip to content

Commit c7998fe

Browse files
Added check for empty chunk ids
1 parent d9a38fa commit c7998fe

File tree

1 file changed

+17
-8
lines changed

1 file changed

+17
-8
lines changed

backend/src/chunkid_entities.py

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -102,14 +102,23 @@ def get_entities_from_chunkids(uri, username, password, chunk_ids):
102102
"""
103103
try:
104104
logging.info(f"Starting graph query process for chunk ids")
105-
chunk_ids_list = chunk_ids.split(",")
106-
driver = get_graphDB_driver(uri, username, password)
107-
records, summary, keys = driver.execute_query(CHUNK_QUERY, chunksIds=chunk_ids_list)
108-
result = process_records(records)
109-
logging.info(f"Nodes and relationships are processed")
110-
result["chunk_data"] = process_chunk_data(records)
111-
logging.info(f"Query process completed successfully for chunk ids")
112-
return result
105+
if chunk_ids:
106+
chunk_ids_list = chunk_ids.split(",")
107+
driver = get_graphDB_driver(uri, username, password)
108+
records, summary, keys = driver.execute_query(CHUNK_QUERY, chunksIds=chunk_ids_list)
109+
result = process_records(records)
110+
logging.info(f"Nodes and relationships are processed")
111+
result["chunk_data"] = process_chunk_data(records)
112+
logging.info(f"Query process completed successfully for chunk ids")
113+
return result
114+
else:
115+
logging.info(f"chunkid_entities module: No chunk ids are passed")
116+
result = {
117+
"nodes": [],
118+
"relationships": [],
119+
"chunk_data":[]
120+
}
121+
return result
113122

114123
except Exception as e:
115124
logging.error(f"chunkid_entities module: An error occurred in get_entities_from_chunkids. Error: {str(e)}")

0 commit comments

Comments
 (0)