Merge pull request codeflash-ai#516 from codeflash-ai/add-one-more-exception-guard

misrasaurabh1 · web-flow · commit e56428e374c1 · 2025-07-05T17:07:39.000-07:00
catch yet another exception
diff --git a/codeflash/code_utils/code_extractor.py b/codeflash/code_utils/code_extractor.py
@@ -324,7 +324,11 @@ def add_needed_imports_from_module(
             full_package_name=src_module_and_package.package,
         )
     )
-    cst.parse_module(src_module_code).visit(gatherer)
+    try:
+        cst.parse_module(src_module_code).visit(gatherer)
+    except Exception as e:
+        logger.error(f"Error parsing source module code: {e}")
+        return dst_module_code
     try:
         for mod in gatherer.module_imports:
             AddImportsVisitor.add_needed_import(dst_context, mod)

Original file line number	Diff line number	Diff line change
`@@ -324,7 +324,11 @@ def add_needed_imports_from_module(`
`324`	`324`	`full_package_name=src_module_and_package.package,`
`325`	`325`	`)`
`326`	`326`	`)`
`327`		`- cst.parse_module(src_module_code).visit(gatherer)`
	`327`	`+ try:`
	`328`	`+ cst.parse_module(src_module_code).visit(gatherer)`
	`329`	`+ except Exception as e:`
	`330`	`+ logger.error(f"Error parsing source module code: {e}")`
	`331`	`+ return dst_module_code`
`328`	`332`	`try:`
`329`	`333`	`for mod in gatherer.module_imports:`
`330`	`334`	`AddImportsVisitor.add_needed_import(dst_context, mod)`