tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

Merge pull request #465 from anuraagdjain/fix/sanitized-model

Browse source 
fix: sanitized model name
This commit is contained in:
Timothy Jaeryang Baek 2024-01-13 21:29:27 -08:00 committed by GitHub
commit bc3df5ac51
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/lib/components/chat/SettingsModal.svelte
View file

@ -314,8 +314,9 @@
}; };
const pullModelHandler = async () => { const pullModelHandler = async () => {
if (modelDownloadStatus[modelTag]) { const sanitizedModelTag = modelTag.trim();
toast.error(`Model '${modelTag}' is already in queue for downloading.`); if (modelDownloadStatus[sanitizedModelTag]) {
toast.error(`Model '${sanitizedModelTag}' is already in queue for downloading.`);
return; return;
} }
if (Object.keys(modelDownloadStatus).length === 3) { if (Object.keys(modelDownloadStatus).length === 3) {
@ -326,7 +327,7 @@
modelTransferring = true; modelTransferring = true;
modelDownloadQueue.push( modelDownloadQueue.push(
{ modelName: modelTag }, { modelName: sanitizedModelTag },
async (data: { modelName: string; success: boolean; error?: Error }) => { async (data: { modelName: string; success: boolean; error?: Error }) => {
const { modelName } = data; const { modelName } = data;
// Remove the downloaded model // Remove the downloaded model