Python调用Unit闲聊对话API的应用

文章目录

1 机器人的创建
2 Python调用API的代码实现

简介：利用百度大脑平台（百度智能对话定制与服务平台，即UNIT）提供的API接口，实现可以完成闲聊功能的智能对话机器人

1 机器人的创建

访问百度大脑网址：https://ai.baidu.com/unit/home
点击进入UNIT：

然后登陆百度账号，进入到以下页面，点击我的机器人：

点击“+”号创建机器人，我们设置机器人名称为“闲聊机器人”，描述里写“闲聊”，然后创建：

点击新创建好的机器人界面，点击技能管理界面：

然后点击蓝字“技能管理页”，再点击“添加预置技能”，拉到最下面，找到闲聊功能并创建

然后新建我的闲聊技能，命名为“闲聊机器人”

然后回到我们的“我的机器人”界面中的技能管理，添加技能，我们可以看到我们创建的“闲聊机器人”功能，将其添加至机器人即可。

然后点击“发布上线”中的“研发/生产环境”，点击蓝色按钮“获取API Key/Secret Key”

我们可以看到如下界面，其中有两个重要的参数——API Key和Secret Key

以上过程我们创建好了闲聊机器人，下面我们需要做的就是在Python中调用该机器人的API接口，实现闲聊功能。

2 Python调用API的代码实现

导入相关包：

import json
import random
import requests

设置好相关参数。这里的id对应的即上面的API Key，secret对应的是上面的Secret Key

client_id = "byKLF3DzDpYGQDPifrymq14j"
client_secret = "dZGTviCbBKr8FWzo2HjL7Qs5LMGFAki1"

编写函数unit_chat。其功能是调用百度大脑UNIT接口。包括两个参数：chat_input代表的是用户输入的内容，user_id表示用户的id号，我们默认为“88888”；函数的返回值是闲聊机器人的回复内容。

def unit_chat(chat_input, user_id="88888"):# 设置默认回复chat_reply = "不好意思，我正在学习中，随后回复你"# 固定的url格式url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s"%(client_id, client_secret)res = requests.get(url)access_token = eval(res.text)["access_token"]unit_chatbot_url = "https://aip.baidubce.com/rpc/2.0/unit/service/chat?access_token=" + access_token# 拼装聊天接口对应请求post_data = {"log_id": str(random.random()),  #登陆的id，是什么不重要，我们用随机数生成一个id即可"request": {"query": chat_input,  #用户输入的内容"user_id": user_id  #用户id},"session_id": "","service_id": "S53842",  #!!!!这个很重要，必须对应我们创建的机器人的id号，id号在百度大脑中我们创建的闲聊机器人中可见"version": "2.0"}# 将聊天接口对应请求数据转为json数据res = requests.post(url=unit_chatbot_url, json=post_data)# 获取聊天接口返回数据unit_chat_obj = json.loads(res.content)# 判断聊天接口返回数据是否出错(error_code == 0则表示请求正确)if unit_chat_obj["error_code"] != 0:return chat_reply# 解析聊天接口返回数据，找到返回文本内容 result -> response_list -> schema -> intent_confidence(>0) -> action_list -> sayunit_chat_obj_result = unit_chat_obj["result"]unit_chat_response_list = unit_chat_obj_result["response_list"]# 随机选取一个"意图置信度"[+response_list[].schema.intent_confidence]不为0的技能作为回答unit_chat_response_obj = random.choice([unit_chat_response for unit_chat_response in unit_chat_response_list ifunit_chat_response["schema"]["intent_confidence"] > 0.0])unit_chat_response_action_list = unit_chat_response_obj["action_list"]unit_chat_response_action_obj = random.choice(unit_chat_response_action_list)unit_chat_response_say = unit_chat_response_action_obj["say"]return unit_chat_response_say

测试我们的闲聊机器人。我们建立一个while循环，当用户输入Bye或bye后退出循环。

if __name__ == "__main__":while True:chat_input = input("请输入:")if chat_input == 'Bye' or chat_input == "bye":breakchat_reply = unit_chat(chat_input)print("Unit:", chat_reply)

测试结果如下：