Jupyter reply types

#README.md

https://jupyter-client.readthedocs.io/en/stable/messaging.html#execution-results

https://github.com/rgbkrk/nbformatjs/blob/6bad9a20b4e12b1434c2c876340bdbcc0284a0d5/v4/nbformat.v4.schema.json#L230

messaging.py

MT = TypeVar("MT", bound=LiteralString)
C = TypeVar("C", bound="_ContentOkBase")


class _ReplyBase(TypedDict, Generic[MT, C]):
    """Jupyter reply message."""

    header: ReplyHeader[MT]
    content: C | ContentError | ContentAbort


class ReplyHeader(TypedDict, Generic[MT]):
    """Jupyter reply message header."""

    msg_type: MT


class _ContentOkBase(TypedDict):
    """Jupyter successful message reply content."""

    status: Literal["ok"]


class ContentAbort(TypedDict):
    """Jupyter abort message reply content."""

    status: Literal["abort"]


class ContentError(TypedDict):
    """Jupyter error message reply content."""

    status: Literal["error"]
    ename: str
    evalue: str
    traceback: list[str]


class ExecuteReplyContent(_ContentOkBase):
    execution_count: int


ExecuteReply = _ReplyBase[Literal["execute_result"], ExecuteReplyContent]


class StreamContent(_ContentOkBase):
    name: Literal["stdout", "stderr"]
    text: str


class DisplayDataContent(_ContentOkBase):
    data: dict[str, str]
    metadata: dict[str, str]


class ExecuteResultContent(DisplayDataContent):
    execution_count: int


Stream = _ReplyBase[Literal["stream"], StreamContent]
DisplayData = _ReplyBase[Literal["display_data"], DisplayDataContent]
ExecuteResult = _ReplyBase[Literal["execute_result"], ExecuteResultContent]
IoPubMsg = Stream | DisplayData | ExecuteResult