node-red-contrib-ollama 0.2.0 → 0.2.3

package/nodes/ollama-config-server.html

@@ -0,0 +1,24 @@
+<script type="text/javascript">
+    RED.nodes.registerType( 'ollama-config-server', {
+        category: 'config',
+        defaults: {
+            host: { value: 'localhost', required: true },
+            port: { value: 11434, required: true, validate: RED.validators.number() }
+        },
+        label: function() {
+            return this.host + ':' + this.port;
+        }
+    } )
+</script>
+
+<script type="text/html" data-template-name="ollama-config-server">
+    <div class="form-row">
+        <label for="node-config-input-host"><i class="fa fa-server"></i> Host</label>
+        <input type="text" id="node-config-input-host" placeholder="localhost" />
+    </div>
+
+    <div class="form-row">
+        <label for="node-config-input-port"><i class="fa fa-port"></i> Port</label>
+        <input type="text" id="node-config-input-port" placeholder="11434" />
+    </div>
+</script>

package/nodes/ollama-config-server.js

@@ -0,0 +1,10 @@
+module.exports = function( RED ) {
+    function OllamaConfigServerNode( config ) {
+        RED.nodes.createNode( this, config )
+
+        this.host = config.host
+        this.port = config.port
+    }
+
+    RED.nodes.registerType( 'ollama-config-server', OllamaConfigServerNode )
+}

package/nodes/ollama-copy.html

@@ -3,7 +3,8 @@
         category: 'AI',
         color: 'rgb(186 230 253)',
         defaults: {
-            name: { value: '' },
+            name: { value: 'Ollama Copy' },
+            server: { value: '', type: 'ollama-config-server', required: false }
         },
         inputs: 1,
         outputs: 1,
@@ -19,6 +20,11 @@
         <label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
         <input type="text" id="node-input-name" placeholder="Name" />
     </div>
+
+    <div class="form-row">
+        <label for="node-input-server"><i class="fa fa-server"></i> Server</label>
+        <input type="text" id="node-input-server" />
+    </div>
 </script>
 
 <script type="text/html" data-help-name="ollama-copy">
@@ -27,26 +33,33 @@
     <h3>Inputs</h3>
 
     <dl class="message-properties">
-        <dt>
-            payload.host
-
-            <span class="property-type">string</span>
+        <dt class="optional">
+            payload <span class="property-type">object</span>
         </dt>
-        <dd>The host of the Ollama instance.</dd>
 
-        <dt>
-            payload.source
+        <dd>
+            The payload object.
 
-            <span class="property-type">string</span>
-        </dt>
-        <dd>The name of the model to copy.</dd>
+            <dl class="message-properties">
+                <dt class="optional">
+                    payload.host <span class="property-type">string</span>
+                </dt>
+                <dd>
+                    The host of the Ollama instance.<br />
+                    <em>Alternative to the <code>Server</code> field.</em>
+                </dd>
 
-        <dt>
-            payload.destination
+                <dt>
+                    payload.source <span class="property-type">string</span>
+                </dt>
+                <dd>The name of the model to copy.</dd>
 
-            <span class="property-type">string</span>
-        </dt>
-        <dd>The name of the new model.</dd>
+                <dt>
+                    payload.destination <span class="property-type">string</span>
+                </dt>
+                <dd>The name of the new model.</dd>
+            </dl>
+        </dd>
     </dl>
 
     <h3>Outputs</h3>

package/nodes/ollama-copy.js

@@ -6,7 +6,15 @@ module.exports = function( RED ) {
         const node = this
 
         node.on( 'input', async function( msg ) {
-            const { host, source, destination } = msg.payload
+            const {
+                host: payloadHost,
+                source,
+                destination
+            } = msg.payload
+
+            const server = RED.nodes.getNode( config.server )
+
+            const host = ( server ) ? server.host + ':' + server.port : payloadHost
 
             const ollama = new Ollama( { host } )
 

package/nodes/ollama-create.html

@@ -3,7 +3,9 @@
         category: 'AI',
         color: 'rgb(186 230 253)',
         defaults: {
-            name: { value: '' },
+            name: { value: 'Ollama Create' },
+            server: { value: '', type: 'ollama-config-server', required: false },
+            model: { value: '', type: 'ollama-config-model', required: false },
         },
         inputs: 1,
         outputs: 1,
@@ -19,6 +21,16 @@
         <label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
         <input type="text" id="node-input-name" placeholder="Name" />
     </div>
+
+    <div class="form-row">
+        <label for="node-input-server"><i class="fa fa-server"></i> Server</label>
+        <input type="text" id="node-input-server" />
+    </div>
+
+    <div class="form-row">
+        <label for="node-input-model"><i class="fa fa-bookmark"></i> Model</label>
+        <input type="text" id="node-input-model" />
+    </div>
 </script>
 
 <script type="text/html" data-help-name="ollama-create">
@@ -27,40 +39,46 @@
     <h3>Inputs</h3>
 
     <dl class="message-properties">
-        <dt>
-            payload.host
-
-            <span class="property-type">string</span>
-        </dt>
-        <dd>The host of the Ollama instance.</dd>
-
-        <dt>
-            payload.model
-
-            <span class="property-type">string</span>
-        </dt>
-        <p>The name of the model to create.</p>
-
         <dt class="optional">
-            payload.path
-
-            <span class="property-type">string</span>
+            payload <span class="property-type">object</span>
         </dt>
-        <p>The path to the Modelfile to use for creating the model.</p>
-
-        <dt class="optional">
-            payload.modelfile
 
-            <span class="property-type">string</span>
-        </dt>
-        <p>The content of the Modelfile to create.</p>
-
-        <dt class="optional">
-            payload.stream
-
-            <span class="property-type">boolean</span>
-        </dt>
-        <p>When true an AsyncGenerator is returned.</p>
+        <dd>
+            The payload object.
+
+            <dl class="message-properties">
+                <dt class="optional">
+                    payload.host <span class="property-type">string</span>
+                </dt>
+                <dd>
+                    The host of the Ollama instance.<br />
+                    <em>Alternative to the <code>Server</code> field.</em>
+                </dd>
+
+                <dt class="optional">
+                    payload.model <span class="property-type">string</span>
+                </dt>
+                <dd>
+                    The name of the model to create.<br />
+                    <em>Alternative to the <code>Model</code> field.</em>
+                </dd>
+
+                <dt class="optional">
+                    payload.path <span class="property-type">string</span>
+                </dt>
+                <dd>The path to the Modelfile to use for creating the model.</dd>
+
+                <dt class="optional">
+                    payload.modelfile <span class="property-type">string</span>
+                </dt>
+                <dd>The content of the Modelfile to create.</dd>
+
+                <dt class="optional">
+                    payload.stream <span class="property-type">boolean</span>
+                </dt>
+                <dd>When true an AsyncGenerator is returned.</dd>
+            </dl>
+        </dd>
     </dl>
 
     <h3>Outputs</h3>

package/nodes/ollama-create.js

@@ -7,15 +7,21 @@ module.exports = function( RED ) {
 
         node.on( 'input', async function( msg ) {
             const {
-                host,
-                model,
+                host: payloadHost,
+                model: payloadModel,
                 modelfile,
                 stream,
                 path
             } = msg.payload
 
+            const server = RED.nodes.getNode( config.server )
+            const host = ( server ) ? server.host + ':' + server.port : payloadHost
+
             const ollama = new Ollama( { host } )
 
+            const modelConfig = RED.nodes.getNode( config.model )
+            const model = ( modelConfig ) ? modelConfig.name : payloadModel
+
             const response = await ollama.create( {
                 model,
                 modelfile,

package/nodes/ollama-delete.html

@@ -3,7 +3,9 @@
         category: 'AI',
         color: 'rgb(186 230 253)',
         defaults: {
-            name: { value: '' },
+            name: { value: 'Ollama Delete' },
+            server: { value: '', type: 'ollama-config-server', required: false },
+            model: { value: '', type: 'ollama-config-model', required: false },
         },
         inputs: 1,
         outputs: 1,
@@ -19,6 +21,16 @@
         <label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
         <input type="text" id="node-input-name" placeholder="Name" />
     </div>
+
+    <div class="form-row">
+        <label for="node-input-server"><i class="fa fa-server"></i> Server</label>
+        <input type="text" id="node-input-server" />
+    </div>
+
+    <div class="form-row">
+        <label for="node-input-model"><i class="fa fa-bookmark"></i> Model</label>
+        <input type="text" id="node-input-model" />
+    </div>
 </script>
 
 <script type="text/html" data-help-name="ollama-delete">
@@ -27,19 +39,31 @@
     <h3>Inputs</h3>
 
     <dl class="message-properties">
-        <dt>
-            payload.host
-
-            <span class="property-type">string</span>
+        <dt class="optional">
+            payload <span class="property-type">object</span>
         </dt>
-        <dd>The host of the Ollama instance.</dd>
 
-        <dt>
-            payload.model
+        <dd>
+            The payload object.
 
-            <span class="property-type">string</span>
-        </dt>
-        <dd>The name of the model to delete.</dd>
+            <dl class="message-properties">
+                <dt class="optional">
+                    payload.host <span class="property-type">string</span>
+                </dt>
+                <dd>
+                    The host of the Ollama instance.<br />
+                    <em>Alternative to the <code>Server</code> field.</em>
+                </dd>
+
+                <dt class="optional">
+                    payload.model <span class="property-type">string</span>
+                </dt>
+                <dd>
+                    The name of the model to delete.<br />
+                    <em>Alternative to the <code>Model</code> field.</em>
+                </dd>
+            </dl>
+        </dd>
     </dl>
 
     <h3>Outputs</h3>

package/nodes/ollama-delete.js

@@ -6,10 +6,19 @@ module.exports = function( RED ) {
         const node = this
 
         node.on( 'input', async function( msg ) {
-            const { host, model } = msg.payload
+            const {
+                host: payloadHost,
+                model: payloadModel
+            } = msg.payload
+
+            const server = RED.nodes.getNode( config.server )
+            const host = ( server ) ? server.host + ':' + server.port : payloadHost
 
             const ollama = new Ollama( { host } )
 
+            const modelConfig = RED.nodes.getNode( config.model )
+            const model = ( modelConfig ) ? modelConfig.name : payloadModel
+
             const response = await ollama.delete( { model } )
                 .catch( error => {
                     node.error( error )