connect ai-worker to flow-engine and be able to update ui nodes in the new structure

Author: devhelpr

apps/vps-web/src/app/ai-flow-engine-worker/ai-flow-engine-worker.ts

@@ -95,7 +95,25 @@ self.addEventListener('message', (event: MessageEvent<AIWorkerMessage>) => {
           } as IComputeResult,
         });
       };
-      flowEngine.initialize(data.flow.flows['flow'].nodes, registerWorkerNodes);
+      flowEngine.initialize(
+        data.flow.flows['flow'].nodes,
+        registerWorkerNodes,
+        {
+          sendOutputToNode: (data, node) => {
+            //console.log('sendOutputToNode', data, node);
+            if (node.nodeInfo?.shouldNotSendOutputFromWorkerToMainThread) {
+              return;
+            }
+            self.postMessage({
+              message: 'node-update',
+              result: {
+                result: node.id,
+                output: data,
+              } as IComputeResult,
+            });
+          },
+        }
+      );
       flowEngine.canvasApp.elements.forEach((node) => {
         if (node && node.nodeInfo) {
           node.nodeInfo.offscreenCanvas = undefined;

apps/vps-web/src/app/flow-engine/flow-engine.ts

@@ -30,7 +30,7 @@ import {
 import { CreateRunCounterContext } from '../custom-nodes/classes/base-rect-node-class';
 
 export class RuntimeFlowEngine {
-  public canvasApp: IFlowCanvasBase<NodeInfo>;
+  public canvasApp: IFlowCanvasBase<NodeInfo, FlowEngine>;
   constructor() {
     this.canvasApp = createRuntimeFlowContext<NodeInfo>();
   }
@@ -43,7 +43,8 @@ export class RuntimeFlowEngine {
       registerNodeFactory: RegisterNodeFactoryFunction,
       createRunCounterContext: CreateRunCounterContext,
       flowEngine?: FlowEngine
-    ) => void
+    ) => void,
+    flowEngine?: FlowEngine
   ) {
     if (!this.canvasApp) {
       throw new Error('CanvasAppInstance not initialized');
@@ -73,7 +74,9 @@ export class RuntimeFlowEngine {
       },
       undefined,
       0,
-      getNodeTaskFactory
+      getNodeTaskFactory,
+      undefined,
+      flowEngine
     );
 
     this.canvasApp.setAnimationFunctions({