Functions.svelte 10.8 KB
Newer Older
Timothy J. Baek's avatar
Timothy J. Baek committed
1
2
3
4
5
<script lang="ts">
	import { toast } from 'svelte-sonner';
	import fileSaver from 'file-saver';
	const { saveAs } = fileSaver;

Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
6
	import { WEBUI_NAME, functions, models } from '$lib/stores';
Timothy J. Baek's avatar
Timothy J. Baek committed
7
8
9
10
11
	import { onMount, getContext } from 'svelte';
	import { createNewPrompt, deletePromptByCommand, getPrompts } from '$lib/apis/prompts';

	import { goto } from '$app/navigation';
	import {
Timothy J. Baek's avatar
Timothy J. Baek committed
12
13
14
15
16
17
18
		createNewFunction,
		deleteFunctionById,
		exportFunctions,
		getFunctionById,
		getFunctions
	} from '$lib/apis/functions';

Timothy J. Baek's avatar
Timothy J. Baek committed
19
20
21
	import ArrowDownTray from '../icons/ArrowDownTray.svelte';
	import Tooltip from '../common/Tooltip.svelte';
	import ConfirmDialog from '../common/ConfirmDialog.svelte';
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
22
	import { getModels } from '$lib/apis';
Timothy J. Baek's avatar
Timothy J. Baek committed
23
24
	import FunctionMenu from './Functions/FunctionMenu.svelte';
	import EllipsisHorizontal from '../icons/EllipsisHorizontal.svelte';
Timothy J. Baek's avatar
Timothy J. Baek committed
25
26
27

	const i18n = getContext('i18n');

Timothy J. Baek's avatar
Timothy J. Baek committed
28
	let functionsImportInputElement: HTMLInputElement;
Timothy J. Baek's avatar
Timothy J. Baek committed
29
30
31
32
	let importFiles;

	let showConfirm = false;
	let query = '';
Timothy J. Baek's avatar
Timothy J. Baek committed
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

	const shareHandler = async (tool) => {
		console.log(tool);
	};

	const cloneHandler = async (func) => {
		const _function = await getFunctionById(localStorage.token, func.id).catch((error) => {
			toast.error(error);
			return null;
		});

		if (_function) {
			sessionStorage.function = JSON.stringify({
				..._function,
				id: `${_function.id}_clone`,
				name: `${_function.name} (Clone)`
			});
			goto('/workspace/functions/create');
		}
	};

	const exportHandler = async (func) => {
		const _function = await getFunctionById(localStorage.token, func.id).catch((error) => {
			toast.error(error);
			return null;
		});

		if (_function) {
			let blob = new Blob([JSON.stringify([_function])], {
				type: 'application/json'
			});
			saveAs(blob, `function-${_function.id}-export-${Date.now()}.json`);
		}
	};

	const deleteHandler = async (func) => {
		const res = await deleteFunctionById(localStorage.token, func.id).catch((error) => {
			toast.error(error);
			return null;
		});

		if (res) {
			toast.success('Function deleted successfully');

			functions.set(await getFunctions(localStorage.token));
			models.set(await getModels(localStorage.token));
		}
	};
Timothy J. Baek's avatar
Timothy J. Baek committed
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
</script>

<svelte:head>
	<title>
		{$i18n.t('Functions')} | {$WEBUI_NAME}
	</title>
</svelte:head>

<div class="mb-3 flex justify-between items-center">
	<div class=" text-lg font-semibold self-center">{$i18n.t('Functions')}</div>
</div>

<div class=" flex w-full space-x-2">
	<div class="flex flex-1">
		<div class=" self-center ml-1 mr-3">
			<svg
				xmlns="http://www.w3.org/2000/svg"
				viewBox="0 0 20 20"
				fill="currentColor"
				class="w-4 h-4"
			>
				<path
					fill-rule="evenodd"
					d="M9 3.5a5.5 5.5 0 100 11 5.5 5.5 0 000-11zM2 9a7 7 0 1112.452 4.391l3.328 3.329a.75.75 0 11-1.06 1.06l-3.329-3.328A7 7 0 012 9z"
					clip-rule="evenodd"
				/>
			</svg>
		</div>
		<input
			class=" w-full text-sm pr-4 py-1 rounded-r-xl outline-none bg-transparent"
			bind:value={query}
			placeholder={$i18n.t('Search Functions')}
		/>
	</div>

	<div>
		<a
			class=" px-2 py-2 rounded-xl border border-gray-200 dark:border-gray-600 dark:border-0 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700 transition font-medium text-sm flex items-center space-x-1"
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
119
			href="/workspace/functions/create"
Timothy J. Baek's avatar
Timothy J. Baek committed
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
		>
			<svg
				xmlns="http://www.w3.org/2000/svg"
				viewBox="0 0 16 16"
				fill="currentColor"
				class="w-4 h-4"
			>
				<path
					d="M8.75 3.75a.75.75 0 0 0-1.5 0v3.5h-3.5a.75.75 0 0 0 0 1.5h3.5v3.5a.75.75 0 0 0 1.5 0v-3.5h3.5a.75.75 0 0 0 0-1.5h-3.5v-3.5Z"
				/>
			</svg>
		</a>
	</div>
</div>
<hr class=" dark:border-gray-850 my-2.5" />

<div class="my-3 mb-5">
Timothy J. Baek's avatar
Timothy J. Baek committed
137
	{#each $functions.filter((f) => query === '' || f.name
Timothy J. Baek's avatar
Timothy J. Baek committed
138
				.toLowerCase()
Timothy J. Baek's avatar
Timothy J. Baek committed
139
				.includes(query.toLowerCase()) || f.id.toLowerCase().includes(query.toLowerCase())) as func}
Timothy J. Baek's avatar
Timothy J. Baek committed
140
		<div
Timothy J. Baek's avatar
Timothy J. Baek committed
141
142
			class=" flex space-x-4 cursor-pointer w-full px-3 py-2 dark:hover:bg-white/5 hover:bg-black/5 rounded-xl"
		>
Timothy J. Baek's avatar
Timothy J. Baek committed
143
144
145
146
147
			<a
				class=" flex flex-1 space-x-3.5 cursor-pointer w-full"
				href={`/workspace/functions/edit?id=${encodeURIComponent(func.id)}`}
			>
				<div class="flex items-center text-left">
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
148
					<div class=" flex-1 self-center pl-1">
Timothy J. Baek's avatar
Timothy J. Baek committed
149
						<div class=" font-semibold flex items-center gap-1.5">
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
150
151
152
153
154
155
							<div
								class=" text-xs font-black px-1 rounded uppercase line-clamp-1 bg-gray-500/20 text-gray-700 dark:text-gray-200"
							>
								{func.type}
							</div>

Timothy J. Baek's avatar
Timothy J. Baek committed
156
							<div class=" line-clamp-1">
Timothy J. Baek's avatar
Timothy J. Baek committed
157
								{func.name}
Timothy J. Baek's avatar
Timothy J. Baek committed
158
159
							</div>
						</div>
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
160
161

						<div class="flex gap-1.5 px-1">
Timothy J. Baek's avatar
Timothy J. Baek committed
162
							<div class=" text-gray-500 text-xs font-medium flex-shrink-0">{func.id}</div>
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
163
164
165
166

							<div class=" text-xs overflow-hidden text-ellipsis line-clamp-1">
								{func.meta.description}
							</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
167
168
						</div>
					</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
169
170
				</div>
			</a>
Timothy J. Baek's avatar
Timothy J. Baek committed
171
			<div class="flex flex-row gap-0.5 self-center">
Timothy J. Baek's avatar
Timothy J. Baek committed
172
173
174
175
				<Tooltip content="Edit">
					<a
						class="self-center w-fit text-sm px-2 py-2 dark:text-gray-300 dark:hover:text-white hover:bg-black/5 dark:hover:bg-white/5 rounded-xl"
						type="button"
Timothy J. Baek's avatar
Timothy J. Baek committed
176
						href={`/workspace/functions/edit?id=${encodeURIComponent(func.id)}`}
Timothy J. Baek's avatar
Timothy J. Baek committed
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
					>
						<svg
							xmlns="http://www.w3.org/2000/svg"
							fill="none"
							viewBox="0 0 24 24"
							stroke-width="1.5"
							stroke="currentColor"
							class="w-4 h-4"
						>
							<path
								stroke-linecap="round"
								stroke-linejoin="round"
								d="M16.862 4.487l1.687-1.688a1.875 1.875 0 112.652 2.652L6.832 19.82a4.5 4.5 0 01-1.897 1.13l-2.685.8.8-2.685a4.5 4.5 0 011.13-1.897L16.863 4.487zm0 0L19.5 7.125"
							/>
						</svg>
					</a>
				</Tooltip>

Timothy J. Baek's avatar
Timothy J. Baek committed
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
				<FunctionMenu
					shareHandler={() => {
						shareHandler(func);
					}}
					cloneHandler={() => {
						cloneHandler(func);
					}}
					exportHandler={() => {
						exportHandler(func);
					}}
					deleteHandler={async () => {
						deleteHandler(func);
					}}
					onClose={() => {}}
				>
Timothy J. Baek's avatar
Timothy J. Baek committed
210
					<button
Timothy J. Baek's avatar
Timothy J. Baek committed
211
						class="self-center w-fit text-sm p-1.5 dark:text-gray-300 dark:hover:text-white hover:bg-black/5 dark:hover:bg-white/5 rounded-xl"
Timothy J. Baek's avatar
Timothy J. Baek committed
212
213
						type="button"
					>
Timothy J. Baek's avatar
Timothy J. Baek committed
214
						<EllipsisHorizontal className="size-5" />
Timothy J. Baek's avatar
Timothy J. Baek committed
215
					</button>
Timothy J. Baek's avatar
Timothy J. Baek committed
216
				</FunctionMenu>
Timothy J. Baek's avatar
Timothy J. Baek committed
217
			</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
218
		</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
219
220
221
222
223
224
225
226
227
228
229
230
231
	{/each}
</div>

<!-- <div class=" text-gray-500 text-xs mt-1 mb-2">
	ⓘ {$i18n.t(
		'Admins have access to all tools at all times; users need tools assigned per model in the workspace.'
	)}
</div> -->

<div class=" flex justify-end w-full mb-2">
	<div class="flex space-x-2">
		<input
			id="documents-import-input"
Timothy J. Baek's avatar
Timothy J. Baek committed
232
			bind:this={functionsImportInputElement}
Timothy J. Baek's avatar
Timothy J. Baek committed
233
234
235
236
237
238
239
240
241
242
243
244
245
			bind:files={importFiles}
			type="file"
			accept=".json"
			hidden
			on:change={() => {
				console.log(importFiles);
				showConfirm = true;
			}}
		/>

		<button
			class="flex text-xs items-center space-x-1 px-3 py-1.5 rounded-xl bg-gray-50 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700 dark:text-gray-200 transition"
			on:click={() => {
Timothy J. Baek's avatar
Timothy J. Baek committed
246
				functionsImportInputElement.click();
Timothy J. Baek's avatar
Timothy J. Baek committed
247
248
			}}
		>
Timothy J. Baek's avatar
Timothy J. Baek committed
249
			<div class=" self-center mr-2 font-medium line-clamp-1">{$i18n.t('Import Functions')}</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269

			<div class=" self-center">
				<svg
					xmlns="http://www.w3.org/2000/svg"
					viewBox="0 0 16 16"
					fill="currentColor"
					class="w-4 h-4"
				>
					<path
						fill-rule="evenodd"
						d="M4 2a1.5 1.5 0 0 0-1.5 1.5v9A1.5 1.5 0 0 0 4 14h8a1.5 1.5 0 0 0 1.5-1.5V6.621a1.5 1.5 0 0 0-.44-1.06L9.94 2.439A1.5 1.5 0 0 0 8.878 2H4Zm4 9.5a.75.75 0 0 1-.75-.75V8.06l-.72.72a.75.75 0 0 1-1.06-1.06l2-2a.75.75 0 0 1 1.06 0l2 2a.75.75 0 1 1-1.06 1.06l-.72-.72v2.69a.75.75 0 0 1-.75.75Z"
						clip-rule="evenodd"
					/>
				</svg>
			</div>
		</button>

		<button
			class="flex text-xs items-center space-x-1 px-3 py-1.5 rounded-xl bg-gray-50 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700 dark:text-gray-200 transition"
			on:click={async () => {
Timothy J. Baek's avatar
Timothy J. Baek committed
270
				const _functions = await exportFunctions(localStorage.token).catch((error) => {
Timothy J. Baek's avatar
Timothy J. Baek committed
271
272
273
274
					toast.error(error);
					return null;
				});

Timothy J. Baek's avatar
Timothy J. Baek committed
275
276
				if (_functions) {
					let blob = new Blob([JSON.stringify(_functions)], {
Timothy J. Baek's avatar
Timothy J. Baek committed
277
278
						type: 'application/json'
					});
Timothy J. Baek's avatar
Timothy J. Baek committed
279
					saveAs(blob, `functions-export-${Date.now()}.json`);
Timothy J. Baek's avatar
Timothy J. Baek committed
280
281
282
				}
			}}
		>
Timothy J. Baek's avatar
Timothy J. Baek committed
283
			<div class=" self-center mr-2 font-medium line-clamp-1">{$i18n.t('Export Functions')}</div>
Timothy J. Baek's avatar
Timothy J. Baek committed
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302

			<div class=" self-center">
				<svg
					xmlns="http://www.w3.org/2000/svg"
					viewBox="0 0 16 16"
					fill="currentColor"
					class="w-4 h-4"
				>
					<path
						fill-rule="evenodd"
						d="M4 2a1.5 1.5 0 0 0-1.5 1.5v9A1.5 1.5 0 0 0 4 14h8a1.5 1.5 0 0 0 1.5-1.5V6.621a1.5 1.5 0 0 0-.44-1.06L9.94 2.439A1.5 1.5 0 0 0 8.878 2H4Zm4 3.5a.75.75 0 0 1 .75.75v2.69l.72-.72a.75.75 0 1 1 1.06 1.06l-2 2a.75.75 0 0 1-1.06 0l-2-2a.75.75 0 0 1 1.06-1.06l.72.72V6.25A.75.75 0 0 1 8 5.5Z"
						clip-rule="evenodd"
					/>
				</svg>
			</div>
		</button>
	</div>
</div>

Timothy J. Baek's avatar
Timothy J. Baek committed
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
<div class=" my-16">
	<div class=" text-lg font-semibold mb-3 line-clamp-1">
		{$i18n.t('Made by OpenWebUI Community')}
	</div>

	<a
		class=" flex space-x-4 cursor-pointer w-full mb-2 px-3 py-2"
		href="https://openwebui.com/"
		target="_blank"
	>
		<div class=" self-center w-10 flex-shrink-0">
			<div
				class="w-full h-10 flex justify-center rounded-full bg-transparent dark:bg-gray-700 border border-dashed border-gray-200"
			>
				<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="currentColor" class="w-6">
					<path
						fill-rule="evenodd"
						d="M12 3.75a.75.75 0 01.75.75v6.75h6.75a.75.75 0 010 1.5h-6.75v6.75a.75.75 0 01-1.5 0v-6.75H4.5a.75.75 0 010-1.5h6.75V4.5a.75.75 0 01.75-.75z"
						clip-rule="evenodd"
					/>
				</svg>
			</div>
		</div>

		<div class=" self-center">
			<div class=" font-bold line-clamp-1">{$i18n.t('Discover a function')}</div>
			<div class=" text-sm line-clamp-1">
				{$i18n.t('Discover, download, and explore custom functions')}
			</div>
		</div>
	</a>
</div>

Timothy J. Baek's avatar
Timothy J. Baek committed
336
337
338
339
340
<ConfirmDialog
	bind:show={showConfirm}
	on:confirm={() => {
		const reader = new FileReader();
		reader.onload = async (event) => {
Timothy J. Baek's avatar
Timothy J. Baek committed
341
342
			const _functions = JSON.parse(event.target.result);
			console.log(_functions);
Timothy J. Baek's avatar
Timothy J. Baek committed
343

Timothy J. Baek's avatar
Timothy J. Baek committed
344
345
			for (const func of _functions) {
				const res = await createNewFunction(localStorage.token, func).catch((error) => {
Timothy J. Baek's avatar
Timothy J. Baek committed
346
347
348
349
350
					toast.error(error);
					return null;
				});
			}

Timothy J. Baek's avatar
Timothy J. Baek committed
351
			toast.success('Functions imported successfully');
Timothy J. Baek's avatar
Timothy J. Baek committed
352
			functions.set(await getFunctions(localStorage.token));
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
353
			models.set(await getModels(localStorage.token));
Timothy J. Baek's avatar
Timothy J. Baek committed
354
355
356
357
358
359
360
361
362
363
		};

		reader.readAsText(importFiles[0]);
	}}
>
	<div class="text-sm text-gray-500">
		<div class=" bg-yellow-500/20 text-yellow-700 dark:text-yellow-200 rounded-lg px-4 py-3">
			<div>Please carefully review the following warnings:</div>

			<ul class=" mt-1 list-disc pl-4 text-xs">
Timothy J. Baek's avatar
Timothy J. Baek committed
364
365
				<li>Functions allow arbitrary code execution.</li>
				<li>Do not install functions from sources you do not fully trust.</li>
Timothy J. Baek's avatar
Timothy J. Baek committed
366
367
368
369
370
371
372
373
374
375
			</ul>
		</div>

		<div class="my-3">
			I acknowledge that I have read and I understand the implications of my action. I am aware of
			the risks associated with executing arbitrary code and I have verified the trustworthiness of
			the source.
		</div>
	</div>
</ConfirmDialog>