File size: 2,514 Bytes
a111a56 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
from .Extension import Extension
import requests
# 扩展的配置信息，用于ai理解扩展的功能 *必填*
ext_config:dict = {
"name": "AnimePic", # 扩展名称，用于标识扩展
"arguments": {
'keyword': 'str', # 关键字
},
"description": "send 1 anime picture by keyword. (usage in response: /#AnimePic&萝莉#/)",
# 参考词，用于上下文参考使用，为空则每次都会被参考(消耗token)
"refer_word": ["图", "pic", "Pic", "再", "还", "涩", "色", "萝莉", "元"],
# 每次消息回复中最大调用次数，不填则默认为99
"max_call_times_per_msg": 3,
# 作者信息
"author": "KroMiose",
# 版本
"version": "0.0.1",
# 扩展简介
"intro": "Sexnyan二次元图片",
}
class CustomExtension(Extension):
async def call(self, arg_dict: dict, ctx_data: dict) -> dict:
""" 当扩展被调用时执行的函数 *由扩展自行实现*
参数:
arg_dict: dict, 由ai解析的参数字典 {参数名: 参数值(类型为str)}
"""
custom_config:dict = self.get_custom_config() # 获取yaml中的配置信息
keyword = arg_dict.get('keyword', '')
r18 = custom_config.get('r18', False)
params = {
'r18': r18,
'keyword': keyword,
} if keyword else {
'r18': r18,
}
url = "https://sex.nyan.xyz/api/v2"
try:
res = requests.get(url, params=params, verify=False, timeout=10).json().get("data")
if not res:
params['keyword'] = None
res = requests.get(url, params=params, verify=False, timeout=10).json().get("data")[0]
else:
res = res[0]
img_data = {
'url': res.get('url'),
'title': res.get('title'),
'author': res.get('author'),
}
return {
'text': f"{img_data['title']} by: {img_data['author']}", # 文本信息
'image': img_data['url'], # 图片url
'voice': None, # 语音url
}
except Exception as e:
return {
'text': f"[sexnyan] 找不到关于 \"{keyword}\" 的图片",
'image': None, # 图片url
'voice': None, # 语音url
}
def __init__(self, custom_config: dict):
super().__init__(ext_config.copy(), custom_config)
|