node-red-contrib-ollama 0.2.3 → 0.2.4

package/nodes/ollama-chat.html

@@ -6,6 +6,7 @@
             name: { value: 'Ollama Chat' },
             server: { value: '', type: 'ollama-config-server', required: false },
             model: { value: '', type: 'ollama-config-model', required: true },
+            tools: { value: '', type: 'ollama-config-tools', required: false },
         },
         inputs: 1,
         outputs: 1,
@@ -31,6 +32,11 @@
         <label for="node-input-model"><i class="fa fa-bookmark"></i> Model</label>
         <input type="text" id="node-input-model" />
     </div>
+
+    <div class="form-row">
+        <label for="node-input-tools"><i class="fa fa-cog"></i> Tools</label>
+        <input type="text" id="node-input-tools" />
+    </div>
 </script>
 
 <script type="text/html" data-help-name="ollama-chat">
@@ -114,6 +120,7 @@
                 <dd>
                     Tools for the model to use if supported.<br />
                     Requires <code>stream</code> to be set to <code>false</code>.
+                    <em>Alternative to the <code>Model</code> field.</em>
 
                     <p>
                         For more information, see the Ollama API documentation:<br />

package/nodes/ollama-chat.js

@@ -13,7 +13,7 @@ module.exports = function( RED ) {
                 format,
                 stream,
                 keep_alive,
-                tools,
+                tools: payloadTools,
                 options
             } = msg.payload
 
@@ -25,6 +25,9 @@ module.exports = function( RED ) {
             const modelConfig = RED.nodes.getNode( config.model )
             const model = ( modelConfig ) ? modelConfig.name : payloadModel
 
+            const toolsConfig = RED.nodes.getNode( config.tools )
+            const tools = ( toolsConfig ) ? toolsConfig.json : payloadTools
+
             const response = await ollama.chat( {
                     model,
                     messages,

package/nodes/ollama-config-tools.html

@@ -0,0 +1,30 @@
+<script type="text/javascript">
+    RED.nodes.registerType( 'ollama-config-tools', {
+        category: 'config',
+        defaults: {
+            name: { value: 'Ollama Tools Config' },
+            json: { value: '' }
+        },
+        label: function() {
+            return this.name;
+        },
+        oneditprepare: function() {
+            $( '#node-config-input-json' ).typedInput( {
+                type: 'json',
+                types: [ 'json' ]
+            } )
+        }
+    } )
+</script>
+
+<script type="text/html" data-template-name="ollama-config-tools">
+    <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="Ollama Tools Config" />
+    </div>
+
+    <div class="form-row">
+        <label for="node-config-input-json"><i class="fa fa-cog"></i> JSON</label>
+        <input type="text" id="node-config-input-json" placeholder="" />
+    </div>
+</script>

package/nodes/ollama-config-tools.js

@@ -0,0 +1,10 @@
+module.exports = function( RED ) {
+    function OllamaConfigToolsNode( config ) {
+        RED.nodes.createNode( this, config )
+
+        this.name = config.name
+        this.json = config.json
+    }
+
+    RED.nodes.registerType( 'ollama-config-tools', OllamaConfigToolsNode )
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-red-contrib-ollama",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "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,8 @@
       "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-model": "nodes/ollama-config-model.js",
+      "ollama-config-tools": "nodes/ollama-config-tools.js"
     }
   },
   "engines": {