Add more nodes for agents, loaders

packages/server/src/ChatflowPool.ts

@@ -0,0 +1,43 @@
+import { INodeData } from 'flowise-components'
+import { IActiveChatflows } from './Interface'
+
+/**
+ * This pool is to keep track of active test triggers (event listeners),
+ * so we can clear the event listeners whenever user refresh or exit page
+ */
+export class ChatflowPool {
+    activeChatflows: IActiveChatflows = {}
+
+    /**
+     * Add to the pool
+     * @param {string} chatflowid
+     * @param {INodeData} endingNodeData
+     */
+    add(chatflowid: string, endingNodeData: INodeData) {
+        this.activeChatflows[chatflowid] = {
+            endingNodeData,
+            inSync: true
+        }
+    }
+
+    /**
+     * Update to the pool
+     * @param {string} chatflowid
+     * @param {boolean} inSync
+     */
+    updateInSync(chatflowid: string, inSync: boolean) {
+        if (Object.prototype.hasOwnProperty.call(this.activeChatflows, chatflowid)) {
+            this.activeChatflows[chatflowid].inSync = inSync
+        }
+    }
+
+    /**
+     * Remove from the pool
+     * @param {string} chatflowid
+     */
+    async remove(chatflowid: string) {
+        if (Object.prototype.hasOwnProperty.call(this.activeChatflows, chatflowid)) {
+            delete this.activeChatflows[chatflowid]
+        }
+    }
+}