Pārlūkot izejas kodu

refac: ollama.db renamed to webui.db

Timothy J. Baek 1 gadu atpakaļ
vecāks
revīzija
6d854125cc
1 mainītis faili ar 11 papildinājumiem un 1 dzēšanām
  1. 11 1
      backend/apps/web/internal/db.py

+ 11 - 1
backend/apps/web/internal/db.py

@@ -1,6 +1,16 @@
 from peewee import *
 from config import DATA_DIR
+import os
 
 
-DB = SqliteDatabase(f"{DATA_DIR}/ollama.db")
+# Check if the file exists
+if os.path.exists(f"{DATA_DIR}/ollama.db"):
+    # Rename the file
+    os.rename(f"{DATA_DIR}/ollama.db", f"{DATA_DIR}/webui.db")
+    print("File renamed successfully.")
+else:
+    pass
+
+
+DB = SqliteDatabase(f"{DATA_DIR}/webui.db")
 DB.connect()