Skip to main content
Open In ColabOpen on GitHub

ValyuContext

Valyu 允许 AI 应用程序和代理搜索互联网和专有数据源,以获取相关的 LLM 就绪信息。

本笔记将介绍如何在 LangChain 中使用 Valyu context 工具。

首先,获取 Valyu API 密钥并将其添加为环境变量。通过在此处注册即可获得 10 美元的免费积分。

概览

集成详情

ClassPackageSerializableJS supportPackage latest
Valyu Searchlangchain-valyuPyPI - Version

设置

该集成位于 langchain-valyu 包中。

%pip install -qU langchain-valyu

为了使用该包,您还需要将 VALYU_API_KEY 环境变量设置为您的 Valyu API 密钥。

import getpass
import os

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

实例化

这里我们展示如何实例化 Valyu 搜索工具。该工具允许您通过 Valyu 的 Context API 端点完成搜索查询。

from langchain_valyu import ValyuSearchTool

tool = ValyuSearchTool()

调用

直接使用参数调用

Valyu 搜索工具在调用时接受以下参数:

  • query (必需): 自然语言搜索查询
  • search_type (可选): 搜索类型,例如 "all"
  • max_num_results (可选): 返回结果的最大数量
  • similarity_threshold (可选): 结果的相似度阈值
  • query_rewrite (可选): 是否重写查询
  • max_price (可选): 搜索的最大价格

出于可靠性和性能原因,某些参数可能是必需的或受限的。有关详情,请参阅 Valyu API 文档

search_results = tool._run(
query="What are agentic search-enhanced large reasoning models?",
search_type="all",
max_num_results=5,
similarity_threshold=0.4,
query_rewrite=False,
max_price=20.0,
)

print("Search Results:", search_results)

在 Agent 中使用

我们可以通过将工具绑定到 agent 来直接在 agent executor 中使用我们的工具。这使得 agent 能够动态地为 Valyu 搜索工具设置可用参数。

if not os.environ.get("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = getpass.getpass("OPENAI_API_KEY:\n")
from langchain_valyu import ValyuSearchTool
from langgraph.prebuilt import create_react_agent

valyu_search_tool = ValyuSearchTool()

agent = create_react_agent(llm, [valyu_search_tool])

user_input = "What are the key factors driving recent stock market volatility, and how do macroeconomic indicators influence equity prices across different sectors?"

for step in agent.stream(
{"messages": user_input},
stream_mode="values",
):
step["messages"][-1].pretty_print()
API Reference:create_react_agent

API 参考

有关所有 Valyu Context API 功能和配置的详细文档,请参阅 API 参考:https://docs.valyu.network/overview