Segment people based on messages in an automation

When you create a segment, select the message type under Message Data. Then you can click Any <message type>, and search for the automations or individual messages that you want to segment on by name.