如果您在colab上打开这个笔记本,您可能需要安装LlamaIndex 🦙。
%pip install llama-index-readers-mbox
!pip install llama-index
%env OPENAI_API_KEY=sk-************
from llama_index.readers.mbox import MboxReader
from llama_index.core import VectorStoreIndex
documents = MboxReader().load_data(
"mbox_data_dir", max_count=1000
) # 返回文档列表
index = VectorStoreIndex.from_documents(
documents
) # 使用文档初始化索引
query_engine = index.as_query_engine()
res = query_engine.query("When did i have that call with the London office?")
> [query] Total LLM token usage: 100 tokens > [query] Total embedding token usage: 10 tokens
res.response
> There is a call scheduled with the London office at 12am GMT on the 10th of February.