Clémentine
commited on
Commit
·
06c0fa4
1
Parent(s):
0b4b222
fix
Browse files- app.py +1 -1
- utils/io.py +8 -1
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import time
|
|
| 3 |
from apscheduler.schedulers.background import BackgroundScheduler
|
| 4 |
import threading
|
| 5 |
import globals
|
| 6 |
-
from utils.io import
|
| 7 |
from utils.jobs import run_single_job, run_multiple_jobs, launch_jobs, update_job_statuses, relaunch_failed_jobs
|
| 8 |
from typing import List, Optional
|
| 9 |
|
|
|
|
| 3 |
from apscheduler.schedulers.background import BackgroundScheduler
|
| 4 |
import threading
|
| 5 |
import globals
|
| 6 |
+
from utils.io import save_results, load_results, load_models_providers, get_results_table, load_models_providers_str, get_summary_stats
|
| 7 |
from utils.jobs import run_single_job, run_multiple_jobs, launch_jobs, update_job_statuses, relaunch_failed_jobs
|
| 8 |
from typing import List, Optional
|
| 9 |
|
utils/io.py
CHANGED
|
@@ -56,9 +56,14 @@ def load_models_providers_str(file_path: str = globals.LOCAL_CONFIG_FILE) -> str
|
|
| 56 |
|
| 57 |
|
| 58 |
def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tuple[str, str]]:
|
| 59 |
-
"""Load models and providers from text file."""
|
| 60 |
models_providers = []
|
| 61 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
with open(file_path, 'r') as f:
|
| 63 |
for line in f:
|
| 64 |
line = line.strip()
|
|
@@ -71,6 +76,8 @@ def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tu
|
|
| 71 |
models_providers.append((model, provider))
|
| 72 |
except Exception as e:
|
| 73 |
print(f"Error loading model providers: {str(e)}")
|
|
|
|
|
|
|
| 74 |
return models_providers
|
| 75 |
|
| 76 |
|
|
|
|
| 56 |
|
| 57 |
|
| 58 |
def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tuple[str, str]]:
|
| 59 |
+
"""Load models and providers from text file. Creates file if it doesn't exist."""
|
| 60 |
models_providers = []
|
| 61 |
try:
|
| 62 |
+
# Check if file exists, if not, create it
|
| 63 |
+
if not os.path.exists(file_path):
|
| 64 |
+
print(f"Config file {file_path} not found. Initializing...")
|
| 65 |
+
initialize_models_providers_file(file_path)
|
| 66 |
+
|
| 67 |
with open(file_path, 'r') as f:
|
| 68 |
for line in f:
|
| 69 |
line = line.strip()
|
|
|
|
| 76 |
models_providers.append((model, provider))
|
| 77 |
except Exception as e:
|
| 78 |
print(f"Error loading model providers: {str(e)}")
|
| 79 |
+
import traceback
|
| 80 |
+
traceback.print_exc()
|
| 81 |
return models_providers
|
| 82 |
|
| 83 |
|