File size: 3,622 Bytes
031beb8 |
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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
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
}
|