Fix hang in predict subcommand

Max Brunsfeld , Ben Kunkle , and Agus Zubiaga created 1 day ago

Co-authored-by: Ben Kunkle <[e-mail]>
Co-authored-by: Agus Zubiaga <[e-mail]>

Change summary

crates/zeta_cli/src/main.rs | 2 ++
1 file changed, 2 insertions(+)

Detailed changes

crates/zeta_cli/src/main.rs

@@ -450,6 +450,7 @@ async fn zeta2_predict(
                         &mut excerpts_text,
                     );
                 }
+                break;
             }
             _ => {}
         }
@@ -771,6 +772,7 @@ fn main() {
                     Zeta2Command::Predict { example_path } => {
                         let example = NamedExample::load(example_path).unwrap();
                         zeta2_predict(example, &app_state, cx).await.unwrap();
+                        let _ = cx.update(|cx| cx.quit());
                         return;
                     }
                     Zeta2Command::Syntax {