Flush typeahead at the OSS picker boundary

codex-rs/tui/src/lib.rs

@@ -1018,7 +1018,6 @@ pub async fn run_main(
             Some(provider)
         } else {
             // No provider configured, prompt the user
-            tui::discard_terminal_input();
             let selection = oss_selection::select_oss_provider().await?;
             let provider = selection.provider;
             if provider == "__CANCELLED__" {

codex-rs/tui/src/oss_selection.rs

@@ -341,6 +341,7 @@ pub async fn select_oss_provider() -> io::Result<OssProviderSelection> {
 
     let mut widget = OssSelectionWidget::new(lmstudio_status, ollama_status)?;
 
+    crate::tui::discard_terminal_input();
     enable_raw_mode()?;
     let mut stdout = io::stdout();
     execute!(stdout, EnterAlternateScreen)?;