feat: Add message and text conversion blocks for workflow system

Author: lss233

framework/workflow/implementations/blocks/im/messages.py

@@ -3,7 +3,7 @@
 
 from framework.im.adapter import IMAdapter
 from framework.im.manager import IMManager
-from framework.im.message import IMMessage
+from framework.im.message import IMMessage, MessageElement, TextMessage
 from framework.im.sender import ChatSender
 from framework.ioc.container import DependencyContainer
 from framework.workflow.core.block import Block, Input, Output, ParamMeta
@@ -53,9 +53,58 @@ def execute(
         if not self.im_name:
             adapter = self.container.resolve(IMAdapter)
         else:
-            adapter = self.container.resolve(IMManager).get_adapter(self.im_name)
+            adapter = self.container.resolve(
+                IMManager).get_adapter(self.im_name)
         loop: asyncio.AbstractEventLoop = self.container.resolve(
             asyncio.AbstractEventLoop
         )
         loop.create_task(adapter.send_message(msg, target or src_msg.sender))
         # return {"ok": True}
+
+# IMMessage 转纯文本
+
+
+class IMMessageToText(Block):
+    """IMMessage 转纯文本"""
+
+    name = "im_message_to_text"
+    container: DependencyContainer
+    inputs = {"msg": Input("msg", "IM 消息", IMMessage, "IM 消息")}
+    outputs = {"text": Output("text", "纯文本", str, "纯文本")}
+
+    def execute(self, msg: IMMessage) -> Dict[str, Any]:
+        return {"text": msg.content}
+
+
+# 纯文本转 IMMessage
+class TextToIMMessage(Block):
+    """纯文本转 IMMessage"""
+
+    name = "text_to_im_message"
+    container: DependencyContainer
+    inputs = {"text": Input("text", "纯文本", str, "纯文本")}
+    outputs = {"msg": Output("msg", "IM 消息", IMMessage, "IM 消息")}
+
+    def __init__(self, split_by: Annotated[Optional[str], ParamMeta(label="分段符")] = None):
+        self.split_by = split_by
+
+    def execute(self, text: str) -> Dict[str, Any]:
+        if self.split_by:
+            return {"msg": IMMessage(sender=ChatSender.get_bot_sender(), message_elements=[TextMessage(line) for line in text.split(self.split_by)])}
+        else:
+            return {"msg": IMMessage(sender=ChatSender.get_bot_sender(), message_elements=[TextMessage(text)])}
+
+# 补充 IMMessage 消息
+class AppendIMMessage(Block):
+    """补充 IMMessage 消息"""
+
+    name = "concat_im_message"
+    container: DependencyContainer
+    inputs = {
+        "base_msg": Input("base_msg", "IM 消息", IMMessage, "IM 消息"),
+        "append_msg": Input("append_msg", "新消息片段", MessageElement, "新消息片段"),
+    }
+    outputs = {"msg": Output("msg", "IM 消息", IMMessage, "IM 消息")}
+
+    def execute(self, base_msg: IMMessage, append_msg: MessageElement) -> Dict[str, Any]:
+        return {"msg": IMMessage(sender=base_msg.sender, message_elements=base_msg.message_elements + [append_msg])}

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

@@ -0,0 +1,25 @@
+
+# LLM 响应转纯文本
+from typing import Any, Dict
+
+from framework.ioc.container import DependencyContainer
+from framework.llm.format.response import LLMChatResponse
+from framework.workflow.core.block.base import Block
+from framework.workflow.core.block.input_output import Input, Output
+
+
+class LLMResponseToText(Block):
+    """LLM 响应转纯文本"""
+
+    name = "llm_response_to_text"
+    container: DependencyContainer
+    inputs = {"response": Input("response", "LLM 响应", LLMChatResponse, "LLM 响应")}
+    outputs = {"text": Output("text", "纯文本", str, "纯文本")}
+
+    def execute(self, response: LLMChatResponse) -> Dict[str, Any]:
+        content = ""
+        if response.choices and response.choices[0].message:
+            content = content +response.choices[0].message.content
+
+        return {"text": content}
+

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

@@ -61,12 +61,13 @@ def __init__(
     ):
         self.regex = regex
 
-    def execute(self) -> Dict[str, Any]:
+    def execute(self, text: str) -> Dict[str, Any]:
         # 使用正则表达式提取文本
-        regex = re.compile(self.regex.value)
-        match = regex.search(self.text.value)
-        if match:
-            return {"text": match.group(0)}
+        regex = re.compile(self.regex)
+        match = regex.search(text)
+        # 如果匹配到，则返回匹配到的文本，否则返回空字符串
+        if match and len(match.groups()) > 0:
+            return {"text": match.group(1)}
         else:
             return {"text": ""}
 

framework/workflow/implementations/blocks/system_blocks.py

@@ -1,13 +1,14 @@
 from framework.workflow.core.block.registry import BlockRegistry
 from framework.workflow.implementations.blocks.im.basic import ExtractChatSender
+from framework.workflow.implementations.blocks.llm.basic import LLMResponseToText
 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, TextConcatBlock, TextExtractByRegexBlock,
                                                                     TextReplaceBlock)
 
 from .game.dice import DiceRoll
 from .game.gacha import GachaSimulator
-from .im.messages import GetIMMessage, SendIMMessage
+from .im.messages import AppendIMMessage, GetIMMessage, IMMessageToText, SendIMMessage, TextToIMMessage
 from .im.states import ToggleEditState
 from .llm.chat import ChatCompletion, ChatMessageConstructor, ChatResponseConverter
 from .memory.chat_memory import ChatMemoryQuery, ChatMemoryStore
@@ -31,6 +32,9 @@ def register_system_blocks(registry: BlockRegistry):
     registry.register(
         "extract_chat_sender", "internal", ExtractChatSender, "IM: 提取消息发送者"
     )
+    registry.register("append_im_message", "internal", AppendIMMessage, "IM: 补充消息")
+    registry.register("im_message_to_text", "internal", IMMessageToText, "IM: 消息转文本")
+    registry.register("text_to_im_message", "internal", TextToIMMessage, "文本: 文本转消息")
 
     # LLM 相关 blocks
     registry.register("chat_memory_query", "internal", ChatMemoryQuery, "LLM: 查询记忆")
@@ -48,6 +52,7 @@ def register_system_blocks(registry: BlockRegistry):
         "LLM->IM: 转换消息",
     )
     registry.register("chat_memory_store", "internal", ChatMemoryStore, "LLM: 存储记忆")
+    registry.register("llm_response_to_text", "internal", LLMResponseToText, "LLM: 响应转文本")
 
     # 画图相关 blocks
     registry.register(