Update beta.py

KRRT7 · KRRT7 · commit 6af7d4cfdea0 · 2025-06-11T18:24:54.000-07:00
diff --git a/codeflash/lsp/beta.py b/codeflash/lsp/beta.py
@@ -147,6 +147,8 @@ def third_step_in_optimize_function(server: CodeflashLanguageServer, params: Opt
             "message": f"No best optimizations found for function {function_to_optimize_qualified_name}",
         }
 
+    optimized_source = best_optimization.candidate.source_code  # noqa: F841
+
     return {
         "functionName": params.functionName,
         "status": "success",

Original file line number	Diff line number	Diff line change
`@@ -147,6 +147,8 @@ def third_step_in_optimize_function(server: CodeflashLanguageServer, params: Opt`
`147`	`147`	`"message": f"No best optimizations found for function {function_to_optimize_qualified_name}",`
`148`	`148`	`}`
`149`	`149`
	`150`	`+ optimized_source = best_optimization.candidate.source_code # noqa: F841`
	`151`	`+`
`150`	`152`	`return {`
`151`	`153`	`"functionName": params.functionName,`
`152`	`154`	`"status": "success",`