open-webui/backend/apps/web/internal/db.py
2024-03-02 00:19:24 -08:00

16 lines
331 B
Python

from peewee import *
from config import DATA_DIR
import os
# 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()