pass the generation_config when using stream function

[BigicecreamTaken]

Fix for the issue: #308

[yair-the-super-wise]

+1 really need this, took me a long time to figure it out can you merge it?