update v0.0.5.0

plugins/ai/__init__.py

@@ -1,20 +1,33 @@
-from .data_source import get_chat_result, hello, no_result
-from services.log import logger
 from nonebot import on_message
-from nonebot.rule import to_me
-from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import (
     Bot,
-    PrivateMessageEvent,
+    GroupMessageEvent,
     Message,
     MessageEvent,
 )
-from utils.utils import get_message_text, get_message_imgs
+from nonebot.rule import to_me
+from nonebot.typing import T_State
+
 from models.friend_user import FriendUser
 from models.group_member_info import GroupInfoUser
+from services.log import logger
+from utils.utils import get_message_text, get_message_imgs
+from .data_source import get_chat_result, hello, no_result
+from configs.config import NICKNAME
 
-__plugin_name__ = "AI [Hidden]"
-
+__zx_plugin_name__ = "AI"
+__plugin_usage__ = f"""
+usage：
+    与{NICKNAME}普普通通的对话吧！
+"""
+__plugin_version__ = 0.1
+__plugin_author__ = 'HibiKier'
+__plugin_settings__ = {
+    "level": 5,
+    "default_status": True,
+    "limit_superuser": False,
+    "cmd": ["Ai", "ai", "AI", "aI"],
+}
 
 ai = on_message(rule=to_me(), priority=8)
 
@@ -38,20 +51,20 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
     ]:
         await ai.finish(hello())
     img = imgs[0] if imgs else ""
-    if isinstance(event, PrivateMessageEvent):
-        nickname = await FriendUser.get_friend_nickname(event.user_id)
-    else:
+    if isinstance(event, GroupMessageEvent):
         nickname = await GroupInfoUser.get_group_member_nickname(
             event.user_id, event.group_id
         )
+    else:
+        nickname = await FriendUser.get_friend_nickname(event.user_id)
     if not nickname:
-        if isinstance(event, PrivateMessageEvent):
-            nickname = event.sender.nickname
-        else:
+        if isinstance(event, GroupMessageEvent):
             nickname = event.sender.card if event.sender.card else event.sender.nickname
+        else:
+            nickname = event.sender.nickname
     result = await get_chat_result(msg, img, event.user_id, nickname)
     logger.info(
-        f"USER {event.user_id} GROUP {event.group_id if not isinstance(event, PrivateMessageEvent) else ''} "
+        f"USER {event.user_id} GROUP {event.group_id if isinstance(event, GroupMessageEvent) else ''} "
         f"问题：{msg} ---- 回答：{result}"
     )
     if result: