diff --git a/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageAssistant.svelte b/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageAssistant.svelte index 79c43e5cbc..2cc13fcf1b 100644 --- a/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageAssistant.svelte +++ b/tools/server/webui/src/lib/components/app/chat/ChatMessages/ChatMessageAssistant.svelte @@ -20,6 +20,8 @@ import Label from '$lib/components/ui/label/label.svelte'; import { config } from '$lib/stores/settings.svelte'; import { isRouterMode } from '$lib/stores/server.svelte'; + import { modelsStore } from '$lib/stores/models.svelte'; + import { ServerModelStatus } from '$lib/enums'; import { AGENTIC_TAGS, REASONING_TAGS } from '$lib/constants/agentic'; interface Props { @@ -183,7 +185,20 @@ {#if displayedModel}
{#if isRouter} - + { + const status = modelsStore.getModelStatus(modelId); + + if (status !== ServerModelStatus.LOADED) { + await modelsStore.loadModel(modelId); + } + + onRegenerate(modelName); + return true; + }} + /> {:else} {/if}