Showing the price per million (or thousand) tokens to the right of each model you have access to (replacing the “Upgrade” text) would be useful to let people know how expensive each model will be and let people plan model usage accordingly. Being able to sort models by Lowest to Highest cost and vice versa would also be useful.

I would expect the text to be slightly smaller and lower contrast (grey) than the main font, formatted like “$0.0005/K” for a thousand tokens or “$0.50/M” for a million tokens. For the sorting, I would expect an icon similar to 🔃 next to the search bar (so it wouldn’t take up the entire width of the top any more) with the relevant options on a submenu that pops up from the bottom of the screen, similar to the AI model selector itself.