Skip to main content

ChromaDB

Embedded vector database for local development.

Setup

pip install chromadb

Usage

from praisonai.persistence.factory import create_knowledge_store

# Local file storage
store = create_knowledge_store("chroma", path="./chroma_data")

# Create collection
store.create_collection("documents", dimension=384)

# Insert documents
from praisonai.persistence.knowledge.base import KnowledgeDocument

doc = KnowledgeDocument(
    content="PraisonAI is an AI agent framework",
    embedding=[0.1] * 384,
    metadata={"source": "docs"}
)
store.insert("documents", [doc])

# Search
results = store.search("documents", query_embedding, limit=5)

Configuration

OptionDescription
pathData directory path
collection_metadataDefault collection metadata