NIUCLOUD是一款SaaS管理后台框架多应用插件+云编译。上千名开发者、服务商正在积极拥抱开发者生态。欢迎开发者们免费入驻。一起助力发展! 广告
[TOC] > [home](https://github.com/asg017/sqlite-vec) ## 概述 主要特点: * 轻量级:基于 SQLite,部署简单 * 高性能:支持向量相似度搜索 * 易用性:SQL 语法操作 * 可扩展:支持自定义距离函数 ## 示例 ``` import sqlite3 import numpy as np # 连接数据库 conn = sqlite3.connect('vector.db') # 创建向量表 conn.execute(''' CREATE VIRTUAL TABLE IF NOT EXISTS items USING vss0( id INTEGER PRIMARY KEY, title TEXT, embedding VECTOR(384) # 384维向量 ) ''') # 插入向量数据 vector = np.random.rand(384) # 生成随机向量 conn.execute(''' INSERT INTO items (title, embedding) VALUES (?, ?) ''', ('测试文档', vector)) # 向量相似度搜索 query_vector = np.random.rand(384) results = conn.execute(''' SELECT title, distance FROM items ORDER BY embedding <-> ? LIMIT 5 ''', (query_vector,)) ```