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)