feat: Add text manipulation blocks for workflow system

Author: lss233

framework/workflow/implementations/blocks/system/basic.py

@@ -36,18 +36,18 @@ class TextReplaceBlock(Block):
     name = "text_replace_block"
     inputs = {
         "text": Input("text", "原始文本", str, "原始文本"),
-        "variable": Input("variable", "被替换的文本", Any, "被替换的文本"),
+        "new_text": Input("new_text", "新文本", Any, "新文本"),
     }
     outputs = {"text": Output("text", "替换后的文本", str, "替换后的文本")}
 
     def __init__(
-        self, name: Annotated[str, ParamMeta(label="变量名称", description="变量名称")]
+        self, variable: Annotated[str, ParamMeta(label="被替换的文本", description="被替换的文本")]
     ):
-        self.name = name
+        self.variable = variable
 
-    def execute(self) -> Dict[str, Any]:
+    def execute(self, text: str, new_text: Any) -> Dict[str, Any]:
         return {
-            "text": self.text.value.replace(self.variable.value, str(self.variable))
+            "text": text.replace(self.variable, str(new_text))
         }
 
 
@@ -56,6 +56,10 @@ class TextExtractByRegexBlock(Block):
     name = "text_extract_by_regex_block"
     inputs = {"text": Input("text", "原始文本", str, "原始文本")}
     outputs = {"text": Output("text", "提取后的文本", str, "提取后的文本")}
+    def __init__(
+        self, regex: Annotated[str, ParamMeta(label="正则表达式", description="正则表达式")]
+    ):
+        self.regex = regex
 
     def execute(self) -> Dict[str, Any]:
         # 使用正则表达式提取文本
@@ -65,3 +69,4 @@ def execute(self) -> Dict[str, Any]:
             return {"text": match.group(0)}
         else:
             return {"text": ""}
+        

framework/workflow/implementations/blocks/system_blocks.py

@@ -2,7 +2,8 @@
 from framework.workflow.implementations.blocks.im.basic import ExtractChatSender
 from framework.workflow.implementations.blocks.llm.image import SimpleStableDiffusionWebUI
 from framework.workflow.implementations.blocks.memory.clear_memory import ClearMemory
-from framework.workflow.implementations.blocks.system.basic import TextBlock
+from framework.workflow.implementations.blocks.system.basic import (TextBlock, TextConcatBlock, TextExtractByRegexBlock,
+                                                                    TextReplaceBlock)
 
 from .game.dice import DiceRoll
 from .game.gacha import GachaSimulator
@@ -17,6 +18,9 @@ def register_system_blocks(registry: BlockRegistry):
     """注册系统自带的 block"""
     # 基础 blocks
     registry.register("text_block", "internal", TextBlock, "基础：文本")
+    registry.register("text_concat_block", "internal", TextConcatBlock, "基础：拼接文本")
+    registry.register("text_replace_block", "internal", TextReplaceBlock, "基础：替换文本")
+    registry.register("text_extract_by_regex_block", "internal", TextExtractByRegexBlock, "基础：正则表达式提取文本")
 
     # IM 相关 blocks
     registry.register("get_message", "internal", GetIMMessage, "IM: 获取最新消息")