Class MessageBlock

entity/messages.py:104–165

Single block of multimodal content.

@dataclass
class MessageBlock:
    """Single block of multimodal content."""

    type: MessageBlockType
    text: Optional[str] = None
    attachment: Optional[AttachmentRef] = None
    data: Dict[str, Any] = field(default_factory=dict)

    def to_dict(self, include_data: bool = True) -> Dict[str, Any]:
        payload: Dict[str, Any] = {
            "type": self.type.value,
        }
        if self.text is not None:
            payload["text"] = self.text
        if self.attachment:
            payload["attachment"] = self.attachment.to_dict(include_data=include_data)
        if self.data:
            payload["data"] = self.data
        return payload

    @classmethod
    def from_dict(cls, data: Dict[str, Any]) -> "MessageBlock":
        raw_type = data.get("type") or MessageBlockType.TEXT.value
        try:
            block_type = MessageBlockType(raw_type)
        except ValueError:
            block_type = MessageBlockType.DATA
        attachment_data = data.get("attachment")
        attachment = None
        if isinstance(attachment_data, dict):
            attachment = AttachmentRef.from_dict(attachment_data)
        return cls(
            type=block_type,
            text=data.get("text"),
            attachment=attachment,
            data=data.get("data") or {},
        )

    @classmethod
    def text_block(cls, text: str) -> "MessageBlock":
        return cls(type=MessageBlockType.TEXT, text=text)

    def describe(self) -> str:
        """Human-friendly summary for logging."""
        if self.type is MessageBlockType.TEXT and self.text:
            return self.text
        if self.attachment:
            name = self.attachment.name or self.attachment.attachment_id
            return f"[{self.type.value} attachment: {name}]"
        if self.text:
            return self.text
        if "text" in self.data:
            return str(self.data["text"])
        return f"[{self.type.value} block]"

    def copy(self) -> "MessageBlock":
        return MessageBlock(
            type=self.type,