File size: 2,393 Bytes
0b1aeae |
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 |
from .Extension import Extension
import requests
# 拓展的配置信息，用于ai理解拓展的功能 *必填*
ext_config:dict = {
"name": "LoliconPic", # 拓展名称，用于标识拓展
"arguments": {
"tag": "str", # 关键字
},
"description": "send 1 specified tag anime picture. (usage in response: /#LoliconPic&萝莉#/)",
# 参考词，用于上下文参考使用，为空则每次都会被参考(消耗token)
"refer_word": ["图", "pic", "Pic", "再", "还", "涩", "色"],
# 每次消息回复中最大调用次数，不填则默认为99
"max_call_times_per_msg": 3,
# 作者信息
"author": "CCYellowStar",
# 版本
"version": "0.0.1",
# 拓展简介
"intro": "发送指定tag二次元图片",
}
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中的配置信息
r18 = 0 #添加r18参数 0为否，1为是，2为混合
tag = arg_dict.get('tag', None)
url = f"https://api.lolicon.app/setu/v2?tag={tag}&r18={r18}"
res = requests.get(url, verify=False, timeout=10)
data=res.json()
if not data["error"]:
if data["data"]:
data = data["data"][0]
img_src = data["urls"]["original"]
else:
return {
'text': f"[来自拓展] 没有这个tag的图片...",
'image': None, # 图片url
'voice': None, # 语音url
}
if img_src is None:
return {
'text': f"[来自拓展] 发送图片错误或超时...",
'image': None, # 图片url
'voice': None, # 语音url
}
else:
# 返回的信息将会被发送到会话中
return {
'text': None, # 文本信息
'image': img_src, # 图片url
'voice': None, # 语音url
}
def __init__(self, custom_config: dict):
super().__init__(ext_config.copy(), custom_config) |