Release 0.6.0 (#206)

Author: syvb

pgvectorscale/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "vectorscale"
-version = "0.6.0-dev"
+version = "0.6.0"
 edition = "2021"
 
 [lib]

pgvectorscale/sql/vectorscale--0.0.2--0.6.0.sql

@@ -0,0 +1 @@
+vectorscale--0.5.1--0.6.0.sql

pgvectorscale/sql/vectorscale--0.2.0--0.6.0.sql

@@ -0,0 +1 @@
+vectorscale--0.5.1--0.6.0.sql

pgvectorscale/sql/vectorscale--0.3.0--0.6.0.sql

@@ -0,0 +1 @@
+vectorscale--0.5.1--0.6.0.sql

pgvectorscale/sql/vectorscale--0.4.0--0.6.0.sql

@@ -0,0 +1 @@
+vectorscale--0.5.1--0.6.0.sql

pgvectorscale/sql/vectorscale--0.5.0--0.6.0.sql

@@ -0,0 +1 @@
+vectorscale--0.5.1--0.6.0.sql

pgvectorscale/sql/vectorscale--0.5.1--0.6.0.sql

@@ -0,0 +1,127 @@
+/* <begin connected objects> */
+/*
+This file is auto generated by pgrx.
+
+The ordering of items is not stable, it is driven by a dependency graph.
+*/
+/* </end connected objects> */
+
+/* <begin connected objects> */
+-- pgvectorscale/src/access_method/mod.rs:33
+-- vectorscale::access_method::amhandler
+
+    CREATE OR REPLACE FUNCTION diskann_amhandler(internal) RETURNS index_am_handler PARALLEL SAFE IMMUTABLE STRICT COST 0.0001 LANGUAGE c AS '$libdir/vectorscale-0.6.0', 'amhandler_wrapper';
+
+    DO $$
+    DECLARE
+        c int;
+    BEGIN
+        SELECT count(*)
+        INTO c
+        FROM pg_catalog.pg_am a
+        WHERE a.amname = 'diskann';
+
+        IF c = 0 THEN
+            CREATE ACCESS METHOD diskann TYPE INDEX HANDLER diskann_amhandler;
+        END IF;
+    END;
+    $$;
+/* </end connected objects> */
+
+/* <begin connected objects> */
+-- pgvectorscale/src/access_method/distance.rs:47
+-- vectorscale::access_method::distance::distance_type_cosine
+CREATE OR REPLACE FUNCTION "distance_type_cosine"() RETURNS smallint /* i16 */
+IMMUTABLE STRICT PARALLEL SAFE
+LANGUAGE c /* Rust */
+AS '$libdir/vectorscale-0.6.0', 'distance_type_cosine_wrapper';
+/* </end connected objects> */
+
+/* <begin connected objects> */
+-- pgvectorscale/src/access_method/distance.rs:57
+-- vectorscale::access_method::distance::distance_type_inner_product
+CREATE OR REPLACE FUNCTION "distance_type_inner_product"() RETURNS smallint /* i16 */
+IMMUTABLE STRICT PARALLEL SAFE
+LANGUAGE c /* Rust */
+AS '$libdir/vectorscale-0.6.0', 'distance_type_inner_product_wrapper';
+/* </end connected objects> */
+
+/* <begin connected objects> */
+-- pgvectorscale/src/access_method/distance.rs:52
+-- vectorscale::access_method::distance::distance_type_l2
+CREATE OR REPLACE FUNCTION "distance_type_l2"() RETURNS smallint /* i16 */
+IMMUTABLE STRICT PARALLEL SAFE
+LANGUAGE c /* Rust */
+AS '$libdir/vectorscale-0.6.0', 'distance_type_l2_wrapper';
+/* </end connected objects> */
+
+/* <begin connected objects> */
+-- pgvectorscale/src/access_method/mod.rs:167
+-- requires:
+--   amhandler
+--   distance_type_cosine
+--   distance_type_l2
+--   distance_type_inner_product
+
+
+DO $$
+DECLARE
+  have_cos_ops int;
+  have_l2_ops int;
+  have_ip_ops int;
+BEGIN
+    -- Has cosine operator class been installed previously?
+    SELECT count(*)
+    INTO have_cos_ops
+    FROM pg_catalog.pg_opclass c
+    WHERE c.opcname = 'vector_cosine_ops'
+    AND c.opcmethod = (SELECT oid FROM pg_catalog.pg_am am WHERE am.amname = 'diskann')
+    AND c.opcnamespace = (SELECT oid FROM pg_catalog.pg_namespace where nspname='@extschema@');
+
+    -- Has L2 operator class been installed previously?
+    SELECT count(*)
+    INTO have_l2_ops
+    FROM pg_catalog.pg_opclass c
+    WHERE c.opcname = 'vector_l2_ops'
+    AND c.opcmethod = (SELECT oid FROM pg_catalog.pg_am am WHERE am.amname = 'diskann')
+    AND c.opcnamespace = (SELECT oid FROM pg_catalog.pg_namespace where nspname='@extschema@');
+
+    -- Has inner product operator class been installed previously?
+    SELECT count(*)
+    INTO have_ip_ops
+    FROM pg_catalog.pg_opclass c
+    WHERE c.opcname = 'vector_ip_ops'
+    AND c.opcmethod = (SELECT oid FROM pg_catalog.pg_am am WHERE am.amname = 'diskann')
+    AND c.opcnamespace = (SELECT oid FROM pg_catalog.pg_namespace where nspname='@extschema@');
+
+    IF have_cos_ops = 0 THEN
+        CREATE OPERATOR CLASS vector_cosine_ops DEFAULT
+        FOR TYPE vector USING diskann AS
+	        OPERATOR 1 <=> (vector, vector) FOR ORDER BY float_ops,
+            FUNCTION 1 distance_type_cosine();
+    ELSIF have_l2_ops = 0 THEN
+        -- Upgrade from 0.4.0 to 0.5.0.  Update cosine opclass to include
+        -- the distance_type_cosine function.
+        INSERT INTO pg_amproc (oid, amprocfamily, amproclefttype, amprocrighttype, amprocnum, amproc)
+        SELECT  (select (max(oid)::int + 1)::oid from pg_amproc), c.opcfamily, c.opcintype, c.opcintype, 1, '@extschema@.distance_type_l2'::regproc
+        FROM pg_opclass c, pg_am a
+        WHERE a.oid = c.opcmethod AND c.opcname = 'vector_cosine_ops' AND a.amname = 'diskann';
+    END IF;
+
+    IF have_l2_ops = 0 THEN
+        CREATE OPERATOR CLASS vector_l2_ops
+        FOR TYPE vector USING diskann AS
+            OPERATOR 1 <-> (vector, vector) FOR ORDER BY float_ops,
+            FUNCTION 1 distance_type_l2();
+    END IF;
+
+    IF have_ip_ops = 0 THEN
+        CREATE OPERATOR CLASS vector_ip_ops
+        FOR TYPE vector USING diskann AS
+            OPERATOR 1 <#> (vector, vector) FOR ORDER BY float_ops,
+            FUNCTION 1 distance_type_inner_product();
+    END IF;
+END;
+$$;
+/* </end connected objects> */
+

pgvectorscale/src/access_method/upgrade_test.rs

@@ -278,4 +278,11 @@ pub mod tests {
     fn test_upgrade_from_0_5_0() {
         test_upgrade_base("0.5.0", "0.12.5", "pgvectorscale", "vectorscale", "diskann");
     }
+
+    #[ignore]
+    #[serial]
+    #[test]
+    fn test_upgrade_from_0_5_1() {
+        test_upgrade_base("0.5.1", "0.12.5", "pgvectorscale", "vectorscale", "diskann");
+    }
 }