Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update data_source.py

Browse Source
This commit is contained in:
HibiKier 2021-07-06 21:29:23 +08:00 committed by GitHub
parent 54928478a3
commit 1be9305755
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 51 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
plugins/ai/data_source.py
View File

@ -32,54 +32,58 @@ async def get_qqbot_chat_result(text: str, img_url: str, user_id: int, user_name
for key in keys: for key in keys:
if text.find(key) != -1: if text.find(key) != -1:
return random.choice(anime_data[key]).replace('你', user_name) return random.choice(anime_data[key]).replace('你', user_name)
if text: try:
req = { if text:
"perception": req = {
{ "perception":
"inputText": {
{ "inputText":
"text": text {
}, "text": text
"selfInfo": },
{ "selfInfo":
"location": {
{ "location":
"city": "陨石坑", {
"province": "火星", "city": "陨石坑",
"street": "第5坑位" "province": "火星",
} "street": "第5坑位"
} }
}, }
"userInfo":
{
"apiKey": TL_KEY[index],
"userId": str(user_id)
}
}
elif img_url:
req = {
"reqType": 1,
"perception":
{
"inputImage": {
"url": img_url
}, },
"selfInfo": "userInfo":
{ {
"location": "apiKey": TL_KEY[index],
{ "userId": str(user_id)
"city": "陨石坑", }
"province": "火星", }
"street": "第5坑位" elif img_url:
} req = {
} "reqType": 1,
}, "perception":
"userInfo": {
{ "inputImage": {
"apiKey": TL_KEY[index], "url": img_url
"userId": str(user_id) },
} "selfInfo":
} {
"location":
{
"city": "陨石坑",
"province": "火星",
"street": "第5坑位"
}
}
},
"userInfo":
{
"apiKey": TL_KEY[index],
"userId": str(user_id)
}
}
except IndexError:
index = 0
return no_result()
async with aiohttp.ClientSession() as sess: async with aiohttp.ClientSession() as sess:
async with sess.post(url, json=req) as response: async with sess.post(url, json=req) as response:
if response.status != 200: if response.status != 200: