1. 添加 Qanything对话接口相关代码

kbqa-graph/src/main/java/com/supervision/qanything/QanythingService.java

@@ -0,0 +1,23 @@
+package com.supervision.qanything;
+
+import com.supervision.qanything.dto.ChatResult;
+import com.supervision.qanything.dto.ResultWrapper;
+
+import java.security.NoSuchAlgorithmException;
+import java.util.List;
+
+/**
+ * Qanything 接口服务
+ */
+public interface QanythingService {
+
+
+    /**
+     * 聊天接口
+     * @param question 问题
+     * @param kbIds 知识库ID
+     * @return
+     * @throws NoSuchAlgorithmException
+     */
+    ResultWrapper<ChatResult> chat(String question,List<String> kbIds) throws NoSuchAlgorithmException;
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/BaseParam.java

@@ -0,0 +1,38 @@
+package com.supervision.qanything.dto;
+
+import lombok.Data;
+
+@Data
+public class BaseParam {
+
+    /**
+     * 应用标识（应用 ID）
+     */
+    private String appKey;
+
+    /**
+     * 时间戳(秒)
+     */
+    private String curtime;
+
+    /**
+     * 随机字符串 建议使用UUID
+     */
+    private String salt;
+
+    /**
+     * 签名信息：sha256(应用ID+input+salt+curtime+应用密钥)
+     */
+    private String sign;
+
+    /**
+     * 签名类型 v3
+     */
+    private String signType;
+
+    private String q;
+
+    private String[] img;
+
+
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/ChatParam.java

@@ -0,0 +1,21 @@
+package com.supervision.qanything.dto;
+
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.util.List;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+public class ChatParam extends BaseParam{
+
+    /**
+     * json数组类型,多知识库问答
+     */
+    private List<String> kbIds;
+
+    /**
+     * json数组类型,最多支持两轮对话历史消息
+     */
+    private List<HistoryDTO> history;
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/ChatResult.java

@@ -0,0 +1,30 @@
+package com.supervision.qanything.dto;
+
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+public class ChatResult {
+
+    /**
+     * 本轮提问
+     */
+    private String question;
+
+    /**
+     * 本轮回答
+     */
+    private String response;
+
+    /**
+     * 历史问答
+     */
+    private List<HistoryDTO> history;
+
+    /**
+     * 文档出处
+     */
+    private List<SourceDTO> source;
+
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/HistoryDTO.java

@@ -0,0 +1,19 @@
+package com.supervision.qanything.dto;
+
+import lombok.Data;
+
+/**
+ * 历史问答
+ */
+@Data
+public class HistoryDTO {
+    /**
+     * 历史提问
+     */
+    private String question;
+
+    /**
+     * 历史回答
+     */
+    private String response;
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/ResultWrapper.java

@@ -0,0 +1,24 @@
+package com.supervision.qanything.dto;
+
+import lombok.Data;
+
+@Data
+public class ResultWrapper<T> {
+
+    /**
+     * 错误码
+     */
+    private String errorCode;
+
+    /**
+     * 描述
+     */
+    private String msg;
+
+    /**
+     * 请求id
+     */
+    private String requestId;
+
+    private T result;
+}

kbqa-graph/src/main/java/com/supervision/qanything/dto/SourceDTO.java

@@ -0,0 +1,30 @@
+package com.supervision.qanything.dto;
+
+
+import lombok.Data;
+
+/**
+ * 文档出处
+ */
+@Data
+public class SourceDTO {
+    /**
+     * 文档id
+     */
+    private String fileId;
+
+    /**
+     * 文档名称
+     */
+    private String fileName;
+
+    /**
+     * 原文内容
+     */
+    private String content;
+
+    /**
+     * 可信度
+     */
+    private String source;
+}

kbqa-graph/src/main/java/com/supervision/qanything/impl/QanythingServiceImpl.java

@@ -0,0 +1,37 @@
+package com.supervision.qanything.impl;
+
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import cn.hutool.json.JSONUtil;
+import com.supervision.qanything.QanythingService;
+import com.supervision.qanything.dto.ChatParam;
+import com.supervision.qanything.dto.ChatResult;
+import com.supervision.qanything.dto.ResultWrapper;
+import com.supervision.qanything.util.AuthV3Util;
+import lombok.extern.slf4j.Slf4j;
+import java.security.NoSuchAlgorithmException;
+import java.util.List;
+
+@Slf4j
+public class QanythingServiceImpl implements QanythingService {
+
+    private static final String APP_KEY = "071053d9bc4b544c";     // 您的应用ID
+    private static final String APP_SECRET = "nObcggsh0li2eGtyII6XOGtSCo6nFgdD";  // 您的应用密钥
+
+    private String baseUrl = "https://openapi.youdao.com";
+    @Override
+    public ResultWrapper<ChatResult> chat(String question, List<String> kbIds) throws NoSuchAlgorithmException {
+        ChatParam chatParam = new ChatParam();
+        chatParam.setQ(question);
+        chatParam.setKbIds(kbIds);
+        // 添加鉴权相关参数
+        AuthV3Util.addAuthParams(APP_KEY, APP_SECRET, chatParam);
+        HttpRequest request = HttpRequest.post(baseUrl + "/q_anything/paas/chat")
+                .body(JSONUtil.toJsonStr(chatParam));
+        try (HttpResponse response = request.execute()){
+            return JSONUtil.toBean(response.body(), ResultWrapper.class);
+        }
+    }
+
+
+}

kbqa-graph/src/main/java/com/supervision/qanything/util/AuthV3Util.java

@@ -0,0 +1,119 @@
+package com.supervision.qanything.util;
+
+import com.supervision.qanything.dto.BaseParam;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.UUID;
+
+public class AuthV3Util {
+
+
+    /**
+     * 添加鉴权相关参数 -
+     * appKey : 应用ID
+     * salt : 随机值
+     * curtime : 当前时间戳(秒)
+     * signType : 签名版本
+     * sign : 请求签名
+     *
+     * @param appKey    您的应用ID
+     * @param appSecret 您的应用密钥
+     * @param baseParam 请求参数表
+     */
+    public static void addAuthParams(String appKey, String appSecret, BaseParam baseParam)
+            throws NoSuchAlgorithmException {
+       /* String[] qArray = baseParam.getQ();
+        if (qArray == null) {
+            qArray = baseParam.getImg();
+        }
+        StringBuilder q = new StringBuilder();
+        for (String item : qArray) {
+            q.append(item);
+        }*/
+        String salt = UUID.randomUUID().toString();
+        String curtime = String.valueOf(System.currentTimeMillis() / 1000);
+        String sign = calculateSign(appKey, appSecret, baseParam.getQ().toString(), salt, curtime);
+        baseParam.setAppKey(appKey);
+        baseParam.setSalt(salt);
+        baseParam.setCurtime(curtime);
+        baseParam.setSignType("v3");
+        baseParam.setSign(sign);
+    }
+
+
+    /**
+     * 添加鉴权相关参数 -
+     *
+     * @param appKey
+     * @param appSecret
+     * @param q         用于生成签名的字段
+     * @return
+     * @throws NoSuchAlgorithmException
+     */
+    public static BaseParam addAuthParams(String appKey, String appSecret, String q) throws NoSuchAlgorithmException {
+        String salt = UUID.randomUUID().toString();
+        String curtime = String.valueOf(System.currentTimeMillis() / 1000);
+        String sign = calculateSign(appKey, appSecret, q, salt, curtime);
+
+        BaseParam baseParam = new BaseParam();
+        baseParam.setAppKey(appKey);
+        baseParam.setSalt(salt);
+        baseParam.setCurtime(curtime);
+        baseParam.setSignType("v3");
+        baseParam.setSign(sign);
+
+        return baseParam;
+
+
+    }
+
+    /**
+     * 计算鉴权签名 -
+     * 计算方式 : sign = sha256(appKey + input(q) + salt + curtime + appSecret)
+     *
+     * @param appKey    您的应用ID
+     * @param appSecret 您的应用密钥
+     * @param q         请求内容
+     * @param salt      随机值
+     * @param curtime   当前时间戳(秒)
+     * @return 鉴权签名sign
+     */
+    public static String calculateSign(String appKey, String appSecret, String q, String salt, String curtime)
+            throws NoSuchAlgorithmException {
+        String strSrc = appKey + getInput(q) + salt + curtime + appSecret;
+        return encrypt(strSrc);
+    }
+
+    private static String encrypt(String strSrc) throws NoSuchAlgorithmException {
+        byte[] bt = strSrc.getBytes();
+        MessageDigest md = MessageDigest.getInstance("SHA-256");
+        md.update(bt);
+        byte[] bts = md.digest();
+        StringBuilder des = new StringBuilder();
+        for (byte b : bts) {
+            String tmp = (Integer.toHexString(b & 0xFF));
+            if (tmp.length() == 1) {
+                des.append("0");
+            }
+            des.append(tmp);
+        }
+        return des.toString();
+    }
+
+    private static String getInput(String input) {
+        if (input == null) {
+            return null;
+        }
+        String result;
+        int len = input.length();
+        if (len <= 20) {
+            result = input;
+        } else {
+            String startStr = input.substring(0, 10);
+            String endStr = input.substring(len - 10, len);
+            result = startStr + len + endStr;
+        }
+        return result;
+    }
+}

kbqa-graph/src/main/java/com/supervision/qanything/util/AuthV4Util.java

@@ -0,0 +1,65 @@
+package com.supervision.qanything.util;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.Map;
+import java.util.UUID;
+
+public class AuthV4Util {
+
+    /**
+     * 添加鉴权相关参数 -
+     * appKey : 应用ID
+     * salt : 随机值
+     * curtime : 当前时间戳(秒)
+     * signType : 签名版本
+     * sign : 请求签名
+     *
+     * @param appKey    您的应用ID
+     * @param appSecret 您的应用密钥
+     * @param paramsMap 请求参数表
+     */
+    public static void addAuthParams(String appKey, String appSecret, Map<String, String[]> paramsMap)
+            throws NoSuchAlgorithmException {
+        String salt = UUID.randomUUID().toString();
+        String curtime = String.valueOf(System.currentTimeMillis() / 1000);
+        String sign = calculateSign(appKey, appSecret, salt, curtime);
+        paramsMap.put("appKey", new String[]{appKey});
+        paramsMap.put("salt", new String[]{salt});
+        paramsMap.put("curtime", new String[]{curtime});
+        paramsMap.put("signType", new String[]{"v4"});
+        paramsMap.put("sign", new String[]{sign});
+    }
+
+    /**
+     * 计算鉴权签名 -
+     * 计算方式 : sign = sha256(appKey + salt + curtime + appSecret)
+     *
+     * @param appKey    您的应用ID
+     * @param appSecret 您的应用密钥
+     * @param salt      随机值
+     * @param curtime   当前时间戳(秒)
+     * @return 鉴权签名sign
+     */
+    public static String calculateSign(String appKey, String appSecret, String salt, String curtime)
+            throws NoSuchAlgorithmException {
+        String strSrc = appKey + salt + curtime + appSecret;
+        return encrypt(strSrc);
+    }
+
+    private static String encrypt(String strSrc) throws NoSuchAlgorithmException {
+        byte[] bt = strSrc.getBytes();
+        MessageDigest md = MessageDigest.getInstance("SHA-256");
+        md.update(bt);
+        byte[] bts = md.digest();
+        StringBuilder des = new StringBuilder();
+        for (byte b : bts) {
+            String tmp = (Integer.toHexString(b & 0xFF));
+            if (tmp.length() == 1) {
+                des.append("0");
+            }
+            des.append(tmp);
+        }
+        return des.toString();
+    }
+}