refac: ollama.db renamed to webui.db

6d854125 · Timothy J. Baek · 74809e73 · 6d854125
Commit 6d854125 authored Mar 02, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 1 deletion

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

No files found.
--- a/backend/apps/web/internal/db.py
+++ b/backend/apps/web/internal/db.py
 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()