up

src/Graph/StellaOps.Graph.Indexer/Documents/GraphSnapshotBuilder.cs

@@ -289,8 +289,31 @@ public sealed class GraphSnapshotBuilder
         out string sourceNodeId,
         out string targetNodeId)
     {
-        var kind = edge["kind"]!.GetValue<string>();
-        var canonicalKey = edge["canonical_key"]!.AsObject();
+        // Handle simple edge format with direct source/target properties
+        if (!edge.TryGetPropertyValue("kind", out var kindNode) || kindNode is null)
+        {
+            if (edge.TryGetPropertyValue("source", out var simpleSource) && simpleSource is not null &&
+                edge.TryGetPropertyValue("target", out var simpleTarget) && simpleTarget is not null)
+            {
+                sourceNodeId = simpleSource.GetValue<string>();
+                targetNodeId = simpleTarget.GetValue<string>();
+                return nodesById.ContainsKey(sourceNodeId) && nodesById.ContainsKey(targetNodeId);
+            }
+
+            sourceNodeId = string.Empty;
+            targetNodeId = string.Empty;
+            return false;
+        }
+
+        var kind = kindNode.GetValue<string>();
+        if (!edge.TryGetPropertyValue("canonical_key", out var canonicalKeyNode) || canonicalKeyNode is null)
+        {
+            sourceNodeId = string.Empty;
+            targetNodeId = string.Empty;
+            return false;
+        }
+
+        var canonicalKey = canonicalKeyNode.AsObject();
 
         string? source = null;
         string? target = null;