Skip to main content

High-performance Python Vector Database & Memory Engine with RESP2 support.

Project description

⚡ PulseDB

An enterprise-grade, in-memory database with a native AI Vector Engine.

Built for developers who need Redis-compatible storage and lightning-fast semantic search — without running two separate systems.

CI Python License: BSL 1.1


What is PulseDB?

PulseDB is a high-performance, open-source database that combines:

  • A Redis-compatible KV store — Strings, Lists, Hashes with TTL, LRU eviction, and RESP2 wire protocol
  • An AI Memory Engine — HNSW-based vector search with native C++ pre-filtering callbacks
  • A Python SDK — Ergonomic db.vectors.upsert() / db.vectors.search() API
  • A LangChain Integration — Drop-in PulseDBVectorStore for RAG pipelines with metadata filtering

One server, one protocol, one SDK. No Pinecone. No Weaviate. No Redis Stack.


Features

Category Capability
KV Store SET, GET, DEL, EXPIRE, TTL, MSET, MGET, INCR, APPEND
Data Types Strings · Lists (LPUSH/RPOP/LRANGE) · Hashes (HSET/HGET/HGETALL)
Vector Engine HNSW cosine similarity, O(log N) search, dynamic resizing
Hybrid Search Native C++ pre-filter callbacks — filter by metadata during graph traversal
Persistence Write-Ahead Log (WAL) + JSON snapshots + HNSW binary graph snapshots
Protocol RESP2 TCP (port 6379) — works with redis-cli, redis-py, ioredis
Cluster Consistent hashing, multi-node routing
Auth API Key (HTTP) + REQUIREPASS (TCP) + optional TLS/SSL
Observability Prometheus /metrics endpoint, structured /health and /ready
LangChain PulseDBVectorStore with similarity_search(filter={...})

Quickstart

1. Run the Server (Docker)

docker run -d \
  -p 6379:6379 \
  -p 8000:8000 \
  -v pulsedb_data:/app/data \
  --name pulsedb \
  ghcr.io/gkavinrajancodes/pulsedb:latest

Or use Docker Compose for a 3-node cluster:

git clone https://github.com/gkavinrajanCodes/pulseDB.git
cd pulseDB && docker-compose up --build

2. Install the SDK

pip install pulsedb

3. Use It

from pulsedb import PulseDB

db = PulseDB(host="localhost", port=6379)

# Standard KV Store
db.set("session:abc", "user_data", ttl=3600)
print(db.get("session:abc"))  # "user_data"

# AI Memory Engine — insert vectors with metadata
db.vectors.upsert("article:1", [0.12, 0.98, 0.34], metadata={"category": "sports", "year": 2024})
db.vectors.upsert("article:2", [0.91, 0.11, 0.67], metadata={"category": "tech", "year": 2023})

# Semantic similarity search — optionally filter by metadata
results = db.vectors.search([0.10, 0.95, 0.40], top_k=5, filter={"category": "sports"})
# → [{"id": "article:1", "score": 0.997}]

LangChain Integration

PulseDB works natively as a LangChain VectorStore, giving your RAG pipeline blazing fast retrieval with hybrid metadata filtering.

from langchain_openai import OpenAIEmbeddings
from sdk.langchain_pulsedb.vectorstore import PulseDBVectorStore

store = PulseDBVectorStore(
    embedding=OpenAIEmbeddings(),
    host="localhost",
    port=6379,
)

# Ingest documents — metadata is automatically stored for hybrid filtering
store.add_texts(
    texts=["PulseDB is fast", "Redis is popular", "Pinecone is expensive"],
    metadatas=[{"source": "blog"}, {"source": "wiki"}, {"source": "review"}]
)

# Hybrid search — find similar docs but only from the blog source
docs = store.similarity_search("fast database", k=2, filter={"source": "blog"})

How the AI Memory Engine Works

Standard vector databases do post-filtering: search all vectors, get K results, then throw away the ones that don't match the filter. This degrades accuracy.

PulseDB does true pre-filtering using native hnswlib C++ filter callbacks. The filter function is evaluated inside the graph traversal — so the C++ engine skips disqualified nodes entirely before scoring them.

Query Vector → HNSW Graph Traversal → [Filter Callback runs on every node visited]
                                        ↓ Pass → included in result set
                                        ↓ Fail → skipped immediately
                                       Top-K results returned

This means your effective top_k is always accurate, even with highly restrictive filters.


Architecture

graph TD
    Client["Client (SDK / redis-cli)"] -->|RESP2 Binary Protocol| TCP["asyncio TCP Server :6379"]
    Client -->|HTTP REST| HTTP["FastAPI Gateway :8000"]
    TCP --> Router["Command Router"]
    HTTP --> Router
    Router --> KV["16-Shard KV Store (LRU + TTL)"]
    Router --> VE["AI Vector Engine (hnswlib HNSW)"]
    Router --> DT["Data Types (Lists, Hashes)"]
    Router --> PS["Pub/Sub Engine"]
    KV --> WAL["Write-Ahead Log"]
    VE --> Snap["HNSW Binary Snapshot"]
    WAL --> Snap

Run Locally (From Source)

# 1. Clone and install
git clone https://github.com/gkavinrajanCodes/pulseDB.git
cd pulseDB
python3.10 -m venv workenv && source workenv/bin/activate
pip install -r requirements.txt

# 2. Start the server
NODE_ID=node1 CLUSTER_NODES=node1 uvicorn server.main:app --host 0.0.0.0 --port 8000

# 3. Install the SDK (in another terminal)
pip install -e sdk/

Contributing

  1. Fork the repository
  2. Create a feature branch: git checkout -b feature/sorted-sets
  3. Commit your changes: git commit -m "feat: add ZADD/ZRANGE sorted set commands"
  4. Push: git push origin feature/sorted-sets
  5. Open a Pull Request

All PRs are validated against our CI matrix (Python 3.10, 3.11, 3.12 with flake8, mypy, and pytest).


License

Distributed under the Business Source License (BSL 1.1). See LICENSE for details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pulsedb-1.0.0.tar.gz (18.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pulsedb-1.0.0-py3-none-any.whl (13.5 kB view details)

Uploaded Python 3

File details

Details for the file pulsedb-1.0.0.tar.gz.

File metadata

  • Download URL: pulsedb-1.0.0.tar.gz
  • Upload date:
  • Size: 18.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.3

File hashes

Hashes for pulsedb-1.0.0.tar.gz
Algorithm Hash digest
SHA256 a0ea02e25785499e62b0c5962cadc6e90f419390e5b1a225b05a360a82c1de85
MD5 f4105e03fd6d20c956c76e093248a219
BLAKE2b-256 e30f3ff350d06731684bd84023653d88905fbf36aff7a63ad6380593d5acebf3

See more details on using hashes here.

File details

Details for the file pulsedb-1.0.0-py3-none-any.whl.

File metadata

  • Download URL: pulsedb-1.0.0-py3-none-any.whl
  • Upload date:
  • Size: 13.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.3

File hashes

Hashes for pulsedb-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 2ce50438428dd371b8d8750f981cbcdff4ff8b122b483daa6380518f0b133683
MD5 26afafdf74ca8d4728f6c7d85cd29f37
BLAKE2b-256 e1f28e140ea3294733a5331988aa953bd54acac36d444133d5051f285af299ce

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page