From 727f8ad4849e62fd94e725433c04c264a0ed4e9f Mon Sep 17 00:00:00 2001
From: bymyself <abolkonsky.rem@gmail.com>
Date: Mon, 17 Feb 2025 20:55:46 -0700
Subject: [PATCH] retain initial node_versions on serialize

---
 js/workflow-metadata.js | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/js/workflow-metadata.js b/js/workflow-metadata.js
index 906abb78..34d5742e 100644
--- a/js/workflow-metadata.js
+++ b/js/workflow-metadata.js
@@ -65,6 +65,15 @@ class WorkflowMetadataExtension {
     return nodeVersions;
   }
 
+  /**
+   * The node versions embedded in the workflow json's initial state.
+   * @returns {Record<string, string> | undefined} The mapping from node name to version
+   */
+  get workflowNodeVersions() {
+    return app.extensionManager?.workflow?.activeWorkflow?.initialState?.extra
+      ?.node_versions;
+  }
+
   async init() {
     const extension = this;
     this.installedNodes = await this.getInstalledNodes();
@@ -81,7 +90,10 @@ class WorkflowMetadataExtension {
       }
       const graph = this;
       try {
-        workflow.extra["node_versions"] = extension.getGraphNodeVersions(graph);
+        workflow.extra["node_versions"] = {
+          ...extension.workflowNodeVersions,
+          ...extension.getGraphNodeVersions(graph),
+        };
       } catch (e) {
         console.error(e);
       }