using existing ensure_codeflash_api_key method

Author: mohamedashrraf222

codeflash/api/cfapi.py

@@ -121,53 +121,6 @@ def get_user_id(api_key: Optional[str] = None) -> Optional[str]:
     return None
 
 
-def validate_api_key() -> None:
-    """Validate the API key by making a request to the /cfapi/cli-get-user endpoint.
-
-    Raises SystemExit if the API key is invalid (403) or missing.
-    This should be called early in the CLI flow before starting optimization.
-    """
-    logger.debug("validate_api_key: Starting API key validation")
-    api_key = get_codeflash_api_key()
-
-    response = make_cfapi_request(
-        endpoint="/cli-get-user",
-        method="GET",
-        extra_headers={"cli_version": __version__},
-        api_key=api_key,
-        suppress_errors=True,
-    )
-
-    if response.status_code == 403:
-        error_message = "Invalid API key"
-        try:
-            json_response = response.json()
-            if "error" in json_response:
-                error_message = json_response["error"]
-            elif "message" in json_response:
-                error_message = json_response["message"]
-        except (ValueError, TypeError):
-            error_message = response.text or "Invalid API key"
-
-        msg = (
-            f"Invalid Codeflash API key. {error_message}\n"
-            "You can generate a valid API key at https://app.codeflash.ai/app/apikeys,\n"
-            "then set it as a CODEFLASH_API_KEY environment variable."
-        )
-        logger.error(f"validate_api_key: API key validation failed with 403 - {error_message}")
-        exit_with_message(msg, error_on_exit=True)
-
-    if response.status_code != 200:
-        msg = (
-            f"Failed to validate API key (status {response.status_code}: {response.reason})\n"
-            "Please check your API key at https://app.codeflash.ai/app/apikeys"
-        )
-        logger.error(f"validate_api_key: API key validation failed with status {response.status_code}")
-        exit_with_message(msg, error_on_exit=True)
-
-    logger.debug("validate_api_key: API key validation successful")
-
-
 def suggest_changes(
     owner: str,
     repo: str,

codeflash/main.py

@@ -6,7 +6,7 @@
 
 from pathlib import Path
 
-from codeflash.api.cfapi import validate_api_key
+from codeflash.code_utils.env_utils import ensure_codeflash_api_key
 from codeflash.cli_cmds.cli import parse_args, process_pyproject_config
 from codeflash.cli_cmds.cmd_init import CODEFLASH_LOGO, ask_run_end_to_end_test
 from codeflash.cli_cmds.console import paneled_text
@@ -44,7 +44,8 @@ def main() -> None:
         if not env_utils.check_formatter_installed(args.formatter_cmds):
             return
         # Validate API key early before starting optimization
-        validate_api_key()
+        if not ensure_codeflash_api_key():
+            return
         args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args)
         init_sentry(not args.disable_telemetry, exclude_errors=True)
         posthog_cf.initialize_posthog(not args.disable_telemetry)