From dbdbbc02713ea516dbf8ea241d4bd17808c50dce Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Mon, 7 Oct 2024 12:15:55 -0500 Subject: [PATCH] remove sqlite_vec from requirements and load the compiled version --- docker/main/requirements-wheels.txt | 1 - frigate/db/sqlitevecq.py | 6 ++++-- 2 files changed, 4 insertions(+), 3 deletions(-) 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)