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:

plugins/ai/data_source.py

@@ -1,19 +1,21 @@
+import os
+import random
+import re
+
+import aiohttp
+from aiohttp.client import ClientSession
+
 from configs.config import TL_KEY, ALAPI_TOKEN, ALAPI_AI_CHECK, NICKNAME
 from configs.path_config import IMAGE_PATH, DATA_PATH
-from aiohttp.client import ClientSession
 from services.log import logger
 from utils.message_builder import image, face
-from utils.utils import get_bot
-import re
-import aiohttp
-import random
-import os
 
 try:
     import ujson as json
 except ModuleNotFoundError:
     import json
 
+
 url = "http://openapi.tuling123.com/openapi/api/v2"
 
 check_url = "https://v2.alapi.cn/api/censor/text"
@@ -24,6 +26,14 @@ anime_data = json.load(open(DATA_PATH + "anime.json", "r", encoding="utf8"))
 
 
 async def get_chat_result(text: str, img_url: str, user_id: int, nickname: str) -> str:
+    """
+    获取 AI 返回值，顺序：图灵 -> 青云客
+    :param text: 问题
+    :param img_url: 图片链接
+    :param user_id: 用户id
+    :param nickname: 用户昵称
+    :return: 回答
+    """
     global index
     if index == 5:
         index = 0
@@ -50,7 +60,15 @@ async def get_chat_result(text: str, img_url: str, user_id: int, nickname: str)
 
 
 # 图灵接口
-async def tu_ling(text: str, img_url: str, user_id: int, sess: ClientSession):
+async def tu_ling(text: str, img_url: str, user_id: int, sess: ClientSession) -> str:
+    """
+    获取图灵接口的回复
+    :param text: 问题
+    :param img_url: 图片链接
+    :param user_id: 用户id
+    :param sess: AIOHTTP SESSION
+    :return: 图灵回复
+    """
     global index
     try:
         if text:
@@ -94,7 +112,13 @@ async def tu_ling(text: str, img_url: str, user_id: int, sess: ClientSession):
 
 
 # 屑 AI
-async def xie_ai(text: str, sess: ClientSession):
+async def xie_ai(text: str, sess: ClientSession) -> str:
+    """
+    获取青云客回复
+    :param text: 问题
+    :param sess: AIOHTTP SESSION
+    :return: 青云可回复
+    """
     async with sess.get(
         f"http://api.qingyunke.com/api.php?key=free&appid=0&msg={text}"
     ) as res:
@@ -104,8 +128,8 @@ async def xie_ai(text: str, sess: ClientSession):
             content = data["content"]
             if "菲菲" in content:
                 content = content.replace("菲菲", NICKNAME)
-            if '艳儿' in content:
-                content = content.replace('艳儿', NICKNAME)
+            if "艳儿" in content:
+                content = content.replace("艳儿", NICKNAME)
             if "公众号" in content:
                 content = ""
             if "{br}" in content:
@@ -123,11 +147,17 @@ async def xie_ai(text: str, sess: ClientSession):
                     )
                 else:
                     break
-        return content if not content and not ALAPI_AI_CHECK else await check_text(content, sess)
+        return (
+            content
+            if not content and not ALAPI_AI_CHECK
+            else await check_text(content, sess)
+        )
 
 
-# 打招呼内容
 def hello() -> str:
+    """
+    一些打招呼的内容
+    """
     result = random.choice(
         (
             "哦豁？！",
@@ -147,6 +177,9 @@ def hello() -> str:
 
 # 没有回答时回复内容
 def no_result() -> str:
+    """
+    没有回答时的回复
+    """
     return (
         random.choice(
             [
@@ -161,8 +194,12 @@ def no_result() -> str:
     )
 
 
-# 检测屑AI回复的文本是否是 *话
 async def check_text(text: str, sess: ClientSession) -> str:
+    """
+    ALAPI文本检测，主要针对青云客API，检测为恶俗文本改为无回复的回答
+    :param text: 回复
+    :param sess: AIOHTTP SESSION
+    """
     if not ALAPI_TOKEN:
         return text
     params = {"token": ALAPI_TOKEN, "text": text}
@@ -170,8 +207,8 @@ async def check_text(text: str, sess: ClientSession) -> str:
         async with sess.get(check_url, timeout=2, params=params) as response:
             data = await response.json()
             if data["code"] == 200:
-                if data['data']["conclusion_type"] == 2:
-                    return ''
+                if data["data"]["conclusion_type"] == 2:
+                    return ""
     except Exception as e:
         logger.error(f"检测违规文本错误...{type(e)}：{e}")
     return text