29 - more improvements for revision

Signed-off-by: marijnvg-tng <[email protected]>
TNG · Nov 22, 2024 · 160f92a · 160f92a
1 parent 28cca0f
commit 160f92a
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 27 deletions.
diff --git a/WebUI/src/components/AddLLMDialog.vue b/WebUI/src/components/AddLLMDialog.vue
@@ -41,7 +41,9 @@ function fastGenerate(e: KeyboardEvent) {
       modelRequest.value += "\n";
     } else {
       e.preventDefault();
-      addModel()
+      if (modelRequest.value !== "") {
+        addModel()
+      }
     }
   }
 }
@@ -85,23 +87,13 @@ async function performDownload() {
 }
 
 async function urlExists(repo_id: string) {
-  const response = await fetch(`${globalSetup.apiHost}/api/checkURLExists`, {
-    method: "POST",
-    body: JSON.stringify(repo_id),
-    headers: {
-      "Content-Type": "application/json"
-    }})
+  const response = await fetch(`${globalSetup.apiHost}/api/checkURLExists?repo_id=${repo_id}`)
   const data = await response.json()
   return data.exists;
 }
 
 async function isLLM(repo_id: string) {
-  const response = await fetch(`${globalSetup.apiHost}/api/isLLM`, {
-    method: "POST",
-    body: JSON.stringify(repo_id),
-    headers: {
-      "Content-Type": "application/json"
-    }})
+  const response = await fetch(`${globalSetup.apiHost}/api/isLLM?repo_id=${repo_id}`)
   const data = await response.json()
   return data.isllm
 }

diff --git a/WebUI/src/views/Answer.vue b/WebUI/src/views/Answer.vue
@@ -388,14 +388,16 @@ async function simulatedInput() {
 }
 
 function fastGenerate(e: KeyboardEvent) {
-    if (e.code == "Enter") {
-        if (e.ctrlKey || e.shiftKey || e.altKey) {
-            question.value += "\n";
-        } else {
-            e.preventDefault();
-            newPromptGenerate();
-        }
+  if (e.code == "Enter") {
+    if (e.ctrlKey || e.shiftKey || e.altKey) {
+      question.value += "\n";
+    } else {
+      e.preventDefault();
+      if (question.value !== "") {
+        newPromptGenerate()
+      }
     }
+  }
 }
 
 async function newPromptGenerate() {

diff --git a/service/model_downloader.py b/service/model_downloader.py
@@ -1,5 +1,5 @@
 from huggingface_hub import HfFileSystem, hf_hub_url, model_info
-from huggingface_hub.utils._errors import RepositoryNotFoundError
+from huggingface_hub.utils import RepositoryNotFoundError
 from typing import Any, Callable, Dict, List
 from os import path, makedirs, rename
 import requests

diff --git a/service/web_api.py b/service/web_api.py
@@ -195,24 +195,24 @@ def check_model_exist():
     return jsonify({"code": 0, "message": "success", "exists": result_list})
 
 
-@app.route("/api/checkURLExists", methods=["POST"])
+@app.route("/api/checkURLExists", methods=["GET"])
 def check_url_exists():
-    address = request.get_json()
+    repo_id = request.args.get('repo_id')
     downloader = HFPlaygroundDownloader()
-    exists = downloader.hf_url_exists(address)
+    exists = downloader.hf_url_exists(repo_id)
 
     return jsonify(
             {
                 "exists": exists
             }
         )
 
-@app.route("/api/isLLM", methods=["POST"])
+@app.route("/api/isLLM", methods=["GET"])
 def is_llm():
-    address = request.get_json()
+    repo_id = request.args.get('repo_id')
     downloader = HFPlaygroundDownloader()
     try:
-        model_type_hf = downloader.probe_type(address)
+        model_type_hf = downloader.probe_type(repo_id)
     except Exception:
         model_type_hf = "undefined"
     return jsonify(