node-red-contrib-ollama 0.2.6 → 0.4.0

package/nodes/ollama-show.js

@@ -3,37 +3,75 @@ module.exports = function( RED ) {
 
     function OllamaShowNode( config )  {
         RED.nodes.createNode( this, config )
-        const node = this
 
-        node.on( 'input', async function( msg ) {
-            const {
-                host: payloadHost,
-                model: payloadModel,
-                system,
-                template,
-                options: payloadOptions
-            } = msg.payload
+        this.modelType = config.modelType || 'str'
+        this.systemType = config.systemType || 'str'
+        this.templateType = config.templateType || 'str'
 
+        const node = this
+        node.on( 'input', async function( msg ) {
             const server = RED.nodes.getNode( config.server )
-            const host = ( server ) ? server.host + ':' + server.port : payloadHost
+            const host = msg?.payload?.host || ( server ) ? server.host + ':' + server.port : null
 
             const ollama = new Ollama( { host } )
 
-            const modelConfig = RED.nodes.getNode( config.model )
-            const model = ( modelConfig ) ? modelConfig.name : payloadModel
+            let model = null
+            if( msg?.payload?.model ) {
+                model = msg?.payload?.model
+            } else if( !!config.model ) {
+                if( node.modelType === 'str' ) {
+                    model = config.model
+                } else if( node.modelType === 'msg' ) {
+                    model = msg[ config.model ]
+                } else if( node.modelType === 'flow' ) {
+                    model = node.context().flow.get( config.model )
+                } else if( node.modelType === 'global' ) {
+                    model = node.context().global.get( config.model )
+                }
+            }
+
+            let system = null
+            if( msg?.payload?.system ) {
+                system = msg?.payload?.system
+            } else if( !!config.system ) {
+                if( node.systemType === 'str' ) {
+                    system = config.system
+                } else if( node.systemType === 'msg' ) {
+                    system = msg[ config.system ]
+                } else if( node.systemType === 'flow' ) {
+                    system = node.context().flow.get( config.system )
+                } else if( node.systemType === 'global' ) {
+                    system = node.context().global.get( config.system )
+                }
+            }
+
+            let template = null
+            if( msg?.payload?.template ) {
+                template = msg?.payload?.template
+            } else if( !!config.template ) {
+                if( node.templateType === 'str' ) {
+                    template = config.template
+                } else if( node.templateType === 'msg' ) {
+                    template = msg[ config.template ]
+                } else if( node.templateType === 'flow' ) {
+                    template = node.context().flow.get( config.template )
+                } else if( node.templateType === 'global' ) {
+                    template = node.context().global.get( config.template )
+                }
+            }
 
             const optionsConfig = RED.nodes.getNode( config.options )
-            const options = ( optionsConfig ) ? optionsConfig.json : payloadOptions
+            const options = msg?.payload?.options || ( optionsConfig ) ? JSON.parse( optionsConfig.json ) : null
 
             const response = await ollama.show( {
-                model,
-                system,
-                template,
-                options
-            } )
-            .catch( error => {
-                node.error( error )
-            } )
+                    model,
+                    system,
+                    template,
+                    options
+                } )
+                .catch( error => {
+                    node.error( error )
+                } )
 
             msg.payload = response
             node.send( msg )

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-red-contrib-ollama",
-  "version": "0.2.6",
+  "version": "0.4.0",
   "description": "Add AI functionality to your flows! This module includes a set of nodes that enable easy communication with Ollama, enriching your projects with intelligent solutions.",
   "keywords": [
     "ai",
@@ -30,7 +30,6 @@
       "ollama-ps": "nodes/ollama-ps.js",
       "ollama-abort": "nodes/ollama-abort.js",
       "ollama-config-server": "nodes/ollama-config-server.js",
-      "ollama-config-model": "nodes/ollama-config-model.js",
       "ollama-config-format": "nodes/ollama-config-format.js",
       "ollama-config-tools": "nodes/ollama-config-tools.js",
       "ollama-config-options": "nodes/ollama-config-options.js"
@@ -40,7 +39,7 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
-    "ollama": "^0.5.11"
+    "ollama": "^0.5.16"
   },
   "bugs": {
     "url": "https://github.com/jakubburkiewicz/node-red-contrib-ollama/issues"

package/nodes/ollama-config-model.html

@@ -1,18 +0,0 @@
-<script type="text/javascript">
-    RED.nodes.registerType( 'ollama-config-model', {
-        category: 'config',
-        defaults: {
-            name: { value: '', required: true }
-        },
-        label: function() {
-            return this.name;
-        }
-    } )
-</script>
-
-<script type="text/html" data-template-name="ollama-config-model">
-    <div class="form-row">
-        <label for="node-config-input-name"><i class="fa fa-server"></i> Name</label>
-        <input type="text" id="node-config-input-name" placeholder="llama3.3" />
-    </div>
-</script>

package/nodes/ollama-config-model.js

@@ -1,9 +0,0 @@
-module.exports = function( RED ) {
-    function OllamaConfigModelNode( config ) {
-        RED.nodes.createNode( this, config )
-
-        this.name = config.name
-    }
-
-    RED.nodes.registerType( 'ollama-config-model', OllamaConfigModelNode )
-}