@export_module("autogen.messages")
def get_annotated_type_for_message_classes() -> type[Any]:
# this is a dynamic type so we need to disable the type checker
union_type = Union[tuple(_message_classes.values())] # type: ignore[valid-type]
return Annotated[union_type, Field(discriminator="type")] # type: ignore[return-value]