Fix module name

Author: lukehinds

scripts/import_packages.py

@@ -5,7 +5,7 @@
 import sqlite3
 
 import numpy as np
-import sqlite_vec
+import sqlite_vec_sl_tmp
 
 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", db_path="./sqlite_data/vectordb.db"):
     def _get_connection(self):
         conn = sqlite3.connect(self.db_path)
         conn.enable_load_extension(True)
-        sqlite_vec.load(conn)
+        sqlite_vec_sl_tmp.load(conn)
         conn.enable_load_extension(False)
         return conn
 

src/codegate/storage/storage_engine.py

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

tests/vectordb/test_sqlitevec.py

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