模块
自定义代理(Custom Agent)

自定义代理人

这个笔记本将介绍如何创建自己的自定义代理人。

一个代理人由三个部分组成:

  • 工具:代理可以使用的工具。
  • 代理人类本身:这决定了采取哪些行动。

在这个笔记本里,我们将介绍如何创建自定义代理人。

from langchain.agents import Tool, AgentExecutor, BaseSingleActionAgent
from langchain import OpenAI, SerpAPIWrapper
search = SerpAPIWrapper()
tools = [
    Tool(
        name = "Search",
        func=search.run,
        description="useful for when you need to answer questions about current events",
        return_direct=True
    )
]
from typing import List, Tuple, Any, Union
from langchain.schema import AgentAction, AgentFinish

class FakeAgent(BaseSingleActionAgent):
 """Fake Custom Agent."""
    
    @prop