fix validation error for python code

Author: mohammedahmed18

codeflash/models/models.py

@@ -273,10 +273,10 @@ def parse_markdown_code(markdown_code: str) -> CodeStringsMarkdown:
         """
         matches = markdown_pattern.findall(markdown_code)
         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()
+                code_string_list.append(CodeString(code=code, file_path=Path(path)))
             return CodeStringsMarkdown(code_strings=code_string_list)
         except ValidationError:
             # if any file is invalid, return an empty CodeStringsMarkdown for the entire context

tests/test_validate_python_code.py

@@ -69,3 +69,5 @@ def test_generated_candidates_validation():
     ]
     candidates = ai_service._get_valid_candidates(mock_generate_candidates, OptimizedCandidateSource.OPTIMIZE)
     assert len(candidates) == 1
+    assert candidates[0].source_code.code_strings[0].code == "print('Hello, World!')"
+    assert candidates[0].source_code.code_strings[0].file_path.as_posix() == "file.py"