codeflash-ai/codeflash
1
0
Fork 0
Code Issues 18 Pull requests 33 Projects Releases Packages Wiki Activity Actions 3.7k

fix: normalize HASHING context in re_extract_from_cache

Browse source 
re_extract_from_cache was always calling add_needed_imports_from_module,
but HASHING context should use ast.unparse(ast.parse()) to normalize
whitespace for consistent hashing, matching extract_all_contexts_from_files.
This commit is contained in:
Kevin Turcios 2026-03-16 14:59:01 -06:00
parent 2cafadb980
commit 85d0d7c526
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
codeflash/languages/python/context/code_context_extractor.py
View file

@ -446,14 +446,17 @@ def re_extract_from_cache(
except ValueError:
continue
if pruned.code.strip():
code = add_needed_imports_from_module(
src_module_code=cache.original_module,
dst_module_code=pruned,
src_path=cache.file_path,
dst_path=cache.file_path,
project_root=project_root_path,
helper_functions=cache.helper_functions,
)
if code_context_type == CodeContextType.HASHING:
code = ast.unparse(ast.parse(pruned.code))
else:
code = add_needed_imports_from_module(
src_module_code=cache.original_module,
dst_module_code=pruned,
src_path=cache.file_path,
dst_path=cache.file_path,
project_root=project_root_path,
helper_functions=cache.helper_functions,
)
result.code_strings.append(CodeString(code=code, file_path=cache.relative_path))
return result