Revert back to sqlite-vec from fork now upstream is fixed

asg017/sqlite-vec@cc7c29e

Author: lukehinds

poetry.lock

@@ -26,9 +26,9 @@ tree-sitter-java = "==0.23.5"
 tree-sitter-javascript = "==0.23.1"
 tree-sitter-python = "==0.23.6"
 tree-sitter-rust = "==0.23.2"
-sqlite-vec-sl-tmp = "==0.0.4"
 alembic = "==1.14.0"
 pygments = "==2.19.1"
+sqlite-vec = "^0.1.6"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "==8.3.4"

pyproject.toml

@@ -5,7 +5,7 @@
 import sqlite3
 
 import numpy as np
-import sqlite_vec_sl_tmp
+import sqlite_vec
 
 from codegate.inference.inference_engine import LlamaCppInferenceEngine
 from codegate.utils.utils import generate_vector_string
@@ -27,7 +27,7 @@ def __init__(self, jsonl_dir="data", vec_db_path="./sqlite_data/vectordb.db"):
     def _get_connection(self):
         conn = sqlite3.connect(self.vec_db_path)
         conn.enable_load_extension(True)
-        sqlite_vec_sl_tmp.load(conn)
+        sqlite_vec.load(conn)
         conn.enable_load_extension(False)
         return conn
 

scripts/import_packages.py

@@ -4,7 +4,7 @@
 from typing import List
 
 import numpy as np
-import sqlite_vec_sl_tmp
+import sqlite_vec
 import structlog
 
 from codegate.config import Config
@@ -69,7 +69,7 @@ def _get_connection(self):
         try:
             conn = sqlite3.connect(self.db_path)
             conn.enable_load_extension(True)
-            sqlite_vec_sl_tmp.load(conn)
+            sqlite_vec.load(conn)
             conn.enable_load_extension(False)
             return conn
         except Exception as e:

src/codegate/storage/storage_engine.py

@@ -1,10 +1,9 @@
-import datetime
 from unittest.mock import AsyncMock, patch
 
 import pytest
 
-from codegate.db.models import Session, Workspace, WorkspaceActive
-from codegate.pipeline.workspace.commands import WorkspaceCommands, WorkspaceCrud
+from codegate.db.models import WorkspaceActive
+from codegate.pipeline.workspace.commands import WorkspaceCommands
 
 
 @pytest.mark.asyncio

tests/pipeline/workspace/test_workspace.py

@@ -10,7 +10,7 @@
 
 @pytest.fixture(scope="module")
 def mock_sqlite_vec():
-    with patch("sqlite_vec_sl_tmp.load") as mock_load:
+    with patch("sqlite_vec.load") as mock_load:
         # Mock the vector similarity extension loading
         def setup_vector_similarity(conn):
             cursor = conn.cursor()