gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Fixes edge case handling

Browse source
This commit is contained in:
hajdul88 2024-11-13 17:18:07 +01:00
parent bf4eedd20e
commit d8024db002
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/graph/cognee_graph/CogneeGraph.py
View file

@ -54,6 +54,10 @@ class CogneeGraph(CogneeAbstractGraph):
directed = True, directed = True,
node_dimension = 1, node_dimension = 1,
edge_dimension = 1) -> None: edge_dimension = 1) -> None:
if node_dimension < 1 or edge_dimension < 1:
raise ValueError("Dimensions must be positive integers")
try: try:
nodes_data, edges_data = await adapter.get_graph_data() nodes_data, edges_data = await adapter.get_graph_data()

4
cognee/modules/graph/cognee_graph/CogneeGraphElements.py
View file

@ -17,6 +17,8 @@ class Node:
status: np.ndarray status: np.ndarray
def __init__(self, node_id: str, attributes: Optional[Dict[str, Any]] = None, dimension: int = 1): def __init__(self, node_id: str, attributes: Optional[Dict[str, Any]] = None, dimension: int = 1):
if dimension <= 0:
raise ValueError("Dimension must be a positive integer")
self.id = node_id self.id = node_id
self.attributes = attributes if attributes is not None else {} self.attributes = attributes if attributes is not None else {}
self.skeleton_neighbours = [] self.skeleton_neighbours = []
@ -104,6 +106,8 @@ class Edge:
return hash(frozenset({self.node1, self.node2})) return hash(frozenset({self.node1, self.node2}))
def __eq__(self, other: "Edge") -> bool: def __eq__(self, other: "Edge") -> bool:
if not isinstance(other, Edge):
return False
if self.directed: if self.directed:
return self.node1 == other.node1 and self.node2 == other.node2 return self.node1 == other.node1 and self.node2 == other.node2
else: else: