fix: community level 2 for better search performance

aymenfurter · Aug 3, 2024 · d45943d · d45943d
1 parent ad899e6
commit d45943d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/query/graphrag_query.py b/app/query/graphrag_query.py
@@ -37,7 +37,7 @@ def read_parquet_from_blob(blob_path):
     async def global_query(self, query: str):
         ENTITY_TABLE = "output/create_final_nodes.parquet"
         COMMUNITY_REPORT_TABLE = "output/create_final_community_reports.parquet"
-        COMMUNITY_LEVEL = 1
+        COMMUNITY_LEVEL = 2
 
         entity_table_path = f"abfs://{self.config.prefix}-{self.config.index_name}-grdata/{ENTITY_TABLE}"
         community_report_table_path = f"abfs://{self.config.prefix}-{self.config.index_name}-grdata/{COMMUNITY_REPORT_TABLE}"
@@ -124,4 +124,4 @@ async def global_query(self, query: str):
             else:
                 serializable_context_data[key] = value
 
-        return result.response, serializable_context_data
+        return result.response, serializable_context_data