thvinhtruong / hyperDB

A hyper-fast local vector database for use with LLM Agents. Now accepting SAFEs at $135M cap.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

HyperDB

HyperDB Logo

A hyper-fast local vector database for use with LLM Agents.

Want to invest? Now accepting SAFEs ($135M cap minimum.)

Join the HyperDBiscord

Advantages

  • Simple interface compatible with all large language model agents.
  • Highly optimized C++ backend vector store with HW accelerated operations via MKL BLAS.
  • Enables users to index documents with advanced features such as ids and metadata.

Installation

Install the package from PyPI:

pip install hyperdb-python

Usage

Here's an example of using HyperDB to store and query documents with information about all 151 original pokemon in an instant:

import json
from hyperdb import HyperDB

# Load documents from the JSONL file
documents = []

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

# Instantiate HyperDB with the list of documents
db = HyperDB(documents, key="info.description")

# Save the HyperDB instance to a file
db.save("demo/pokemon_hyperdb.pickle.gz")

# Load the HyperDB instance from the save file
db.load("demo/pokemon_hyperdb.pickle.gz")

# Query the HyperDB instance with a text input
results = db.query("Likes to sleep.", top_k=5)

Returns

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.

*Benchmark Credit: Benim Kıçım

About

A hyper-fast local vector database for use with LLM Agents. Now accepting SAFEs at $135M cap.

License:MIT License


Languages

Language:Python 100.0%