From 565f4d07fba55ad09130aca5a658b9990bd630ba Mon Sep 17 00:00:00 2001
From: Ben Kunkle <ben@zed.dev>
Date: Wed, 22 Apr 2026 07:48:01 -0500
Subject: [PATCH] Fix missing V0331SeedModelPy references (#54500)

Self-Review Checklist:

- [x] I've reviewed my own diff for quality, security, and reliability
- [x] Unsafe blocks (if any) have justifying comments
- [x] The content is consistent with the [UI/UX
checklist](https://github.com/zed-industries/zed/blob/main/CONTRIBUTING.md#uiux-checklist)
- [x] Tests cover the new/changed behavior
- [x] Performance impact has been considered and is acceptable

Closes #ISSUE

Release Notes:

- N/A or Added/Fixed/Improved ...
---
 crates/zeta_prompt/src/zeta_prompt.rs | 2 ++
 1 file changed, 2 insertions(+)

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)