mirror of
https://github.com/open-webui/open-webui
synced 2025-06-26 18:26:48 +00:00
enh: alert private task model
This commit is contained in:
parent
30cc51ec82
commit
de69c08a1d
@ -2,9 +2,9 @@
|
|||||||
import { v4 as uuidv4 } from 'uuid';
|
import { v4 as uuidv4 } from 'uuid';
|
||||||
import { toast } from 'svelte-sonner';
|
import { toast } from 'svelte-sonner';
|
||||||
|
|
||||||
import { getBackendConfig, getTaskConfig, updateTaskConfig } from '$lib/apis';
|
import { getBackendConfig, getModels, getTaskConfig, updateTaskConfig } from '$lib/apis';
|
||||||
import { setDefaultPromptSuggestions } from '$lib/apis/configs';
|
import { setDefaultPromptSuggestions } from '$lib/apis/configs';
|
||||||
import { config, models, settings, user } from '$lib/stores';
|
import { config, settings, user } from '$lib/stores';
|
||||||
import { createEventDispatcher, onMount, getContext } from 'svelte';
|
import { createEventDispatcher, onMount, getContext } from 'svelte';
|
||||||
|
|
||||||
import { banners as _banners } from '$lib/stores';
|
import { banners as _banners } from '$lib/stores';
|
||||||
@ -15,6 +15,8 @@
|
|||||||
import Tooltip from '$lib/components/common/Tooltip.svelte';
|
import Tooltip from '$lib/components/common/Tooltip.svelte';
|
||||||
import Switch from '$lib/components/common/Switch.svelte';
|
import Switch from '$lib/components/common/Switch.svelte';
|
||||||
import Textarea from '$lib/components/common/Textarea.svelte';
|
import Textarea from '$lib/components/common/Textarea.svelte';
|
||||||
|
import Spinner from '$lib/components/common/Spinner.svelte';
|
||||||
|
import { getBaseModels } from '$lib/apis/models';
|
||||||
|
|
||||||
const dispatch = createEventDispatcher();
|
const dispatch = createEventDispatcher();
|
||||||
|
|
||||||
@ -49,6 +51,7 @@
|
|||||||
};
|
};
|
||||||
|
|
||||||
onMount(async () => {
|
onMount(async () => {
|
||||||
|
await init();
|
||||||
taskConfig = await getTaskConfig(localStorage.token);
|
taskConfig = await getTaskConfig(localStorage.token);
|
||||||
|
|
||||||
promptSuggestions = $config?.default_prompt_suggestions ?? [];
|
promptSuggestions = $config?.default_prompt_suggestions ?? [];
|
||||||
@ -58,9 +61,40 @@
|
|||||||
const updateBanners = async () => {
|
const updateBanners = async () => {
|
||||||
_banners.set(await setBanners(localStorage.token, banners));
|
_banners.set(await setBanners(localStorage.token, banners));
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let workspaceModels = null;
|
||||||
|
let baseModels = null;
|
||||||
|
|
||||||
|
let models = null;
|
||||||
|
|
||||||
|
const init = async () => {
|
||||||
|
workspaceModels = await getBaseModels(localStorage.token);
|
||||||
|
baseModels = await getModels(localStorage.token, null, true);
|
||||||
|
|
||||||
|
models = baseModels.map((m) => {
|
||||||
|
const workspaceModel = workspaceModels.find((wm) => wm.id === m.id);
|
||||||
|
|
||||||
|
if (workspaceModel) {
|
||||||
|
return {
|
||||||
|
...m,
|
||||||
|
...workspaceModel
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
return {
|
||||||
|
...m,
|
||||||
|
id: m.id,
|
||||||
|
name: m.name,
|
||||||
|
|
||||||
|
is_active: true
|
||||||
|
};
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('models', models);
|
||||||
|
};
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
{#if taskConfig}
|
{#if models !== null && taskConfig}
|
||||||
<form
|
<form
|
||||||
class="flex flex-col h-full justify-between space-y-3 text-sm"
|
class="flex flex-col h-full justify-between space-y-3 text-sm"
|
||||||
on:submit|preventDefault={() => {
|
on:submit|preventDefault={() => {
|
||||||
@ -105,9 +139,27 @@
|
|||||||
class="w-full rounded-lg py-2 px-4 text-sm bg-gray-50 dark:text-gray-300 dark:bg-gray-850 outline-hidden"
|
class="w-full rounded-lg py-2 px-4 text-sm bg-gray-50 dark:text-gray-300 dark:bg-gray-850 outline-hidden"
|
||||||
bind:value={taskConfig.TASK_MODEL}
|
bind:value={taskConfig.TASK_MODEL}
|
||||||
placeholder={$i18n.t('Select a model')}
|
placeholder={$i18n.t('Select a model')}
|
||||||
|
on:change={() => {
|
||||||
|
if (taskConfig.TASK_MODEL) {
|
||||||
|
const model = models.find((m) => m.id === taskConfig.TASK_MODEL);
|
||||||
|
if (model) {
|
||||||
|
if (model?.access_control !== null) {
|
||||||
|
toast.error(
|
||||||
|
$i18n.t(
|
||||||
|
'This model is not publicly available. Please select another model.'
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
taskConfig.TASK_MODEL = model.id;
|
||||||
|
} else {
|
||||||
|
taskConfig.TASK_MODEL = '';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}}
|
||||||
>
|
>
|
||||||
<option value="" selected>{$i18n.t('Current Model')}</option>
|
<option value="" selected>{$i18n.t('Current Model')}</option>
|
||||||
{#each $models.filter((m) => m.owned_by === 'ollama') as model}
|
{#each models.filter((m) => m.owned_by === 'ollama') as model}
|
||||||
<option value={model.id} class="bg-gray-100 dark:bg-gray-700">
|
<option value={model.id} class="bg-gray-100 dark:bg-gray-700">
|
||||||
{model.name}
|
{model.name}
|
||||||
</option>
|
</option>
|
||||||
@ -121,9 +173,27 @@
|
|||||||
class="w-full rounded-lg py-2 px-4 text-sm bg-gray-50 dark:text-gray-300 dark:bg-gray-850 outline-hidden"
|
class="w-full rounded-lg py-2 px-4 text-sm bg-gray-50 dark:text-gray-300 dark:bg-gray-850 outline-hidden"
|
||||||
bind:value={taskConfig.TASK_MODEL_EXTERNAL}
|
bind:value={taskConfig.TASK_MODEL_EXTERNAL}
|
||||||
placeholder={$i18n.t('Select a model')}
|
placeholder={$i18n.t('Select a model')}
|
||||||
|
on:change={() => {
|
||||||
|
if (taskConfig.TASK_MODEL_EXTERNAL) {
|
||||||
|
const model = models.find((m) => m.id === taskConfig.TASK_MODEL_EXTERNAL);
|
||||||
|
if (model) {
|
||||||
|
if (model?.access_control !== null) {
|
||||||
|
toast.error(
|
||||||
|
$i18n.t(
|
||||||
|
'This model is not publicly available. Please select another model.'
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
taskConfig.TASK_MODEL_EXTERNAL = model.id;
|
||||||
|
} else {
|
||||||
|
taskConfig.TASK_MODEL_EXTERNAL = '';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}}
|
||||||
>
|
>
|
||||||
<option value="" selected>{$i18n.t('Current Model')}</option>
|
<option value="" selected>{$i18n.t('Current Model')}</option>
|
||||||
{#each $models as model}
|
{#each models as model}
|
||||||
<option value={model.id} class="bg-gray-100 dark:bg-gray-700">
|
<option value={model.id} class="bg-gray-100 dark:bg-gray-700">
|
||||||
{model.name}
|
{model.name}
|
||||||
</option>
|
</option>
|
||||||
@ -480,4 +550,8 @@
|
|||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</form>
|
</form>
|
||||||
|
{:else}
|
||||||
|
<div class=" h-full w-full flex justify-center items-center">
|
||||||
|
<Spinner />
|
||||||
|
</div>
|
||||||
{/if}
|
{/if}
|
||||||
|
Loading…
Reference in New Issue
Block a user