332 lines
15 KiB
Java
332 lines
15 KiB
Java
// AuditContent4TargetServiceImpl.java
|
||
package com.gxwebsoft.ai.service.impl;
|
||
|
||
import com.alibaba.fastjson.JSONArray;
|
||
import com.alibaba.fastjson.JSONObject;
|
||
import com.gxwebsoft.ai.constants.AuditContent4TargetConstants;
|
||
import com.gxwebsoft.ai.service.AuditContent4TargetService;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.stereotype.Service;
|
||
|
||
import cn.hutool.core.util.StrUtil;
|
||
|
||
import java.util.*;
|
||
import java.util.stream.Collectors;
|
||
|
||
@Slf4j
|
||
@Service
|
||
public class AuditContent4TargetServiceImpl extends AbstractAuditContentService implements AuditContent4TargetService {
|
||
|
||
// 工作流配置
|
||
private static final String DIFY_WORKFLOW_TOKEN = "Bearer app-Z6YPjz3nAB46roFnZSfCTk3w";
|
||
|
||
@Override
|
||
public JSONObject generateTargetAuditTableData(String kbIds, String libraryKbIds, String projectLibrary,
|
||
String userName, String history, String suggestion) {
|
||
log.info("开始生成目标责任制完成情况审计表数据 - 用户: {}, kbIds: {}, libraryIds: {}, projectLibrary: {}",
|
||
userName, kbIds, libraryKbIds, projectLibrary);
|
||
|
||
long startTime = System.currentTimeMillis();
|
||
|
||
try {
|
||
// 1. 检索相关知识
|
||
Map<String, List<String>> knowledgeSources = retrieveKnowledgeForTargetResponsibility(
|
||
kbIds, libraryKbIds, projectLibrary
|
||
);
|
||
|
||
// 2. 构建完整的知识上下文
|
||
String knowledgeContext = buildCompleteKnowledgeContext(
|
||
knowledgeSources, history, suggestion
|
||
);
|
||
|
||
// 3. 调用工作流生成数据
|
||
JSONObject requestBody = buildWorkflowRequest(knowledgeContext, userName);
|
||
JSONArray auditData = callWorkflow(DIFY_WORKFLOW_URL, DIFY_WORKFLOW_TOKEN, requestBody, "目标责任制审计");
|
||
|
||
// 4. 后处理：确保数据格式正确
|
||
auditData = processAuditData(auditData);
|
||
|
||
log.info("目标责任制完成情况审计表生成成功 - 记录数: {}, 处理时间: {}ms",
|
||
auditData.size(), (System.currentTimeMillis() - startTime));
|
||
|
||
return buildSuccessResponse(auditData, startTime, "target_audit");
|
||
|
||
} catch (Exception e) {
|
||
log.error("生成目标责任制完成情况审计表失败", e);
|
||
return buildErrorResponse("生成目标责任制完成情况审计表失败: " + e.getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检索目标责任制相关知识
|
||
*/
|
||
private Map<String, List<String>> retrieveKnowledgeForTargetResponsibility(String kbIds, String libraryKbIds, String projectLibrary) {
|
||
Map<String, List<String>> knowledgeSources = new HashMap<>();
|
||
knowledgeSources.put("enterprise", new ArrayList<>());
|
||
knowledgeSources.put("regulation", new ArrayList<>());
|
||
knowledgeSources.put("auditCase", new ArrayList<>());
|
||
|
||
// 构建查询词
|
||
List<String> queries = buildTargetResponsibilityQueries();
|
||
|
||
// 企业单位库检索
|
||
if (StrUtil.isNotBlank(kbIds)) {
|
||
Arrays.stream(kbIds.split(","))
|
||
.map(String::trim)
|
||
.filter(StrUtil::isNotBlank)
|
||
.forEach(kbId -> knowledgeSources.get("enterprise")
|
||
.addAll(queryKnowledgeBase(kbId, queries, 100)));
|
||
}
|
||
|
||
// 公共法律法规库检索
|
||
if (StrUtil.isNotBlank(libraryKbIds)) {
|
||
Arrays.stream(libraryKbIds.split(","))
|
||
.map(String::trim)
|
||
.filter(StrUtil::isNotBlank)
|
||
.forEach(libId -> knowledgeSources.get("regulation")
|
||
.addAll(queryKnowledgeBase(libId, queries, 80)));
|
||
}
|
||
|
||
// 审计案例库检索
|
||
if (StrUtil.isNotBlank(projectLibrary)) {
|
||
knowledgeSources.get("auditCase").addAll(
|
||
queryKnowledgeBase(projectLibrary, queries, 60));
|
||
}
|
||
|
||
// 智能去重和排序
|
||
knowledgeSources.forEach((key, list) -> {
|
||
List<String> processed = list.stream()
|
||
.distinct()
|
||
.sorted(this::targetResponsibilityComparator)
|
||
.limit(getLimitBySourceType(key))
|
||
.collect(Collectors.toList());
|
||
knowledgeSources.put(key, processed);
|
||
});
|
||
|
||
log.debug("目标责任制知识检索完成 - 企业: {}条, 法规: {}条, 案例: {}条",
|
||
knowledgeSources.get("enterprise").size(),
|
||
knowledgeSources.get("regulation").size(),
|
||
knowledgeSources.get("auditCase").size());
|
||
|
||
return knowledgeSources;
|
||
}
|
||
|
||
/**
|
||
* 构建目标责任制查询词
|
||
*/
|
||
private List<String> buildTargetResponsibilityQueries() {
|
||
return Arrays.asList(
|
||
"目标责任 目标责任制 目标任务",
|
||
"下达文件 计划文件 考核办法",
|
||
"完成情况 未完成原因 考核结果",
|
||
"地矿局 上级主管 主管部门",
|
||
"年度计划 工作目标 考核指标",
|
||
"责任书 责任状 目标考核",
|
||
"经营目标 工作计划 绩效考核",
|
||
"重点工作 专项任务 年度总结"
|
||
);
|
||
}
|
||
|
||
/**
|
||
* 构建完整的知识上下文
|
||
*/
|
||
private String buildCompleteKnowledgeContext(Map<String, List<String>> knowledgeSources,
|
||
String history, String suggestion) {
|
||
StringBuilder context = new StringBuilder();
|
||
|
||
// 1. 核心审计任务
|
||
context.append("## 核心审计任务\n");
|
||
context.append("审计分类：").append(AuditContent4TargetConstants.CATEGORY_TARGET_RESPONSIBILITY).append("\n");
|
||
context.append("审计描述：检查被审计领导干部任职期间，对照上级主管部门下达的目标责任或单位自定的目标责任，检查其完成情况\n\n");
|
||
|
||
// 2. 审计框架（审计规则）
|
||
context.append("## 审计框架（审计规则）\n");
|
||
context.append("以下审计框架定义了审计范围和要点，请基于此框架开展工作：\n");
|
||
context.append(AuditContent4TargetConstants.AUDIT_FRAMEWORK).append("\n\n");
|
||
|
||
// 3. 审计目标
|
||
context.append("## 审计目标\n");
|
||
context.append(AuditContent4TargetConstants.AUDIT_OBJECTIVE).append("\n\n");
|
||
|
||
// 4. 审计工作原则
|
||
context.append("## 审计工作原则\n");
|
||
context.append(AuditContent4TargetConstants.AUDIT_PRINCIPLES).append("\n\n");
|
||
|
||
// 5. 企业单位知识（主要考察内容）
|
||
if (!knowledgeSources.get("enterprise").isEmpty()) {
|
||
context.append("## 企业单位知识（具体考察内容）\n");
|
||
context.append("以下是企业单位的实际资料，请基于审计框架，结合这些具体内容生成审计记录：\n\n");
|
||
knowledgeSources.get("enterprise").forEach(knowledge ->
|
||
context.append("• ").append(knowledge).append("\n"));
|
||
context.append("\n");
|
||
}
|
||
|
||
// 6. 审计要点提示
|
||
context.append("## 审计要点提示\n");
|
||
context.append(AuditContent4TargetConstants.AUDIT_KEY_POINTS).append("\n\n");
|
||
|
||
// 7. 数据格式要求
|
||
context.append("## 数据格式要求\n");
|
||
context.append("请生成JSON数组格式的审计表数据，每个审计条目包含以下字段：\n");
|
||
context.append(AuditContent4TargetConstants.OUTPUT_FORMAT).append("\n\n");
|
||
|
||
// 8. 审计证据要求
|
||
context.append("## 审计证据要求\n");
|
||
context.append(AuditContent4TargetConstants.AUDIT_EVIDENCE_REQUIREMENTS).append("\n\n");
|
||
|
||
// 9. 特别提醒（新增关键优化点）
|
||
context.append("## 特别提醒\n");
|
||
context.append("1. 审计证据必须包含具体的查阅过程：查阅了哪些文件、文件中发现了什么内容\n");
|
||
context.append("2. workPaperIndex必须填写实际存在的完整文件名，确保能在文件夹中搜索到\n");
|
||
context.append("3. 完成情况判定必须基于充分证据：只有证据充分且完全符合要求才能判定为已完成\n");
|
||
context.append("4. 对于执行不到位、效果不佳、证据不充分的情况必须在备注中说明\n");
|
||
context.append("5. 尽可能识别知识库中所有相关年度，生成对应的审计记录\n");
|
||
context.append("6. 如果知识库中没有单位自定目标文件，请根据企业职责和上级目标合理推断单位自定计划内容\n");
|
||
context.append("7. 不能填写简单的\"无\"，要提供有意义的描述\n");
|
||
context.append("8. 对于正在进行的年度，完成情况应填写\"年度未结束无法评估\"\n");
|
||
context.append("9. 如果知识库中有多个目标责任文件，每个文件都应生成独立的审计记录\n");
|
||
context.append("10. 同一年度可能有不同类型的多个目标责任，应尽可能多地生成审计记录\n");
|
||
context.append("11. 不能填写简单的\"无\"，要提供有意义的描述\n");
|
||
context.append("12. 对于正在进行的年度，完成情况应填写\"无法评估\"\n\n");
|
||
|
||
// 10. 法规和案例参考
|
||
if (!knowledgeSources.get("regulation").isEmpty()) {
|
||
context.append("## 法律法规参考\n");
|
||
knowledgeSources.get("regulation").forEach(knowledge ->
|
||
context.append("• ").append(knowledge).append("\n"));
|
||
context.append("\n");
|
||
}
|
||
|
||
if (!knowledgeSources.get("auditCase").isEmpty()) {
|
||
context.append("## 审计案例参考\n");
|
||
knowledgeSources.get("auditCase").forEach(knowledge ->
|
||
context.append("• ").append(knowledge).append("\n"));
|
||
context.append("\n");
|
||
}
|
||
|
||
// 11. 历史内容（如果有）
|
||
if (StrUtil.isNotBlank(history)) {
|
||
context.append("## 历史生成内容\n");
|
||
context.append(history).append("\n\n");
|
||
}
|
||
|
||
// 12. 用户建议（如果有）
|
||
if (StrUtil.isNotBlank(suggestion)) {
|
||
context.append("## 用户建议\n");
|
||
context.append(suggestion).append("\n\n");
|
||
}
|
||
|
||
log.debug("目标责任制审计知识上下文长度: {} 字符", context.length());
|
||
|
||
return context.toString();
|
||
}
|
||
|
||
/**
|
||
* 目标责任制审计相关性比较器
|
||
*/
|
||
private int targetResponsibilityComparator(String reg1, String reg2) {
|
||
int score1 = calculateTargetResponsibilityRelevanceScore(reg1);
|
||
int score2 = calculateTargetResponsibilityRelevanceScore(reg2);
|
||
return Integer.compare(score2, score1); // 降序排序
|
||
}
|
||
|
||
/**
|
||
* 计算目标责任制审计相关性分数
|
||
*/
|
||
private int calculateTargetResponsibilityRelevanceScore(String content) {
|
||
return AuditContent4TargetConstants.KEYWORD_WEIGHTS.entrySet().stream()
|
||
.filter(entry -> content.contains(entry.getKey()))
|
||
.mapToInt(Map.Entry::getValue)
|
||
.sum();
|
||
}
|
||
|
||
private int getLimitBySourceType(String sourceType) {
|
||
switch (sourceType) {
|
||
case "enterprise": return 200; // 企业知识数量
|
||
case "regulation": return 80; // 法规数量
|
||
case "auditCase": return 60; // 案例数量
|
||
default: return 50;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 后处理审计数据（优化字段处理逻辑）
|
||
*/
|
||
private JSONArray processAuditData(JSONArray auditData) {
|
||
if (auditData == null || auditData.isEmpty()) {
|
||
return auditData;
|
||
}
|
||
|
||
JSONArray processedData = new JSONArray();
|
||
|
||
for (int i = 0; i < auditData.size(); i++) {
|
||
JSONObject item = auditData.getJSONObject(i);
|
||
|
||
// 确保序号存在
|
||
if (!item.containsKey("index") || item.getInteger("index") == null) {
|
||
item.put("index", i + 1);
|
||
}
|
||
|
||
// 确保年度存在
|
||
if (!item.containsKey("year") || StrUtil.isBlank(item.getString("year"))) {
|
||
item.put("year", "未知年度");
|
||
}
|
||
|
||
// 优化单位自定计划字段：不能是简单的"无"
|
||
String selfPlan = item.getString("selfPlan");
|
||
if (StrUtil.isBlank(selfPlan) || "无".equals(selfPlan)) {
|
||
String year = item.getString("year");
|
||
item.put("selfPlan", year + "年度经营工作计划");
|
||
}
|
||
|
||
// 优化自定完成情况：不能是简单的"无"
|
||
String selfCompletion = item.getString("selfCompletion");
|
||
if (StrUtil.isBlank(selfCompletion) || "无".equals(selfCompletion)) {
|
||
String superiorCompletion = item.getString("superiorCompletion");
|
||
if ("已完成".equals(superiorCompletion)) {
|
||
item.put("selfCompletion", "已完成");
|
||
} else if ("部分完成".equals(superiorCompletion)) {
|
||
item.put("selfCompletion", "部分完成");
|
||
} else if ("未完成".equals(superiorCompletion)) {
|
||
item.put("selfCompletion", "未完成");
|
||
} else if (superiorCompletion != null &&
|
||
(superiorCompletion.contains("尚未完成") || superiorCompletion.contains("年度未结束"))) {
|
||
item.put("selfCompletion", "年度未结束无法评估");
|
||
} else {
|
||
item.put("selfCompletion", "未提供完成情况数据");
|
||
}
|
||
}
|
||
|
||
// 优化自定未完成原因：不能是简单的"无"
|
||
String selfReason = item.getString("selfReason");
|
||
if (StrUtil.isBlank(selfReason) || "无".equals(selfReason)) {
|
||
String selfComp = item.getString("selfCompletion");
|
||
if ("已完成".equals(selfComp)) {
|
||
item.put("selfReason", "无未完成原因");
|
||
} else if ("部分完成".equals(selfComp)) {
|
||
item.put("selfReason", "部分指标未达预期");
|
||
} else if ("未完成".equals(selfComp)) {
|
||
item.put("selfReason", "经营目标未实现");
|
||
} else if ("年度未结束无法评估".equals(selfComp)) {
|
||
item.put("selfReason", "年度尚未结束，无法评估");
|
||
} else {
|
||
item.put("selfReason", "未提供原因分析");
|
||
}
|
||
}
|
||
|
||
// 确保工作底稿索引是数组格式
|
||
if (!item.containsKey("workPaperIndex") || !(item.get("workPaperIndex") instanceof JSONArray)) {
|
||
JSONArray workPaperIndex = new JSONArray();
|
||
String year = item.getString("year");
|
||
workPaperIndex.add(year + "年度目标责任书");
|
||
workPaperIndex.add(year + "年度工作总结报告");
|
||
item.put("workPaperIndex", workPaperIndex);
|
||
}
|
||
|
||
processedData.add(item);
|
||
}
|
||
|
||
return processedData;
|
||
}
|
||
|
||
} |