Mitigate some of the jank in opening/closeing search.

There is still some jank on first open, but it's much less janky
on close and consecutive open calls.

Test: existing tests confirm no regressions
PiperOrigin-RevId: 189802733
Change-Id: I49d6c2def7c33be0fdbb79c89772a5e4513cde3b