Perplexity 搜索集成

pip install -qU langchain-perplexity

# 以及此笔记本的一些依赖项
pip install -qU langchain langchain-openai

import getpass
import os

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

from langchain_perplexity import PerplexitySearchResults

# 初始化 PerplexitySearchResults 工具
search_tool = PerplexitySearchResults(max_results=5)

# 执行搜索查询
search_results = search_tool.invoke("When was the last time the New York Knicks won the NBA Championship?")

print("Search Results:", search_results)

from langchain_perplexity import PerplexitySearchResults

# 将结果限制在最近的时间范围和特定的域名集合内
search_tool = PerplexitySearchResults(
    max_results=10,
    country="US",
    search_recency_filter="week",
    search_domain_filter=["nytimes.com", "reuters.com", "-pinterest.com"],
)

results = search_tool.invoke("Latest AI research papers")
print(results)

from langchain_perplexity import PerplexitySearchResults

# 将结果限制在特定的日期范围内
search_tool = PerplexitySearchResults(
    max_results=5,
    search_after_date="01/01/2025",
    search_before_date="06/30/2025",
)

results = search_tool.invoke("US Federal Reserve interest rate decisions")
print(results)

import getpass

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

from langchain.agents import create_agent
from langchain.chat_models import init_chat_model
from langchain_perplexity import PerplexitySearchResults


# 初始化语言模型
model = init_chat_model(model="gpt-5.5", temperature=0)

# 初始化 Perplexity 搜索工具
perplexity_search = PerplexitySearchResults(
    max_results=5,
    search_recency_filter="month",
)

# 创建智能体
agent = create_agent(model, [perplexity_search])

# 运行智能体
user_input = "What are the latest developments in quantum computing?"

for step in agent.stream(
    {"messages": [{"role": "user", "content": user_input}]},
    stream_mode="values",
):
    step["messages"][-1].pretty_print()