LlamaIndex API Example - 2

关联知识库：LlamaIndex API Example - 2

create retriever by index

from llama_index.core import SummaryIndex, SimpleDirectoryReader

documents = SimpleDirectoryReader("files").load_data()
summary_index = SummaryIndex.from_documents(documents)
retriever = summary_index.as_retriever(
retriever_mode='embedding'
)
result = retriever.retrieve("Tell me about ancient Rome")
print(result[0].text)

prefer this one

from llama_index.core import SummaryIndex, SimpleDirectoryReader
from llama_index.core.retrievers import SummaryIndexEmbeddingRetriever

documents = SimpleDirectoryReader("files").load_data()
summary_index = SummaryIndex.from_documents(documents)
retriever = SummaryIndexEmbeddingRetriever(
index=summary_index
)
result = retriever.retrieve("Tell me about ancient Rome")
print(result[0].text)

async retriever

import asyncio
from llama_index.core import KeywordTableIndex
from llama_index.core import SimpleDirectoryReader

async def retrieve(retriever, query, label):
response = await retriever.aretrieve(query)
print(f"{label} retrieved {str(len(response))} nodes")

async def main():
reader = SimpleDirectoryReader('files')
documents = reader.load_data()
index = KeywordTableIndex.from_documents(documents)
retriever1 = index.as_retriever(
retriever_mode='default'
)
retriever2 = index.as_retriever(
retriever_mode='simple'
)
query = "Where is the Colosseum?"

await asyncio.gather(
    retrieve(retriever1, query, '<llm>'),
    retrieve(retriever2, query, '<simple>')
)

asyncio.run(main())