SystemPromptGenerator.java 2.44 KB
package com.xly.agent;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.json.JSONConfig;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.alibaba.fastjson2.JSONWriter;
import com.xly.entity.ParamRule;
import com.xly.entity.UserSceneSession;

import java.util.List;

public class SystemPromptGenerator {

    public static String generate(String toolName, UserSceneSession session) {
        StringBuilder sb = new StringBuilder();

        sb.append("你是一个无情的参数提取器,不是聊天助手。\n\n");
        sb.append("【核心任务】\n");
        sb.append("从用户输入中提取参数,并立即调用工具 ").append(toolName).append("。\n\n");
        sb.append("【绝对指令】\n");
        sb.append("1. 严禁与用户对话,严禁输出任何纯文本回复。\n");
        sb.append("2. 无论参数缺失多么严重,都必须直接调用工具,缺失字段填空字符串或0。\n");
        sb.append("3. 不要引导用户,后端程序会处理引导逻辑。\n\n");
        sb.append("【参数提取规则】\n");
        sb.append("用户输入格式为 \"参数名: 值\" 或 \"参数名 值\",你必须识别参数名并提取对应的值。\n\n");
        sb.append("【可提取的参数列表】\n");
        List<ParamRule> dynamicParamRuleAll = session.getCurrentTool().getParamRuleListAll();
        // 从参数定义中提取
        for (ParamRule paramRule : dynamicParamRuleAll) {
            sb.append("- ").append(paramRule.getSParam()).append("\n");
        }
        sb.append("\n【提取示例】\n");
        // 动态生成示例(取前3个参数作为示例)
        int count = 0;
        for (ParamRule paramRule : dynamicParamRuleAll) {
            if (count++ >= 3) break;
            sb.append("- 用户输入 \"").append(paramRule.getSParam()).append(": {值}\" → 提取 ").append(paramRule.getSParam()).append("\n");
        }

        if(ObjectUtil.isNotEmpty(session.getArgs())){
            sb.append("【已提取参数】\n");
            sb.append(JSON.toJSONString(session.getArgs(), JSONWriter.Feature.PrettyFormat));
        }
        sb.append("\n【工具调用规则】\n");
        sb.append("1. 每次只调用一次工具\n");
        sb.append("2. 调用后立即停止,不要重复调用\n");
        sb.append("3. 不要总结,不要回复用户,只调用工具");

        return sb.toString();
    }
}