From 7dc02263e78055fa032541ed90e070853625b2ce Mon Sep 17 00:00:00 2001 From: Oleksiy Syvokon Date: Wed, 6 May 2026 18:03:49 +0300 Subject: [PATCH] ep: Don't check for special tokens on client (#55899) Closes #55692 Release Notes: - N/A Co-authored-by: Ben Kunkle --- crates/edit_prediction/src/zeta.rs | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/crates/edit_prediction/src/zeta.rs b/crates/edit_prediction/src/zeta.rs index 7f9f12884709d480bf3360efc012a1838dbc17a9..33b347be1757678e733bb56ec9becf904a1fa791 100644 --- a/crates/edit_prediction/src/zeta.rs +++ b/crates/edit_prediction/src/zeta.rs @@ -24,8 +24,7 @@ use zeta_prompt::{ParsedOutput, ZetaPromptInput}; use std::{env, ops::Range, path::Path, sync::Arc}; use zeta_prompt::{ - ZetaFormat, format_zeta_prompt, get_prefill, parse_zeta2_model_output, - prompt_input_contains_special_tokens, stop_tokens_for_format, + ZetaFormat, format_zeta_prompt, get_prefill, parse_zeta2_model_output, stop_tokens_for_format, zeta1::{self, EDITABLE_REGION_END_MARKER}, }; @@ -120,10 +119,6 @@ pub fn request_prediction_with_zeta( repo_url, ); - if prompt_input_contains_special_tokens(&prompt_input, zeta_version) { - return Err(anyhow::anyhow!("prompt contains special tokens")); - } - let formatted_prompt = format_zeta_prompt(&prompt_input, zeta_version); if let Some(debug_tx) = &debug_tx {