1  from llama_index.core import Document, VectorStoreIndex
2  
3  import mlflow
4  
5  index = VectorStoreIndex.from_documents(documents=[Document.example()])
6  retriever = index.as_retriever()
7  
8  mlflow.models.set_model(retriever)