Fix modeling bug (#143)

* fix bug where patching an invalid model would throw error * Update CHANGELOG.md
a-s-g93 · Oct 31, 2024 · 3358139 · 3358139
1 parent 7201dde
commit 3358139
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,8 @@
 
 ### Fixed
 
+* Fix bug where patching previous valid nodes into an invalid data model would throw an error
+
 ### Changed
 
 * Add comma separator for numbers in `GraphEDA` report

diff --git a/neo4j_runway/llm/base.py b/neo4j_runway/llm/base.py
@@ -192,8 +192,14 @@ def _get_initial_data_model_response(
             apply_neo4j_naming_conventions=apply_neo4j_naming_conventions,
         )
 
-        if not hasattr(initial_data_model, "nodes") and hasattr(nodes, "nodes"):
-            initial_data_model.nodes = nodes.nodes
+        if (
+            not hasattr(initial_data_model, "nodes") or initial_data_model.nodes is None
+        ) and hasattr(nodes, "nodes"):
+            try:
+                print(initial_data_model)
+                initial_data_model.nodes = nodes.nodes
+            except Exception as e:
+                pass
 
         return initial_data_model