Bug fixes.

Allow asynchronous streaming generators:

model_name = MODEL_NAME

async def my_async_stream(*args, **kwargs):
        astream_generator = await sales_agent.astep(
            return_streaming_generator=True, model_name=model_name
        )
        async for chunk in astream_generator:
              # do async work