entari-plugin-hyw 3.5.0rc3__tar.gz → 3.5.0rc6__tar.gz

{entari_plugin_hyw-3.5.0rc3/src/entari_plugin_hyw.egg-info → entari_plugin_hyw-3.5.0rc6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: entari_plugin_hyw
-Version: 3.5.0rc3
+Version: 3.5.0rc6
 Summary: Use large language models to interpret chat messages
 Author-email: kumoSleeping <zjr2992@outlook.com>
 License: MIT

{entari_plugin_hyw-3.5.0rc3 → entari_plugin_hyw-3.5.0rc6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "entari_plugin_hyw"
-version = "3.5.0-rc3"
+version = "3.5.0-rc6"
 description = "Use large language models to interpret chat messages"
 authors = [{name = "kumoSleeping", email = "zjr2992@outlook.com"}]
 dependencies = [

{entari_plugin_hyw-3.5.0rc3 → entari_plugin_hyw-3.5.0rc6}/src/entari_plugin_hyw/__init__.py

@@ -22,7 +22,7 @@ from arclet.entari.event.command import CommandReceive
 from .pipeline import ProcessingPipeline
 from .history import HistoryManager
 from .render_vue import ContentRenderer
-from .misc import process_onebot_json, process_images, resolve_model_name
+from .misc import process_onebot_json, process_images, resolve_model_name, render_refuse_answer, REFUSE_ANSWER_MARKDOWN
 from arclet.entari.event.lifespan import Cleanup
 
 import os
@@ -288,18 +288,27 @@ async def process_request(
 
         # Use stats_list if available, otherwise standard stats
         stats_to_render = final_resp.get("stats_list", final_resp.get("stats", {}))
-             
-        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,
-        )
+        
+        # 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:
@@ -397,5 +406,4 @@ metadata("hyw", author=[{"name": "kumoSleeping", "email": "zjr2992@outlook.com"}
 
 @listen(CommandReceive)
 async def remove_at(content: MessageChain):
-    logger.info(f"remove_at: {content}")
     return content.lstrip(At)