From a3b36ae9a4d0abf148ddbbf9705fb339ee4d06ed Mon Sep 17 00:00:00 2001
From: Oleksiy Syvokon <oleksiy@zed.dev>
Date: Thu, 5 Feb 2026 14:29:27 +0200
Subject: [PATCH] ep: Fix trailing newlines parsing (#48473)

Release Notes:

- N/A
---
 crates/edit_prediction_cli/src/parse_output.rs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/crates/edit_prediction_cli/src/parse_output.rs b/crates/edit_prediction_cli/src/parse_output.rs
index c965f1785a4b63568372058e3fd6ba9c1861b856..884f4c340960d1525b6d6b080e4440f1a2de03f2 100644
--- a/crates/edit_prediction_cli/src/parse_output.rs
+++ b/crates/edit_prediction_cli/src/parse_output.rs
@@ -75,7 +75,6 @@ fn extract_zeta2_current_region(prompt: &str, version: ZetaVersion) -> Result<St
         + start;
 
     let region = &prompt[start..end];
-    let region = region.strip_suffix('\n').unwrap_or(region);
     let region = region.replace(CURSOR_MARKER, "");
 
     Ok(region)
@@ -176,7 +175,7 @@ mod tests {
         "};
 
         let region = extract_zeta2_current_region(prompt, ZetaVersion::V0113Ordered).unwrap();
-        assert_eq!(region, "println!(\"hello\");");
+        assert_eq!(region, "println!(\"hello\");\n");
     }
 
     #[test]
@@ -193,7 +192,7 @@ mod tests {
         "};
 
         let region = extract_zeta2_current_region(prompt, ZetaVersion::V0112MiddleAtEnd).unwrap();
-        assert_eq!(region, "println!(\"hello\");");
+        assert_eq!(region, "println!(\"hello\");\n");
     }
 
     #[test]
@@ -210,7 +209,7 @@ mod tests {
         "};
 
         let region = extract_zeta2_current_region(prompt, ZetaVersion::V0113Ordered).unwrap();
-        assert_eq!(region, "println!(\"hello\");");
+        assert_eq!(region, "println!(\"hello\");\n");
     }
 
     #[test]
@@ -228,7 +227,7 @@ mod tests {
 
         let region =
             extract_zeta2_current_region(prompt, ZetaVersion::V0120GitMergeMarkers).unwrap();
-        assert_eq!(region, "println!(\"hello\");");
+        assert_eq!(region, "println!(\"hello\");\n");
     }
 
     #[test]
@@ -246,6 +245,6 @@ mod tests {
 
         let region =
             extract_zeta2_current_region(prompt, ZetaVersion::V0120GitMergeMarkers).unwrap();
-        assert_eq!(region, "println!(\"hello\");");
+        assert_eq!(region, "println!(\"hello\");\n");
     }
 }