mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-10 08:17:28 +08:00
26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
# encoding:utf-8
|
|
|
|
import requests
|
|
from bot.bot import Bot
|
|
|
|
|
|
class BaiduUnitBot(Bot):
|
|
def reply(self, query, context=None):
|
|
token = self.get_token()
|
|
url = 'https://aip.baidubce.com/rpc/2.0/unit/service/v3/chat?access_token=' + token
|
|
post_data = "{\"version\":\"3.0\",\"service_id\":\"S73177\",\"session_id\":\"\",\"log_id\":\"7758521\",\"skill_ids\":[\"1221886\"],\"request\":{\"terminal_id\":\"88888\",\"query\":\"" + query + "\", \"hyper_params\": {\"chat_custom_bot_profile\": 1}}}"
|
|
print(post_data)
|
|
headers = {'content-type': 'application/x-www-form-urlencoded'}
|
|
response = requests.post(url, data=post_data.encode(), headers=headers)
|
|
if response:
|
|
return response.json()['result']['context']['SYS_PRESUMED_HIST'][1]
|
|
|
|
def get_token(self):
|
|
access_key = 'wdNwpvzh4fBr8xzTuPePOWhU'
|
|
secret_key = 'EUtLXyrogfTQV8nnjncnqNEfeyoHDUqS'
|
|
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + access_key + '&client_secret=' + secret_key
|
|
response = requests.get(host)
|
|
if response:
|
|
print(response.json())
|
|
return response.json()['access_token']
|