在 LangChain 中,如何使用 AgentExecutor 替代已被禁用的 initialize_agent 函數?

在 LangChain 中,如何使用 AgentExecutor 替代已被禁用的 initialize_agent 函數?

LangChain 中 initialize_agent 函數的替代方案:使用 AgentExecutor

LangChain 的 initialize_agent 函數已被棄用,推薦使用更靈活的 AgentExecutor 類來初始化和運行代理。 AgentExecutor 提供了更精細的控制和定制選項。 下面是使用 AgentExecutor 的步驟:

1. 導入必要的模塊:

from langchain.agents import AgentExecutor from langchain.agents.tools import Tool

2. 定義工具 (Tools):

代理需要一系列工具來完成任務。這些工具可以是函數、API 調用或其他可執行單元。 例如:

def search_tool(query):     #  此處實現你的搜索邏輯,例如使用 SerpAPI 或其他搜索引擎     #  ...     return "搜索結果"  tools = [     Tool(         name="Search",         func=search_tool,         description="用于搜索信息的工具,輸入查詢關鍵詞。"     ) ]

3. 創建代理 (Agent):

選擇合適的代理類型,例如 ZeroShotAgent 或其他。 以下示例使用 ZeroShotAgent:

from langchain.agents import ZeroShotAgent from langchain import LLMChain, SerpAPIWrapper  # 替換為你的LLM和搜索工具  #  替換為你的LLM llm = ...   search = SerpAPIWrapper() # 或其他搜索工具 tools = [     Tool(         name="Calculator",         func=lambda x: str(eval(x)), #簡單的計算器示例,實際應用中需更完善的處理         description="用于進行簡單數學計算的工具,輸入數學表達式。"     ),     Tool(         name="Search",         func=search.run,         description="用于搜索信息的工具,輸入查詢關鍵詞。"     ) ]  prompt = ZeroShotAgent.create_prompt(     tools,     prefix="請根據以下工具回答問題:",     suffix="開始!" )  llm_chain = LLMChain(llm=llm, prompt=prompt) agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools)

4. 初始化并使用 AgentExecutor:

agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)  response = agent_executor.run("東京的人口是多少?") print(response)

通過以上步驟,你便可以使用 AgentExecutor 來替代被禁用的 initialize_agent 函數,并更有效地管理和運行你的 LangChain 代理。 記住替換示例代碼中的占位符 (…) 為你實際使用的 LLM 和工具。 verbose=True 將打印代理的推理過程,方便調試。

? 版權聲明
THE END
喜歡就支持一下吧
點贊9 分享