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

feat: return file_name for model name

Browse source
This commit is contained in:
Timothy J. Baek 2023-12-24 00:40:59 -08:00
parent 232401a042
commit ee83791596
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
backend/apps/web/routers/utils.py
View file

@ -42,7 +42,7 @@ def parse_huggingface_url(hf_url):
return None return None
async def download_file_stream(url, file_path, chunk_size=1024 * 1024): async def download_file_stream(url, file_path, file_name, chunk_size=1024 * 1024):
done = False done = False
if os.path.exists(file_path): if os.path.exists(file_path):
@ -79,7 +79,7 @@ async def download_file_stream(url, file_path, chunk_size=1024 * 1024):
res = { res = {
"done": done, "done": done,
"blob": f"sha256:{hashed}", "blob": f"sha256:{hashed}",
"name": file.name, "name": file_name,
} }
os.remove(file_path) os.remove(file_path)
@ -93,14 +93,15 @@ async def download(
url: str, url: str,
): ):
# url = "https://huggingface.co/TheBloke/stablelm-zephyr-3b-GGUF/resolve/main/stablelm-zephyr-3b.Q2_K.gguf" # url = "https://huggingface.co/TheBloke/stablelm-zephyr-3b-GGUF/resolve/main/stablelm-zephyr-3b.Q2_K.gguf"
model_file = parse_huggingface_url(url) file_name = parse_huggingface_url(url)
if model_file: if file_name:
os.makedirs("./uploads", exist_ok=True) os.makedirs("./uploads", exist_ok=True)
file_path = os.path.join("./uploads", f"{model_file}") file_path = os.path.join("./uploads", f"{file_name}")
return StreamingResponse( return StreamingResponse(
download_file_stream(url, file_path), media_type="text/event-stream" download_file_stream(url, file_path, file_name),
media_type="text/event-stream",
) )
else: else:
return None return None

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

@ -276,6 +276,7 @@
const uploadModelHandler = async () => { const uploadModelHandler = async () => {
modelTransferring = true; modelTransferring = true;
uploadProgress = 0;
let uploaded = false; let uploaded = false;
let fileResponse = null; let fileResponse = null;
let name = ''; let name = '';
@ -422,6 +423,8 @@
modelFileUrl = ''; modelFileUrl = '';
modelInputFile = ''; modelInputFile = '';
modelTransferring = false; modelTransferring = false;
uploadProgress = null;
models.set(await getModels()); models.set(await getModels());
}; };