from llama_index import ServiceContext from llama_index.embeddings.openai import OpenAIEmebedding embed_model = OpenAIEmbedding(embed_batch_size=100) service_context = ServiceContext.from_defaults(..., embed_model=embed_model)