diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt index 4db0a935a..5db2e8886 100644 --- a/docker/main/requirements-wheels.txt +++ b/docker/main/requirements-wheels.txt @@ -33,7 +33,6 @@ unidecode == 1.3.* # OpenVino (ONNX installed in wheels-post) openvino == 2024.3.* # Embeddings -sqlite_vec == 0.1.3 transformers == 4.45.* onnx_clip == 4.0.* # Generative AI diff --git a/frigate/db/sqlitevecq.py b/frigate/db/sqlitevecq.py index 26d7e921c..8f5b3dae7 100644 --- a/frigate/db/sqlitevecq.py +++ b/frigate/db/sqlitevecq.py @@ -1,6 +1,5 @@ import sqlite3 -import sqlite_vec from playhouse.sqliteq import SqliteQueueDatabase @@ -9,6 +8,9 @@ class SqliteVecQueueDatabase(SqliteQueueDatabase): super().__init__(*args, **kwargs) self.load_vec_extension: bool = load_vec_extension + # no extension necessary, sqlite will load correctly for each platform + self.sqlite_vec_path = "/usr/local/lib/vec0" + def _connect(self, *args, **kwargs) -> sqlite3.Connection: conn: sqlite3.Connection = super()._connect(*args, **kwargs) if self.load_vec_extension: @@ -17,5 +19,5 @@ class SqliteVecQueueDatabase(SqliteQueueDatabase): def _load_vec_extension(self, conn: sqlite3.Connection) -> None: conn.enable_load_extension(True) - sqlite_vec.load(conn) + conn.load_extension(self.sqlite_vec_path) conn.enable_load_extension(False)