Update server config

jhj0517 · Nov 19, 2024 · 6b34bab · 6b34bab
1 parent 912d16e
commit 6b34bab
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/backend/tests/test_backend_config.py b/backend/tests/test_backend_config.py
@@ -6,10 +6,12 @@
 import os
 import requests
 import pytest
+import yaml
 
 from backend.main import backend_app
 from modules.whisper.data_classes import *
 from modules.utils.paths import *
+from modules.utils.files_manager import load_yaml, save_yaml
 
 TEST_PIPELINE_PARAMS = {**WhisperParams(model_size="tiny").model_dump(exclude_none=True),
                         **VadParams().model_dump(exclude_none=True),
@@ -19,9 +21,12 @@
 TEST_BGM_SEPARATION_PARAMS = BGMSeparationParams()
 TEST_FILE_DOWNLOAD_URL = "https://github.com/jhj0517/whisper_flutter_new/raw/main/example/assets/jfk.wav"
 TEST_FILE_PATH = os.path.join(WEBUI_DIR, "backend", "tests", "jfk.wav")
+TEST_WHISPER_MODEL = "tiny"
+TEST_COMPUTE_TYPE = "float32"
 
 
 @pytest.fixture(autouse=True)
+@functools.lru_cache
 def setup_test_file():
     def download_file(url=TEST_FILE_DOWNLOAD_URL, file_path=TEST_FILE_PATH):
         if os.path.exists(file_path):
@@ -39,6 +44,11 @@ def download_file(url=TEST_FILE_DOWNLOAD_URL, file_path=TEST_FILE_PATH):
 
     download_file(TEST_FILE_DOWNLOAD_URL, TEST_FILE_PATH)
 
+    server_config = load_yaml(SERVER_CONFIG_PATH)
+    server_config["whisper"]["model_size"] = TEST_WHISPER_MODEL
+    server_config["whisper"]["compute_type"] = TEST_COMPUTE_TYPE
+    save_yaml(server_config, SERVER_CONFIG_PATH)
+
 
 @pytest.fixture
 def upload_file_instance(filepath: str = TEST_FILE_PATH) -> UploadFile: