Skip to main content

VectorizeDB is a database for vectorized data and metadata, allowing for fast similarity search and retrieval.

Project description

Overview

VectorizeDB is a Python package designed for the efficient storage and retrieval of high-dimensional vectors. It’s particularly useful in applications like machine learning and information retrieval. The package utilizes hnswlib for fast approximate nearest neighbor searches and LMDB for scalable and reliable storage.

Installation

To install VectorizeDB, ensure you have Python 3.10 or higher. It can be installed via pip:

pip install vectorizedb

Usage

Initialization

from vectorizedb import Database

# Initialize a new database
db = Database(path="path/to/db", dim=128, readonly=False, similarity="cosine")

Adding Data

import numpy as np

# Add a vector with an associated key
db.add(key="sample_key", vector=np.random.rand(128))

# Add a vector with metadata
db.add(key="another_key", vector=np.random.rand(128), metadata={"info": "sample metadata"})

# Another way to add data
db["yet_another_key"] = (np.random.rand(128), {"info": "sample metadata"})

Retrieving Data

# Retrieve vector and metadata by key
vector, metadata = db["sample_key"]

# Check if a key exists in the database
exists = "sample_key" in db

Searching

# Search for nearest neighbors of a vector
results = db.search(vector=np.random.rand(128), k=5)
for key, vector, distance, metadata in results:
    print(key, distance, metadata)

Iterating Through Data

# Iterate through all keys, vectors and metadata in the database
for key, vector, metadata in db:
    print(key, metadata)

Updating Data

# Update a vector in the database
db.update_vector("sample_key", np.random.rand(128))

# Update metadata
db.update_metadata("sample_key", {"info": "updated metadata"})

Deleting Data

# Delete a vector from the database by key
del db["sample_key"]

Database Length

# Get the number of entries in the database
length = len(db)

License

VectorizeDB is released under the Apache License. For more details, see the LICENSE file included in the package.

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

vectorizedb-1.0.0.tar.gz (8.3 kB view hashes)

Uploaded Source

Built Distribution

vectorizedb-1.0.0-py3-none-any.whl (9.0 kB view hashes)

Uploaded Python 3

Supported by

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