mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-02-07 12:16:12 +08:00
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
# 主交互
|
||
|
||
from datetime import datetime
|
||
|
||
from nonebot_plugin_marshoai.plugin import String, on_function_call
|
||
|
||
from .pc_token import ERROR_DICT, dict_to_token, token_to_dict
|
||
|
||
|
||
@on_function_call(description="创建一个新的猫对象（养一只新猫）").params(
|
||
name=String(description="猫的名字")
|
||
)
|
||
async def create_cat(name: str) -> str:
|
||
cat_data = {
|
||
"name": name,
|
||
"age": ERROR_DICT["age"],
|
||
"type": ERROR_DICT["type"],
|
||
"health": ERROR_DICT["health"],
|
||
"saturation": ERROR_DICT["saturation"],
|
||
"energy": ERROR_DICT["energy"],
|
||
"skill": ERROR_DICT["skill"],
|
||
"date": (datetime(2025, 1, 1) - datetime.now()).days,
|
||
}
|
||
token = dict_to_token(cat_data)
|
||
return f"猫对象创建成功，Token: {token}"
|
||
|
||
|
||
@on_function_call(description="查询猫对象信息").params(
|
||
token=String(description="猫对象的Token（是一串类似base64的字符串）")
|
||
)
|
||
async def query_cat(token: str) -> str:
|
||
cat_data = token_to_dict(token)
|
||
return f"猫的名字: {cat_data['name']}, 年龄: {cat_data['age']}, 种类: {cat_data['type']}, 生命值: {cat_data['health']}, 饱食度: {cat_data['saturation']}, 活力值: {cat_data['energy']}, 技能: {cat_data['skill']}, 日期: {cat_data['date']}"
|