add id attr

explodinggradients · Jun 18, 2024 · c417a61 · c417a61
1 parent c8e94be
commit c417a61
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/ragas/testsetv3/graph.py b/src/ragas/testsetv3/graph.py
@@ -34,11 +34,17 @@ class Relationship(ObjectType):
  for relationship in node if target.id == node.id then it means that the relationship is incoming to the node
  """
 
+ id = String(required=True)
  source = Field(lambda: Node)
  target = Field(lambda: Node)
  label = String()
  properties = Field(JSONString)
 
+ def __init__(self, id=None, **kwargs):
+ if id is None:
+ id = str(uuid.uuid4())
+ super().__init__(id=id, **kwargs)
+
 
 class Node(ObjectType):
  """Represents a node in a graph with associated properties.