Files
leocrm/app/plugins/builtins/unified_search/migrations/0002_embeddings.sql
T

46 lines
1.8 KiB
SQL
Raw Normal View History

-- Unified Search: Embedding columns and HNSW indexes
-- Note: These columns are also added by Alembic migration 0104.
-- This migration is idempotent (IF NOT EXISTS) and skips non-existent tables.
-- Ensure pgvector extension is installed
CREATE EXTENSION IF NOT EXISTS vector;
-- ─── Mails ───
DO $$ BEGIN
IF EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'mails') THEN
ALTER TABLE mails ADD COLUMN IF NOT EXISTS embedding vector(768);
CREATE INDEX IF NOT EXISTS ix_mails_embedding ON mails USING hnsw(embedding vector_cosine_ops);
END IF;
END $$;
-- ─── Contacts ───
DO $$ BEGIN
IF EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'contacts') THEN
ALTER TABLE contacts ADD COLUMN IF NOT EXISTS embedding vector(768);
CREATE INDEX IF NOT EXISTS ix_contacts_embedding ON contacts USING hnsw(embedding vector_cosine_ops);
END IF;
END $$;
-- ─── Files ───
DO $$ BEGIN
IF EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'files') THEN
ALTER TABLE files ADD COLUMN IF NOT EXISTS embedding vector(768);
CREATE INDEX IF NOT EXISTS ix_files_embedding ON files USING hnsw(embedding vector_cosine_ops);
END IF;
END $$;
-- ─── Calendar Entries ───
DO $$ BEGIN
IF EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'calendar_entries') THEN
ALTER TABLE calendar_entries ADD COLUMN IF NOT EXISTS embedding vector(768);
CREATE INDEX IF NOT EXISTS ix_cal_embedding ON calendar_entries USING hnsw(embedding vector_cosine_ops);
END IF;
END $$;
-- ─── Tags (shorter dimension for short text) ───
DO $$ BEGIN
IF EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'tags') THEN
ALTER TABLE tags ADD COLUMN IF NOT EXISTS embedding vector(384);
END IF;
END $$;