-- Initialize PostgreSQL with pgvector for SFDX RAG
CREATE EXTENSION IF NOT EXISTS vector;
CREATE EXTENSION IF NOT EXISTS pg_trgm;
-- Verify extensions are loaded
SELECT extname FROM pg_extension WHERE extname IN ('vector', 'pg_trgm');
-- Test vector functionality
DO $$
BEGIN
-- Test creating a vector column
EXECUTE 'CREATE TEMP TABLE test_vector (id int, embedding vector(3))';
-- Test inserting vector data
EXECUTE 'INSERT INTO test_vector VALUES (1, ''[1,2,3]'')';
-- Test vector similarity
EXECUTE 'SELECT id, embedding <-> ''[1,2,3]'' as distance FROM test_vector';
RAISE NOTICE 'pgvector extension is working correctly';
END
$$;
