KBQA/kbqa-graph/src/main/java/com/supervision/handler/gpt/IdentifyIntentHandler.java

package com.supervision.handler.gpt;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.supervision.ai.AiUtil;
import com.supervision.ai.dto.MessageDTO;
import com.supervision.enums.IdentifyIntentEnum;
import com.supervision.exception.IdentifyIntentException;
import io.lettuce.core.dynamic.annotation.CommandNaming;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;

/**
 * 意图识别handler
 */
@Component
public class IdentifyIntentHandler {

    public String identifyIntent(String question) {
        // 首先生成提示词
        List<MessageDTO> messageList = new ArrayList<>();
        messageList.add(new MessageDTO("system", "假设你是一个意图识别模型,且精通社会保险业务中的事务分类，" +
                "现在我会给你一个意图列表以及意图列表可能的一些典型问法,你首先要根据意图列表和典型问法进行学习(典型问法只是参考,并不一定局限于典型问法范围)，" +
                "然后我再给你一个句子,请帮我的要求识别出这个句子中的意图.如果该句子中不含有意图列表中任意意图，你可以输出:未匹配。不要回答其他的内容！！"));
        messageList.add(new MessageDTO("assistant", "好的"));
        // 构建学习案例
        StringBuilder stringBuilder = new StringBuilder();
        for (IdentifyIntentEnum value : IdentifyIntentEnum.values()) {
            stringBuilder.append("意图:").append(value.getIntent()).append(";").append("典型问法:").append(CollUtil.join(value.getExplainList(), ",")).append("\n");
        }
        messageList.add(new MessageDTO("user", stringBuilder.toString()));
        messageList.add(new MessageDTO("assistant", "已学习"));
        // 构建问题
        messageList.add(new MessageDTO("user", "问题是:" + question + ",现在请用列表的形式展示，如果该句子中不含有指定的意图，你可以输出:未识别。" +
                "如果有，你就选一个列表中的词进行回答！输出格式形为：意图名称。除了这个列表以外请不要输出别的多余的话。"));
        // 进行提问
        String intent = AiUtil.chatByMessage(messageList);
        // 尝试转为JSON的形式
        if (StrUtil.isBlank(intent) || StrUtil.equals("未识别", intent)) {
            throw new IdentifyIntentException("意图未识别");
        }
        return intent;
    }
}