From 1661672569a221c1b38ce910076d875ecb078128 Mon Sep 17 00:00:00 2001 From: Alonso Guevara Date: Wed, 6 Nov 2024 17:22:11 -0600 Subject: [PATCH] Fix optional covariates check in incremental indexing (#1374) * Fix optional covariates check in incremental indexing * Oopsie fix --- .../next-release/patch-20241106225803494336.json | 4 ++++ graphrag/index/update/incremental_index.py | 8 ++++++-- 2 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 .semversioner/next-release/patch-20241106225803494336.json diff --git a/.semversioner/next-release/patch-20241106225803494336.json b/.semversioner/next-release/patch-20241106225803494336.json new file mode 100644 index 00000000..e61c2630 --- /dev/null +++ b/.semversioner/next-release/patch-20241106225803494336.json @@ -0,0 +1,4 @@ +{ + "type": "patch", + "description": "Fix optional covariates update in incremental indexing" +} diff --git a/graphrag/index/update/incremental_index.py b/graphrag/index/update/incremental_index.py index e947fb92..57d3cd2b 100644 --- a/graphrag/index/update/incremental_index.py +++ b/graphrag/index/update/incremental_index.py @@ -120,8 +120,12 @@ async def update_dataframe_outputs( ) # Merge final covariates - progress_reporter.info("Updating Final Covariates") - await _update_covariates(dataframe_dict, storage, update_storage) + if ( + await storage.has("create_final_covariates.parquet") + and "create_final_covariates" in dataframe_dict + ): + progress_reporter.info("Updating Final Covariates") + await _update_covariates(dataframe_dict, storage, update_storage) # Merge final nodes and update community ids progress_reporter.info("Updating Final Nodes")