Google may be working on an improved Search widget for Pixel phones with AI Mode integration. More details here.