-- 创建扩展
CREATE EXTENSION vector;

-- 测试向量类型
SELECT '[1,2,3]'::vector;

-- 创建测试表
CREATE TABLE items (
    id BIGSERIAL PRIMARY KEY,
    name TEXT,
    embedding vector(3)
);

-- 插入测试数据
INSERT INTO items (name, embedding) VALUES 
    ('item1', '[1,1,1]'),
    ('item2', '[2,2,2]'),
    ('item3', '[3,3,3]');

-- 测试相似性搜索
SELECT 
    id,
    name,
    embedding,
    embedding <-> '[1,1,1]' as distance
FROM items 
ORDER BY embedding <-> '[1,1,1]' 
LIMIT 3;