Update Neo4jPathsModel with relationships (#100)

* Update Neo4jPathsModel with relationships --------- Co-authored-by: Valerii Mironchenko <[email protected]>
opendatadiscovery · Jul 15, 2024 · 79dac43 · 79dac43
1 parent b9af751
commit 79dac43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/oddrn_generator/path_models.py b/oddrn_generator/path_models.py
@@ -529,13 +529,15 @@ class Neo4jPathsModel(BasePathsModel):
     databases: str
     nodes: Optional[str] = None
     fields: Optional[str] = None
+    relationships: Optional[str] = None
 
     @classmethod
     def _dependencies_map_factory(cls):
         return {
             "databases": ("databases",),
             "nodes": ("databases", "nodes"),
             "fields": ("databases", "nodes", "fields"),
+            "relationships": ("databases", "nodes", "relationships"),
         }
 
     data_source_path: str = "databases"

diff --git a/tests/params.py b/tests/params.py
@@ -258,6 +258,7 @@
                 "databases": "some_database",
                 "nodes": "some_node",
                 "fields": "some_field",
+                "relationships": "some_relationship",
             },
             "aliases": {},
         },