Serializing computed fields returning enums about pydantic HOT 4 OPEN

class MyModel(BaseModel):
    static_a_or_b: MyEnum | None = None
    ...

from enum import Enum
from pydantic import BaseModel, computed_field, WrapSerializer
from typing import Annotated, Any

class MyEnum(Enum):
    A = "a"
    B = "b"

def serialize_my_enum(value: Any, handler, info) ->  str:
    partial_result = handler(value, info)
    return partial_result.value

MySerializableEnum = Annotated[MyEnum, WrapSerializer(serialize_my_enum)]

class MyModel(BaseModel):    
    @computed_field
    @property
    def computed_a_or_b(self) -> MySerializableEnum:
        return MyEnum.B

if __name__ == "__main__":
    m = MyModel()
    assert m.model_dump()['computed_a_or_b'] == "b"