Trace grpc stream message

What is the best way to trace grpc stream message ?
For the unary request Span context is attached to Metadata, but stream message does not have headers or metadata. Will I have to modify proto files adding fields for the Span context to be included in each stream message ?