From cf6e0070b40875cace5f7c574185cc0165ebb443 Mon Sep 17 00:00:00 2001
From: Shyju Krishnankutty <connectshyju@gmail.com>
Date: Mon, 26 Jan 2026 09:44:08 -0800
Subject: [PATCH] Cleanup

---
 .../WorkflowHelper.cs                         |  2 +-
 .../Microsoft.Agents.AI.Workflows/Workflow.cs | 31 +++++++++++++------
 2 files changed, 22 insertions(+), 11 deletions(-)

diff --git a/dotnet/src/Microsoft.Agents.AI.DurableTask/WorkflowHelper.cs b/dotnet/src/Microsoft.Agents.AI.DurableTask/WorkflowHelper.cs
index fe2da3d28d..1237ad7081 100644
--- a/dotnet/src/Microsoft.Agents.AI.DurableTask/WorkflowHelper.cs
+++ b/dotnet/src/Microsoft.Agents.AI.DurableTask/WorkflowHelper.cs
@@ -65,7 +65,7 @@ public static class WorkflowHelper
 
         Dictionary<string, ExecutorBinding> executors = workflow.ReflectExecutors();
         Dictionary<string, HashSet<EdgeInfo>> edges = workflow.ReflectEdges();
-        Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> edgeConditions = workflow.GetEdgeConditions();
+        Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> edgeConditions = workflow.ReflectEdgeConditions();
 
         WorkflowExecutionPlan plan = new();
 
diff --git a/dotnet/src/Microsoft.Agents.AI.Workflows/Workflow.cs b/dotnet/src/Microsoft.Agents.AI.Workflows/Workflow.cs
index 28edf94943..0c54d3b8c7 100644
--- a/dotnet/src/Microsoft.Agents.AI.Workflows/Workflow.cs
+++ b/dotnet/src/Microsoft.Agents.AI.Workflows/Workflow.cs
@@ -37,20 +37,31 @@ public class Workflow
     }
 
     /// <summary>
-    /// Gets the condition functions for direct edges, keyed by (sourceId, targetId) tuple.
+    /// Gets a mapping of edge identifiers to their associated condition delegates.
     /// </summary>
-    /// <returns>A dictionary mapping edge connections to their condition functions (null if no condition).</returns>
-    /// <remarks>This method creates a new dictionary each time it is called to ensure thread safety.</remarks>
-    [System.Diagnostics.CodeAnalysis.SuppressMessage("Design", "CA1024:Use properties where appropriate", Justification = "Method creates a new collection on each call.")]
-    public Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> GetEdgeConditions()
+    /// <remarks>The returned dictionary includes only edges that have direct edge data. The condition
+    /// delegate can be used to determine whether the edge is active or valid for a given input. Edges without direct
+    /// edge data are not included in the result.</remarks>
+    /// <returns>A dictionary where each key is a tuple containing the source and target node identifiers, and each value is a
+    /// delegate that evaluates the condition for the corresponding edge. The value is <see langword="null"/> if the
+    /// edge has no associated condition.</returns>
+    [System.Diagnostics.CodeAnalysis.SuppressMessage("Design", "CA1024:Use properties where appropriate", Justification = "<Pending>")]
+    public Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> ReflectEdgeConditions()
     {
-        Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> conditions = [];
-
-        foreach (KeyValuePair<string, HashSet<Edge>> edgeGroup in this.Edges)
+        int capacity = 0;
+        foreach (HashSet<Edge> edgeSet in this.Edges.Values)
         {
-            foreach (Edge edge in edgeGroup.Value)
+            capacity += edgeSet.Count;
+        }
+
+        Dictionary<(string SourceId, string TargetId), Func<object?, bool>?> conditions = new(capacity);
+
+        foreach (HashSet<Edge> edgeSet in this.Edges.Values)
+        {
+            foreach (Edge edge in edgeSet)
             {
-                if (edge.DirectEdgeData is DirectEdgeData directEdge)
+                DirectEdgeData? directEdge = edge.DirectEdgeData;
+                if (directEdge is not null)
                 {
                     conditions[(directEdge.SourceId, directEdge.SinkId)] = directEdge.Condition;
                 }