entari-plugin-hyw 2.2.5__py3-none-any.whl → 3.5.0rc6__py3-none-any.whl

entari_plugin_hyw/__init__.py

@@ -1,353 +1,409 @@
-from dataclasses import dataclass
-import html
+from dataclasses import dataclass, field
+from importlib.metadata import version as get_version
+from typing import List, Dict, Any, Optional, Union
 import time
-from collections import deque
-from typing import Any, Deque, Dict, List, Optional, Set, Text, Tuple, Union, TYPE_CHECKING, cast
-from typing_extensions import override
-from arclet.entari import metadata
-from arclet.entari import MessageChain, Session
-from arclet.entari.event.base import MessageEvent
+import asyncio
+
+# 从 pyproject.toml 读取版本号，避免重复维护
+try:
+    __version__ = get_version("entari_plugin_hyw")
+except Exception:
+    __version__ = "0.0.0"
+
+from arclet.alconna import Alconna, Args, AllParam, CommandMeta, Option, Arparma, MultiVar, store_true
+from arclet.entari import metadata, listen, Session, plugin_config, BasicConfModel, plugin, command
+from arclet.letoderea import on
+from arclet.entari import MessageChain, Text, Image, MessageCreatedEvent, Quote, At
+from satori.element import Custom
 from loguru import logger
-from satori.exception import ActionFailed
-from arclet.entari import MessageChain, Image, Quote, Text
 import arclet.letoderea as leto
-from arclet.entari import MessageCreatedEvent, Session
-from arclet.entari import BasicConfModel, metadata, plugin_config
-import httpx
-import asyncio
-import json
-import re
-from arclet.alconna import (
-    Args,
-    Alconna,
-    AllParam,
-    MultiVar,
-    CommandMeta,
-    Option,
-)
-from arclet.entari import MessageChain, Session, command
-from arclet.entari import plugin, Ready, Cleanup, Startup
-from satori.element import Custom, E
-from .hyw_core import HYW, HYWConfig
-
-# 全局变量
-hyw_core = None
-
-class HistoryManager:
-    def __init__(self, max_records: int = 20):
-        self.max_records = max_records
-        self._order: Deque[str] = deque()
-        self._store: Dict[str, List[dict]] = {}
-        self._bindings: Dict[str, Set[str]] = {}
-        self._msg_map: Dict[str, str] = {}
-
-    def extract_message_id(self, message_like: Any) -> Optional[str]:
-        if message_like is None:
-            return None
-        if isinstance(message_like, (list, tuple)):
-            for item in message_like:
-                mid = self.extract_message_id(item)
-                if mid:
-                    return mid
-            return None
-        if isinstance(message_like, dict):
-            for key in ("message_id", "id"):
-                value = message_like.get(key)
-                if value:
-                    return str(value)
-        for attr in ("message_id", "id"):
-            value = getattr(message_like, attr, None)
-            if value:
-                return str(value)
-        nested = getattr(message_like, "message", None)
-        if nested is not None and nested is not message_like:
-            return self.extract_message_id(nested)
-        return None
-
-    def remove(self, conversation_id: Optional[str], *, remove_from_order: bool = True) -> None:
-        if not conversation_id:
-            return
-        cid = str(conversation_id)
-        if remove_from_order:
-            try:
-                self._order.remove(cid)
-            except ValueError:
-                pass
-        bindings = self._bindings.pop(cid, set())
-        for msg_id in bindings:
-            self._msg_map.pop(msg_id, None)
-        self._store.pop(cid, None)
-
-    def _enforce_limit(self) -> None:
-        while len(self._order) > self.max_records:
-            obsolete = self._order.popleft()
-            self.remove(obsolete, remove_from_order=False)
-
-    def remember(self, conversation_id: Optional[str], history: Optional[List[dict]], related_ids: List[Optional[str]]) -> None:
-        if not conversation_id or not history:
-            return
-        cid = str(conversation_id)
-        self._store[cid] = list(history)
-        binding_ids = {str(mid) for mid in related_ids if mid}
-        self._bindings[cid] = binding_ids
-        for mid in binding_ids:
-            self._msg_map[mid] = cid
-        self._order.append(cid)
-        self._enforce_limit()
-
-    def get_history(self, msg_id: str) -> Optional[List[dict]]:
-        cid = self._msg_map.get(msg_id)
-        if cid:
-            return list(self._store.get(cid, []))
-        return None
-    
-    def get_conversation_id(self, msg_id: str) -> Optional[str]:
-        return self._msg_map.get(msg_id)
+from arclet.entari.event.command import CommandReceive
 
-history_manager = HistoryManager()
+from .pipeline import ProcessingPipeline
+from .history import HistoryManager
+from .render_vue import ContentRenderer
+from .misc import process_onebot_json, process_images, resolve_model_name, render_refuse_answer, REFUSE_ANSWER_MARKDOWN
+from arclet.entari.event.lifespan import Cleanup
 
-# Request lock for HYW agent
-_hyw_request_lock: Optional[asyncio.Lock] = None
+import os
+import secrets
+import base64
 
-def _get_hyw_request_lock() -> asyncio.Lock:
-    global _hyw_request_lock
-    if _hyw_request_lock is None:
-        _hyw_request_lock = asyncio.Lock()
-    return _hyw_request_lock
+import re
 
 
+def parse_color(color: str) -> str:
+    """
+    Parse color from hex or RGB tuple to hex format.
+    Supports: #ff0000, ff0000, (255, 0, 0), 255,0,0
+    """
+    if not color:
+        return "#ef4444"
+    
+    color = str(color).strip()
+    
+    # Hex format: #fff or #ffffff or ffffff
+    if color.startswith('#') and len(color) in [4, 7]:
+        return color
+    if re.match(r'^[0-9a-fA-F]{6}$', color):
+        return f'#{color}'
+    
+    # RGB tuple: (r, g, b) or r,g,b
+    rgb_match = re.match(r'^\(?(\d+)[,\s]+(\d+)[,\s]+(\d+)\)?$', color)
+    if rgb_match:
+        r, g, b = (max(0, min(255, int(x))) for x in rgb_match.groups())
+        return f'#{r:02x}{g:02x}{b:02x}'
+    
+    logger.warning(f"Invalid color '{color}', using default #ef4444")
+    return "#ef4444"
+
+class _RecentEventDeduper:
+    def __init__(self, ttl_seconds: float = 30.0, max_size: int = 2048):
+        self.ttl_seconds = ttl_seconds
+        self.max_size = max_size
+        self._seen: Dict[str, float] = {}
+
+    def seen_recently(self, key: str) -> bool:
+        now = time.time()
+        if len(self._seen) > self.max_size:
+            self._prune(now)
+        ts = self._seen.get(key)
+        if ts is None or now - ts > self.ttl_seconds:
+            self._seen[key] = now
+            return False
+        return True
+
+    def _prune(self, now: float):
+        expired = [k for k, ts in self._seen.items() if now - ts > self.ttl_seconds]
+        for k in expired:
+            self._seen.pop(k, None)
+        if len(self._seen) > self.max_size:
+            for k, _ in sorted(self._seen.items(), key=lambda kv: kv[1])[: len(self._seen) - self.max_size]:
+                self._seen.pop(k, None)
+
+_event_deduper = _RecentEventDeduper()
+
+@dataclass
 class HywConfig(BasicConfModel):
-    command_name_list: Union[str, List[str]] = "hyw"
-    model_name: str
-    api_key: str
+    admins: List[str] = field(default_factory=list)
+    models: List[Dict[str, Any]] = field(default_factory=list)
+    question_command: str = "/q"
+    model_name: Optional[str] = None
+    api_key: Optional[str] = None
     base_url: str = "https://openrouter.ai/api/v1"
+    vision_model_name: Optional[str] = None
+    vision_api_key: Optional[str] = None
+    language: str = "Simplified Chinese"
+    vision_base_url: Optional[str] = None
+    instruct_model_name: Optional[str] = None
+    instruct_api_key: Optional[str] = None
+    instruct_base_url: Optional[str] = None
+    search_base_url: str = "https://lite.duckduckgo.com/lite/?q={query}"
+    image_search_base_url: str = "https://duckduckgo.com/?q={query}&iax=images&ia=images"
     headless: bool = False
     save_conversation: bool = False
+    icon: str = "openai"
+    render_timeout_ms: int = 6000
+    render_image_timeout_ms: int = 3000
+    extra_body: Optional[Dict[str, Any]] = None
+    vision_extra_body: Optional[Dict[str, Any]] = None
+    instruct_extra_body: Optional[Dict[str, Any]] = None
+    enable_browser_fallback: bool = False
+    reaction: bool = False
+    quote: bool = True
+    temperature: float = 0.4
+    # Billing configuration (price per million tokens)
+    input_price: Optional[float] = None  # $ per 1M input tokens
+    output_price: Optional[float] = None  # $ per 1M output tokens
+    # Vision model pricing overrides (defaults to main model pricing if not set)
+    vision_input_price: Optional[float] = None
+    vision_output_price: Optional[float] = None
+    # Instruct model pricing overrides (defaults to main model pricing if not set)
+    instruct_input_price: Optional[float] = None
+    instruct_output_price: Optional[float] = None
+    # Provider Names
+    search_name: str = "DuckDuckGo"
+    search_provider: str = "crawl4ai"  # crawl4ai | httpx | ddgs
+    fetch_provider: str = "crawl4ai"  # crawl4ai | jinaai
+    jina_api_key: Optional[str] = None  # Optional API key for Jina AI
+    model_provider: Optional[str] = None
+    vision_model_provider: Optional[str] = None
+    instruct_model_provider: Optional[str] = None
+    # UI Theme
+    theme_color: str = "#ef4444"  # Tailwind red-500, supports hex/RGB/color names
     
-    browser_tool: str = "jina"
-    jina_api_key: Optional[str] = None
-    
-    vision_model_name: Optional[str] = None
-    vision_base_url: Optional[str] = None
-    vision_api_key: Optional[str] = None
-    # verbose: bool = False
-
-metadata(
-    "hyw",
-    author=[{"name": "kumoSleeping", "email": "zjr2992@outlook.com"}],
-    version="0.1.0",
-    description="",
-    config=HywConfig,
-)
-
-conf = plugin_config(HywConfig)
-alc = Alconna(
-    conf.command_name_list,
-    Option("-t|--text", dest="text_only", default=False, help_text="仅文本模式(禁用图片识别)"),
-    Args["all_param", AllParam],
-    # Option("-v|--verbose", dest="verbose", default=False, help_text="启用详细日志输出"),
-    meta=CommandMeta(compact=False)
-)
-
-# Create HYW configuration
-hyw_config = HYWConfig(
-    api_key=conf.api_key,
-    model_name=conf.model_name,
-    base_url=conf.base_url,
-    save_conversation=conf.save_conversation,
-    headless=conf.headless,
-    browser_tool=conf.browser_tool,
-    jina_api_key=conf.jina_api_key,
-    vision_model_name=conf.vision_model_name,
-    vision_base_url=conf.vision_base_url,
-    vision_api_key=conf.vision_api_key
-)
-
-hyw = HYW(config=hyw_config)
+    def __post_init__(self):
+        """Parse and normalize theme color after initialization."""
+        self.theme_color = parse_color(self.theme_color)
 
 
 
-# Emoji到代码的映射字典
-EMOJI_TO_CODE = {
-    "🐳": "128051",
-    "❌": "10060",
-    "🍧": "127847",
-    "✨": "10024",
-    "📫": "128235"
-}
-
-async def download_image(url: str) -> bytes:
-    """下载图片"""
-    try:
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            resp = await client.get(url)
-            if resp.status_code == 200:
-                return resp.content
-            else:
-                raise ActionFailed(f"下载图片失败，状态码: {resp.status_code}")
-    except Exception as e:
-        raise ActionFailed(f"下载图片失败: {url}, 错误: {str(e)}")
+conf = plugin_config(HywConfig)
+history_manager = HistoryManager()
+renderer = ContentRenderer()
 
 
-def process_onebot_json(json_data_str: str) -> str:
-    try:
-        # 解码HTML实体
-        json_str = html.unescape(json_data_str)
-        return json_str
-    except Exception as e:
-        return json_data_str
+class GlobalCache:
+    models_image_path: Optional[str] = None
 
+global_cache = GlobalCache()
 
 async def react(session: Session, emoji: str):
+    if not conf.reaction: return
     try:
-        if session.event.login.platform == "onebot":
-            code = EMOJI_TO_CODE.get(emoji, "10024")
-            await session.account.protocol.call_api("internal/set_group_reaction", {"group_id": int(session.guild.id), "message_id": int(session.event.message.id), "code": code, "is_add": True})
-        else:
-            await session.reaction_create(emoji=emoji)
-    except ActionFailed:
-        pass
-
-def handle_shortcut(message_chain: MessageChain) -> Tuple[bool, str]:
-    current_msg_text = str(message_chain.get(Text)) if message_chain.get(Text) else ""
-    is_shortcut = False
-    shortcut_replacement = ""
-    if current_msg_text.strip().startswith("/"):
-        is_shortcut = True
-        shortcut_replacement = current_msg_text.strip()[1:]
-    return is_shortcut, shortcut_replacement
-
-async def process_images(mc: MessageChain, parse_result: Any) -> Tuple[List[str], Optional[str]]:
-    is_text_only = False
-    if parse_result.matched:
-        def get_bool_value(val):
-            if hasattr(val, 'value'):
-                return bool(val.value)
-            return bool(val)
-        is_text_only = get_bool_value(getattr(parse_result, 'text_only', False))
-    
-    text_str = str(mc.get(Text) or "")
-    if not is_text_only and re.search(r'(?:^|\s)(-t|--text)(?:$|\s)', text_str):
-        is_text_only = True
-    
-    if is_text_only:
-        logger.info("检测到仅文本模式参数，跳过图片分析")
-        return [], None
-
-    has_images = bool(mc.get(Image))
-    images = []
-    if has_images:
-        urls = mc[Image].map(lambda x: x.src)
-        tasks = [download_image(url) for url in urls]
-        raw_images = await asyncio.gather(*tasks)
-        import base64
-        images = [base64.b64encode(img).decode('utf-8') for img in raw_images]
-    
-    return images, None
-
-@leto.on(MessageCreatedEvent)
-async def on_message_created(message_chain: MessageChain, session: Session[MessageEvent]):
-    # Skip if no substantial content in original message
-    original_text = str(message_chain.get(Text)).strip()
-    has_images = bool(message_chain.get(Image))
-    has_custom = bool(message_chain.get(Custom))
-    if not original_text and not has_images and not has_custom:
-        return
-    
+        await session.reaction_create(emoji=emoji)
+    except Exception as e:
+        logger.warning(f"Reaction failed: {e}")
+
+async def process_request(
+    session: Session[MessageCreatedEvent],
+    all_param: Optional[MessageChain] = None,
+    selected_model: Optional[str] = None,
+    selected_vision_model: Optional[str] = None,
+    conversation_key_override: Optional[str] = None,
+    local_mode: bool = False,
+) -> None:
+    logger.info(f"Processing request: {all_param}")
+    mc = MessageChain(all_param)
+    logger.info(f"reply: {session.reply}")
     if session.reply:
         try:
-            message_chain.extend(MessageChain(" ") + session.reply.origin.message)
-        except Exception:
-            pass
+            # Check if reply is from self (the bot)
+            # 1. Check by Message ID (reliable for bot's own messages if recorded)
+            reply_msg_id = str(session.reply.origin.id) if hasattr(session.reply.origin, 'id') else None
+            is_bot = False
             
-    message_chain = message_chain.get(Text) + message_chain.get(Image) + message_chain.get(Custom)
-    
-    quoted_message_id: Optional[str] = None
-    conversation_history_key: Optional[str] = None
-    conversation_history_payload: List[dict] = []
-    
-    if session.reply:
-        try:
-            quoted_message_id = str(session.reply.origin.id) if hasattr(session.reply.origin, 'id') else None
+            if reply_msg_id and history_manager.is_bot_message(reply_msg_id):
+                is_bot = True
+                logger.info(f"Reply target {reply_msg_id} identified as bot message via history")
+
+            if is_bot:
+                logger.info("Reply is from me - ignoring content")
+            else:
+                logger.info(f"Reply is from user (or unknown) - including content")
+                mc.extend(MessageChain(" ") + session.reply.origin.message)
         except Exception as e:
-            logger.warning(f"提取引用消息ID失败: {e}")
-            quoted_message_id = None
-            
-        if quoted_message_id:
-            conversation_history_key = history_manager.get_conversation_id(quoted_message_id)
-            if conversation_history_key:
-                conversation_history_payload = history_manager.get_history(quoted_message_id) or []
-                logger.info(f"继续对话模式触发, 引用消息ID: {quoted_message_id}, 历史长度: {len(conversation_history_payload)}")
-
-    parse_result = alc.parse(message_chain)
-    is_shortcut, shortcut_replacement = handle_shortcut(message_chain)
-    
-    should_process = parse_result.matched or (bool(conversation_history_key) and is_shortcut)
+            logger.warning(f"Failed to process reply origin: {e}")
+            mc.extend(MessageChain(" ") + session.reply.origin.message)
     
-    if not should_process:
+    # Filter and reconstruct MessageChain
+    filtered_elements = mc.get(Text) + mc.get(Image) + mc.get(Custom)
+    mc = MessageChain(filtered_elements)
+    logger.info(f"mc: {mc}")
+
+    text_content = str(mc.get(Text)).strip()
+    # Remove HTML image tags from text content to prevent "unreasonable code behavior"
+    text_content = re.sub(r'<img[^>]+>', '', text_content, flags=re.IGNORECASE)
+
+    if not text_content and not mc.get(Image) and not mc.get(Custom):
         return
 
-    raw_param_chain: MessageChain = parse_result.all_param if parse_result.matched else message_chain  # type: ignore
-    if not parse_result.matched and is_shortcut:
-        logger.debug(f"触发快捷指令，替换内容: {shortcut_replacement}")
-        
-    mc = MessageChain(raw_param_chain)
+    # History & Context
+    hist_key = conversation_key_override
+    if not hist_key and session.reply and hasattr(session.reply.origin, 'id'):
+        hist_key = history_manager.get_conversation_id(str(session.reply.origin.id))
     
-    async def process_request() -> None:
-        await react(session, "✨")
+    hist_payload = history_manager.get_history(hist_key) if hist_key else []
+    meta = history_manager.get_metadata(hist_key) if hist_key else {}
+    context_id = f"guild_{session.guild.id}" if session.guild else f"user_{session.user.id}"
+
+    if conf.reaction: await react(session, "✨")
+
+    try:
+        msg_text = str(mc.get(Text)).strip() if mc.get(Text) else ""
+        msg_text = re.sub(r'<img[^>]+>', '', msg_text, flags=re.IGNORECASE)
+        
+        # If message is empty but has images, use a placeholder
+        if not msg_text and (mc.get(Image) or mc.get(Custom)):
+             msg_text = "[图片]"
+        
+        for custom in [e for e in mc if isinstance(e, Custom)]:
+            if custom.tag == 'onebot:json':
+                if decoded := process_onebot_json(custom.attributes()): msg_text += f"\n{decoded}"
+                break
+        
+        # Model Selection (Step 1)
+        # Resolve model names from config if they are short names/keywords
+        model = selected_model or meta.get("model")
+        if model and model != "off":
+            resolved, err = resolve_model_name(model, conf.models)
+            if resolved:
+                model = resolved
+            elif err:
+                logger.warning(f"Model resolution warning for {model}: {err}")
+
+        vision_model = selected_vision_model or meta.get("vision_model")
+        if vision_model and vision_model != "off":
+            resolved_v, err_v = resolve_model_name(vision_model, conf.models)
+            if resolved_v:
+                vision_model = resolved_v
+            elif err_v:
+                logger.warning(f"Vision model resolution warning for {vision_model}: {err_v}")
+
+        images, err = await process_images(mc, vision_model)
+
+        # Call Pipeline directly
+        safe_input = msg_text
+        pipeline = ProcessingPipeline(conf)
         try:
-            if is_shortcut and not parse_result.matched:
-                msg = shortcut_replacement
-            else:
-                msg = mc.get(Text).strip() if mc.get(Text) else ""
-            
-            if mc.get(Custom): # type: ignore
-                custom_elements = [e for e in mc if isinstance(e, Custom)]
-                for custom in custom_elements:
-                    if custom.tag == 'onebot:json':
-                        decoded_json = process_onebot_json(custom.attributes())
-                        msg += decoded_json
-                        break
-            
-            time_start = time.perf_counter()
-            images, error_msg = await process_images(mc, parse_result)
-            
-            if error_msg:
-                await session.send(error_msg)
-                return
+            resp = await pipeline.execute(
+                safe_input,
+                hist_payload,
+                model_name=model,
+                images=images,
+                selected_vision_model=vision_model,
+            )
+        finally:
+            await pipeline.close()
+        
+        # Step 1 Results
+        step1_vision_model = resp.get("vision_model_used")
+        step1_model = resp.get("model_used")
+        step1_history = resp.get("conversation_history", [])
+        step1_stats = resp.get("stats", {})
+        
+        final_resp = resp
+        
+        # Step 2 (Optional)
 
-            lock = _get_hyw_request_lock()
-            async with lock:
-                response = await hyw.agent(str(msg), conversation_history=conversation_history_payload, images=images)
             
-            response_content = response.get("llm_response", "") if isinstance(response, dict) else ""
-            new_history = response.get("conversation_history", []) if isinstance(response, dict) else []
+        
+        # Extract Response Data
+        content = final_resp.get("llm_response", "")
+        structured = final_resp.get("structured_response", {})
+        
+        # Render
+        import tempfile
+        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tf:
+            output_path = tf.name
+        model_used = final_resp.get("model_used")
+
+        # Determine session short code
+        if hist_key:
+            display_session_id = history_manager.get_code_by_key(hist_key)
+            if not display_session_id:
+                display_session_id = history_manager.generate_short_code()
+        else:
+            display_session_id = history_manager.generate_short_code()
+
+        # Use stats_list if available, otherwise standard stats
+        stats_to_render = final_resp.get("stats_list", final_resp.get("stats", {}))
+        
+        # Check if refuse_answer was triggered
+        if final_resp.get("refuse_answer"):
+            logger.info(f"Refuse answer triggered. Rendering refuse image. Reason: {final_resp.get('refuse_reason', '')}")
+            render_ok = await render_refuse_answer(
+                renderer=renderer,
+                output_path=output_path,
+                theme_color=conf.theme_color,
+            )
+        else:
+            render_ok = await renderer.render(
+                markdown_content=content,
+                output_path=output_path,
+                stats=stats_to_render,
+                references=structured.get("references", []),
+                page_references=structured.get("page_references", []),
+                image_references=structured.get("image_references", []),
+                stages_used=final_resp.get("stages_used", []),
+                image_timeout=conf.render_image_timeout_ms,
+                theme_color=conf.theme_color,
+            )
+        
+        # Send & Save
+        if not render_ok:
+            logger.error("Render failed; skipping reply. Check Crawl4AI rendering status.")
+            if os.path.exists(output_path):
+                try:
+                    os.remove(output_path)
+                except Exception as exc:
+                    logger.warning(f"Failed to delete render output {output_path}: {exc}")
+            sent = None
+        else:
+            # Convert to base64
+            with open(output_path, "rb") as f:
+                img_data = base64.b64encode(f.read()).decode()
+                
+            # Build single reply chain (image only now)
+            elements = []
+            elements.append(Image(src=f'data:image/png;base64,{img_data}'))
+
+            msg_chain = MessageChain(*elements)
             
+            if conf.quote:
+                msg_chain = MessageChain(Quote(session.event.message.id)) + msg_chain
+                
+            # Use reply_to instead of manual Quote insertion to avoid ActionFailed errors
+            sent = await session.send(msg_chain)
+        
+        sent_id = next((str(e.id) for e in sent if hasattr(e, 'id')), None) if sent else None
+        msg_id = str(session.event.message.id) if hasattr(session.event, 'message') else str(session.event.id)
+        related = [msg_id] + ([str(session.reply.origin.id)] if session.reply and hasattr(session.reply.origin, 'id') else [])
+        
+        history_manager.remember(
+            sent_id,
+            final_resp.get("conversation_history", []),
+            related,
+            {
+                "model": model_used,
+                "trace_markdown": final_resp.get("trace_markdown"),
+            },
+            context_id,
+            code=display_session_id,
+        )
+        
+        if conf.save_conversation and sent_id:
+            history_manager.save_to_disk(sent_id)
+
+
+    except Exception as e:
+        logger.exception(f"Error: {e}")
+        err_msg = f"Error: {e}"
+        if conf.quote:
+             await session.send([Quote(session.event.message.id), err_msg])
+        else:
+             await session.send(err_msg)
+        
+        # Save conversation on error if response was generated
+        if 'resp' in locals() and resp and conf.save_conversation:
             try:
-                send_result = await session.send([Quote(session.event.message.id), response_content])
-            except ActionFailed as e:
-                if "9057" in str(e):
-                    logger.warning(f"发送消息失败(9057)，尝试截断发送: {e}")
-                    truncated_content = response_content[:1000] + "\n\n[...内容过长，已大幅截断...]"
-                    send_result = await session.send([Quote(session.event.message.id), truncated_content])
-                else:
-                    raise e
-            
-            sent_message_id = history_manager.extract_message_id(send_result)
-            current_user_message_id = str(session.event.message.id)
-            related_ids: List[Optional[str]] = [current_user_message_id, sent_message_id]
-            
-            if conversation_history_key:
-                history_manager.remove(conversation_history_key)
-                related_ids.append(quoted_message_id)
-            
-            history_manager.remember(sent_message_id, new_history, related_ids)
-            
-        except Exception as exc:
-            await react(session, "❌")
-            logger.exception("处理HYW消息失败: {}", exc)
+                # Use a temporary ID for error cases
+                error_id = f"error_{int(time.time())}_{secrets.token_hex(4)}"
+                history_manager.remember(error_id, resp.get("conversation_history", []), [], {"model": model_used if 'model_used' in locals() else "unknown", "error": str(e)}, context_id, code=display_session_id if 'display_session_id' in locals() else None)
+                history_manager.save_to_disk(error_id)
+                logger.info(f"Saved error conversation to {error_id}")
+            except Exception as save_err:
+                logger.error(f"Failed to save error conversation: {save_err}")
+
+
+alc = Alconna(
+    conf.question_command,
+    Args["all_param;?", AllParam],
+)
+
+@command.on(alc)    
+async def handle_question_command(session: Session[MessageCreatedEvent], result: Arparma):
+    """Handle main Question command"""
+    try:
+        logger.info(f"Question Command Triggered. Message: {result}")
+        mid = str(session.event.message.id) if getattr(session.event, "message", None) else str(session.event.id)
+        dedupe_key = f"{getattr(session.account, 'id', 'account')}:{mid}"
+        if _event_deduper.seen_recently(dedupe_key):
+            logger.warning(f"Duplicate command event ignored: {dedupe_key}")
+            return
+    except Exception:
+        pass
+
+    logger.info(f"Question Command Triggered. Message: {session.event.message}")
+    
+    args = result.all_matched_args
+    logger.info(f"Matched Args: {args}")
+    
+    await process_request(session, args.get("all_param"), selected_model=None, selected_vision_model=None, conversation_key_override=None, local_mode=False)
 
-    asyncio.create_task(process_request())
-    return
+metadata("hyw", author=[{"name": "kumoSleeping", "email": "zjr2992@outlook.com"}], version=__version__, config=HywConfig)
 
 
+@listen(CommandReceive)
+async def remove_at(content: MessageChain):
+    return content.lstrip(At)