node-red-contrib-ollama 0.1.10 → 0.2.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +5 -1
- package/nodes/ollama-abort.html +53 -0
- package/nodes/ollama-abort.js +30 -0
- package/nodes/ollama-chat.html +214 -253
- package/nodes/ollama-chat.js +23 -2
- package/nodes/ollama-config-server.html +24 -0
- package/nodes/ollama-config-server.js +10 -0
- package/nodes/ollama-copy.html +29 -16
- package/nodes/ollama-copy.js +9 -1
- package/nodes/ollama-create.html +41 -53
- package/nodes/ollama-create.js +20 -5
- package/nodes/ollama-delete.html +26 -11
- package/nodes/ollama-delete.js +8 -1
- package/nodes/ollama-embed.html +238 -0
- package/nodes/ollama-embed.js +41 -0
- package/nodes/ollama-generate.html +50 -242
- package/nodes/ollama-generate.js +34 -5
- package/nodes/ollama-list.html +23 -13
- package/nodes/ollama-list.js +7 -1
- package/nodes/ollama-ps.html +153 -0
- package/nodes/{ollama-embeddings.js → ollama-ps.js} +10 -4
- package/nodes/ollama-pull.html +37 -26
- package/nodes/ollama-pull.js +10 -1
- package/nodes/ollama-push.html +37 -26
- package/nodes/ollama-push.js +10 -1
- package/nodes/ollama-show.html +36 -27
- package/nodes/ollama-show.js +20 -5
- package/package.json +7 -4
- package/nodes/ollama-embeddings.html +0 -275
package/nodes/ollama-copy.html
CHANGED
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
category: 'AI',
|
|
4
4
|
color: 'rgb(186 230 253)',
|
|
5
5
|
defaults: {
|
|
6
|
-
name: { value: '' },
|
|
6
|
+
name: { value: 'Ollama Copy' },
|
|
7
|
+
server: { value: '', type: 'ollama-config-server', required: false }
|
|
7
8
|
},
|
|
8
9
|
inputs: 1,
|
|
9
10
|
outputs: 1,
|
|
@@ -19,6 +20,11 @@
|
|
|
19
20
|
<label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
|
|
20
21
|
<input type="text" id="node-input-name" placeholder="Name" />
|
|
21
22
|
</div>
|
|
23
|
+
|
|
24
|
+
<div class="form-row">
|
|
25
|
+
<label for="node-input-server"><i class="fa fa-server"></i> Server</label>
|
|
26
|
+
<input type="text" id="node-input-server" />
|
|
27
|
+
</div>
|
|
22
28
|
</script>
|
|
23
29
|
|
|
24
30
|
<script type="text/html" data-help-name="ollama-copy">
|
|
@@ -27,26 +33,33 @@
|
|
|
27
33
|
<h3>Inputs</h3>
|
|
28
34
|
|
|
29
35
|
<dl class="message-properties">
|
|
30
|
-
<dt>
|
|
31
|
-
payload
|
|
32
|
-
|
|
33
|
-
<span class="property-type">string</span>
|
|
36
|
+
<dt class="optional">
|
|
37
|
+
payload <span class="property-type">object</span>
|
|
34
38
|
</dt>
|
|
35
|
-
<dd>The host of the Ollama instance.</dd>
|
|
36
39
|
|
|
37
|
-
<
|
|
38
|
-
payload.
|
|
40
|
+
<dd>
|
|
41
|
+
The payload object.
|
|
39
42
|
|
|
40
|
-
<
|
|
41
|
-
|
|
42
|
-
|
|
43
|
+
<dl class="message-properties">
|
|
44
|
+
<dt class="optional">
|
|
45
|
+
payload.host <span class="property-type">string</span>
|
|
46
|
+
</dt>
|
|
47
|
+
<dd>
|
|
48
|
+
The host of the Ollama instance.<br />
|
|
49
|
+
<em>Alternative to the <code>Server</code> field.</em>
|
|
50
|
+
</dd>
|
|
43
51
|
|
|
44
|
-
|
|
45
|
-
|
|
52
|
+
<dt>
|
|
53
|
+
payload.source <span class="property-type">string</span>
|
|
54
|
+
</dt>
|
|
55
|
+
<dd>The name of the model to copy.</dd>
|
|
46
56
|
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
57
|
+
<dt>
|
|
58
|
+
payload.destination <span class="property-type">string</span>
|
|
59
|
+
</dt>
|
|
60
|
+
<dd>The name of the new model.</dd>
|
|
61
|
+
</dl>
|
|
62
|
+
</dd>
|
|
50
63
|
</dl>
|
|
51
64
|
|
|
52
65
|
<h3>Outputs</h3>
|
package/nodes/ollama-copy.js
CHANGED
|
@@ -6,7 +6,15 @@ module.exports = function( RED ) {
|
|
|
6
6
|
const node = this
|
|
7
7
|
|
|
8
8
|
node.on( 'input', async function( msg ) {
|
|
9
|
-
const {
|
|
9
|
+
const {
|
|
10
|
+
host: payloadHost,
|
|
11
|
+
source,
|
|
12
|
+
destination
|
|
13
|
+
} = msg.payload
|
|
14
|
+
|
|
15
|
+
const server = RED.nodes.getNode( config.server )
|
|
16
|
+
|
|
17
|
+
const host = ( server ) ? server.host + ':' + server.port : payloadHost
|
|
10
18
|
|
|
11
19
|
const ollama = new Ollama( { host } )
|
|
12
20
|
|
package/nodes/ollama-create.html
CHANGED
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
category: 'AI',
|
|
4
4
|
color: 'rgb(186 230 253)',
|
|
5
5
|
defaults: {
|
|
6
|
-
name: { value: '' },
|
|
6
|
+
name: { value: 'Ollama Create' },
|
|
7
|
+
server: { value: '', type: 'ollama-config-server', required: false }
|
|
7
8
|
},
|
|
8
9
|
inputs: 1,
|
|
9
10
|
outputs: 1,
|
|
@@ -19,6 +20,11 @@
|
|
|
19
20
|
<label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
|
|
20
21
|
<input type="text" id="node-input-name" placeholder="Name" />
|
|
21
22
|
</div>
|
|
23
|
+
|
|
24
|
+
<div class="form-row">
|
|
25
|
+
<label for="node-input-server"><i class="fa fa-server"></i> Server</label>
|
|
26
|
+
<input type="text" id="node-input-server" />
|
|
27
|
+
</div>
|
|
22
28
|
</script>
|
|
23
29
|
|
|
24
30
|
<script type="text/html" data-help-name="ollama-create">
|
|
@@ -27,40 +33,43 @@
|
|
|
27
33
|
<h3>Inputs</h3>
|
|
28
34
|
|
|
29
35
|
<dl class="message-properties">
|
|
30
|
-
<dt>
|
|
31
|
-
payload.host
|
|
32
|
-
|
|
33
|
-
<span class="property-type">string</span>
|
|
34
|
-
</dt>
|
|
35
|
-
<dd>The host of the Ollama instance.</dd>
|
|
36
|
-
|
|
37
|
-
<dt>
|
|
38
|
-
payload.model
|
|
39
|
-
|
|
40
|
-
<span class="property-type">string</span>
|
|
41
|
-
</dt>
|
|
42
|
-
<p>The name of the model to create.</p>
|
|
43
|
-
|
|
44
|
-
<dt class="optional">
|
|
45
|
-
payload.path
|
|
46
|
-
|
|
47
|
-
<span class="property-type">string</span>
|
|
48
|
-
</dt>
|
|
49
|
-
<p>The path to the Modelfile to use for creating the model.</p>
|
|
50
|
-
|
|
51
36
|
<dt class="optional">
|
|
52
|
-
payload
|
|
53
|
-
|
|
54
|
-
<span class="property-type">string</span>
|
|
37
|
+
payload <span class="property-type">object</span>
|
|
55
38
|
</dt>
|
|
56
|
-
<p>The content of the Modelfile to create.</p>
|
|
57
39
|
|
|
58
|
-
<
|
|
59
|
-
payload.
|
|
60
|
-
|
|
61
|
-
<
|
|
62
|
-
|
|
63
|
-
|
|
40
|
+
<dd>
|
|
41
|
+
The payload object.
|
|
42
|
+
|
|
43
|
+
<dl class="message-properties">
|
|
44
|
+
<dt class="optional">
|
|
45
|
+
payload.host <span class="property-type">string</span>
|
|
46
|
+
</dt>
|
|
47
|
+
<dd>
|
|
48
|
+
The host of the Ollama instance.<br />
|
|
49
|
+
<em>Alternative to the <code>Server</code> field.</em>
|
|
50
|
+
</dd>
|
|
51
|
+
|
|
52
|
+
<dt>
|
|
53
|
+
payload.model <span class="property-type">string</span>
|
|
54
|
+
</dt>
|
|
55
|
+
<p>The name of the model to create.</p>
|
|
56
|
+
|
|
57
|
+
<dt class="optional">
|
|
58
|
+
payload.path <span class="property-type">string</span>
|
|
59
|
+
</dt>
|
|
60
|
+
<p>The path to the Modelfile to use for creating the model.</p>
|
|
61
|
+
|
|
62
|
+
<dt class="optional">
|
|
63
|
+
payload.modelfile <span class="property-type">string</span>
|
|
64
|
+
</dt>
|
|
65
|
+
<p>The content of the Modelfile to create.</p>
|
|
66
|
+
|
|
67
|
+
<dt class="optional">
|
|
68
|
+
payload.stream <span class="property-type">boolean</span>
|
|
69
|
+
</dt>
|
|
70
|
+
<p>When true an AsyncGenerator is returned.</p>
|
|
71
|
+
</dl>
|
|
72
|
+
</dd>
|
|
64
73
|
</dl>
|
|
65
74
|
|
|
66
75
|
<h3>Outputs</h3>
|
|
@@ -72,26 +81,5 @@
|
|
|
72
81
|
<span class="property-type">string</span>
|
|
73
82
|
</dt>
|
|
74
83
|
<dd>The progress status.</dd>
|
|
75
|
-
|
|
76
|
-
<dt>
|
|
77
|
-
payload.digest
|
|
78
|
-
|
|
79
|
-
<span class="property-type">string</span>
|
|
80
|
-
</dt>
|
|
81
|
-
<dd>The progress digest.</dd>
|
|
82
|
-
|
|
83
|
-
<dt>
|
|
84
|
-
payload.total
|
|
85
|
-
|
|
86
|
-
<span class="property-type">number</span>
|
|
87
|
-
</dt>
|
|
88
|
-
<dd>The total progress.</dd>
|
|
89
|
-
|
|
90
|
-
<dt>
|
|
91
|
-
payload.completed
|
|
92
|
-
|
|
93
|
-
<span class="property-type">number</span>
|
|
94
|
-
</dt>
|
|
95
|
-
<dd>The completed progress.</dd>
|
|
96
84
|
</dl>
|
|
97
85
|
</script>
|
package/nodes/ollama-create.js
CHANGED
|
@@ -6,14 +6,29 @@ module.exports = function( RED ) {
|
|
|
6
6
|
const node = this
|
|
7
7
|
|
|
8
8
|
node.on( 'input', async function( msg ) {
|
|
9
|
-
const {
|
|
9
|
+
const {
|
|
10
|
+
host: payloadHost,
|
|
11
|
+
model,
|
|
12
|
+
modelfile,
|
|
13
|
+
stream,
|
|
14
|
+
path
|
|
15
|
+
} = msg.payload
|
|
16
|
+
|
|
17
|
+
const server = RED.nodes.getNode( config.server )
|
|
18
|
+
|
|
19
|
+
const host = ( server ) ? server.host + ':' + server.port : payloadHost
|
|
10
20
|
|
|
11
21
|
const ollama = new Ollama( { host } )
|
|
12
22
|
|
|
13
|
-
const response = await ollama.create( {
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
23
|
+
const response = await ollama.create( {
|
|
24
|
+
model,
|
|
25
|
+
modelfile,
|
|
26
|
+
stream,
|
|
27
|
+
path
|
|
28
|
+
} )
|
|
29
|
+
.catch( error => {
|
|
30
|
+
node.error( error )
|
|
31
|
+
} )
|
|
17
32
|
|
|
18
33
|
msg.payload = response
|
|
19
34
|
node.send( msg )
|
package/nodes/ollama-delete.html
CHANGED
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
category: 'AI',
|
|
4
4
|
color: 'rgb(186 230 253)',
|
|
5
5
|
defaults: {
|
|
6
|
-
name: { value: '' },
|
|
6
|
+
name: { value: 'Ollama Delete' },
|
|
7
|
+
server: { value: '', type: 'ollama-config-server', required: false }
|
|
7
8
|
},
|
|
8
9
|
inputs: 1,
|
|
9
10
|
outputs: 1,
|
|
@@ -19,6 +20,11 @@
|
|
|
19
20
|
<label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
|
|
20
21
|
<input type="text" id="node-input-name" placeholder="Name" />
|
|
21
22
|
</div>
|
|
23
|
+
|
|
24
|
+
<div class="form-row">
|
|
25
|
+
<label for="node-input-server"><i class="fa fa-server"></i> Server</label>
|
|
26
|
+
<input type="text" id="node-input-server" />
|
|
27
|
+
</div>
|
|
22
28
|
</script>
|
|
23
29
|
|
|
24
30
|
<script type="text/html" data-help-name="ollama-delete">
|
|
@@ -27,19 +33,28 @@
|
|
|
27
33
|
<h3>Inputs</h3>
|
|
28
34
|
|
|
29
35
|
<dl class="message-properties">
|
|
30
|
-
<dt>
|
|
31
|
-
payload
|
|
32
|
-
|
|
33
|
-
<span class="property-type">string</span>
|
|
36
|
+
<dt class="optional">
|
|
37
|
+
payload <span class="property-type">object</span>
|
|
34
38
|
</dt>
|
|
35
|
-
<dd>The host of the Ollama instance.</dd>
|
|
36
39
|
|
|
37
|
-
<
|
|
38
|
-
payload.
|
|
40
|
+
<dd>
|
|
41
|
+
The payload object.
|
|
39
42
|
|
|
40
|
-
<
|
|
41
|
-
|
|
42
|
-
|
|
43
|
+
<dl class="message-properties">
|
|
44
|
+
<dt class="optional">
|
|
45
|
+
payload.host <span class="property-type">string</span>
|
|
46
|
+
</dt>
|
|
47
|
+
<dd>
|
|
48
|
+
The host of the Ollama instance.<br />
|
|
49
|
+
<em>Alternative to the <code>Server</code> field.</em>
|
|
50
|
+
</dd>
|
|
51
|
+
|
|
52
|
+
<dt>
|
|
53
|
+
payload.model <span class="property-type">string</span>
|
|
54
|
+
</dt>
|
|
55
|
+
<dd>The name of the model to delete.</dd>
|
|
56
|
+
</dl>
|
|
57
|
+
</dd>
|
|
43
58
|
</dl>
|
|
44
59
|
|
|
45
60
|
<h3>Outputs</h3>
|
package/nodes/ollama-delete.js
CHANGED
|
@@ -6,7 +6,14 @@ module.exports = function( RED ) {
|
|
|
6
6
|
const node = this
|
|
7
7
|
|
|
8
8
|
node.on( 'input', async function( msg ) {
|
|
9
|
-
const {
|
|
9
|
+
const {
|
|
10
|
+
host: payloadHost,
|
|
11
|
+
model
|
|
12
|
+
} = msg.payload
|
|
13
|
+
|
|
14
|
+
const server = RED.nodes.getNode( config.server )
|
|
15
|
+
|
|
16
|
+
const host = ( server ) ? server.host + ':' + server.port : payloadHost
|
|
10
17
|
|
|
11
18
|
const ollama = new Ollama( { host } )
|
|
12
19
|
|
|
@@ -0,0 +1,238 @@
|
|
|
1
|
+
<script type="text/javascript">
|
|
2
|
+
RED.nodes.registerType( 'ollama-embed', {
|
|
3
|
+
category: 'AI',
|
|
4
|
+
color: 'rgb(186 230 253)',
|
|
5
|
+
defaults: {
|
|
6
|
+
name: { value: 'Ollama Embed' },
|
|
7
|
+
server: { value: '', type: 'ollama-config-server', required: false }
|
|
8
|
+
},
|
|
9
|
+
inputs: 1,
|
|
10
|
+
outputs: 1,
|
|
11
|
+
icon: 'ollama.png',
|
|
12
|
+
label: function() {
|
|
13
|
+
return this.name || 'ollama-embed';
|
|
14
|
+
},
|
|
15
|
+
} )
|
|
16
|
+
</script>
|
|
17
|
+
|
|
18
|
+
<script type="text/html" data-template-name="ollama-embed">
|
|
19
|
+
<div class="form-row">
|
|
20
|
+
<label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
|
|
21
|
+
<input type="text" id="node-input-name" placeholder="Name" />
|
|
22
|
+
</div>
|
|
23
|
+
|
|
24
|
+
<div class="form-row">
|
|
25
|
+
<label for="node-input-server"><i class="fa fa-server"></i> Server</label>
|
|
26
|
+
<input type="text" id="node-input-server" />
|
|
27
|
+
</div>
|
|
28
|
+
</script>
|
|
29
|
+
|
|
30
|
+
<script type="text/html" data-help-name="ollama-embed">
|
|
31
|
+
<p>Generate embeddings from a model</p>
|
|
32
|
+
|
|
33
|
+
<h3>Inputs</h3>
|
|
34
|
+
|
|
35
|
+
<dl class="message-properties">
|
|
36
|
+
<dt class="optional">
|
|
37
|
+
payload <span class="property-type">object</span>
|
|
38
|
+
</dt>
|
|
39
|
+
|
|
40
|
+
<dd>
|
|
41
|
+
The payload object.
|
|
42
|
+
|
|
43
|
+
<dl class="message-properties">
|
|
44
|
+
<dt class="optional">
|
|
45
|
+
payload.host <span class="property-type">string</span>
|
|
46
|
+
</dt>
|
|
47
|
+
<dd>
|
|
48
|
+
The host of the Ollama instance.<br />
|
|
49
|
+
<em>Alternative to the <code>Server</code> field.</em>
|
|
50
|
+
</dd>
|
|
51
|
+
|
|
52
|
+
<dt>
|
|
53
|
+
payload.model <span class="property-type">string</span>
|
|
54
|
+
</dt>
|
|
55
|
+
<dd>The name of the model used to generate the embeddings.</dd>
|
|
56
|
+
|
|
57
|
+
<dt>
|
|
58
|
+
payload.input <span class="property-type">string | array[string]</span>
|
|
59
|
+
</dt>
|
|
60
|
+
<dd>The input used to generate the embeddings.</dd>
|
|
61
|
+
|
|
62
|
+
<dt class="optional">
|
|
63
|
+
payload.truncate <span class="property-type">boolean</span>
|
|
64
|
+
</dt>
|
|
65
|
+
<dd>Truncate the input to fit the maximum context length supported by the model.</dd>
|
|
66
|
+
|
|
67
|
+
<dt class="optional">
|
|
68
|
+
payload.keep_alive <span class="property-type">string | number</span>
|
|
69
|
+
</dt>
|
|
70
|
+
<dd>How long to keep the model loaded.</dd>
|
|
71
|
+
|
|
72
|
+
<dt class="optional">
|
|
73
|
+
payload.options <span class="property-type">object</span>
|
|
74
|
+
</dt>
|
|
75
|
+
<dd>
|
|
76
|
+
Options to configure the runtime.<br />
|
|
77
|
+
|
|
78
|
+
<h4>Options</h4>
|
|
79
|
+
<dl class="message-properties">
|
|
80
|
+
<dt>
|
|
81
|
+
numa <span class="property-type">boolean</span>
|
|
82
|
+
</dt>
|
|
83
|
+
<dd>Use NUMA for the runtime.</dd>
|
|
84
|
+
|
|
85
|
+
<dt>
|
|
86
|
+
num_ctx <span class="property-type">number</span>
|
|
87
|
+
</dt>
|
|
88
|
+
<dd>The number of contexts to use.</dd>
|
|
89
|
+
|
|
90
|
+
<dt>
|
|
91
|
+
num_batch <span class="property-type">number</span>
|
|
92
|
+
</dt>
|
|
93
|
+
<dd>The number of batches to use.</dd>
|
|
94
|
+
|
|
95
|
+
<dt>
|
|
96
|
+
main_gpu <span class="property-type">number</span>
|
|
97
|
+
</dt>
|
|
98
|
+
<dd>The main GPU to use.</dd>
|
|
99
|
+
|
|
100
|
+
<dt>
|
|
101
|
+
low_vram <span class="property-type">boolean</span>
|
|
102
|
+
</dt>
|
|
103
|
+
<dd>Use low VRAM.</dd>
|
|
104
|
+
|
|
105
|
+
<dt>
|
|
106
|
+
f16_kv <span class="property-type">boolean</span>
|
|
107
|
+
</dt>
|
|
108
|
+
<dd>Use f16 key-value.</dd>
|
|
109
|
+
|
|
110
|
+
<dt>
|
|
111
|
+
logits_all <span class="property-type">boolean</span>
|
|
112
|
+
</dt>
|
|
113
|
+
<dd>Return all logits.</dd>
|
|
114
|
+
|
|
115
|
+
<dt>
|
|
116
|
+
vocab_only <span class="property-type">boolean</span>
|
|
117
|
+
</dt>
|
|
118
|
+
<dd>Return only the vocabulary.</dd>
|
|
119
|
+
|
|
120
|
+
<dt>
|
|
121
|
+
use_nmap <span class="property-type">boolean</span>
|
|
122
|
+
</dt>
|
|
123
|
+
<dd>Use NMAP.</dd>
|
|
124
|
+
|
|
125
|
+
<dt>
|
|
126
|
+
use_mlock <span class="property-type">boolean</span>
|
|
127
|
+
</dt>
|
|
128
|
+
<dd>Use mlock.</dd>
|
|
129
|
+
|
|
130
|
+
<dt>
|
|
131
|
+
embedding_only <span class="property-type">boolean</span>
|
|
132
|
+
</dt>
|
|
133
|
+
<dd>Return only the embeddings.</dd>
|
|
134
|
+
|
|
135
|
+
<dt>
|
|
136
|
+
num_thread <span class="property-type">number</span>
|
|
137
|
+
</dt>
|
|
138
|
+
<dd>The number of threads to use.</dd>
|
|
139
|
+
|
|
140
|
+
<dt>
|
|
141
|
+
num_keep <span class="property-type">number</span>
|
|
142
|
+
</dt>
|
|
143
|
+
<dd>The number of models to keep.</dd>
|
|
144
|
+
|
|
145
|
+
<dt>
|
|
146
|
+
seed <span class="property-type">number</span>
|
|
147
|
+
</dt>
|
|
148
|
+
<dd>The seed to use.</dd>
|
|
149
|
+
|
|
150
|
+
<dt>
|
|
151
|
+
num_predict <span class="property-type">number</span>
|
|
152
|
+
</dt>
|
|
153
|
+
<dd>The number of predictions to use.</dd>
|
|
154
|
+
|
|
155
|
+
<dt>
|
|
156
|
+
top_k <span class="property-type">number</span>
|
|
157
|
+
</dt>
|
|
158
|
+
<dd>The top k to use.</dd>
|
|
159
|
+
|
|
160
|
+
<dt>
|
|
161
|
+
top_p <span class="property-type">number</span>
|
|
162
|
+
</dt>
|
|
163
|
+
<dd>The top p to use.</dd>
|
|
164
|
+
|
|
165
|
+
<dt>
|
|
166
|
+
typical_p <span class="property-type">number</span>
|
|
167
|
+
</dt>
|
|
168
|
+
<dd>The typical p to use.</dd>
|
|
169
|
+
|
|
170
|
+
<dt>
|
|
171
|
+
repeat_last_n <span class="property-type">number</span>
|
|
172
|
+
</dt>
|
|
173
|
+
<dd>The repeat last n to use.</dd>
|
|
174
|
+
|
|
175
|
+
<dt>
|
|
176
|
+
temperature <span class="property-type">number</span>
|
|
177
|
+
</dt>
|
|
178
|
+
<dd>The temperature to use.</dd>
|
|
179
|
+
|
|
180
|
+
<dt>
|
|
181
|
+
repeat_pentalty <span class="property-type">number</span>
|
|
182
|
+
</dt>
|
|
183
|
+
<dd>The repeat pentalty to use.</dd>
|
|
184
|
+
|
|
185
|
+
<dt>
|
|
186
|
+
presence_penalty <span class="property-type">number</span>
|
|
187
|
+
</dt>
|
|
188
|
+
<dd>The presence penalty to use.</dd>
|
|
189
|
+
|
|
190
|
+
<dt>
|
|
191
|
+
frequency_penalty <span class="property-type">number</span>
|
|
192
|
+
</dt>
|
|
193
|
+
<dd>The frequency penalty to use.</dd>
|
|
194
|
+
|
|
195
|
+
<dt>
|
|
196
|
+
mirostat <span class="property-type">boolean</span>
|
|
197
|
+
</dt>
|
|
198
|
+
<dd>Use mirostat.</dd>
|
|
199
|
+
|
|
200
|
+
<dt>
|
|
201
|
+
mirostat_tau <span class="property-type">number</span>
|
|
202
|
+
</dt>
|
|
203
|
+
|
|
204
|
+
<dt>
|
|
205
|
+
mirostat_eta <span class="property-type">number</span>
|
|
206
|
+
</dt>
|
|
207
|
+
|
|
208
|
+
<dt>
|
|
209
|
+
penalize_newline <span class="property-type">boolean</span>
|
|
210
|
+
</dt>
|
|
211
|
+
<dd>Penalize newline.</dd>
|
|
212
|
+
|
|
213
|
+
<dt>
|
|
214
|
+
stop <span class="property-type">array</span>
|
|
215
|
+
</dt>
|
|
216
|
+
<dd>The stop to use.</dd>
|
|
217
|
+
</dl>
|
|
218
|
+
</dd>
|
|
219
|
+
</dl>
|
|
220
|
+
</dd>
|
|
221
|
+
</dl>
|
|
222
|
+
|
|
223
|
+
<h3>Outputs</h3>
|
|
224
|
+
|
|
225
|
+
<dl class="message-properties">
|
|
226
|
+
<dt>
|
|
227
|
+
payload.model
|
|
228
|
+
</dt>
|
|
229
|
+
<dd>Name of the model.</dd>
|
|
230
|
+
|
|
231
|
+
<dt>
|
|
232
|
+
payload.embeddings
|
|
233
|
+
|
|
234
|
+
<span class="property-type">array</span>
|
|
235
|
+
</dt>
|
|
236
|
+
<dd>The embeddings generated from the model.</dd>
|
|
237
|
+
</dl>
|
|
238
|
+
</script>
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
module.exports = function( RED ) {
|
|
2
|
+
const { Ollama } = require( 'ollama' )
|
|
3
|
+
|
|
4
|
+
function OllamaEmbedNode( config ) {
|
|
5
|
+
RED.nodes.createNode( this, config )
|
|
6
|
+
const node = this
|
|
7
|
+
|
|
8
|
+
node.on( 'input', async function( msg ) {
|
|
9
|
+
const {
|
|
10
|
+
host: payloadHost,
|
|
11
|
+
model,
|
|
12
|
+
input,
|
|
13
|
+
truncate,
|
|
14
|
+
keep_alive,
|
|
15
|
+
options
|
|
16
|
+
} = msg.payload
|
|
17
|
+
|
|
18
|
+
const server = RED.nodes.getNode( config.server )
|
|
19
|
+
|
|
20
|
+
const host = ( server ) ? server.host + ':' + server.port : payloadHost
|
|
21
|
+
|
|
22
|
+
const ollama = new Ollama( { host } )
|
|
23
|
+
|
|
24
|
+
const response = await ollama.embed( {
|
|
25
|
+
model,
|
|
26
|
+
input,
|
|
27
|
+
truncate,
|
|
28
|
+
keep_alive,
|
|
29
|
+
options
|
|
30
|
+
} )
|
|
31
|
+
.catch( error => {
|
|
32
|
+
node.error( error )
|
|
33
|
+
} )
|
|
34
|
+
|
|
35
|
+
msg.payload = response
|
|
36
|
+
node.send( msg )
|
|
37
|
+
} )
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
RED.nodes.registerType( 'ollama-embed', OllamaEmbedNode )
|
|
41
|
+
}
|