✨ 暗语初步支持

nonebot_plugin_marshoai/__init__.py

@@ -26,17 +26,19 @@ from nonebot.plugin import require
 
 require("nonebot_plugin_alconna")
 require("nonebot_plugin_localstore")
+require("nonebot_plugin_argot")
 
 import nonebot_plugin_localstore as store  # type: ignore
 from nonebot import get_driver, logger  # type: ignore
 
 from .config import config
-
-# from .hunyuan import *
 from .dev import *
 from .marsho import *
 from .metadata import metadata
 
+# from .hunyuan import *
+
+
 __author__ = "Asankilp"
 __plugin_meta__ = metadata
 

nonebot_plugin_marshoai/dev.py

@@ -9,7 +9,7 @@ from nonebot.typing import T_State
 from nonebot_plugin_marshoai.plugin.load import reload_plugin
 
 from .config import config
-from .marsho import context
+from .instances import context
 from .plugin.func_call.models import SessionContext
 
 require("nonebot_plugin_alconna")

nonebot_plugin_marshoai/handler.py

@@ -1,4 +1,5 @@
 import json
+from datetime import timedelta
 from typing import Optional, Tuple, Union
 
 from azure.ai.inference.models import (
@@ -17,7 +18,14 @@ from nonebot.matcher import (
     current_event,
     current_matcher,
 )
-from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg, get_message_id, get_target
+from nonebot_plugin_alconna.uniseg import (
+    Text,
+    UniMessage,
+    UniMsg,
+    get_message_id,
+    get_target,
+)
+from nonebot_plugin_argot import Argot
 from openai import AsyncOpenAI, AsyncStream
 from openai.types.chat import ChatCompletion, ChatCompletionChunk, ChatCompletionMessage
 
@@ -35,7 +43,7 @@ from .util import (
     make_chat_openai,
     parse_richtext,
 )
-from .utils.processor import process_chat_stream
+from .utils.processor import process_chat_stream, process_completion_to_details
 
 
 class MarshoHandler:
@@ -230,12 +238,20 @@ class MarshoHandler:
                 target_list.append([self.target.id, self.target.private])
 
             # 对话成功发送消息
+            send_message = UniMessage()
             if config.marshoai_enable_richtext_parse:
-                await (await parse_richtext(str(choice_msg_content))).send(
-                    reply_to=True
-                )
+                send_message = await parse_richtext(str(choice_msg_content))
             else:
-                await UniMessage(str(choice_msg_content)).send(reply_to=True)
+                send_message = UniMessage(str(choice_msg_content))
+            send_message.append(
+                Argot(
+                    "detail",
+                    Text(await process_completion_to_details(response)),
+                    command=f"detail",
+                    expired_at=timedelta(minutes=5),
+                )
+            )
+            await send_message.send(reply_to=True)
             return UserMessage(content=user_message), choice_msg_after
         elif choice.finish_reason == CompletionsFinishReason.CONTENT_FILTERED:
 

nonebot_plugin_marshoai/marsho.py

@@ -23,6 +23,7 @@ from nonebot_plugin_alconna import (
     message_reaction,
     on_alconna,
 )
+from nonebot_plugin_argot.extension import ArgotExtension
 
 from .config import config
 from .constants import INTRODUCTION, SUPPORT_IMAGE_MODELS
@@ -63,6 +64,7 @@ marsho_cmd = on_alconna(
     aliases=tuple(config.marshoai_aliases),
     priority=96,
     block=True,
+    extensions=[ArgotExtension()],
 )
 resetmem_cmd = on_alconna(
     Alconna(

nonebot_plugin_marshoai/utils/processor.py

@@ -69,3 +69,19 @@ async def process_chat_stream(
             object="chat.completion",
             usage=None,
         )
+
+
+async def process_completion_to_details(completion: ChatCompletion) -> str:
+    usage_text = ""
+    usage = completion.usage
+    if usage is None:
+        usage_text = "无"
+    else:
+        usage_text = str(usage)
+
+    details_text = f"""=========消息详情=========
+    模型: {completion.model}
+    消息 ID: {completion.id}
+    用量信息: {usage_text}"""
+    print(details_text)
+    return details_text