swarmauri_vectorstore_mlm 0.6.1.dev6

Swarmauri MLM Vector Store

---

MLM Vector Store

A vector store implementation using MLM (Masked Language Model) embeddings for document storage and retrieval.

Installation

pip install swarmauri_vectorstore_mlm

Usage

Here's a basic example of how to use the MLM Vector Store:

from swarmauri.documents.Document import Document
from swarmauri.vector_stores.MlmVectorStore import MlmVectorStore

# Initialize the vector store
vs = MlmVectorStore()

# Create some documents
documents = [
    Document(content="first document"),
    Document(content="second document"),
    Document(content="third document")
]

# Add documents to the store
vs.add_documents(documents)

# Retrieve similar documents
results = vs.retrieve(query="document", top_k=2)

Want to help?

If you want to contribute to swarmauri-sdk, read up on our guidelines for contributing that will help you get started.