From 9dea3f48cb2c26beffd53f3cf297851a6674403b Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 20 Aug 2024 13:26:27 -0700 Subject: [PATCH] feat: Add support for scanning and yielding ```bash blocks in find_original_update_blocks --- aider/coders/editblock_coder.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/aider/coders/editblock_coder.py b/aider/coders/editblock_coder.py index 781b1530c..8f22fcc02 100644 --- a/aider/coders/editblock_coder.py +++ b/aider/coders/editblock_coder.py @@ -406,6 +406,16 @@ def find_original_update_blocks(content, fence=DEFAULT_FENCE): while pieces: cur = pieces.pop() + # Check for ```bash blocks + if cur.strip().startswith("```bash"): + bash_content = [] + while pieces and not pieces[-1].strip().startswith("```"): + bash_content.append(pieces.pop()) + if pieces and pieces[-1].strip().startswith("```"): + pieces.pop() # Remove the closing ``` + yield "bash_command", "".join(bash_content), "" + continue + if cur in (DIVIDER, UPDATED): processed.append(cur) raise ValueError(f"Unexpected {cur}")