Llama-index RAG: как отобразить полученный контекст? ⇐ Python

[Anonymous]

Я использую LlamaIndex для выполнения расширенной генерации поиска (RAG).
В настоящее время я могу извлекать вопросы и отвечать на них, используя следующий пример, состоящий минимум из 5 строк, из https://docs.llamaindex.ai/en/stable/ge ... r_example/:

Код: Выделить всё

from llama_index import VectorStoreIndex, SimpleDirectoryReader

documents = SimpleDirectoryReader("data").load_data()
index = VectorStoreIndex.from_documents(documents)
query_engine = index.as_query_engine()
response = query_engine.query("What did the author do growing up?")
print(response)

Это возвращает ответ, но я хотел бы отобразить полученный контекст (например, фрагменты документа или источники) перед ответом.
Желаемый формат вывода будет выглядеть примерно так:

Код: Выделить всё

Here's my retrieved context:
[x]
[y]
[z]

And here's my answer:
[answer]

Каков самый простой воспроизводимый способ изменить пример из 5 строк для достижения этой цели?

Подробнее здесь: https://stackoverflow.com/questions/792 ... ed-context