From 2991b0593cb4212687c1cd8c456947906e0e04da Mon Sep 17 00:00:00 2001
From: Oleksiy Syvokon <oleksiy@zed.dev>
Date: Thu, 5 Mar 2026 09:16:09 +0200
Subject: [PATCH] ep: Make --repetitions work with Anthropic and OpenAI batches
 (#50789)

Release Notes:

- N/A
---
 crates/edit_prediction_cli/src/predict.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/crates/edit_prediction_cli/src/predict.rs b/crates/edit_prediction_cli/src/predict.rs
index bd89d54ab37521ecb9661b6f1bb0156f30ba1acb..94e28d00da2d61f63b59364304c3b9b4276e15f7 100644
--- a/crates/edit_prediction_cli/src/predict.rs
+++ b/crates/edit_prediction_cli/src/predict.rs
@@ -390,7 +390,7 @@ async fn predict_anthropic(
             .await?
         else {
             // Request stashed for batched processing
-            return Ok(());
+            continue;
         };
 
         let actual_output = response
@@ -464,7 +464,7 @@ async fn predict_openai(
             .await?
         else {
             // Request stashed for batched processing
-            return Ok(());
+            continue;
         };
 
         let actual_output = response