[Agent]-----MRKLAgentForChatModels组件开发

参考资料:

https://python.langchain.com/docs/modules/agents/agent_types/react
https://python.langchain.com/docs/modules/agents/how_to/custom_mrkl_agent
https://python.langchain.com/docs/modules/agents/how_to/mrkl

该agent主要使用ReAct框架来决定操作的代理,从而优化聊天模型。
agent需要的组件:tools,chatmodels
实现步骤:
第一步:定义一个tool,这里采用计算器

llm_math_chain = LLMMathChain(llm=llm, verbose=True)
tool =  Tool(
        name="Calculator",
        func=llm_math_chain.run,
        description="useful for when you need to answer questions about math"
    )

第二步,定义agent方法,这里的AgentType采用的是Mrkl的chatReAct模型,使用聊天模型来代替llm去调用ReAct代理

param_dict = {
    "tools":[tool],
    "model":llm
    }

from langchain.agents import initialize_agent,AgentType

agentExcutor = initialize_agent(
    llm=param_dict.get("model"),
    tools=param_dict.get("tools"),
    agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
    verbose=True
    )
resp = agentExcutor.run("写一个芒果的故事 ")
print(resp)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值