Please improve AI Assistants Streaming template - BuildShip