llamaindex 检索器路由查询引擎

检索器路由查询引擎

概述

在本教程中，我们定义了一个基于检索器（retriever）的路由查询引擎。检索器将选择一组节点，然后我们选择正确的查询引擎（QueryEngine）。

我们使用新的 ToolRetrieverRouterQueryEngine 类来实现这一点！

安装依赖

首先，我们需要安装 LlamaIndex：

!pip install llama-index

在 Jupyter Notebook 中，我们需要应用 nest_asyncio 以允许嵌套的异步查询：

import nest_asyncio
nest_asyncio.apply()

设置日志记录：

import logging
import sys

logging.basicConfig(stream=sys.stdout, level=logging.INFO)
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))

下载数据

下载 Paul Graham 的散文：

!mkdir -p 'data/paul_graham/'
!wget 'https://raw.githubusercontent.com/run-llama/llama_index/main/doc