File size: 906 Bytes
1a252b6 fc2f017 1a252b6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | import aiosqlite
from app.core.config import settings
from custom_logger import logger_config as logger
async def init_db():
logger.info(f"Initializing database at {settings.DATABASE_FILE}")
async with aiosqlite.connect(settings.DATABASE_FILE) as db:
await db.execute('''CREATE TABLE IF NOT EXISTS text_tasks
(id TEXT PRIMARY KEY,
input_text TEXT NOT NULL,
system_prompt TEXT,
model TEXT DEFAULT 'qwen',
status TEXT NOT NULL,
result TEXT,
created_at TEXT NOT NULL,
processed_at TEXT,
progress INTEGER DEFAULT 0,
progress_text TEXT,
hide_from_ui INTEGER DEFAULT 0)''')
await db.commit()
logger.info("Database initialized successfully.")
|