diff --git a/sql/vector--0.6.1--0.7.0.sql b/sql/vector--0.6.1--0.7.0.sql index 2b3fa8e..7c32f37 100644 --- a/sql/vector--0.6.1--0.7.0.sql +++ b/sql/vector--0.6.1--0.7.0.sql @@ -4,3 +4,7 @@ CREATE OPERATOR CLASS vector_integer_ops DEFAULT FOR TYPE integer USING hnsw AS OPERATOR 2 = (integer, integer); + +CREATE OPERATOR CLASS vector_text_ops + DEFAULT FOR TYPE text USING hnsw AS + OPERATOR 2 = (text, text); diff --git a/sql/vector.sql b/sql/vector.sql index bc5ede2..3a844cd 100644 --- a/sql/vector.sql +++ b/sql/vector.sql @@ -296,3 +296,7 @@ CREATE OPERATOR CLASS vector_cosine_ops CREATE OPERATOR CLASS vector_integer_ops DEFAULT FOR TYPE integer USING hnsw AS OPERATOR 2 = (integer, integer); + +CREATE OPERATOR CLASS vector_text_ops + DEFAULT FOR TYPE text USING hnsw AS + OPERATOR 2 = (text, text); diff --git a/test/t/020_hnsw_filtering.pl b/test/t/020_hnsw_filtering.pl index f85f9a3..a0e4987 100644 --- a/test/t/020_hnsw_filtering.pl +++ b/test/t/020_hnsw_filtering.pl @@ -21,7 +21,7 @@ sub test_recall my $explain = $node->safe_psql("postgres", qq( SET enable_seqscan = off; - EXPLAIN ANALYZE SELECT i FROM tst WHERE c = $cs[0] ORDER BY v $operator '$queries[0]' LIMIT $limit; + EXPLAIN ANALYZE SELECT i FROM tst WHERE c = '$cs[0]' ORDER BY v $operator '$queries[0]' LIMIT $limit; )); like($explain, qr/Index Cond/); @@ -29,7 +29,7 @@ sub test_recall { my $actual = $node->safe_psql("postgres", qq( SET enable_seqscan = off; - SELECT i FROM tst WHERE c = $cs[$i] ORDER BY v $operator '$queries[$i]' LIMIT $limit; + SELECT i FROM tst WHERE c = '$cs[$i]' ORDER BY v $operator '$queries[$i]' LIMIT $limit; )); my @actual_ids = split("\n", $actual); my %actual_set = map { $_ => 1 } @actual_ids; @@ -58,13 +58,13 @@ $node->start; # Create table $node->safe_psql("postgres", "CREATE EXTENSION vector;"); -$node->safe_psql("postgres", "CREATE TABLE tst (i int4, v vector($dim), c int4);"); +$node->safe_psql("postgres", "CREATE TABLE tst (i int4, v vector($dim), c text);"); $node->safe_psql("postgres", "INSERT INTO tst SELECT i, ARRAY[$array_sql], i % $nc FROM generate_series(1, 10000) i;" ); $node->safe_psql("postgres", "CREATE INDEX ON tst USING hnsw (v vector_l2_ops, c);"); $node->safe_psql("postgres", - "INSERT INTO tst SELECT i, ARRAY[$array_sql], i % $nc FROM generate_series(1, 10000) i;" + "INSERT INTO tst SELECT i, ARRAY[$array_sql], (i % $nc)::text FROM generate_series(1, 10000) i;" ); # Generate queries @@ -85,7 +85,7 @@ for my $i (0 .. $#queries) { my $res = $node->safe_psql("postgres", qq( SET enable_indexscan = off; - SELECT i FROM tst WHERE c = $cs[$i] ORDER BY v <-> '$queries[$i]' LIMIT $limit; + SELECT i FROM tst WHERE c = '$cs[$i]' ORDER BY v <-> '$queries[$i]' LIMIT $limit; )); push(@expected, $res); } @@ -94,7 +94,7 @@ for my $i (0 .. $#queries) test_recall(0.99, '<->'); # Test vacuum -$node->safe_psql("postgres", "DELETE FROM tst WHERE c > 5;"); +$node->safe_psql("postgres", "DELETE FROM tst WHERE c > '5';"); $node->safe_psql("postgres", "VACUUM tst;"); # Test columns