盘古大模型 PANGULARGEMODELS-实例化Agent

时间:2024-10-15 17:57:21

实例化Agent

Agent实例化过程包括注册LLM和注册工具两个部分。

from pangukitsappdev.agent.react_pangu_agent import ReactPanguAgent
from pangukitsappdev.api.llms.factory import LLMs

agent = ReactPanguAgent(LLMs.of("pangu", llm_config=LLMConfig(llm_param_config=LLMParamConfig(with_prompt=True), llm_module_config=LLMModuleConfig(model_version="N2_agent_v2"))))
agent.set_max_iterations(5)
agent.add_tool(ReverseTool())
agent.add_tool(AddTool())
agent.add_tool(SearchTool())
  • 静态工具和动态工具的注册方式相同,通过addTool接口进行注册。
  • 通过set_max_iterations可以设置最大迭代次数,控制Agent子规划的最大迭代步数,防止无限制的迭代或出现死循环情况。
  • Agent使用的模型必须为Pangu-NLP-N2-Agent-L0.C模型,或其衍生模型,使用通用模型或其他模型无法运行。如上例所示,当前的module-version需要配置为“N2_agent_v2”,模型的相关配置需要改为Pangu-NLP-N2-Agent-L0.C模型的地址。
  • with_prompt参数配置为True,prompt的拼接由Agent托管处理。
support.huaweicloud.com/ugobs-obs-pangulm/zh-cn_topic_0000001680598878.html