Merge branch 'lsp/init-flow' of github.com:codeflash-ai/codeflash into lsp/init-flow

Author: Codeflash Bot

codeflash/code_utils/concolic_utils.py

@@ -70,10 +70,11 @@ def __init__(self) -> None:
 
 def clean_concolic_tests(test_suite_code: str) -> str:
     try:
-        can_parse = True
         tree = ast.parse(test_suite_code)
-    except SyntaxError:
+        can_parse = True
+    except Exception:
         can_parse = False
+        tree = None
 
     if not can_parse:
         return AssertCleanup().transform_asserts(test_suite_code)
@@ -87,7 +88,6 @@ def clean_concolic_tests(test_suite_code: str) -> str:
                         new_body.append(ast.Expr(value=stmt.test.left))
                     else:
                         new_body.append(stmt)
-
                 else:
                     new_body.append(stmt)
             node.body = new_body

codeflash/optimization/function_optimizer.py

@@ -19,7 +19,7 @@
 from rich.tree import Tree
 
 from codeflash.api.aiservice import AiServiceClient, AIServiceRefinerRequest, LocalAiServiceClient
-from codeflash.api.cfapi import add_code_context_hash, create_staging, mark_optimization_success
+from codeflash.api.cfapi import CFAPI_BASE_URL, add_code_context_hash, create_staging, mark_optimization_success
 from codeflash.benchmarking.utils import process_benchmark_data
 from codeflash.cli_cmds.console import code_print, console, logger, lsp_log, progress_bar
 from codeflash.code_utils import env_utils
@@ -1472,7 +1472,7 @@ def process_review(
         elif staging_review:
             response = create_staging(**data)
             if response.status_code == 200:
-                staging_url = f"https://app.codeflash.ai/review-optimizations/{self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id}"
+                staging_url = f"{CFAPI_BASE_URL}/review-optimizations/{self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id}"
                 console.print(
                     Panel(
                         f"[bold green]✅ Staging created:[/bold green]\n[link={staging_url}]{staging_url}[/link]",

codeflash/tracer.py

@@ -103,6 +103,11 @@ def main(args: Namespace | None = None) -> ArgumentParser:
             replay_test_paths = []
             if parsed_args.module and unknown_args[0] == "pytest":
                 pytest_splits, test_paths = pytest_split(unknown_args[1:])
+                if pytest_splits is None or test_paths is None:
+                    console.print(f"❌ Could not find test files in the specified paths: {unknown_args[1:]}")
+                    console.print(f"Current working directory: {Path.cwd()}")
+                    console.print("Please ensure the test directory exists and contains test files.")
+                    sys.exit(1)
 
             if len(pytest_splits) > 1:
                 processes = []