package com.ruoyi.mock.prompt;
|
|
import com.ruoyi.project.system.domain.SysUser;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* 各模块的 AI Prompt 模板
|
*/
|
public final class MockDataPrompt {
|
|
private MockDataPrompt() {}
|
|
public static String buildSystemPrompt() {
|
return """
|
你是一个企业ERP系统的数据模拟专家。
|
|
核心规则（必须严格遵守）：
|
1. 输出必须是纯JSON数组，不要用markdown代码块包裹，不要有任何其他文字
|
2. 每个JSON对象必须包含 "entity" 字段，值为实体类型名
|
3. 数据内容要符合指定行业的特征
|
4. 金额、数量等数值字段要合理
|
5. 同一模块内的实体之间要有引用关系
|
6. 不要使用emoji或特殊unicode字符
|
7. 最重要：日期只从"指定时间范围"中取，绝对不要用其他年份！
|
8. 人员字段（如entryPerson、salesman等）必须从"系统用户列表"中选取，根据角色和部门匹配对应模块，不要编造人名！
|
""";
|
}
|
|
public static String buildUserMessage(List<String> modules, List<String> industries,
|
int countMin, int countMax,
|
String dateStart, String dateEnd,
|
String additionalInfo,
|
Map<Long, String> productModelIdToCategory,
|
List<SysUser> systemUsers,
|
List<String> existingCustomerNames,
|
List<String> existingSupplierNames) {
|
StringBuilder sb = new StringBuilder();
|
sb.append("指定行业: ").append(String.join("、", industries)).append("\n");
|
sb.append("每种实体生成: ").append(countMin).append("-").append(countMax).append(" 条\n");
|
sb.append("指定时间范围: ").append(dateStart).append(" 到 ").append(dateEnd);
|
sb.append("（所有日期字段必须在这个范围内！禁止使用2023、2024、2025等任何超出范围的年份！）\n");
|
if (additionalInfo != null && !additionalInfo.isBlank()) {
|
sb.append("补充信息: ").append(additionalInfo).append("\n");
|
}
|
sb.append("\n");
|
|
// 按顶级产品分类（成品/原材料/半成品）分组列出产品规格
|
Map<String, List<Map.Entry<Long, String>>> grouped = productModelIdToCategory.entrySet().stream()
|
.collect(Collectors.groupingBy(
|
e -> categorize(e.getValue()),
|
java.util.LinkedHashMap::new,
|
Collectors.toList()));
|
|
sb.append("=== 产品规格列表 ===\n");
|
for (Map.Entry<String, List<Map.Entry<Long, String>>> entry : grouped.entrySet()) {
|
String label = entry.getKey();
|
List<Map.Entry<Long, String>> items = entry.getValue();
|
sb.append("【").append(label).append("】: ");
|
sb.append(items.stream()
|
.map(e -> e.getKey() + "(" + e.getValue() + ")")
|
.collect(Collectors.joining(", "))).append("\n");
|
}
|
sb.append("销售台账的productData只能选【成品类】规格，采购台账只能选【原材料类】规格。\n\n");
|
|
// 系统用户列表（含角色和部门）
|
sb.append("=== 系统用户列表（人员字段必须从中选取，不要编造人名）===\n");
|
for (SysUser u : systemUsers) {
|
String roleNames = u.getRoleNames() != null && !u.getRoleNames().isBlank() ? u.getRoleNames() : "无角色";
|
String deptNames = u.getDeptNames() != null && !u.getDeptNames().isBlank() ? u.getDeptNames() : "无部门";
|
sb.append(u.getNickName()).append("（角色: ").append(roleNames).append(", 部门: ").append(deptNames).append("）\n");
|
}
|
sb.append("选择规则：\n");
|
sb.append("- 销售模块的人员(entryPerson、salesman)选角色或部门含\"销售\"的用户\n");
|
sb.append("- 采购模块的人员选角色或部门含\"采购\"的用户\n");
|
sb.append("- 生产模块的人员选角色或部门含\"生产\"的用户\n");
|
sb.append("- 质量模块的人员选角色或部门含\"质量\"或\"质检\"的用户\n");
|
sb.append("- 仓库模块的人员选角色或部门含\"仓库\"或\"库存\"的用户\n\n");
|
|
if (!existingCustomerNames.isEmpty()) {
|
sb.append("=== 已有客户（只能引用，不要生成新customer实体）===\n");
|
sb.append(String.join("、", existingCustomerNames)).append("\n\n");
|
}
|
if (!existingSupplierNames.isEmpty()) {
|
sb.append("=== 已有供应商（只能引用，不要生成新supplier实体）===\n");
|
sb.append(String.join("、", existingSupplierNames)).append("\n\n");
|
}
|
|
sb.append("数据模块: ").append(String.join("、", modules)).append("\n\n");
|
|
if (modules.contains("sales")) {
|
sb.append(buildSalesPrompt(countMin, countMax, dateStart, dateEnd, existingCustomerNames));
|
}
|
if (modules.contains("purchase")) {
|
sb.append(buildPurchasePrompt(countMin, countMax, dateStart, dateEnd, existingSupplierNames));
|
}
|
if (modules.contains("quality")) {
|
sb.append(buildQualityPrompt(countMin, countMax, dateStart));
|
}
|
if (modules.contains("production")) {
|
sb.append(buildProductionPrompt(countMin, countMax, dateStart, dateEnd));
|
}
|
if (modules.contains("stock")) {
|
sb.append(buildStockPrompt(countMin, countMax, dateStart));
|
}
|
|
sb.append("\n请直接输出JSON数组:");
|
return sb.toString();
|
}
|
|
/**
|
* 将顶级产品名称映射为分类标签
|
*/
|
private static String categorize(String rootProductName) {
|
if (rootProductName == null) return "其他类";
|
if (rootProductName.contains("成品") || rootProductName.contains("产成品")) return "成品类-销售优先选这些";
|
if (rootProductName.contains("原料") || rootProductName.contains("原材料") || rootProductName.contains("材料")) return "原材料类-采购优先选这些";
|
if (rootProductName.contains("半成品")) return "半成品类";
|
return "其他类";
|
}
|
|
private static String buildSalesPrompt(int min, int max, String dateStart, String dateEnd,
|
List<String> existingCustomerNames) {
|
String namesHint = existingCustomerNames.isEmpty() ? ""
|
: "customerName必须从以下已有客户中选择: " + String.join("、", existingCustomerNames) + "\n";
|
return """
|
销售模块格式（所有日期必须在 %s ~ %s 范围内）:
|
%s注意：不要生成customer实体，只生成salesLedger！
|
productData中的productModelId必须选【成品类】规格ID！
|
entryPerson和salesman必须从系统用户列表中角色或部门含"销售"的用户中选取！
|
{
|
"entity": "salesLedger",
|
"customerName": "引用上面生成的客户名称",
|
"salesContractNo": "XS-YYYYMMDD-001",
|
"projectName": "XX项目",
|
"entryDate": "%s",
|
"entryPerson": "从用户列表中选",
|
"salesman": "从用户列表中选",
|
"contractAmount": 50000.00,
|
"paymentMethod": "月结30天",
|
"executionDate": "%s",
|
"deliveryDate": "%s",
|
"type": 1,
|
"productData": [{
|
"productId": 1, "productModelId": 1, "quantity": 100,
|
"taxInclusiveUnitPrice": 500.00, "taxInclusiveTotalPrice": 50000.00,
|
"taxExclusiveTotalPrice": 44247.79, "taxRate": 13.00, "unit": "件", "type": 1
|
}]
|
}
|
salesContractNo中YYYYMMDD替换为entryDate的日期。%d-%d条。
|
""".formatted(dateStart, dateEnd, namesHint, dateStart, dateStart, dateEnd, min, max);
|
}
|
|
private static String buildPurchasePrompt(int min, int max, String dateStart, String dateEnd,
|
List<String> existingSupplierNames) {
|
String namesHint = existingSupplierNames.isEmpty() ? ""
|
: "supplierName必须从以下已有供应商中选择: " + String.join("、", existingSupplierNames) + "\n";
|
return """
|
采购模块格式（所有日期必须在 %s ~ %s 范围内）:
|
%s注意：不要生成supplier实体，只生成purchaseLedger！
|
productData中的productModelId必须选【原材料类】规格ID！
|
人员字段必须从系统用户列表中角色或部门含"采购"的用户中选取！
|
{
|
"entity": "purchaseLedger",
|
"supplierName": "从已有供应商中选择",
|
"purchaseContractNumber": "CG-YYYYMMDD-001",
|
"projectName": "XX采购项目",
|
"entryDate": "%s",
|
"contractAmount": 30000.00,
|
"paymentMethod": "货到付款",
|
"executionDate": "%s",
|
"productData": [{
|
"productId": 1, "productModelId": 1, "quantity": 50,
|
"taxInclusiveUnitPrice": 600.00, "taxInclusiveTotalPrice": 30000.00,
|
"taxExclusiveTotalPrice": 26548.67, "taxRate": 13.00, "unit": "件", "type": 2
|
}]
|
}
|
purchaseContractNumber中YYYYMMDD替换为entryDate的日期。%d-%d条。
|
""".formatted(dateStart, dateEnd, namesHint, dateStart, dateStart, min, max);
|
}
|
|
private static String buildQualityPrompt(int min, int max, String dateStart) {
|
return """
|
质量模块格式:
|
{
|
"entity": "qualityTestStandard",
|
"standardNo": "QTS-YYYYMMDD-001",
|
"standardName": "XX产品检验标准",
|
"inspectType": 0,
|
"remark": "适用于XX行业的质量检验标准"
|
}
|
inspectType: 0=原材料检验, 1=过程检验, 2=出厂检验。三种类型都要覆盖。
|
standardNo中YYYYMMDD替换为 %s。%d-%d条。
|
{
|
"entity": "qualityTestStandardBinding",
|
"productId": 1,
|
"testStandardId": 1
|
}
|
%d-%d条。
|
""".formatted(dateStart.replace("-", ""), min, max, min, max);
|
}
|
|
private static String buildProductionPrompt(int min, int max, String dateStart, String dateEnd) {
|
return """
|
生产模块格式（所有日期必须在 %s ~ %s 范围内）:
|
{
|
"entity": "productionPlan",
|
"productModelId": 1,
|
"qtyRequired": 200,
|
"requiredDate": "%s",
|
"source": "销售",
|
"promisedDeliveryDate": "%s",
|
"remark": "XX客户订单需求"
|
}
|
{
|
"entity": "productionOrder",
|
"productModelId": 1,
|
"quantity": 200,
|
"planCompleteTime": "%s",
|
"remark": "根据生产计划XX生成"
|
}
|
source可选"销售"或"内部"。%d-%d条。
|
""".formatted(dateStart, dateEnd, dateStart, dateEnd, dateEnd, min, max);
|
}
|
|
private static String buildStockPrompt(int min, int max, String dateStart) {
|
return """
|
库存模块格式:
|
{
|
"entity": "stockInventory",
|
"productModelId": 1,
|
"qualitity": 500,
|
"batchNo": "BATCH-YYYYMMDD-001",
|
"warnNum": 50,
|
"remark": "安全库存"
|
}
|
batchNo中YYYYMMDD替换为 %s。%d-%d条。
|
""".formatted(dateStart.replace("-", ""), min, max);
|
}
|
}
|
