pass reference values to aiservices

Author: saga4

codeflash/api/aiservice.py

@@ -10,6 +10,7 @@
 from pydantic.json import pydantic_encoder
 
 from codeflash.cli_cmds.console import console, logger
+from codeflash.code_utils.config_consts import get_n_candidates, get_n_candidates_lp
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.code_utils.git_utils import get_last_commit_author_if_pr_exists, get_repo_owner_and_name
 from codeflash.lsp.helpers import is_LSP_enabled
@@ -131,6 +132,7 @@ def optimize_python_code(  # noqa: D417
             "current_username": get_last_commit_author_if_pr_exists(None),
             "repo_owner": git_repo_owner,
             "repo_name": git_repo_name,
+            "n_candidates": get_n_candidates(),
         }
 
         logger.info("Generating optimized candidates…")
@@ -192,6 +194,7 @@ def optimize_python_code_line_profiler(  # noqa: D417
             "experiment_metadata": experiment_metadata,
             "codeflash_version": codeflash_version,
             "lsp_mode": is_LSP_enabled(),
+            "n_candidates_lp": get_n_candidates_lp(),
         }
 
         logger.info("Generating optimized candidates…")

codeflash/code_utils/config_consts.py

@@ -11,11 +11,13 @@
 MIN_TESTCASE_PASSED_THRESHOLD = 6
 REPEAT_OPTIMIZATION_PROBABILITY = 0.1
 DEFAULT_IMPORTANCE_THRESHOLD = 0.001
+N_CANDIDATES_LP = 6
 
 # LSP-specific
 N_CANDIDATES_LSP = 3
-N_TESTS_TO_GENERATE_LSP = 1
+N_TESTS_TO_GENERATE_LSP = 2
 TOTAL_LOOPING_TIME_LSP = 10.0  # Kept same timing for LSP mode to avoid in increase in performance reporting
+N_CANDIDATES_LP_LSP = 3
 
 
 def get_n_candidates() -> int:
@@ -24,6 +26,12 @@ def get_n_candidates() -> int:
     return N_CANDIDATES_LSP if is_LSP_enabled() else N_CANDIDATES
 
 
+def get_n_candidates_lp() -> int:
+    from codeflash.lsp.helpers import is_LSP_enabled
+
+    return N_CANDIDATES_LP_LSP if is_LSP_enabled() else N_CANDIDATES_LP
+
+
 def get_n_tests_to_generate() -> int:
     from codeflash.lsp.helpers import is_LSP_enabled
 

codeflash/optimization/function_optimizer.py

@@ -45,6 +45,7 @@
     INDIVIDUAL_TESTCASE_TIMEOUT,
     REPEAT_OPTIMIZATION_PROBABILITY,
     get_n_candidates,
+    get_n_candidates_lp,
     get_n_tests_to_generate,
     get_total_looping_time,
 )
@@ -476,7 +477,7 @@ def determine_best_candidate(
             dependency_code=code_context.read_only_context_code,
             trace_id=self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id,
             line_profiler_results=original_code_baseline.line_profile_results["str_out"],
-            num_candidates=10,
+            num_candidates=get_n_candidates_lp(),
             experiment_metadata=ExperimentMetadata(
                 id=self.experiment_id, group="control" if exp_type == "EXP0" else "experiment"
             )