package com.ard.agent.config;
|
|
import com.alibaba.cloud.ai.graph.agent.ReactAgent;
|
import com.alibaba.cloud.ai.graph.agent.hook.modelcalllimit.ModelCallLimitHook;
|
import com.alibaba.cloud.ai.graph.checkpoint.savers.MemorySaver;
|
import com.ard.agent.tool.KnowledgeTools;
|
import com.ard.agent.tool.MonitorTools;
|
import com.ard.agent.tool.SystemTools;
|
import jakarta.annotation.Resource;
|
import org.springframework.ai.chat.model.ChatModel;
|
import org.springframework.ai.support.ToolCallbacks;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
|
@Configuration
|
public class AgentConfig {
|
|
@Resource(name = "ollamaChatModel")
|
private ChatModel chatModel;
|
@Resource
|
private MonitorTools monitorTools;
|
@Resource
|
private SystemTools systemTools;
|
@Resource
|
private KnowledgeTools knowledgeTools;
|
|
private static final String UNIFIED_AGENT_PROMPT = """
|
你是 ARD 智能业务助手，能使用摄像机管理、系统管理和知识库工具。
|
|
## 路由原则
|
- 查询或修改摄像机：使用摄像机工具。
|
- 查询人员、部门、组织架构：使用系统工具。
|
- 查询说明书、制度、操作指南、故障排查：使用知识库工具。
|
- 复合问题可以连续调用多个工具，再综合回答。
|
|
## 约束
|
- 业务数据必须来自工具调用结果。
|
- 修改类操作要在回答中明确修改对象和结果。
|
- 无法确认的数据要直接说明，不要编造。
|
|
## 输出格式
|
使用 Markdown，给出清晰结论和下一步建议。
|
""";
|
|
private final ModelCallLimitHook modelCallLimitHook = ModelCallLimitHook.builder()
|
.runLimit(8) // 最多调用8次LLM
|
.exitBehavior(ModelCallLimitHook.ExitBehavior.ERROR) // 超限时抛出异常
|
.build();
|
|
@Bean("unifiedAgent")
|
public ReactAgent unifiedAgent() {
|
return buildAgent("unifiedAgent", UNIFIED_AGENT_PROMPT, monitorTools, systemTools, knowledgeTools);
|
}
|
|
private ReactAgent buildAgent(String name, String instruction, Object... tools) {
|
return ReactAgent.builder()
|
.name(name)
|
.model(chatModel)
|
.hooks(modelCallLimitHook)
|
.instruction(instruction)
|
.tools(ToolCallbacks.from(tools))
|
.saver(new MemorySaver())
|
.enableLogging(true)
|
.build();
|
}
|
}
|
