Change to correct alg: vec_distance_cosine

Author: lukehinds

src/codegate/storage/storage_engine.py

@@ -178,7 +178,7 @@ async def search(
                 query_sql = """
                     WITH distances AS (
                         SELECT name, type, status, description,
-                               vss_distance(embedding, ?) as distance
+                               vec_distance_cosine(embedding, ?) as distance
                         FROM packages
                     )
                     SELECT name, type, status, description, distance

tests/vectordb/test_sqlitevec.py

@@ -25,16 +25,16 @@ def setup_vector_similarity(conn):
             # Insert a mock distance value that will be used in searches
             cursor.execute("INSERT INTO vector_distances (distance) VALUES (0.1)")
 
-            # Create a view that simulates the vss_distance function
+            # Create a view that simulates the vec_distance_cosine function
             cursor.execute(
                 """
-            CREATE VIEW IF NOT EXISTS vss_distance_view AS
+            CREATE VIEW IF NOT EXISTS vec_distance_cosine_view AS
             SELECT distance FROM vector_distances WHERE id = 1
             """
             )
 
             # Create a function that returns the mock distance
-            conn.create_function("vss_distance", 2, lambda x, y: 0.1)
+            conn.create_function("vec_distance_cosine", 2, lambda x, y: 0.1)
 
         mock_load.side_effect = setup_vector_similarity
         yield mock_load