feat: 插件新增配置项以支持在渲染新结果前清除旧的 HTML 输出

2025-12-20 15:07:41 +08:00
parent 0ef4d67d09
commit 533a6dee5d
8 changed files with 119 additions and 1 deletions
--- a/plugins/actions/smart-mind-map/smart_mind_map.py
+++ b/plugins/actions/smart-mind-map/smart_mind_map.py
@@ -60,6 +60,7 @@ User Language: {user_language}
 """

 HTML_TEMPLATE_MINDMAP = """
+<!-- OPENWEBUI_PLUGIN_OUTPUT -->
 <!DOCTYPE html>
 <html lang="{user_language}">
 <head>
@@ -369,6 +370,10 @@ class Action:
            default=100,
            description="Minimum text length (character count) required for mind map analysis.",
        )
+        CLEAR_PREVIOUS_HTML: bool = Field(
+            default=False,
+            description="Whether to clear existing plugin-generated HTML content in the message before appending new results (identified by marker).",
+        )

    def __init__(self):
        self.valves = self.Valves()
@@ -393,6 +398,11 @@ class Action:
            extracted_content = llm_output.strip()
        return extracted_content.replace("</script>", "<\\/script>")

+    def _remove_existing_html(self, content: str) -> str:
+        """Removes existing plugin-generated HTML code blocks from the content."""
+        pattern = r"```html\s*<!-- OPENWEBUI_PLUGIN_OUTPUT -->[\s\S]*?```"
+        return re.sub(pattern, "", content).strip()
+
    async def action(
        self,
        body: dict,
@@ -558,6 +568,9 @@ class Action:
                .replace("{markdown_syntax}", markdown_syntax)
            )

+            if self.valves.CLEAR_PREVIOUS_HTML:
+                long_text_content = self._remove_existing_html(long_text_content)
+
            html_embed_tag = f"```html\n{final_html_content}\n```"
            body["messages"][-1]["content"] = f"{long_text_content}\n\n{html_embed_tag}"