plugin_task/plugins.py

import os
from typing import Dict

from plugin_task.model import Plugin

PLUGIN_JSON_SCHEMA = [
    {
        "url": os.environ["WEATHER_PLUGIN_URL"],
        "method": "POST",
        "headers": {},
        "request_body": [
            {
                "enum": [],
                "name": "city_name",
                "description": "需要明确城市名称后再调用接口",
                "is_required": True,
                "parament_type": "str",
            }
        ],
        "name_for_human": "查天气",
        "description_for_human": "查询天气的工具,如果发现用户有查询天气的意图会使用该工具",
        "description_for_model": "It can help users check the weather situation, and if the description includes a city, it can be directly queried.",
        "unique_name_for_model": "query_weather",
    },
    {
        "url": "http://gwgp-pt7hznkstln.n.bdcloudapi.com/expTrack?com=auto",
        "method": "POST",
        "headers": {
            "Content-Type": "application/json",
            "X-Bce-Signature": os.environ["BCE_APP_CODE"],
        },
        "request_body": [
            {
                "enum": [],
                "name": "nu",
                "description": "需要明确快递单号后再调用接口",
                "is_required": True,
                "parament_type": "str",
            }
        ],
        "name_for_human": "查快递",
        "description_for_human": "查询快递的工具,如果发现用户有查询快递的意图会使用该工具",
        "description_for_model": "It can help users check the express delivery, and if the description includes a express delivery, it can be directly queried.",
        "unique_name_for_model": "query_express",
    },
    {
        "url": "https://api.oioweb.cn/api/common/teladress",
        "method": "GET",
        "headers": {},
        "request_body": [
            {
                "enum": [],
                "name": "mobile",
                "description": "需要明确电话号码后再调用接口",
                "is_required": True,
                "parament_type": "str",
            }
        ],
        "name_for_human": "查电话号码归属地",
        "description_for_human": "查询电话号码归属地的工具,如果发现用户有查询电话号码归属地的意图会使用该工具",
        "description_for_model": "It can help users check the phone number attribution, and if the description includes a phone number, it can be directly queried.",
        "unique_name_for_model": "query_teladress",
    },
    {
        "url": "https://tenapi.cn/v2/bing?format=json",
        "method": "POST",
        "headers": {},
        "request_body": [],
        "name_for_human": "bing每日壁纸",
        "description_for_human": "获取bing每日壁纸的工具,如果发现用户有获取bing每日壁纸会使用该工具",
        "description_for_model": "It can help users Get bing daily wallpapers, and if the description Get bing daily wallpapers, it can be directly queried.",
        "unique_name_for_model": "Get bing daily wallpapers",
    },
]


PLUGINS: Dict[str, Plugin] = {
    plugin["unique_name_for_model"]: Plugin(
        url=plugin["url"],
        method=plugin["method"],
        headers=plugin["headers"],
        request_body=plugin["request_body"],
        name_for_human=plugin["name_for_human"],
        description_for_human=plugin["description_for_human"],
        description_for_model=plugin["description_for_model"],
        unique_name_for_model=plugin["unique_name_for_model"],
    )
    for plugin in PLUGIN_JSON_SCHEMA
}