diff --git a/src/recorder/worker.rs b/src/recorder/worker.rs index a6da90f0..cbef6b11 100644 --- a/src/recorder/worker.rs +++ b/src/recorder/worker.rs @@ -11,6 +11,7 @@ // vectorchord-inquiry@tensorchord.ai // // Copyright (c) 2025 TensorChord Inc. + use crate::recorder::types::PgRefCell; use std::cell::RefMut; use std::fs; diff --git a/src/sql/finalize.sql b/src/sql/finalize.sql index a520ba39..c5844c44 100644 --- a/src/sql/finalize.sql +++ b/src/sql/finalize.sql @@ -149,10 +149,11 @@ STRICT LANGUAGE c AS 'MODULE_PATHNAME', '_vchordrq_sampled_vectors_wrapper'; CREATE OR REPLACE FUNCTION vchordrq_sampled_queries(regclass) RETURNS TABLE( schema_name NAME, + index_name NAME, table_name NAME, column_name NAME, operator TEXT, - vector_text TEXT + value TEXT ) LANGUAGE plpgsql STRICT AS $$ @@ -175,6 +176,7 @@ BEGIN WITH index_metadata AS ( SELECT NS.nspname AS schema_name, + I.relname AS index_name, C.relname AS table_name, PA.attname AS column_name, CASE @@ -205,13 +207,14 @@ BEGIN ) SELECT im.schema_name, + im.index_name, im.table_name, im.column_name, im.operator, - s.vector_text + s.value FROM index_metadata im, - LATERAL %2$I.vchordrq_sampled_vectors(%1$s) AS s(vector_text); + LATERAL %2$I.vchordrq_sampled_vectors(%1$s) AS s(value); $q$, $1::oid, ext_schema @@ -223,10 +226,11 @@ $$; CREATE VIEW vchordrq_sampled_queries AS SELECT record.schema_name, + record.index_name, record.table_name, record.column_name, record.operator, - record.vector_text + record.value FROM ( SELECT i.oid diff --git a/tests/vchordrq/recall.slt b/tests/vchordrq/recall.slt index ca3fb095..d75068cc 100644 --- a/tests/vchordrq/recall.slt +++ b/tests/vchordrq/recall.slt @@ -86,12 +86,12 @@ statement ok SELECT * from t_dim4 ORDER BY val <-> '[1.00, 0.50, 0.25, 0]'; query I -SELECT vector_text from vchordrq_sampled_queries('idx1'); +SELECT value from vchordrq_sampled_queries('idx1'); ---- [0.5,0.25,1] query I -SELECT vector_text from vchordrq_sampled_queries('idx2'); +SELECT value from vchordrq_sampled_queries('idx2'); ---- [1,0.5,0.25,0] @@ -106,7 +106,7 @@ SELECT * from t_dim4 ORDER BY val <-> '[2.1, 0.3, 0.7, 0.9]'; query I SELECT * from vchordrq_sampled_queries('idx2'); ---- -public t_dim4 val <-> [2.1,0.3,0.7,0.9] +public idx2 t_dim4 val <-> [2.1,0.3,0.7,0.9] query I SELECT AVG(recall_value) @@ -119,7 +119,7 @@ FROM ( lq.table_name, lq.column_name, lq.operator, - lq.vector_text + lq.value ) ) AS recall_value FROM @@ -146,7 +146,7 @@ SELECT column_name from vchordrq_sampled_queries('idx3'); NULL query I -SELECT vector_text from vchordrq_sampled_queries('idx3'); +SELECT value from vchordrq_sampled_queries('idx3'); ---- [1,0.5,0.25] @@ -154,7 +154,7 @@ statement ok SET search_path='@'; query I -SELECT vector_text from public.vchordrq_sampled_queries('public.idx3'); +SELECT value from public.vchordrq_sampled_queries('public.idx3'); ---- [1,0.5,0.25]