open-webui/backend/apps/web/internal/db.py

17 lines
331 B
Python
Raw Normal View History

2023-12-26 06:44:28 +01:00
from peewee import *
from config import DATA_DIR
2024-03-02 09:19:24 +01:00
import os
2023-12-26 06:44:28 +01:00
2024-03-02 09:19:24 +01:00
# 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.")
2024-03-02 09:19:24 +01:00
else:
pass
DB = SqliteDatabase(f"{DATA_DIR}/webui.db")
2023-12-26 06:44:28 +01:00
DB.connect()