Skip to main content

A high-performance vector database library for Python

Project description

Valori

Author: Varshith
Team: Valori
Contact: varshith.gudur17@gmail.com

PyPI page for Valori

PyPI version Python versions License: MIT Build Status

A high-performance vector database library for Python that provides efficient storage, indexing, and search capabilities for high-dimensional vectors.

Features

  • 🚀 High Performance: Optimized for speed with multiple indexing algorithms
  • 📄 Document Parsing: Support for PDF, Office, text, and advanced parsing with Docling
  • 🔄 Processing Pipeline: Complete document processing with cleaning, chunking, and embedding
  • 💾 Multiple Storage Backends: Memory, disk, and hybrid storage options
  • 🔍 Advanced Indexing: Flat, HNSW, and IVF indices for different use cases
  • 🗜️ Vector Quantization: Scalar and product quantization for memory efficiency
  • 💾 Persistence: Tensor-based and incremental persistence strategies
  • 🏭 Production Ready: Comprehensive logging, monitoring, and error handling
  • 🐍 Python Native: Pure Python implementation with NumPy integration
  • 📊 Extensible: Plugin architecture for custom components

Installation

Install Valori using pip:

pip install valori

Or install from source:

git clone https://github.com/varshith-Git/valori.git
cd valori
pip install -e .

Quick Start

import numpy as np
from valori import VectorDBClient
from valori.storage import MemoryStorage
from valori.indices import FlatIndex
from valori.processors import ProcessingPipeline

# Create components
storage = MemoryStorage({})
index = FlatIndex({"metric": "cosine"})

# Create client
client = VectorDBClient(storage, index)
client.initialize()

# Process documents
pipeline_config = {
    "parsers": {"text": {"chunk_size": 1000}},
    "processors": {
        "cleaning": {"normalize_whitespace": True},
        "chunking": {"strategy": "semantic"},
        "embedding": {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
    }
}

pipeline = ProcessingPipeline(pipeline_config)
pipeline.initialize()

# Process a document
result = pipeline.process_document("document.pdf")
embedding = np.array(result["embedding"]).reshape(1, -1)

# Store in vector database
inserted_ids = client.insert(embedding, [result["metadata"]])

# Search for similar documents
query_text = "machine learning"
query_result = pipeline.process_text(query_text)
query_embedding = np.array(query_result["embedding"])

results = client.search(query_embedding, k=5)
for i, result in enumerate(results):
    print(f"{i+1}. Document: {result['metadata']['file_name']}")

# Clean up
client.close()
pipeline.close()

Components

Storage Backends

Memory Storage: Fast but not persistent

from valori.storage import MemoryStorage
storage = MemoryStorage({})

Disk Storage: Persistent but slower

from valori.storage import DiskStorage
storage = DiskStorage({"data_dir": "./my_vectordb"})

Hybrid Storage: Combines memory and disk for optimal performance

from valori.storage import HybridStorage
storage = HybridStorage({
    "memory": {},
    "disk": {"data_dir": "./my_vectordb"},
    "memory_limit": 10000
})

Index Types

Flat Index: Exhaustive search, accurate but slower for large datasets

from valori.indices import FlatIndex
index = FlatIndex({"metric": "cosine"})  # or "euclidean"

HNSW Index: Fast approximate search for large datasets

from valori.indices import HNSWIndex
index = HNSWIndex({
    "metric": "cosine",
    "m": 16,
    "ef_construction": 200,
    "ef_search": 50
})

IVF Index: Clustering-based index for large datasets

from valori.indices import IVFIndex
index = IVFIndex({
    "metric": "cosine",
    "n_clusters": 100,
    "n_probes": 10
})

LSH Index: Locality sensitive hashing for high-dimensional data

from valori.indices import LSHIndex
index = LSHIndex({
    "metric": "cosine",
    "num_hash_tables": 10,
    "hash_size": 16,
    "num_projections": 64,
    "threshold": 0.3
})

Annoy Index: Approximate nearest neighbors with random projection trees

from valori.indices import AnnoyIndex
index = AnnoyIndex({
    "metric": "angular",
    "num_trees": 10,
    "search_k": -1
})

# Add vectors, then build
index.add(vectors, metadata)
index.build()  # Required for Annoy

Document Parsing

Parse various document formats:

Text and PDF Parsing:

from valori.parsers import TextParser, PDFParser

# Parse text files
text_parser = TextParser({"encoding": "auto", "chunk_size": 1000})
result = text_parser.parse("document.txt")

# Parse PDF files
pdf_parser = PDFParser({"extract_tables": True, "chunk_size": 1000})
result = pdf_parser.parse("document.pdf")

Advanced Parsing with Docling:

from valori.parsers import DoclingParser

# Microsoft Docling for advanced parsing
docling_parser = DoclingParser({"extract_tables": True, "preserve_layout": True})

Document Processing Pipeline

Complete Processing Pipeline:

from valori.processors import ProcessingPipeline

pipeline_config = {
    "parsers": {"text": {"chunk_size": 1000}},
    "processors": {
        "cleaning": {"normalize_whitespace": True, "remove_html": True},
        "chunking": {"strategy": "semantic", "chunk_size": 1000},
        "embedding": {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
    }
}

pipeline = ProcessingPipeline(pipeline_config)
pipeline.initialize()

# Process document end-to-end
result = pipeline.process_document("document.pdf")

Quantization

Reduce memory usage with vector quantization:

Scalar Quantization:

from valori.quantization import ScalarQuantizer
quantizer = ScalarQuantizer({"bits": 8})

Product Quantization:

from valori.quantization import ProductQuantizer
quantizer = ProductQuantizer({"m": 8, "k": 256})

Advanced Usage

Complete Setup with All Components

import numpy as np
from valori import VectorDBClient
from valori.storage import HybridStorage
from valori.indices import HNSWIndex
from valori.quantization import ProductQuantizer
from valori.persistence import TensorPersistence

# Create all components
storage = HybridStorage({
    "memory": {},
    "disk": {"data_dir": "./vectordb_data"},
    "memory_limit": 10000
})

index = HNSWIndex({
    "metric": "cosine",
    "m": 32,
    "ef_construction": 400,
    "ef_search": 100
})

quantizer = ProductQuantizer({
    "m": 16,
    "k": 256
})

persistence = TensorPersistence({
    "data_dir": "./vectordb_persistence",
    "compression": True
})

# Create client
client = VectorDBClient(storage, index, quantizer, persistence)
client.initialize()

# Your vector operations here...
client.close()

Production Setup

import json
from valori.utils.logging import setup_logging

# Setup logging
setup_logging({
    "level": "INFO",
    "log_to_file": True,
    "log_file": "Valori.log"
})

# Load configuration
with open("config.json", "r") as f:
    config = json.load(f)

# Initialize with production config
client = VectorDBClient.from_config(config)
client.initialize()

# Your production code here...
client.close()

Examples

Check out the examples/ directory for comprehensive examples:

  • basic_usage.py - Basic operations and concepts
  • document_processing.py - Complete document parsing and processing workflow
  • advanced_indexing.py - LSH and Annoy indexing algorithms comparison
  • advanced_quantization.py - Quantization techniques and performance
  • production_setup.py - Production deployment and monitoring

Documentation

Full documentation is available at https://github.com/varshith-Git/valori.

Development

Setup Development Environment

# Clone the repository
git clone https://github.com/varshith-Git/valori.git
cd valori

# Setup development environment
bash scripts/install_dev.sh

# Activate virtual environment
source venv/bin/activate

Running Tests

# Run all tests
bash scripts/run_tests.sh

# Run with coverage
bash scripts/run_tests.sh --coverage

# Run specific tests
bash scripts/run_tests.sh tests/test_storage.py

Code Quality

# Format code
black src/ tests/

# Lint code
flake8 src/ tests/

# Type checking
mypy src/

# Security checks
safety check
bandit -r src/

Building Documentation

cd docs
make html

Benchmarking

# Run benchmarks
python scripts/benchmark.py

# Quick benchmarks
python scripts/benchmark.py --quick

Performance

Valori is designed for high performance:

  • Memory Efficiency: Up to 75% memory reduction with quantization
  • Search Speed: Sub-millisecond search times for small datasets
  • Scalability: Handles millions of vectors with appropriate indexing
  • Flexibility: Choose the right components for your use case

Contributing

We welcome contributions! Please see our Contributing Guide for details.

  1. Fork the repository
  2. Create a feature branch
  3. Make your changes
  4. Add tests
  5. Submit a pull request

License

This project is licensed under the MIT License - see the LICENSE file for details.

Support

Roadmap

  • GPU acceleration support
  • Distributed deployment
  • More indexing algorithms (LSH, Annoy)
  • REST API server
  • Web UI for database management
  • Integration with popular ML frameworks

Citation

If you use Valori in your research, please cite:

@software{valori2024,
  title={Valori: A High-Performance Vector Database for Python},
  author={Valori Team},
  year={2024},
  url={https://github.com/varshith-Git/valori}
}

Made with ❤️ by the Valori Team

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

valori-0.1.2.tar.gz (115.8 kB view details)

Uploaded Source

Built Distribution

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

valori-0.1.2-py3-none-any.whl (90.7 kB view details)

Uploaded Python 3

File details

Details for the file valori-0.1.2.tar.gz.

File metadata

  • Download URL: valori-0.1.2.tar.gz
  • Upload date:
  • Size: 115.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.2

File hashes

Hashes for valori-0.1.2.tar.gz
Algorithm Hash digest
SHA256 318329112d80372c39db79d1e04dade5089d3cb1f8937f464c7a1b13216bde99
MD5 c8ac37f980ffca3b6fbf0d67c1109d84
BLAKE2b-256 079f7f3f673aa8f80fe60a0cdfb6a9aed7ac234c5d8e7d1d4ea62b1565fb75f8

See more details on using hashes here.

File details

Details for the file valori-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: valori-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 90.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.2

File hashes

Hashes for valori-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 7a6cc84ebde234644fd5ebe663b4d4abda3ed9f66321ce38ceb102d8c785b12f
MD5 d8a61282a42785b70586ace7d9aec521
BLAKE2b-256 4eb97928f7d1ef31d868c73a53ce9d852f5758d6df24c42c05cab4fe6a26ebbd

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