Merge pull request #465 from stacklok/vec-port

Vec port

Author: Luke Hinds

poetry.lock

@@ -20,7 +20,6 @@ greenlet = "^3.0.3"
 aiosqlite = "^0.20.0"
 ollama = ">=0.4.4"
 pydantic-settings = "^2.7.0"
-sqlite-vec = ">=0.1.0"
 numpy = ">=1.24.0"
 tree-sitter = ">=0.23.2"
 tree-sitter-go = ">=0.23.4"
@@ -29,6 +28,7 @@ tree-sitter-javascript = ">=0.23.1"
 tree-sitter-python = ">=0.23.6"
 tree-sitter-rust = ">=0.23.2"
 
+sqlite-vec-sl-tmp = "0.0.3"
 [tool.poetry.group.dev.dependencies]
 pytest = ">=7.4.0"
 pytest-cov = ">=4.1.0"

pyproject.toml

@@ -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
 

scripts/import_packages.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:

src/codegate/storage/storage_engine.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()