package com.ruoyi.mock.prompt;
|
|
import java.util.List;
|
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. 同一模块内的实体之间要有引用关系（如销售台账引用客户名称）
|
7. 所有字符串字段不要使用emoji或特殊unicode字符
|
""";
|
}
|
|
public static String buildUserMessage(List<String> modules, List<String> industries,
|
int countMin, int countMax,
|
String dateStart, String dateEnd,
|
String additionalInfo,
|
List<Long> productIds,
|
List<Long> productModelIds) {
|
StringBuilder sb = new StringBuilder();
|
sb.append("请为以下行业生成模拟ERP业务数据:\n");
|
sb.append("- 行业: ").append(String.join("、", industries)).append("\n");
|
sb.append("- 数据条数: 每种实体 ").append(countMin).append("-").append(countMax).append(" 条\n");
|
sb.append("- 时间范围: ").append(dateStart).append(" ~ ").append(dateEnd).append("\n");
|
if (additionalInfo != null && !additionalInfo.isBlank()) {
|
sb.append("- 补充信息: ").append(additionalInfo).append("\n");
|
}
|
sb.append("\n");
|
|
sb.append("可用的产品ID列表: ").append(productIds.stream().map(String::valueOf).collect(Collectors.joining(","))).append("\n");
|
sb.append("可用的产品规格ID列表: ").append(productModelIds.stream().map(String::valueOf).collect(Collectors.joining(","))).append("\n");
|
sb.append("\n");
|
|
sb.append("需要生成的模块: ").append(String.join("、", modules)).append("\n\n");
|
|
if (modules.contains("sales")) {
|
sb.append(buildSalesPrompt(countMin, countMax, dateStart, dateEnd));
|
}
|
if (modules.contains("purchase")) {
|
sb.append(buildPurchasePrompt(countMin, countMax, dateStart, dateEnd));
|
}
|
if (modules.contains("quality")) {
|
sb.append(buildQualityPrompt(countMin, countMax));
|
}
|
if (modules.contains("production")) {
|
sb.append(buildProductionPrompt(countMin, countMax, dateStart, dateEnd));
|
}
|
if (modules.contains("stock")) {
|
sb.append(buildStockPrompt(countMin, countMax));
|
}
|
|
sb.append("\n请直接输出JSON数组，不要有任何其他内容:");
|
return sb.toString();
|
}
|
|
private static String buildSalesPrompt(int min, int max, String dateStart, String dateEnd) {
|
return """
|
销售模块 - 按以下格式生成:
|
{
|
"entity": "customer",
|
"customerName": "XX科技有限公司",
|
"customerType": "企业客户",
|
"contactPerson": "张三",
|
"contactPhone": "13800138000",
|
"companyAddress": "XX省XX市XX区XX路XX号",
|
"companyPhone": "0513-XXXXXXXX",
|
"taxpayerIdentificationNumber": "91110108XXXXXXXXXX",
|
"maintainer": "李四",
|
"maintenanceTime": "2026-06-01",
|
"bankAccount": "622202XXXXXXXXXXXX",
|
"basicBankAccount": "XX银行XX支行",
|
"bankCode": "308100XXXXXX"
|
}
|
{
|
"entity": "salesLedger",
|
"customerName": "引用上面生成的客户名称",
|
"salesContractNo": "XS-20260601-001",
|
"projectName": "XX项目",
|
"entryDate": "2026-06-01",
|
"entryPerson": "张三",
|
"salesman": "销售员姓名",
|
"contractAmount": 50000.00,
|
"paymentMethod": "月结30天",
|
"executionDate": "2026-06-01",
|
"deliveryDate": "2026-07-01",
|
"type": 1,
|
"productData": [
|
{
|
"productId": 1,
|
"productModelId": 1,
|
"quantity": 100,
|
"taxInclusiveUnitPrice": 500.00,
|
"taxInclusiveTotalPrice": 50000.00,
|
"taxExclusiveTotalPrice": 44247.79,
|
"taxRate": 13.00,
|
"unit": "件",
|
"type": 1
|
}
|
]
|
}
|
客户名称要像指定行业的公司，合同编号格式XS-年月日-序号，金额合理。
|
注意 taxExclusiveTotalPrice 是不含税总价，需要根据含税总价和税率计算（=含税总价/(1+税率/100)）。
|
日期范围: %s ~ %s，每种实体生成%d-%d条。
|
""".formatted(dateStart, dateEnd, min, max);
|
}
|
|
private static String buildPurchasePrompt(int min, int max, String dateStart, String dateEnd) {
|
return """
|
采购模块 - 按以下格式生成:
|
{
|
"entity": "supplier",
|
"supplierName": "XX原材料有限公司",
|
"supplierType": "原材料供应商",
|
"contactUserName": "李四",
|
"contactUserPhone": "13900139000",
|
"companyAddress": "XX省XX市XX区XX路XX号",
|
"companyPhone": "0513-XXXXXXXX",
|
"taxpayerIdentificationNum": "91110108XXXXXXXXXX",
|
"bankAccountName": "XX银行XX支行",
|
"bankAccountNum": "622202XXXXXXXXXXXX",
|
"isWhite": 0
|
}
|
{
|
"entity": "purchaseLedger",
|
"supplierName": "引用上面生成的供应商名称",
|
"purchaseContractNumber": "CG-20260601-001",
|
"projectName": "XX采购项目",
|
"entryDate": "2026-06-01",
|
"contractAmount": 30000.00,
|
"paymentMethod": "货到付款",
|
"executionDate": "2026-06-01",
|
"productData": [
|
{
|
"productId": 1,
|
"productModelId": 1,
|
"quantity": 50,
|
"taxInclusiveUnitPrice": 600.00,
|
"taxInclusiveTotalPrice": 30000.00,
|
"taxExclusiveTotalPrice": 26548.67,
|
"taxRate": 13.00,
|
"unit": "件",
|
"type": 2
|
}
|
]
|
}
|
供应商名称要像指定行业的供应商，合同编号格式CG-年月日-序号。
|
注意 taxExclusiveTotalPrice 是不含税总价，需要根据含税总价和税率计算（=含税总价/(1+税率/100)）。
|
日期范围: %s ~ %s，每种实体生成%d-%d条。
|
""".formatted(dateStart, dateEnd, min, max);
|
}
|
|
private static String buildQualityPrompt(int min, int max) {
|
return """
|
质量模块 - 按以下格式生成:
|
{
|
"entity": "qualityTestStandard",
|
"standardNo": "QTS-20260601-001",
|
"standardName": "XX产品检验标准",
|
"inspectType": 0,
|
"remark": "适用于XX行业的质量检验标准"
|
}
|
inspectType: 0=原材料检验, 1=过程检验, 2=出厂检验。三种类型都要覆盖。
|
{
|
"entity": "qualityTestStandardBinding",
|
"productId": 1,
|
"testStandardId": 1
|
}
|
每种实体生成%d-%d条。
|
""".formatted(min, max);
|
}
|
|
private static String buildProductionPrompt(int min, int max, String dateStart, String dateEnd) {
|
return """
|
生产模块 - 按以下格式生成:
|
{
|
"entity": "productionPlan",
|
"productModelId": 1,
|
"qtyRequired": 200,
|
"requiredDate": "2026-06-15",
|
"source": "销售",
|
"promisedDeliveryDate": "2026-07-01",
|
"remark": "XX客户订单需求"
|
}
|
{
|
"entity": "productionOrder",
|
"productModelId": 1,
|
"quantity": 200,
|
"planCompleteTime": "2026-06-30",
|
"remark": "根据生产计划XX生成"
|
}
|
productionPlan的source可选"销售"或"内部"。
|
日期范围: %s ~ %s，每种实体生成%d-%d条。
|
""".formatted(dateStart, dateEnd, min, max);
|
}
|
|
private static String buildStockPrompt(int min, int max) {
|
return """
|
库存模块 - 按以下格式生成:
|
{
|
"entity": "stockInventory",
|
"productModelId": 1,
|
"qualitity": 500,
|
"batchNo": "BATCH-20260601-001",
|
"warnNum": 50,
|
"remark": "安全库存"
|
}
|
每种实体生成%d-%d条。batchNo格式BATCH-年月日-序号。
|
""".formatted(min, max);
|
}
|
}
|
