java使用百度UNIT
因为网上不太有java的教学所以写一下以便以后用到或帮助大家一起理解
第一步是注册机器人，官网：https://ai.baidu.com/unit/home。注册和测试技能官网都有，就不多做介绍。
工具类：
1.AuthService
package com.ssx.robot.baidu; import org.springframework.boot.configurationprocessor.json.JSONObject; import java.io.BufferedReader; import java.io.InputStreamReader; import java.net.HttpURLConnection; import java.net.URL; import java.util.List; import java.util.Map; /** * 获取token类 */ public class AuthService { /** * 获取权限token * @return 返回示例： * { * "access_token": "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567", * "expires_in": 2592000 * } */ public static String getAuth() { // 官网获取的 API Key 更新为你注册的 String clientId = ""; // 官网获取的 Secret Key 更新为你注册的 String clientSecret = ""; return getAuth(clientId, clientSecret); } /** * 获取API访问token * 该token有一定的有效期，需要自行管理，当失效时需重新获取. * @param ak - 百度云官网获取的 API Key * @param sk - 百度云官网获取的 Securet Key * @return assess_token 示例： * "24.460da4889caad24cccdb1fea17221975.2592000.1491995545.282335-1234567" */ public static String getAuth(String ak, String sk) { // 获取token地址 String authHost = "https://aip.baidubce.com/oauth/2.0/token?"; String getAccessTokenUrl = authHost // 1. grant_type为固定参数 + "grant_type=client_credentials" // 2. 官网获取的 API Key + "&client_id=" + ak // 3. 官网获取的 Secret Key + "&client_secret=" + sk; try { URL realUrl = new URL(getAccessTokenUrl); // 打开和URL之间的连接 HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection(); connection.setRequestMethod("GET"); connection.connect(); // 获取所有响应头字段 Map<String, List<String>> map = connection.getHeaderFields(); // 遍历所有的响应头字段 for (String key : map.keySet()) { System.err.println(key + "--->" + map.get(key)); } // 定义 BufferedReader输入流来读取URL的响应 BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream())); String result = ""; String line; while ((line = in.readLine()) != null) { result += line; } /** * 返回结果示例 */ System.err.println("result:" + result); JSONObject jsonObject = new JSONObject(result); String access_token = jsonObject.getString("access_token"); return access_token; } catch (Exception e) { System.err.printf("获取token失败！"); e.printStackTrace(System.err); } return null; } }
其中clientId和clientSecret为你注册机器人后获得的API Key / Secret Key。
2.UnitService
package com.ssx.robot.baidu; import net.sf.json.JSONArray; import net.sf.json.JSONObject; import java.util.HashMap; import java.util.Map; /** * 重要提示代码中所需工具类 * FileUtil,Base64Util,HttpUtil,GsonUtils请从 * https://ai.baidu.com/file/658A35ABAB2D404FBF903F64D47C1F72 * https://ai.baidu.com/file/C8D81F3301E24D2892968F09AE1AD6E2 * https://ai.baidu.com/file/544D677F5D4E4F17B4122FBD60DB82B3 * https://ai.baidu.com/file/470B3ACCA3FE43788B5A963BF0B625F3 * 下载 */ public class UnitService { /*sessionid和msg用户的话*/ public static Map<String, String> utterance(String session_id, String content) { Map<String, String> map = new HashMap<String, String>(); try { String talkUrl = "https://aip.baidubce.com/rpc/2.0/unit/service/chat"; // 请求参数 scene_id记得修改 //{"log_id":"UNITTEST_10000","version":"2.0","service_id":"S27823","skill_id":"","session_id":" ","request":{"query":"","user_id":"UNIT_DEV_11"}} String params = "{\"log_id\":\"UNITTEST_10000\",\"version\":\"2.0\",\"service_id\":\"S27823\",\"session_id\":\"" + session_id + " \",\"request\":{\"query\":\"" + content + "\",\"user_id\":\"UNIT_DEV_11\"}}"; String accessToken = AuthService.getAuth(); //获得返回的json数据 String str = HttpUtil.post(talkUrl, accessToken, "application/json", params); System.out.println(str); System.out.println("=========str已获得========"); //获得result JSONObject jsonobject = JSONObject.fromObject(str); String result = jsonobject.getString("result"); System.out.println(result); System.out.println("======result已获得======"); //获得会话session_id JSONObject jsonobject2 = JSONObject.fromObject(result); session_id = jsonobject2.getString("session_id"); System.out.println("======sessionid:====" + session_id); map.put("session_id", session_id); //获得动作列表 JSONArray response_list = jsonobject2.getJSONArray("response_list"); // JSONArray response_list2 = JSONObject.fromObject(response_list); System.out.println("===response_list===:" + response_list); if (response_list.size() > 0) { for (int i = 0; i < response_list.size(); i++) { JSONObject test = response_list.getJSONObject(i); JSONArray action_list = test.getJSONArray("action_list"); JSONObject action_list1 = JSONObject.fromObject(action_list.get(0)); System.out.println("====循环action_list====" + action_list1); String say = ""; if (!action_list1.get("type").toString().equals("failure")) { say = (String) action_list1.get("say");//机器bot返回的话术 map.put("say", say);//循环插入覆盖，但每次调用都不同所以依然是随机的 } System.out.println("======say:====" +map.get("say")); } } } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } return map; } public static void main(String[] args) { utterance("", "你好"); } }
一些其他工具类在注释中可以下载。因为是单论对话所以我session_id就设置为空了，content就是我们用户输入的话了。params是要传上去的数据内容是json字符串，我写的是一些必要的属性（talkUrl可以是机器人也可以是技能api，因此params也会相应不同）。然后我只传了say，也就是机器人回的话，因为偷懒。。。
控制类：
@RequestMapping("/unit")
public void unitchat(HttpServletRequest request, HttpServletResponse response) throws Exception {
//防乱码
request.setCharacterEncoding("UTF-8");
response.setContentType("text/html;charset=utf-8");
JSONObject data = new JSONObject();
Gson gson = new Gson();
String msg = request.getParameter("msg");
//调用unit分析语义
Map<String, String> map = UnitService.utterance("", msg);
String say = gson.toJson(map.get("say"));
data.put("say", say);
HttpUtils.send(response, data.toString());
}
获取前端传的msg并传回机器人的say
前端：
$.ajax({ url: _ctx + "/unit", data: { "msg":txti }, type: "post", dataType: "json", async: true, success: function (data) { var say = JSON.parse(data.say); $('#sended').append('<div id="left"><div id="txt">' + say + '</div></div>'); document.getElementById('sended').scrollTop = document.getElementById('sended').scrollHeight; } });
ajax调用并显示
大概就这些啦
