Fix graph rendering bug due to missing column in updated GraphRAG out…

…put (#16)
noworneverev · Oct 9, 2024 · b35647a · b35647a
1 parent 0d28680
commit b35647a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/app/utils/parquet-utils.ts b/src/app/utils/parquet-utils.ts
@@ -87,9 +87,9 @@ export const readParquetFile = async (file: File, schema?: string): Promise<any[
               id: row[0],
               title: row[1],
               level: parseValue(row[2], 'number'),
-              raw_community: parseValue(row[3], 'number'),
-              relationship_ids: row[4],
-              text_unit_ids: row[5],
+              ...(row.length > 5 ? { raw_community: parseValue(row[3], 'number') } : {}), // From graphrag 0.3.X onwards, raw_community is removed
+              relationship_ids: row[row.length > 5 ? 4 : 3],
+              text_unit_ids: row[row.length > 5 ? 5 : 4],              
             })));
           } else if (schema === 'community_report') {
             resolve(rows.map(row => ({