Skip to content

PistaDB为 LLM 原生应用打造的嵌入式向量数据库

纯 C99 · 零外部依赖 · 单文件存储 · 笔记本上亚毫秒级搜索

60 秒上手

python
import numpy as np
from pistadb import PistaDB, Metric, Index

with PistaDB("docs.pst", dim=1536, metric=Metric.COSINE, index=Index.HNSW) as db:
    db.insert(1, np.random.rand(1536).astype("float32"), label="chunk_0001")
    results = db.search(np.random.rand(1536).astype("float32"), k=10)
    for r in results:
        print(r.id, r.distance, r.label)
    db.save()

每一个 LLM 应用,最终都需要一个向量数据库。 PistaDB 把生产级向量存储浓缩进单个文件和零依赖的 C 库。 内嵌进桌面应用,部署到边缘设备,或直接和你的 Python 脚本放在一起。

基于 MIT 协议发布