support multiple suggestions in next node prediction

Author: Steve-Mcl

lib/assistant.js

@@ -6,7 +6,6 @@ const { getLongestUpstreamPath } = require('./flowGraph')
 const { hasProperty } = require('./utils')
 const semver = require('semver')
 
-const PREDICTION_SEQUENCE_LENGTH = 5 // The length of the input sequence for the TensorFlow model
 const FF_ASSISTANT_USER_AGENT = 'FlowFuse Assistant Plugin/' + require('../package.json').version
 
 /**
@@ -334,16 +333,16 @@ class Assistant {
         },
         /** @type {import('@modelcontextprotocol/sdk/server/mcp.js').ToolCallback} */
         async ({ flow, sourceNode, sourcePort }) => {
-            sourceNode = sourceNode || {}
+            const attachToNode = sourceNode || {}
             /** @type {Array<{type: string}>} */
             let suggestedNodes = []
             /** @type {Array<{type: string}>} */
             const upstreamNodes = []
-            if (flow && flow.length && sourceNode.id) {
-                upstreamNodes.push(...getLongestUpstreamPath(flow, sourceNode.id))
+            if (flow && flow.length && attachToNode.id) {
+                upstreamNodes.push(...getLongestUpstreamPath(flow, attachToNode.id))
             }
             if (this.completionsReady) {
-                const allNodes = [...upstreamNodes, sourceNode].slice(-PREDICTION_SEQUENCE_LENGTH) // Include the last node in the input
+                const allNodes = [...upstreamNodes, attachToNode] // Include the last node in the input
                 const typeNames = allNodes.map(node => node.type) // Extract the type names from the nodes
                 const vectorizedText = this.labeller.encode_sequence(typeNames)
 
@@ -352,7 +351,8 @@ class Assistant {
                 }
 
                 const results = await this._completionsSession.run(feeds)
-                const predictions = this.labeller.decode_predictions(results.probabilities.cpuData)
+                // Get top 3 + 5 additional predictions (only 3 will be used since there are 5 permanent suggestions in the typeSearch, we need enough to fill the suggestions array)
+                const predictions = this.labeller.decode_predictions(results.probabilities.cpuData, 3 + 5)
 
                 suggestedNodes = predictions.map(prediction => ({ type: prediction.className })) // Create new nodes with the predicted types
             }