from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
from llama_index.embeddings.openai import OpenAIEmbedding
from llama_index.llms.openai import OpenAI
docs = SimpleDirectoryReader("./docs").load_data
index = VectorStoreIndex.from_documents(docs, embed_model=OpenAIEmbedding("text-embedding-3-small"))
query_engine = index.as_query_engine(llm=OpenAI(model="gpt-4o-mini"))
print(query_engine.query("What are the key policies?"))