mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-01-19 01:21:01 +08:00
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
# encoding:utf-8
|
|
|
|
import requests
|
|
|
|
from bot.bot import Bot
|
|
from bridge.reply import Reply, ReplyType
|
|
|
|
|
|
# Baidu Unit对话接口 (可用, 但能力较弱)
|
|
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:
|
|
reply = Reply(
|
|
ReplyType.TEXT,
|
|
response.json()["result"]["context"]["SYS_PRESUMED_HIST"][1],
|
|
)
|
|
return reply
|
|
|
|
def get_token(self):
|
|
access_key = "YOUR_ACCESS_KEY"
|
|
secret_key = "YOUR_SECRET_KEY"
|
|
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"]
|