2023-12-26 06:44:28 +01:00
|
|
|
from peewee import *
|
2024-04-02 11:28:04 +02:00
|
|
|
from config import DATA_DIR
|
2024-03-02 09:19:24 +01:00
|
|
|
import os
|
2023-12-26 06:44:28 +01:00
|
|
|
|
2024-01-19 21:13:09 +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")
|
2024-04-02 11:28:04 +02:00
|
|
|
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()
|