Skip to content

Commit 2de45e8

Browse files
committed
disable concolic testing in LSP mode
1 parent 081cc17 commit 2de45e8

File tree

3 files changed

+14
-2
lines changed

3 files changed

+14
-2
lines changed

codeflash/api/aiservice.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from pydantic.json import pydantic_encoder
1111

1212
from codeflash.cli_cmds.console import console, logger
13-
from codeflash.code_utils.env_utils import get_codeflash_api_key
13+
from codeflash.code_utils.env_utils import get_codeflash_api_key, is_LSP_enabled
1414
from codeflash.code_utils.git_utils import get_last_commit_author_if_pr_exists, get_repo_owner_and_name
1515
from codeflash.models.models import OptimizedCandidate
1616
from codeflash.telemetry.posthog_cf import ph
@@ -182,6 +182,7 @@ def optimize_python_code_line_profiler( # noqa: D417
182182
"python_version": platform.python_version(),
183183
"experiment_metadata": experiment_metadata,
184184
"codeflash_version": codeflash_version,
185+
"lsp_mode": is_LSP_enabled(),
185186
}
186187

187188
logger.info("Generating optimized candidates…")

codeflash/code_utils/env_utils.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from pathlib import Path
88
from typing import Any, Optional
99

10-
from codeflash.cli_cmds.console import logger
10+
from codeflash.cli_cmds.console import console, logger
1111
from codeflash.code_utils.code_utils import exit_with_message
1212
from codeflash.code_utils.formatter import format_code
1313
from codeflash.code_utils.shell_utils import read_api_key_from_shell_config
@@ -110,3 +110,8 @@ def get_cached_gh_event_data() -> dict[str, Any] | None:
110110
return None
111111
with Path(event_path).open() as f:
112112
return json.load(f) # type: ignore # noqa
113+
114+
115+
@lru_cache(maxsize=1)
116+
def is_LSP_enabled() -> bool:
117+
return console.quiet

codeflash/verification/concolic_testing.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from codeflash.cli_cmds.console import console, logger
1111
from codeflash.code_utils.compat import SAFE_SYS_EXECUTABLE
1212
from codeflash.code_utils.concolic_utils import clean_concolic_tests
13+
from codeflash.code_utils.env_utils import is_LSP_enabled
1314
from codeflash.code_utils.static_analysis import has_typed_parameters
1415
from codeflash.discovery.discover_unit_tests import discover_unit_tests
1516
from codeflash.telemetry.posthog_cf import ph
@@ -28,6 +29,11 @@ def generate_concolic_tests(
2829
start_time = time.perf_counter()
2930
function_to_concolic_tests = {}
3031
concolic_test_suite_code = ""
32+
33+
if is_LSP_enabled():
34+
logger.debug("Skipping concolic test generation in LSP mode")
35+
return function_to_concolic_tests, concolic_test_suite_code
36+
3137
if (
3238
test_cfg.concolic_test_root_dir
3339
and isinstance(function_to_optimize_ast, (ast.FunctionDef, ast.AsyncFunctionDef))

0 commit comments

Comments
 (0)