diff --git a/codeflash/models/models.py b/codeflash/models/models.py index 0aa8f96e6..e33425607 100644 --- a/codeflash/models/models.py +++ b/codeflash/models/models.py @@ -272,12 +272,12 @@ def parse_markdown_code(markdown_code: str) -> CodeStringsMarkdown: """ matches = markdown_pattern.findall(markdown_code) - results = CodeStringsMarkdown() + code_string_list = [] + for file_path, code in matches: + path = file_path.strip() + code_string_list.append(CodeString(code=code, file_path=Path(path))) try: - for file_path, code in matches: - path = file_path.strip() - results.code_strings.append(CodeString(code=code, file_path=Path(path))) - return results # noqa: TRY300 + return CodeStringsMarkdown(code_strings=code_string_list) except ValidationError: # if any file is invalid, return an empty CodeStringsMarkdown for the entire context return CodeStringsMarkdown()