@flowfuse/nr-assistant
Version:
FlowFuse Node-RED assistant plugin
88 lines (84 loc) • 4.12 kB
JavaScript
module.exports = (RED) => {
const { default: got } = require('got')
RED.plugins.registerPlugin('flowfuse-nr-assistant', {
type: 'assistant',
name: 'Node-RED Assistant Plugin',
icon: 'font-awesome/fa-magic',
settings: {
'*': { exportable: true }
},
onadd: function () {
const assistantSettings = RED.settings.flowforge?.assistant || { enabled: false }
const clientSettings = {
enabled: assistantSettings.enabled !== false && !!assistantSettings.url,
requestTimeout: assistantSettings.requestTimeout || 60000
}
RED.comms.publish('nr-assistant/initialise', clientSettings, true /* retain */)
if (!assistantSettings || !assistantSettings.enabled) {
RED.log.info('FlowFuse Assistant Plugin is disabled')
return
}
if (!assistantSettings.url) {
RED.log.info('FlowFuse Assistant Plugin is missing url')
return
}
RED.log.info('FlowFuse Assistant Plugin loaded')
RED.httpAdmin.post('/nr-assistant/:method', RED.auth.needsPermission('write'), function (req, res) {
const method = req.params.method
// limit method to prevent path traversal
if (!method || typeof method !== 'string' || /[^a-z0-9-_]/.test(method)) {
res.status(400)
res.json({ status: 'error', message: 'Invalid method' })
return
}
const input = req.body
if (!input || !input.prompt || typeof input.prompt !== 'string') {
res.status(400)
res.json({ status: 'error', message: 'prompt is required' })
return
}
const body = {
prompt: input.prompt, // this is the prompt to the AI
promptHint: input.promptHint, // this is used to let the AI know what we are generating (`function node? Node JavaScript? flow?)
context: input.context, // this is used to provide additional context to the AI (e.g. the selected text of the function node)
transactionId: input.transactionId // used to correlate the request with the response
}
// join url & method (taking care of trailing slashes)
const url = `${assistantSettings.url.replace(/\/$/, '')}/${method.replace(/^\//, '')}`
got.post(url, {
headers: {
Accept: '*/*',
'Accept-Language': 'en-GB,en-US;q=0.9,en;q=0.8,es;q=0.7',
Authorization: `Bearer ${assistantSettings.token}`,
'Content-Type': 'application/json'
},
json: body
}).then(response => {
const data = JSON.parse(response.body)
res.json({
status: 'ok',
data
})
}).catch((error) => {
let body = error.response?.body
if (typeof body === 'string') {
try {
body = JSON.parse(body)
} catch (e) {
// ignore
}
}
let message = 'FlowFuse Assistant request was unsuccessful'
const errorData = { status: 'error', message, body }
const errorCode = error.response?.statusCode || 500
res.status(errorCode).json(errorData)
RED.log.trace('nr-assistant error:', error)
if (body && typeof body === 'object' && body.error) {
message = `${message}: ${body.error}`
}
RED.log.warn(message)
})
})
}
})
}