refactor(core): remove ContentBlockType enum and update ContentBlock schema

Author: jinliyl

flowllm/core/enumeration/__init__.py

@@ -1,14 +1,12 @@
 """Core enumeration module."""
 
 from .chunk_enum import ChunkEnum
-from .content_block_type import ContentBlockType
 from .http_enum import HttpEnum
 from .registry_enum import RegistryEnum
 from .role import Role
 
 __all__ = [
     "ChunkEnum",
-    "ContentBlockType",
     "HttpEnum",
     "RegistryEnum",
     "Role",

flowllm/core/enumeration/content_block_type.py

@@ -6,7 +6,7 @@
 from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from .tool_call import ToolCall
-from ..enumeration import Role, ContentBlockType
+from ..enumeration import Role
 
 
 class ContentBlock(BaseModel):
@@ -35,7 +35,7 @@ class ContentBlock(BaseModel):
 
     model_config = ConfigDict(extra="allow")
 
-    type: ContentBlockType = Field(default="")
+    type: str = Field(default="")
     content: str | dict | list = Field(default="")
 
     @model_validator(mode="before")
@@ -51,8 +51,8 @@ def init_block(cls, data: dict):
     def simple_dump(self) -> dict:
         """Convert ContentBlock to a simple dictionary format."""
         result = {
-            "type": self.type.value,
-            self.type.value: self.content,
+            "type": self.type,
+            self.type: self.content,
             **self.model_extra,
         }
 
@@ -95,11 +95,6 @@ def simple_dump(self, add_reasoning: bool = True) -> dict:
 
         return result
 
-    @property
-    def string_buffer(self) -> str:
-        """Get a string representation of the message with role and content."""
-        return f"{self.role.value}: {self.dump_content()}"
-
 
 class Trajectory(BaseModel):
     """Represents a conversation trajectory with messages and optional scoring."""