-- Preset
CREATE TABLE t (id SERIAL PRIMARY KEY, val vector(3));
INSERT INTO t (val)
SELECT ARRAY[i * 0.0001, i * 0.00005, i * 0.0002]::vector(3) FROM generate_series(1, 10000) as s(i);
CREATE INDEX idx ON t USING vchordrq (val vector_l2_ops);
CREATE TABLE t1 (val vector(4), id SERIAL PRIMARY KEY);
INSERT INTO t1 (val)
SELECT ARRAY[i * 0.0001, i * 0.00005, i * 0.0002, i * 0.001]::vector(4) FROM generate_series(1, 10000) as s(i);
CREATE INDEX idx1 ON t1 USING vchordrq (val vector_l2_ops);

SET vchordrq.query_sampling_enable = on;
-- Log up to 2 queries
SET vchordrq.query_sampling_max_records = 2;
-- Record query at 100% rate
SET vchordrq.query_sampling_rate = 1;

-- Record
SELECT * from t ORDER BY val <-> '[0.50, 0.25, 1.00]';
SELECT val from t1 ORDER BY val <-> '[0.50, 0.25, 1.00, 0]';

SELECT * from vchordrq_sampled_queries('idx1');

-- Result
 schema_name | table_name | column_name | operator |            vector_text             
-------------+------------+-------------+----------+------------------------------------
 public      | t1         | val         | <->      | [0.5,0.25,1,0]

SELECT * from vchordrq_sampled_queries;

-- Result
 schema_name | table_name | column_name | operator |            vector_text             
-------------+------------+-------------+----------+------------------------------------
 public      | t          | val         | <->      | [0.5,0.25,1]
 public      | t1         | val         | <->      | [0.5,0.25,1,0]

-- Recall metric
SELECT AVG(recall_value)
FROM (
    SELECT
        vchordrq_evaluate_query_recall(
            query => format(
                'SELECT ctid FROM %I.%I ORDER BY %I %s ''%s'' LIMIT 10',
                lq.schema_name,
                lq.table_name,
                lq.column_name,
                lq.operator,
                lq.vector_text
            )
        ) AS recall_value
    FROM
        vchordrq_sampled_queries('idx') AS lq
) AS eval_results;

-- Result
 avg 
-----
   1