HyperDB

一个专为 LLM Agent 设计的超高速本地向量数据库。这不是开玩笑。

想投资吗？现已接受 SAFE 协议（最低 1.35 亿美元估值上限）

加入 HyperDB Discord

优势

• 简洁的接口，兼容所有大语言模型 Agent。
• 高度优化的 C++ 后端向量存储，通过 MKL BLAS 提供硬件加速运算。
• 支持用户通过 ids 和 metadata 等高级功能索引文档。

安装

从 PyPI 安装该包：

pip install hyperdb-python

本地 embedding 的可选依赖

如果你想在该包中使用 Hugging Face 模型，需要安装 sentence-transformers 包：

pip install sentence-transformers

使用方法

以下是一个使用 HyperDB 存储并查询所有 151 只原始宝可梦信息的示例，瞬间即可完成：

import json
from hyperdb import HyperDB

# 从 JSONL 文件加载文档
documents = []

with open("demo/pokemon.jsonl", "r") as f:
    for line in f:
        documents.append(json.loads(line))

# 使用文档列表实例化 HyperDB
db = HyperDB(documents, key="info.description")

# 将 HyperDB 实例保存到文件
db.save("demo/pokemon_hyperdb.pickle.gz")

# 从保存的文件加载 HyperDB 实例
db.load("demo/pokemon_hyperdb.pickle.gz")

# 使用文本输入查询 HyperDB 实例
results = db.query("Likes to sleep.", top_k=5)

返回结果：

Name: Snorlax
Pokedex ID: 143
HP: 160
Type: normal
Weakness: fighting
Description: Very lazy. Just eats and sleeps. As its rotund bulk builds, it becomes steadily more slothful.

Name: Drowzee
Pokedex ID: 96
HP: 230
Type: psychic
Weakness: dark
Description: Puts enemies to sleep then eats their dreams. Occasionally gets sick from eating bad dreams.

Name: Pinsir
Pokedex ID: 127
HP: 160
Type: bug
Weakness: fire
Description: When the temperature drops at night, it sleeps on treetops or among roots where it is well hidden.

Name: Abra
Pokedex ID: 63
HP: 160
Type: psychic
Weakness: dark
Description: Sleeps 18 hours a day. If it senses danger, it will teleport itself to safety even as it sleeps.

Name: Venonat
Pokedex ID: 48
HP: 160
Type: bug
Weakness: fire
Description: Lives in the shadows of tall trees where it eats insects. It is attracted by light at night.

性能基准测试致谢：Benim Kıçım