Skip to main content

A chromadb embeddings plugin for OVOS

Project description

ChromaEmbeddingsDB Plugin

Overview

The ChromaEmbeddingsDB plugin integrates with the ChromaDB database to provide a robust solution for managing and querying embeddings. This plugin extends the abstract EmbeddingsDB class, allowing you to store, retrieve, and query embeddings efficiently using ChromaDB’s capabilities.

This plugin is meant to be used by other specialized plugins such as:

Features

  • Add Embeddings: Store embeddings with associated keys.
  • Retrieve Embeddings: Fetch embeddings by their keys.
  • Delete Embeddings: Remove embeddings by their keys.
  • Query Embeddings: Find the closest embeddings to a given query, with support for cosine distance.

Example

Here is a full example demonstrating the basic usage of ChromaEmbeddingsDB.

import numpy as np
from chroma_embeddings_db import ChromaEmbeddingsDB

# Initialize the database
db = ChromaEmbeddingsDB(path="path_to_chromadb_storage")

# Add embeddings
embedding1 = np.array([0.1, 0.2, 0.3])
embedding2 = np.array([0.4, 0.5, 0.6])
db.add_embeddings("user1", embedding1)
db.add_embeddings("user2", embedding2)

# Retrieve and print embeddings
print(db.get_embedding("user1"))
print(db.get_embedding("user2"))

# Query embeddings
query_embedding = np.array([0.2, 0.3, 0.4])
results = db.query(query_embedding, top_k=2)
print(results)

# Delete an embedding
db.delete_embedding("user1")

Ensure that the path provided to the ChromaEmbeddingsDB constructor is accessible and writable.

Acknowledgements

  • ChromaDB for providing the database backend.
  • NumPy for numerical operations.

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

ovos-chromadb-embeddings-plugin-0.0.0.tar.gz (4.1 kB view details)

Uploaded Source

Built Distribution

File details

Details for the file ovos-chromadb-embeddings-plugin-0.0.0.tar.gz.

File metadata

File hashes

Hashes for ovos-chromadb-embeddings-plugin-0.0.0.tar.gz
Algorithm Hash digest
SHA256 1fbc58a17071e9d62751751189b3310932ccf3c271fc1773ccf098aeba837cb9
MD5 2a0b97c7d855ffce226aa5ea27490a36
BLAKE2b-256 451e0fdaa673c39a5c8835ba4b74bf1a7dae53b7ccae9bff3bcff4584256aff4

See more details on using hashes here.

File details

Details for the file ovos_chromadb_embeddings_plugin-0.0.0-py3-none-any.whl.

File metadata

File hashes

Hashes for ovos_chromadb_embeddings_plugin-0.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 210ed1ec21674db02de954390ad0fe6c1512e88abfd6fbb9e248e3cb92ca32b4
MD5 88daaec54320b1a139e957fd56a219b1
BLAKE2b-256 3a5bc2836b0962f174ed98a757c54844a3d81b69e8f7993b5dba2ea670af1f1f

See more details on using hashes here.

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