-- 创建 "vector" 扩展（如果不存在）
CREATE EXTENSION IF NOT EXISTS vector;

-- 创建表 modeldata
CREATE TABLE IF NOT EXISTS modeldata (
    id BIGSERIAL PRIMARY KEY,
    vector VECTOR(1536) NOT NULL,
    user_id VARCHAR(50),
    team_id VARCHAR(50),
    kb_id VARCHAR(50),
    tmb_id VARCHAR(50),
    dataset_id VARCHAR(50),
    data_id VARCHAR(50),
    collection_id VARCHAR(50),
    source VARCHAR(100),
    file_id VARCHAR(100),
    q TEXT,
    a TEXT
);

-- 创建索引 "modelData_userId_index"
CREATE INDEX IF NOT EXISTS modelData_userId_index ON modeldata USING HASH (user_id);

-- 创建索引 "modelData_kbId_index"
CREATE INDEX IF NOT EXISTS modelData_kbId_index ON modeldata USING HASH (kb_id);

-- 创建复合索引 "idx_model_data_md5_q_a_user_id_kb_id"
CREATE INDEX IF NOT EXISTS idx_model_data_md5_q_a_user_id_kb_id ON modeldata (md5(q), md5(a), user_id, kb_id);