it

core/graph_manager.py

@@ -163,8 +163,11 @@ class GraphManager:
                     # Skip creating correlation node - would be redundant
                     continue
                 
-                # STEP 2: Filter out node pairs that already have direct edges
-                eligible_nodes = self._filter_nodes_without_direct_edges(set(corr['nodes']))
+                # *** CHANGE START ***
+                # The overly aggressive filtering logic has been removed.
+                # All nodes involved in the correlation will now be used.
+                eligible_nodes = set(corr['nodes'])
+                # *** CHANGE END ***
                 
                 if len(eligible_nodes) < 2:
                     # Need at least 2 nodes to create a correlation
@@ -196,27 +199,10 @@ class GraphManager:
         self.last_modified = datetime.now(timezone.utc).isoformat()
         return is_new_node
 
-    def _filter_nodes_without_direct_edges(self, node_set: set) -> set:
-        """
-        Filter out nodes that already have direct edges between them.
-        Returns set of nodes that should be included in correlation.
-        """
-        nodes_list = list(node_set)
-        eligible_nodes = set(node_set)  # Start with all nodes
-        
-        # Check all pairs of nodes
-        for i in range(len(nodes_list)):
-            for j in range(i + 1, len(nodes_list)):
-                node_a = nodes_list[i]
-                node_b = nodes_list[j]
-                
-                # Check if direct edge exists in either direction
-                if self._has_direct_edge_bidirectional(node_a, node_b):
-                    # Remove both nodes from eligible set since they're already connected
-                    eligible_nodes.discard(node_a)
-                    eligible_nodes.discard(node_b)
-        
-        return eligible_nodes
+    # *** CHANGE START ***
+    # The following function is no longer needed and has been removed to avoid confusion.
+    # def _filter_nodes_without_direct_edges(self, node_set: set) -> set:
+    # *** CHANGE END ***
 
     def _has_direct_edge_bidirectional(self, node_a: str, node_b: str) -> bool:
         """