✨ 支持消息表态 (#51)

Author: KomoriDev

nonebot_plugin_deepseek/utils.py

@@ -8,8 +8,9 @@
 from nonebot.adapters import Event
 from nonebot.permission import User, Permission
 from nonebot_plugin_waiter import Waiter, prompt
+from nonebot_plugin_alconna import SupportAdapter
 from nonebot.matcher import Matcher, current_event, current_matcher
-from nonebot_plugin_alconna.uniseg import UniMsg, UniMessage, get_message_id
+from nonebot_plugin_alconna.uniseg import UniMsg, UniMessage, get_target, get_message_id, message_reaction
 
 from .apis import API
 from .log import tts_logger
@@ -46,6 +47,8 @@ async def handle(self, content: Optional[str]) -> None:
         if content:
             self.context.append({"role": "user", "content": content})
 
+        await self._message_reaction("thinking")
+
         if not self.is_contextual:
             await self._handle_single_conversion()
         else:
@@ -99,6 +102,20 @@ def _prompt_handler(self, msg: UniMsg) -> UniMsg:
         self.message_id = get_message_id()
         return msg
 
+    async def _message_reaction(self, status: Literal["fail", "thinking", "done"]) -> None:
+        emoji_map = {
+            "fail": ["10060", "❌"],
+            "thinking": ["424", "👀"],
+            "done": ["144", "🎉"],
+        }
+        target = get_target(self.event)
+        if target.adapter == SupportAdapter.onebot11:
+            emoji = emoji_map[status][0]
+        else:
+            emoji = emoji_map[status][1]
+
+        await message_reaction(emoji, message_id=self.message_id)
+
     async def _process_waiter_response(self, resp: Union[bool, str]) -> None:
         timeout = ds_config.timeout if isinstance(ds_config.timeout, int) else ds_config.timeout.user_input
 
@@ -112,6 +129,8 @@ async def _process_waiter_response(self, resp: Union[bool, str]) -> None:
                 await UniMessage.text("等待超时").finish(reply_to=self.message_id)
             resp = self._waiter_handler(_resp, skip=True)
 
+        await self._message_reaction("thinking")
+
         if resp is False:
             await UniMessage.text("已结束对话").finish(reply_to=self.message_id)
         elif resp == "rollback":
@@ -138,6 +157,7 @@ async def _handle_rollback(self, steps: int = 1, by_error: bool = False) -> None
             )
         elif by_error and len(self.context) > 0:
             self.context.clear()
+            await self._message_reaction("fail")
             await UniMessage.text("Oops! 连接异常，请重新输入").send(reply_to=self.message_id)
         else:
             await UniMessage.text("无法回滚，当前对话记录为空").send(reply_to=self.message_id)
@@ -167,6 +187,7 @@ async def _get_response_message(self) -> Optional[Message]:
             return completion.choices[0].message
         except (httpx.ReadTimeout, httpx.RequestError):
             if not self.is_contextual:
+                await self._message_reaction("fail")
                 await UniMessage.text("Oops! 网络超时，请稍后重试").finish(reply_to=self.message_id)
             await self._handle_rollback(by_error=True)
         except RequestException as e:
@@ -199,6 +220,9 @@ def _format_output(self, message: Message, with_thinking: bool) -> str:
     async def _send_response(self, message: Message) -> None:
         output = self._format_output(message, ds_config.enable_send_thinking)
         message.reasoning_content = None
+
+        await self._message_reaction("done")
+
         if self.tts_model:
             try:
                 output = self._format_output(message, False)