Function parse_full_rewrite

utils/code_utils.py:109–136

Extract a complete code block from an LLM response for a full rewrite. Args: llm_response: The full text response from the LLM. language: The programming language to look for in the markdown code block. Returns: The extracted code string, or None if no code blo

(llm_response: str, language: str = "python")

def parse_full_rewrite(llm_response: str, language: str = "python") -> Optional[str]:
    """
    Extract a complete code block from an LLM response for a full rewrite.

    Args:
        llm_response: The full text response from the LLM.
        language: The programming language to look for in the markdown code block.

    Returns:
        The extracted code string, or None if no code block is found.
    """
    # First, try to find a code block specifically marked with the given language
    code_block_pattern = r"```" + language + r"\n(.*?)```"
    matches = re.findall(code_block_pattern, llm_response, re.DOTALL)

    if matches:
        # Return the content of the first matching block, stripped of whitespace
        return matches[0].strip()

    # Fallback: if no language-specific block is found, look for any code block
    code_block_pattern = r"```(.*?)```"
    matches = re.findall(code_block_pattern, llm_response, re.DOTALL)

    if matches:
        return matches[0].strip()

    # Final fallback: if no code blocks are found at all, return the raw response
    return llm_response


def format_diff_summary(diff_blocks: List[Tuple[str, str]]) -> str: