File size: 2,353 Bytes
bc27e65
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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
81
82
83
84
85
86
87
88
89
90
91
92
<script>

	import { v4 as uuidv4 } from 'uuid';

	import { toast } from 'svelte-sonner';

	import { goto } from '$app/navigation';

	import { config, models, settings } from '$lib/stores';



	import { onMount, tick, getContext } from 'svelte';

	import { createNewModel, getModelById } from '$lib/apis/models';

	import { getModels } from '$lib/apis';



	import ModelEditor from '$lib/components/workspace/Models/ModelEditor.svelte';



	const i18n = getContext('i18n');



	const onSubmit = async (modelInfo) => {

		if ($models.find((m) => m.id === modelInfo.id)) {

			toast.error(

				`Error: A model with the ID '${modelInfo.id}' already exists. Please select a different ID to proceed.`

			);

			return;

		}



		if (modelInfo.id === '') {

			toast.error('Error: Model ID cannot be empty. Please enter a valid ID to proceed.');

			return;

		}



		if (modelInfo) {

			const res = await createNewModel(localStorage.token, {

				...modelInfo,

				meta: {

					...modelInfo.meta,

					profile_image_url: modelInfo.meta.profile_image_url ?? '/static/favicon.png',

					suggestion_prompts: modelInfo.meta.suggestion_prompts

						? modelInfo.meta.suggestion_prompts.filter((prompt) => prompt.content !== '')

						: null

				},

				params: { ...modelInfo.params }

			}).catch((error) => {

				toast.error(`${error}`);

				return null;

			});



			if (res) {

				await models.set(

					await getModels(

						localStorage.token,

						$config?.features?.enable_direct_connections && ($settings?.directConnections ?? null)

					)

				);

				toast.success($i18n.t('Model created successfully!'));

				await goto('/workspace/models');

			}

		}

	};



	let model = null;



	onMount(async () => {

		window.addEventListener('message', async (event) => {

			if (

				!['https://openwebui.com', 'https://www.openwebui.com', 'http://localhost:5173'].includes(

					event.origin

				)

			) {

				return;

			}



			let data = JSON.parse(event.data);



			if (data?.info) {

				data = data.info;

			}



			model = data;

		});



		if (window.opener ?? false) {

			window.opener.postMessage('loaded', '*');

		}



		if (sessionStorage.model) {

			model = JSON.parse(sessionStorage.model);

			sessionStorage.removeItem('model');

		}

	});

</script>

{#key model}
	<ModelEditor {model} {onSubmit} />
{/key}