File size: 2,361 Bytes
5714f9e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from .Extension import Extension
import requests, random

# 拓展的配置信息,用于ai理解拓展的功能 *必填*
ext_config = {
    "name": "acg_image",
    "arguments": {
        "keyword": "str",   # 关键字
    },
    "description": "Send an ACG image related to the keyword(in Chinese). Usage in response: /#acg_image&动漫#/",
    # 参考词,用于上下文参考使用,为空则每次都会被参考(消耗token)
    "refer_word": [],
    # 每次消息回复中最大调用次数,不填则默认为99
    "max_call_times_per_msg": 3,
    # 作者信息
    "author": "KroMiose",
    # 版本
    "version": "0.0.1",
    # 拓展简介
    "intro": "发送ACG图片",
}


class CustomExtension(Extension):
    async def call(self, arg_dict: dict, ctx_data: dict) -> dict:
        """ 当拓展被调用时执行的函数 *由拓展自行实现*

        参数:
            arg_dict: dict, 由ai解析的参数字典 {参数名: 参数值}
        """
        custom_config: dict = self.get_custom_config()  # 获取yaml中的配置信息

        token = custom_config.get('token', None)
        if token is None:
            return {
                'text': error_msg if self.get_custom_config().get('debug', False) else None
            }

        keyword = arg_dict.get('keyword', '')
        req_args = {
            'token': custom_config['token'],
            'format': 'json',
            'keyword': keyword
        }

        url = "https://v2.alapi.cn/api/acg"
        payload = {'token': custom_config['token'], 'format': 'json'}
        headers = {'Content-Type': 'application/json'}

        try:
            res = requests.post(url, json=req_args, headers=headers, timeout=10).json()
        except Exception as e:
            return {
                'text': f"[ext_acg_image] 访问api时发生错误: {e}"

            }

        if 'data' in res and isinstance(res['data'], list) and len(res['data']) > 0:
            return {
                'image': random.choice(res['data'])
            }
        else:
            error_msg = f"[ext_acg_image] 返回的数据格式不正确: {res}"
            return {
                'text': error_msg if self.get_custom_config().get('debug', False) else None
            }

    def __init__(self, custom_config: dict):
         super().__init__(custom_config=custom_config)