diff --git a/crates/zeta_prompt/src/zeta_prompt.rs b/crates/zeta_prompt/src/zeta_prompt.rs
index 0a0f3e54ed2d21a02315e967692295428218f82d..37799d528b923b3a50bf5b008bcdcee4d952e867 100644
--- a/crates/zeta_prompt/src/zeta_prompt.rs
+++ b/crates/zeta_prompt/src/zeta_prompt.rs
@@ -244,6 +244,7 @@ pub fn format_zeta_prompt(input: &ZetaPromptInput, format: ZetaFormat) -> Option
         | ZetaFormat::V0306SeedMultiRegions
         | ZetaFormat::V0316SeedMultiRegions
         | ZetaFormat::V0317SeedMultiRegions
+        | ZetaFormat::V0331SeedCoderModelPy
         | ZetaFormat::V0318SeedMultiRegions => 4096,
         ZetaFormat::V0327SingleFile => 16384,
     };
@@ -1036,6 +1037,7 @@ pub fn format_expected_output(
         | ZetaFormat::v0226Hashline
         | ZetaFormat::V0304VariableEdit
         | ZetaFormat::V0304SeedNoEdits
+        | ZetaFormat::V0331SeedCoderModelPy
         | ZetaFormat::V0306SeedMultiRegions => {
             let (mut result, first_hunk_offset) = if empty_patch {
                 (old_editable.clone(), None)