no gen test arg

Author: Codeflash Bot

codeflash/cli_cmds/cli.py

@@ -80,6 +80,9 @@ def parse_args() -> Namespace:
     parser.add_argument(
         "--no-pr", action="store_true", help="Do not create a PR for the optimization, only update the code locally."
     )
+    parser.add_argument(
+        "--no-gen-test", action="store_true", help="Do not generate tests, use only existing tests for optimization."
+    )
     parser.add_argument("--staging-review", action="store_true", help="Upload optimizations to staging for review")
     parser.add_argument(
         "--verify-setup",

codeflash/optimization/function_optimizer.py

@@ -1344,7 +1344,10 @@ def generate_tests(
 
         logger.info(f"!lsp|Generated '{count_tests}' tests for '{self.function_to_optimize.function_name}'")
 
-        generated_tests = GeneratedTestsList(generated_tests=tests)
+        if self.args.no_gen_test:
+            generated_tests = GeneratedTestsList(generated_tests=[])
+        else:
+            generated_tests = GeneratedTestsList(generated_tests=tests)
         return Success((count_tests, generated_tests, function_to_concolic_tests, concolic_test_str))
 
     def generate_optimizations(