Spaces:
Running
Running
| """DB initialization entrypoint — run on app startup to create tables.""" | |
| from __future__ import annotations | |
| import logging | |
| import shutil | |
| from pathlib import Path | |
| from app.storage.db import Base, engine | |
| from app.storage.models import Message, Session, SkillPref, ToolRun # noqa: F401 | |
| from app.config import get_settings | |
| logger = logging.getLogger(__name__) | |
| async def init_db() -> None: | |
| settings = get_settings() | |
| # Diagnostic block: log WHERE the DB will live and whether we're | |
| # actually on a persistent volume. If you see "WARNING" below, | |
| # /data is NOT persisted and the user's history will be wiped on | |
| # the next Space restart. Fix: Space Settings → Persistent Storage. | |
| if settings.turso_database_url: | |
| logger.info( | |
| "DB backend: Turso (remote libSQL). Persistent across rebuilds." | |
| ) | |
| elif settings.is_hf_space: | |
| db_path = Path(settings.persistent_db_path) | |
| db_path.parent.mkdir(parents=True, exist_ok=True) | |
| is_real_volume = _is_persistent_mount("/data") | |
| if not is_real_volume: | |
| logger.warning( | |
| "DB backend: HF Space detected, but /data is NOT a persistent " | |
| "mount. History will be lost on every restart. → Fix: open the " | |
| "Space's Settings tab and enable 'Persistent Storage'." | |
| ) | |
| logger.info( | |
| "DB backend: SQLite at %s (HF Space, persistent=%s)", | |
| db_path, | |
| is_real_volume, | |
| ) | |
| else: | |
| logger.info( | |
| "DB backend: SQLite at %s (local dev)", settings.local_sqlite_path | |
| ) | |
| # On HuggingFace Spaces, the project root is wiped on every | |
| # container rebuild, but /data persists. Migrate any pre-existing | |
| # local DB (e.g. from before the persistent path was wired) to | |
| # /data so the user doesn't lose history. | |
| if not settings.turso_database_url: | |
| new_path = Path(settings.persistent_db_path) | |
| old_path = Path(settings.local_sqlite_path) | |
| new_path.parent.mkdir(parents=True, exist_ok=True) | |
| if old_path.exists() and not new_path.exists() and new_path.parent != old_path.parent: | |
| try: | |
| shutil.copy2(old_path, new_path) | |
| logger.info("Migrated local SQLite to persistent path: %s", new_path) | |
| except OSError as exc: | |
| logger.warning("Could not migrate SQLite: %s", exc) | |
| async with engine.begin() as conn: | |
| await conn.run_sync(Base.metadata.create_all) | |
| logger.info("Database tables ready at %s", settings.database_url) | |
| def _is_persistent_mount(path: str) -> bool: | |
| """Heuristic: is `path` mounted on a separate filesystem from /? | |
| On HF Space, the persistent /data volume is its own mount, so | |
| `os.stat(path).st_dev` differs from `os.stat('/').st_dev`. If it | |
| matches, /data is just a directory on the container's root | |
| filesystem and will be wiped on every rebuild. | |
| """ | |
| import os | |
| try: | |
| return os.stat(path).st_dev != os.stat("/").st_dev | |
| except OSError: | |
| return False | |