Add community and covariate records for pipeline parity

Author: KSemenenko

src/ManagedCode.GraphRag/Community/CommunityRecord.cs

@@ -0,0 +1,20 @@
+using System.Collections.Immutable;
+
+namespace GraphRag.Community;
+
+/// <summary>
+/// Represents a finalized community row emitted by the GraphRAG pipeline.
+/// </summary>
+public sealed record CommunityRecord(
+    string Id,
+    int HumanReadableId,
+    int CommunityId,
+    int Level,
+    int? ParentId,
+    ImmutableArray<int> Children,
+    string Title,
+    ImmutableArray<string> EntityIds,
+    ImmutableArray<string> RelationshipIds,
+    ImmutableArray<string> TextUnitIds,
+    string? Period,
+    int Size);

src/ManagedCode.GraphRag/Constants/PipelineTableNames.cs

@@ -6,5 +6,7 @@ public static class PipelineTableNames
     public const string TextUnits = "text_units";
     public const string Entities = "entities";
     public const string Relationships = "relationships";
+    public const string Communities = "communities";
     public const string CommunityReports = "community_reports";
+    public const string Covariates = "covariates";
 }

src/ManagedCode.GraphRag/Covariates/CovariateRecord.cs

@@ -0,0 +1,18 @@
+namespace GraphRag.Covariates;
+
+/// <summary>
+/// Represents a finalized covariate (claim) row emitted by the GraphRAG pipeline.
+/// </summary>
+public sealed record CovariateRecord(
+    string Id,
+    int HumanReadableId,
+    string CovariateType,
+    string? Type,
+    string? Description,
+    string SubjectId,
+    string? ObjectId,
+    string? Status,
+    string? StartDate,
+    string? EndDate,
+    string? SourceText,
+    string TextUnitId);

tests/ManagedCode.GraphRag.Tests/Community/CommunityRecordTests.cs

@@ -0,0 +1,40 @@
+using System.Collections.Immutable;
+using System.Linq;
+using GraphRag.Community;
+using Xunit;
+
+namespace ManagedCode.GraphRag.Tests.Community;
+
+public sealed class CommunityRecordTests
+{
+    [Fact]
+    public void Constructor_PopulatesAllProperties()
+    {
+        var record = new CommunityRecord(
+            Id: "community-uuid",
+            HumanReadableId: 12,
+            CommunityId: 42,
+            Level: 3,
+            ParentId: 7,
+            Children: ImmutableArray.Create(8, 9),
+            Title: "Energy Sector",
+            EntityIds: ImmutableArray.Create("entity-1", "entity-2"),
+            RelationshipIds: ImmutableArray.Create("rel-1"),
+            TextUnitIds: ImmutableArray.Create("unit-1", "unit-2"),
+            Period: "2024-05-01",
+            Size: 18);
+
+        Assert.Equal("community-uuid", record.Id);
+        Assert.Equal(12, record.HumanReadableId);
+        Assert.Equal(42, record.CommunityId);
+        Assert.Equal(3, record.Level);
+        Assert.Equal(7, record.ParentId);
+        Assert.Equal(new[] { 8, 9 }, record.Children.ToArray());
+        Assert.Equal("Energy Sector", record.Title);
+        Assert.Equal(new[] { "entity-1", "entity-2" }, record.EntityIds.ToArray());
+        Assert.Equal(new[] { "rel-1" }, record.RelationshipIds.ToArray());
+        Assert.Equal(new[] { "unit-1", "unit-2" }, record.TextUnitIds.ToArray());
+        Assert.Equal("2024-05-01", record.Period);
+        Assert.Equal(18, record.Size);
+    }
+}

tests/ManagedCode.GraphRag.Tests/Covariates/CovariateRecordTests.cs

@@ -0,0 +1,38 @@
+using GraphRag.Covariates;
+using Xunit;
+
+namespace ManagedCode.GraphRag.Tests.Covariates;
+
+public sealed class CovariateRecordTests
+{
+    [Fact]
+    public void Constructor_PopulatesAllProperties()
+    {
+        var record = new CovariateRecord(
+            Id: "cov-1",
+            HumanReadableId: 5,
+            CovariateType: "claim",
+            Type: "fraud",
+            Description: "Alleged false reporting",
+            SubjectId: "entity-1",
+            ObjectId: "entity-2",
+            Status: "SUSPECTED",
+            StartDate: "2024-03-01",
+            EndDate: "2024-04-01",
+            SourceText: "Entity-1 may have misled Entity-2",
+            TextUnitId: "unit-1");
+
+        Assert.Equal("cov-1", record.Id);
+        Assert.Equal(5, record.HumanReadableId);
+        Assert.Equal("claim", record.CovariateType);
+        Assert.Equal("fraud", record.Type);
+        Assert.Equal("Alleged false reporting", record.Description);
+        Assert.Equal("entity-1", record.SubjectId);
+        Assert.Equal("entity-2", record.ObjectId);
+        Assert.Equal("SUSPECTED", record.Status);
+        Assert.Equal("2024-03-01", record.StartDate);
+        Assert.Equal("2024-04-01", record.EndDate);
+        Assert.Equal("Entity-1 may have misled Entity-2", record.SourceText);
+        Assert.Equal("unit-1", record.TextUnitId);
+    }
+}