chore: add TODO for get_code_optimization_context extraction, update follow_up.md

KRRT7 · KRRT7 · commit 96c1459ca4d9 · 2026-03-01T23:22:15.000-05:00
diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py
@@ -1565,6 +1565,8 @@ def replace_function_and_helpers_with_optimized_code(
             )
         return did_update
 
+    # TODO: Extract into PythonFunctionOptimizer — code_context_extractor is Python-only.
+    # Needs a JS context extractor equivalent first.
     def get_code_optimization_context(self) -> Result[CodeOptimizationContext, str]:
         try:
             new_code_ctx = code_context_extractor.get_code_optimization_context(

Original file line number	Diff line number	Diff line change
`@@ -1565,6 +1565,8 @@ def replace_function_and_helpers_with_optimized_code(`
`1565`	`1565`	`)`
`1566`	`1566`	`return did_update`
`1567`	`1567`
	`1568`	`+ # TODO: Extract into PythonFunctionOptimizer — code_context_extractor is Python-only.`
	`1569`	`+ # Needs a JS context extractor equivalent first.`
`1568`	`1570`	`def get_code_optimization_context(self) -> Result[CodeOptimizationContext, str]:`
`1569`	`1571`	`try:`
`1570`	`1572`	`new_code_ctx = code_context_extractor.get_code_optimization_context(`