From 896654c562e42cb1715a09bae82d78928ba7dad7 Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Mon, 3 Nov 2025 11:35:04 -0800
Subject: [PATCH] Fix hang in predict subcommand

Co-authored-by: Ben Kunkle <ben.kunkle@gmail.com>
Co-authored-by: Agus Zubiaga <agus@zed.dev>
---
 crates/zeta_cli/src/main.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/crates/zeta_cli/src/main.rs b/crates/zeta_cli/src/main.rs
index b2822c2e1efb31dc4149f9e8814ab6664d334796..b8e7c9f5e1df79aaf9ad0485838d7db65dfabce0 100644
--- a/crates/zeta_cli/src/main.rs
+++ b/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 {