natural-agi-common 0.1.31__tar.gz → 0.1.32__tar.gz

{natural_agi_common-0.1.31 → natural_agi_common-0.1.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: natural-agi-common
-Version: 0.1.31
+Version: 0.1.32
 Requires-Dist: pydantic
 Requires-Dist: networkx
 Requires-Dist: neo4j

{natural_agi_common-0.1.31 → natural_agi_common-0.1.32}/common/traversal/graph_traversal.py

@@ -1,6 +1,6 @@
 from typing import Generator, Any, Tuple, Optional
 import networkx as nx
-
+import logging
 from ..model.point import Point
 from ..model.vector import Vector
 
@@ -8,18 +8,17 @@ from ..model.vector import Vector
 class GraphTraversal:
     def __init__(self, graph: nx.Graph):
         self.graph = graph
+        self.logger = logging.getLogger(__name__)
 
     def dfs_traversal(
         self, start_node: Any
     ) -> Generator[Tuple[Point, Optional[Vector]], None, None]:
-        visited = set()
+        visited_vector = set()
         for edge in nx.dfs_edges(self.graph, start_node):
             source_node, target_node = edge
-            if target_node in visited:
+            if source_node in visited_vector or target_node in visited_vector:
                 continue
 
-            visited.add(target_node)
-
             # Convert to your Point and Vector objects
             source_data = self.graph.nodes[source_node]
             target_data = self.graph.nodes[target_node]
@@ -31,11 +30,13 @@ class GraphTraversal:
                 point = Point.from_node_data(source_data)
             elif self._is_vector(source_data):
                 vector = Vector.from_node_data(source_data)
+                visited_vector.add(source_node)
 
             if self._is_point(target_data):
                 point = Point.from_node_data(target_data)
             elif self._is_vector(target_data):
                 vector = Vector.from_node_data(target_data)
+                visited_vector.add(target_node)
 
             if point is None and vector is None:
                 raise ValueError(f"Invalid node data: {source_data} or {target_data}")

{natural_agi_common-0.1.31 → natural_agi_common-0.1.32}/natural_agi_common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: natural-agi-common
-Version: 0.1.31
+Version: 0.1.32
 Requires-Dist: pydantic
 Requires-Dist: networkx
 Requires-Dist: neo4j

{natural_agi_common-0.1.31 → natural_agi_common-0.1.32}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="natural-agi-common",
-    version="0.1.31",
+    version="0.1.32",
     packages=find_packages(include=["common", "common.*"]),
     install_requires=["pydantic", "networkx", "neo4j"],
 )