mirror of
https://github.com/lllin000/PaperForge.git
synced 2026-07-22 06:50:53 +00:00
745c3dd1e1
loadable_path() returns a path without .dll extension on Windows, causing vec0 extension loading to fail. Use sqlite_vec.load(conn) which handles platform-specific paths correctly.
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import sqlite3
|
|
from pathlib import Path
|
|
|
|
from paperforge.config import paperforge_paths
|
|
|
|
|
|
def get_memory_db_path(vault: Path) -> Path:
|
|
"""Return the absolute path to paperforge.db."""
|
|
paths = paperforge_paths(vault)
|
|
index_path = paths.get("index")
|
|
if not index_path:
|
|
raise FileNotFoundError("index path not configured")
|
|
return index_path.parent / "paperforge.db"
|
|
|
|
|
|
def get_connection(db_path: Path, read_only: bool = False) -> sqlite3.Connection:
|
|
"""Open a SQLite connection to paperforge.db with WAL mode.
|
|
|
|
Args:
|
|
db_path: Path to paperforge.db.
|
|
read_only: If True, open in read-only mode (for queries).
|
|
"""
|
|
if read_only:
|
|
uri = "file:" + db_path.as_posix() + "?mode=ro"
|
|
conn = sqlite3.connect(uri, uri=True)
|
|
else:
|
|
db_path.parent.mkdir(parents=True, exist_ok=True)
|
|
conn = sqlite3.connect(str(db_path))
|
|
conn.row_factory = sqlite3.Row
|
|
if not read_only:
|
|
conn.execute("PRAGMA journal_mode=WAL;")
|
|
conn.execute("PRAGMA foreign_keys=ON;")
|
|
return conn
|
|
|
|
|
|
def ensure_vec_extension(conn: sqlite3.Connection) -> None:
|
|
"""Load the sqlite-vec extension if available.
|
|
|
|
Enables vec0 virtual table support for vector similarity search.
|
|
Gracefully no-ops when the extension is not installed.
|
|
"""
|
|
try:
|
|
import sqlite_vec
|
|
|
|
conn.enable_load_extension(True)
|
|
try:
|
|
sqlite_vec.load(conn)
|
|
except Exception:
|
|
pass # extension not available
|
|
finally:
|
|
try:
|
|
conn.enable_load_extension(False)
|
|
except AttributeError:
|
|
pass
|
|
except ImportError:
|
|
pass # sqlite_vec not installed
|