Fix tests listing text generation models (#2593)
* Fix tests listing text generation models
* mark legacy test as flaky
* Revert "Fix tests listing text generation models"
This reverts commit 909361c65fb89200d542d16ff40627098911d426.
* Revert "Revert "Fix tests listing text generation motdels""
This reverts commit 62eaab7a4cbaead3cf478fc799575350b76d6323.
* Revert "mark legacy test as flaky"
This reverts commit 24e7d2c20bc1e181b8f1fb937264c533645df215.
