from docarray.index import MongoAtlasDocumentIndex
import numpy as np

NDIM = 10

class MyDoc(BaseDoc):
    text: str
    embedding: NdArray[NDIM]
    number: int

docs = [MyDoc(text=f'text {i}', embedding=np.random.randn(NDIM), number=i) for i in range(5)]
index = MongoAtlasDocumentIndex[MyDoc](mongo_connection_uri='localhost')
index.index(docs)
    query = (
        index.build_query()
        .find(query=np.ones(NDIM), search_field='embedding')
        .filter(query={"number": {"$gt": 0}})
        .text_search(query="text 3", search_field='text')
        .build(limit=n_docs)