Class AttachmentRecord

utils/attachments.py:19–56

Stores metadata about an attachment tracked inside a workflow run.

@dataclass
class AttachmentRecord:
    """Stores metadata about an attachment tracked inside a workflow run."""

    ref: AttachmentRef
    kind: MessageBlockType = MessageBlockType.FILE
    description: Optional[str] = None
    extra: Dict[str, Any] = field(default_factory=dict)

    def to_dict(self) -> Dict[str, Any]:
        return {
            "ref": self.ref.to_dict(),
            "kind": self.kind.value,
            "description": self.description,
            "extra": self.extra,
        }

    @classmethod
    def from_dict(cls, data: Dict[str, Any]) -> "AttachmentRecord":
        ref_data = data.get("ref") or {}
        raw_kind = data.get("kind", MessageBlockType.FILE.value)
        try:
            kind = MessageBlockType(raw_kind)
        except ValueError:
            kind = MessageBlockType.FILE
        return cls(
            ref=AttachmentRef.from_dict(ref_data),
            kind=kind,
            description=data.get("description"),
            extra=data.get("extra") or {},
        )

    def as_message_block(self) -> MessageBlock:
        """Convert to a MessageBlock referencing this attachment."""
        return MessageBlock(
            type=self.kind,
            attachment=self.ref.copy(),
            data=dict(self.extra),
        )


class AttachmentStore: