百度unit API
1. 获取access_token
参考http://ai.baidu.com/docs#/Auth/top
import requests
import ssl
def get_access_token(api_key, secret_key):
# client_id 为官网获取的AK， client_secret 为官网获取的SK
url = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={api_key}&client_secret={secret_key}'
headers = {'Content-Type': 'application/json; charset=UTF-8'}
response = requests.post(url, headers=headers)
return response.json()['access_token']
api_key = '
secret_key = ''
access_token = get_access_token(api_key, secret_key)
2. unit api
参考https://ai.baidu.com/docs#/UNIT-v2-service-API/c3cd4b5e
import json
import requests
def chat_baidu(access_token, text='你好'):
headers = {'Content-Type':'application/json'}
access_token = access_token
url = f'https://aip.baidubce.com/rpc/2.0/unit/service/chat?access_token={access_token}'
post_data = {
"log_id": "UNITTEST_10000",
"version": "2.0",
"service_id": "S21125",
# "skill_ids": ['73803', '73808', '73809', '73810', '73811'],
"session_id": '',
"request": {"query": text, "user_id": "88888"},
# "dialog_state":{"contexts": {"SYS_REMEMBERED_SKILLS":["1057"]}}
}
try:
response = requests.post(url, data=json.dumps(post_data), headers=headers)
if not response.ok:
raise Exception
response = response.json()
if response['error_code']:
raise Exception(response['error_msg'])
results = response['result']['response_list'][0]['action_list'][0]['say']
print(results)
except Exception as e:
print(f'error: {e}')
else:
return results
