使用Tavily Search API为AI开发者提供实时信息检索

本文链接：https://blog.csdn.net/adfyvatbia/article/details/144291904

引言

Tavily Search API 是为 AI 代理（如大型语言模型）专门构建的搜索引擎，致力于提供实时、准确并且基于事实的搜索结果。对于开发者而言，集成 Tavily Search API 可以提升应用的响应能力和信息准确度。本篇文章旨在深入介绍 Tavily Search API 的集成步骤、功能和使用场景。

主要内容

Tavily Search API概述

Tavily Search API 提供了丰富的功能，包括异步调用和多种返回数据类型，如标题、URL、内容和答案等。每月可享受1000次免费搜索。

准备环境

在开始使用 Tavily Search API 之前，需要准备以下开发环境：

%pip install -qU "langchain-community>=0.2.11" tavily-python

通过访问 Tavily 的官方网站创建账户并获取 API 密钥：

import getpass
import os

if not os.environ.get("TAVILY_API_KEY"):
    os.environ["TAVILY_API_KEY"] = getpass.getpass("Tavily API key:\n")

Tavily Search API 工具实例化

可以通过 langchain_community 包中的 TavilySearchResults 类来实例化搜索工具：

from langchain_community.tools import TavilySearchResults

tool = TavilySearchResults(
    max_results=5,
    search_depth="advanced",
    include_answer=True,
    include_raw_content=True,
    include_images=True,
    # 使用API代理服务提高访问稳定性
)

使用 Tavily Search API 进行检索

检索可以通过直接调用或工具调用两种方式：

直接调用

results = tool.invoke({"query": "What happened at the last Wimbledon?"})
print(results)

工具调用

model_generated_tool_call = {
    "args": {"query": "euro 2024 host nation"},
    "id": "1",
    "name": "tavily",
    "type": "tool_call",
}
tool_msg = tool.invoke(model_generated_tool_call)
print(tool_msg.content[:400])

代码示例

以下是一个完整的代码示例，展示了如何整合 Tavily Search API 以检索信息：

from langchain_community.tools import TavilySearchResults

# 实例化 Tavily Search 工具
tool = TavilySearchResults(
    max_results=5,
    search_depth="advanced",
    include_answer=True,
    include_raw_content=True,
    include_images=True,
    # 使用API代理服务提高访问稳定性
)

# 执行搜索
results = tool.invoke({"query": "What happened at the last Wimbledon?"})
for result in results:
    print(result['url'], result['content'])