package com.ard.agent.tool;
|
|
|
import com.ard.common.core.constant.SecurityConstants;
|
import com.ard.common.core.domain.R;
|
import com.ard.system.api.RemoteDeptService;
|
import com.ard.system.api.RemoteUserService;
|
import com.ard.system.api.domain.SysDept;
|
import com.ard.system.api.domain.SysUser;
|
import com.ard.work.api.domian.ArdCamera;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.ai.tool.annotation.Tool;
|
import org.springframework.ai.tool.annotation.ToolParam;
|
import org.springframework.stereotype.Component;
|
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Component
|
@Slf4j
|
public class SystemTools {
|
|
@Resource
|
private RemoteUserService userService;
|
|
@Resource
|
private RemoteDeptService deptService;
|
|
|
|
/**
|
* 搜索用户信息
|
*/
|
@Tool(description = """
|
检索系统用户信息。根据用户名、手机号、真实姓名等条件查询用户。
|
返回用户的详细信息，包括所属部门、角色等。
|
适用场景：
|
- 查询某个用户的基本信息
|
- 根据手机号查找用户
|
- 列出某部门下的所有用户
|
""")
|
public String searchUsers(
|
@ToolParam(description = "真实姓名（模糊匹配）", required = false) String nickName,
|
@ToolParam(description = "用户名（模糊匹配）", required = false) String userName,
|
@ToolParam(description = "手机号（模糊匹配）", required = false) String phone,
|
@ToolParam(description = "部门名称（模糊匹配）", required = false) String deptName,
|
@ToolParam(description = "最大返回数量，默认10") Integer limit) {
|
|
log.info("AI调用searchUsers工具，参数: nickName={}, userName={}, phone={}, deptName={}, limit={}",
|
nickName, userName, phone, deptName, limit);
|
|
if (limit == null) limit = 10;
|
|
// 构造查询条件
|
SysUser query = new SysUser();
|
if (userName != null && !userName.isEmpty()) {
|
query.setUserName(userName);
|
}
|
if (phone != null && !phone.isEmpty()) {
|
query.setPhonenumber(phone);
|
}
|
if (deptName != null && !deptName.isEmpty()) {
|
SysDept dept = new SysDept();
|
dept.setDeptName(deptName);
|
query.setDept(dept);
|
}
|
|
|
// 调用若依的 Service 查询
|
R<List<SysUser>> result = userService.list(query, SecurityConstants.INNER);
|
if (result == null) {
|
return "错误：服务返回空结果";
|
}
|
|
if (result.getCode() != R.SUCCESS) {
|
return String.format("获取用户列表失败：%s", result.getMsg());
|
}
|
|
List<SysUser> users = result.getData();
|
// 限制返回数量
|
if (users.size() > limit) {
|
users = users.subList(0, limit);
|
}
|
|
if (users.isEmpty()) {
|
return "未找到匹配的用户";
|
}
|
|
// 格式化返回结果
|
return users.stream()
|
.map(user -> String.format("""
|
用户ID: %s
|
用户名: %s
|
真实姓名: %s
|
手机号: %s
|
邮箱: %s
|
部门: %s
|
状态: %s
|
创建时间: %s
|
""",
|
user.getUserId(),
|
user.getUserName(),
|
user.getNickName(),
|
maskPhone(user.getPhonenumber()),
|
user.getEmail(),
|
user.getDept() != null ? user.getDept().getDeptName() : "未分配",
|
"0".equals(user.getStatus()) ? "正常" : "禁用",
|
user.getCreateTime()))
|
.collect(Collectors.joining("\n---\n"));
|
}
|
|
/**
|
* 搜索部门信息
|
*/
|
@Tool(description = """
|
检索系统部门信息。根据部门名称或部门ID查询部门详情。
|
返回部门的层级结构、负责人、状态等信息。
|
适用场景：
|
- 查询某个部门的详细信息
|
- 列出所有部门
|
- 查询部门的层级关系
|
""")
|
public String searchDepts(
|
@ToolParam(description = "部门名称（支持模糊搜索），留空则返回所有部门") String deptName,
|
@ToolParam(description = "最大返回数量，默认20") Integer limit) {
|
|
log.info("AI调用searchDepts工具，参数: deptName={}, limit={}", deptName, limit);
|
|
if (limit == null) limit = 20;
|
|
SysDept query = new SysDept();
|
if (deptName != null && !deptName.isEmpty()) {
|
query.setDeptName(deptName);
|
}
|
|
R<List<SysDept>> result = deptService.list(query, SecurityConstants.INNER);
|
if (result == null) {
|
return "错误：服务返回空结果";
|
}
|
|
if (result.getCode() != R.SUCCESS) {
|
return String.format("获取部门列表失败：%s", result.getMsg());
|
}
|
|
List<SysDept> depts = result.getData();
|
if (depts.size() > limit) {
|
depts = depts.subList(0, limit);
|
}
|
|
if (depts.isEmpty()) {
|
return "未找到匹配的部门";
|
}
|
|
return depts.stream()
|
.map(dept -> String.format("""
|
部门ID: %s
|
部门名称: %s
|
父部门ID: %s
|
负责人: %s
|
联系电话: %s
|
邮箱: %s
|
状态: %s
|
""",
|
dept.getDeptId(),
|
dept.getDeptName(),
|
dept.getParentId(),
|
dept.getLeader(),
|
dept.getPhone(),
|
dept.getEmail(),
|
"0".equals(dept.getStatus()) ? "正常" : "停用"))
|
.collect(Collectors.joining("\n---\n"));
|
}
|
|
|
/**
|
* 手机号脱敏
|
*/
|
private String maskPhone(String phone) {
|
if (phone == null || phone.length() < 7) {
|
return phone;
|
}
|
return phone.substring(0, 3) + "****" + phone.substring(7);
|
}
|
|
}
|
