frontend is more forgivable to predicted semantic model that does not…

… have enough data node

www/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sand",
-  "version": "2.2.1",
+  "version": "2.2.2",
   "private": true,
   "dependencies": {
     "@ant-design/colors": "^6.0.0",

www/src/models/AssistantService.ts

@@ -77,6 +77,37 @@ export class AssistantService extends RStore<number, AssistantRecord> {
       data: rawsm,
       id: "",
     }).graph;
+    // add back any data node that aren't in the graph
+    for (
+      let columnIndex = 0;
+      columnIndex < table.columns.length;
+      columnIndex++
+    ) {
+      if (graph.nodeByColumnIndex(columnIndex) === undefined) {
+        let prefix = "dnode";
+        if (graph.hasNode(`${prefix}:${columnIndex}`)) {
+          for (let counter = 0; counter < 1000; counter++) {
+            prefix = `dnode-p${counter}`;
+            if (!graph.hasNode(`${prefix}:${columnIndex}`)) {
+              break;
+            }
+          }
+        }
+
+        if (graph.hasNode(`${prefix}:${columnIndex}`)) {
+          throw new Error(
+            `Cannot generate a unique id to create a node for column ${columnIndex} in the graph`
+          );
+        }
+
+        graph.addDataNode({
+          id: `${prefix}:${columnIndex}`,
+          label: table.columns[columnIndex],
+          columnIndex: columnIndex,
+          nodetype: "data_node",
+        });
+      }
+    }
 
     // before set a new draft, check if an empty draft is there (as default) and remove it
     const prevDrafts = this.smStore.getCreateDraftsByTable(table);

www/src/models/sm/SMGraph.ts

@@ -455,6 +455,7 @@ export class SMGraph {
       throw new Error("Duplicated id");
     }
     this.nodeId2Index[node.id] = this.nodes.length;
+    this.column2nodeIndex[node.columnIndex] = this.nodes.length;
     this.nodes.push(node);
     this.version += 1;
     this.stale = true;