Supporting beam search in streaming mode

Hi,

My app requires streaming since I wan to stop the generation once a certain (complicated) condition is met.
My decoding method is beam_search with beam_width=2, using greedy decoding or sampling is just not good enough accuracy-wise.
Are you planning to add support to beam_search in streaming mode? 
This is very important to me.