Fix CLI init and optimize flow regressions

Author: mohamedashrraf222

codeflash/api/cfapi.py

@@ -102,10 +102,11 @@ def make_cfapi_request(
 
 
 @lru_cache(maxsize=1)
-def get_user_id(api_key: Optional[str] = None) -> Optional[str]:
+def get_user_id(api_key: Optional[str] = None, *, suppress_errors: bool = False) -> Optional[str]:
     """Retrieve the user's userid by making a request to the /cfapi/cli-get-user endpoint.
 
     :param api_key: The API key to use. If None, uses get_codeflash_api_key().
+    :param suppress_errors: If True, avoid exiting on auth/version errors and return None instead.
     :return: The userid or None if the request fails.
     """
     lsp_enabled = is_LSP_enabled()
@@ -129,6 +130,9 @@ def get_user_id(api_key: Optional[str] = None) -> Optional[str]:
             if min_version and version.parse(min_version) > version.parse(__version__):
                 msg = "Your Codeflash CLI version is outdated. Please update to the latest version using `pip install --upgrade codeflash`."
                 console.print(f"[bold red]{msg}[/bold red]")
+                if suppress_errors:
+                    logger.debug(msg)
+                    return None
                 if lsp_enabled:
                     logger.debug(msg)
                     return f"Error: {msg}"
@@ -140,6 +144,9 @@ def get_user_id(api_key: Optional[str] = None) -> Optional[str]:
 
     if response.status_code == 403:
         error_title = "Invalid Codeflash API key. The API key you provided is not valid."
+        if suppress_errors:
+            logger.debug(error_title)
+            return None
         if lsp_enabled:
             return f"Error: {error_title}"
         msg = (

codeflash/cli_cmds/cli.py

@@ -289,11 +289,15 @@ def _handle_show_config() -> None:
         from codeflash.code_utils.config_parser import parse_config_file
 
         config, config_file_path = parse_config_file()
-        status = "Saved config"
+        is_file_backed_config = config_file_path.is_file()
+        status = "Saved config" if is_file_backed_config else "Auto-detected (zero-config)"
 
         console.print()
         console.print(f"[bold]Codeflash Configuration[/bold] ({status})")
-        console.print(f"[dim]Config file: {config_file_path}[/dim]")
+        if is_file_backed_config:
+            console.print(f"[dim]Config file: {config_file_path}[/dim]")
+        else:
+            console.print(f"[dim]Config source: {config_file_path}[/dim]")
         console.print()
 
         table = Table(show_header=True, header_style="bold cyan")

codeflash/cli_cmds/cmd_init.py

@@ -44,7 +44,7 @@
     from argparse import Namespace
 
 
-def init_codeflash() -> None:
+def init_codeflash(*, skip_confirm: bool = False, skip_api_key: bool = False) -> None:
     try:
         welcome_panel = Panel(
             Text(
@@ -63,34 +63,47 @@ def init_codeflash() -> None:
         project_language = detect_project_language()
 
         if project_language == ProjectLanguage.GO:
-            init_go_project()
+            init_go_project(skip_confirm=skip_confirm, skip_api_key=skip_api_key)
             return
 
         if project_language == ProjectLanguage.JAVA:
-            init_java_project()
+            init_java_project(skip_confirm=skip_confirm, skip_api_key=skip_api_key)
             return
 
         if project_language in (ProjectLanguage.JAVASCRIPT, ProjectLanguage.TYPESCRIPT):
-            init_js_project(project_language)
+            init_js_project(project_language, skip_confirm=skip_confirm, skip_api_key=skip_api_key)
             return
 
         # Python project flow
-        did_add_new_key = prompt_api_key()
-
-        should_modify, config = should_modify_pyproject_toml()
+        did_add_new_key = False if skip_api_key else prompt_api_key()
+        git_remote = "origin"
 
-        git_remote = config.get("git_remote", "origin") if config else "origin"
+        if skip_confirm:
+            from codeflash.setup import detect_project, write_config
 
-        if should_modify:
-            setup_info: CLISetupInfo = collect_setup_info()
-            git_remote = setup_info.git_remote
-            configured = configure_pyproject_toml(setup_info)
-            if not configured:
+            detected = detect_project()
+            configured, message = write_config(detected)
+            if configured:
+                click.echo(message)
+                click.echo()
+            else:
+                click.echo(message)
                 apologize_and_exit()
+        else:
+            should_modify, config = should_modify_pyproject_toml()
+
+            git_remote = config.get("git_remote", "origin") if config else "origin"
+
+            if should_modify:
+                setup_info = collect_setup_info()
+                git_remote = setup_info.git_remote
+                configured = configure_pyproject_toml(setup_info)
+                if not configured:
+                    apologize_and_exit()
 
         install_github_app(git_remote)
 
-        install_github_actions(override_formatter_check=True)
+        install_github_actions(override_formatter_check=True, skip_confirm=skip_confirm)
 
         install_vscode_extension()
 

codeflash/cli_cmds/console.py

@@ -41,8 +41,43 @@
 
 DEBUG_MODE = logging.getLogger().getEffectiveLevel() == logging.DEBUG
 
+
+def _configure_stdio_for_unicode_safety() -> None:
+    """Avoid hard failures when the active console encoding can't represent Unicode."""
+    for stream in (sys.stdout, sys.stderr):
+        reconfigure = getattr(stream, "reconfigure", None)
+        if callable(reconfigure):
+            with contextlib.suppress(OSError, ValueError):
+                reconfigure(errors="replace")
+
+
+def _can_encode(text: str) -> bool:
+    encoding = getattr(sys.stdout, "encoding", None) or "utf-8"
+    try:
+        text.encode(encoding)
+    except UnicodeEncodeError:
+        return False
+    return True
+
+
+_configure_stdio_for_unicode_safety()
+
 console = Console(highlighter=NullHighlighter())
 
+_original_console_rule = console.rule
+
+
+def _safe_console_rule(title: str = "", *args: object, **kwargs: object) -> None:
+    if "characters" not in kwargs and not _can_encode("─"):
+        kwargs["characters"] = "-"
+    if title and not _can_encode(title):
+        encoding = getattr(sys.stdout, "encoding", None) or "utf-8"
+        title = title.encode(encoding, errors="replace").decode(encoding, errors="replace")
+    _original_console_rule(title, *args, **kwargs)
+
+
+console.rule = _safe_console_rule  # type: ignore[method-assign]
+
 if is_LSP_enabled() or is_subagent_mode():
     console.quiet = True
 

codeflash/cli_cmds/github_workflow.py

@@ -34,7 +34,7 @@ class DependencyManager(Enum):
     UNKNOWN = auto()
 
 
-def install_github_actions(override_formatter_check: bool = False) -> None:
+def install_github_actions(override_formatter_check: bool = False, *, skip_confirm: bool = False) -> None:
     try:
         config, _config_file_path = parse_config_file(override_formatter_check=override_formatter_check)
 
@@ -100,12 +100,15 @@ def install_github_actions(override_formatter_check: bool = False) -> None:
             console.print(benchmark_panel)
             console.print()
 
-            benchmark_questions = [
-                inquirer.Confirm("benchmark_mode", message="Run GitHub Actions in benchmark mode?", default=True)
-            ]
+            if skip_confirm:
+                benchmark_mode = True
+            else:
+                benchmark_questions = [
+                    inquirer.Confirm("benchmark_mode", message="Run GitHub Actions in benchmark mode?", default=True)
+                ]
 
-            benchmark_answers = inquirer.prompt(benchmark_questions, theme=CodeflashTheme())
-            benchmark_mode = benchmark_answers["benchmark_mode"] if benchmark_answers else False
+                benchmark_answers = inquirer.prompt(benchmark_questions, theme=CodeflashTheme())
+                benchmark_mode = benchmark_answers["benchmark_mode"] if benchmark_answers else False
 
         # Show prompt only if workflow doesn't exist locally
         actions_panel = Panel(
@@ -121,26 +124,28 @@ def install_github_actions(override_formatter_check: bool = False) -> None:
         console.print(actions_panel)
         console.print()
 
-        creation_questions = [
-            inquirer.Confirm(
-                "confirm_creation",
-                message="Set up GitHub Actions for continuous optimization? We'll open a pull request with the workflow file.",
-                default=True,
-            )
-        ]
+        if skip_confirm:
+            confirm_creation = True
+        else:
+            creation_questions = [
+                inquirer.Confirm(
+                    "confirm_creation",
+                    message="Set up GitHub Actions for continuous optimization? We'll open a pull request with the workflow file.",
+                    default=True,
+                )
+            ]
+
+            creation_answers = inquirer.prompt(creation_questions, theme=CodeflashTheme())
+            confirm_creation = bool(creation_answers and creation_answers["confirm_creation"])
 
-        creation_answers = inquirer.prompt(creation_questions, theme=CodeflashTheme())
-        if not creation_answers or not creation_answers["confirm_creation"]:
+        if not confirm_creation:
             skip_panel = Panel(
                 Text("⏩️ Skipping GitHub Actions setup.", style="yellow"), title="⏩️ Skipped", border_style="yellow"
             )
             console.print(skip_panel)
             ph("cli-github-workflow-skipped")
             return
-        ph(
-            "cli-github-optimization-confirm-workflow-creation",
-            {"confirm_creation": creation_answers["confirm_creation"]},
-        )
+        ph("cli-github-optimization-confirm-workflow-creation", {"confirm_creation": confirm_creation})
 
         # Generate workflow content AFTER user confirmation
         logger.info("[github_workflow.py:install_github_actions] User confirmed, generating workflow content...")
@@ -423,6 +428,11 @@ def install_github_actions(override_formatter_check: bool = False) -> None:
                 f"🚀 Codeflash is now configured to automatically optimize new Github PRs!{LF}"
             )
 
+        if skip_confirm:
+            click.echo("Add your CODEFLASH_API_KEY as a GitHub secret before running this workflow.")
+            ph("cli-github-workflow-created")
+            return
+
         # Show GitHub secrets setup panel (needed in both cases - PR created via API or local file)
         try:
             existing_api_key = get_codeflash_api_key()
@@ -555,8 +565,11 @@ def get_github_action_working_directory(toml_path: Path, git_root: Path) -> str:
 def detect_project_language_for_workflow(project_root: Path) -> str:
     """Detect the primary language of the project for workflow generation.
 
-    Returns: 'python', 'javascript', 'typescript', or 'java'
+    Returns: 'python', 'javascript', 'typescript', 'java', or 'go'
     """
+    if (project_root / "go.mod").exists():
+        return "go"
+
     # Check for Java build tools first (pom.xml or build.gradle)
     if (
         (project_root / "pom.xml").exists()
@@ -693,9 +706,9 @@ def generate_dynamic_workflow_content(
     # Detect project language
     project_language = detect_project_language_for_workflow(Path.cwd())
 
-    # For JavaScript/TypeScript and Java projects, use static template customization
+    # For JavaScript/TypeScript, Java, and Go projects, use static template customization
     # (AI-generated steps are currently Python-only)
-    if project_language in ("javascript", "typescript", "java"):
+    if project_language in ("javascript", "typescript", "java", "go"):
         return customize_codeflash_yaml_content(optimize_yml_content, config, git_root, benchmark_mode)
 
     # Python project - try AI-generated steps
@@ -824,6 +837,9 @@ def customize_codeflash_yaml_content(
     if project_language in ("javascript", "typescript"):
         return _customize_js_workflow_content(optimize_yml_content, git_root, benchmark_mode)
 
+    if project_language == "go":
+        return _customize_go_workflow_content(optimize_yml_content, git_root, benchmark_mode)
+
     # Python project (default)
     return _customize_python_workflow_content(optimize_yml_content, git_root, benchmark_mode)
 
@@ -948,3 +964,40 @@ def _customize_java_workflow_content(optimize_yml_content: str, git_root: Path)
     # Install dependencies command
     install_deps = get_java_dependency_installation_commands(build_tool)
     return optimize_yml_content.replace("{{ install_dependencies_command }}", install_deps)
+
+
+def _customize_go_workflow_content(
+    optimize_yml_content: str, git_root: Path, benchmark_mode: bool = False
+) -> str:
+    """Customize workflow content for Go projects."""
+    from codeflash.cli_cmds.init_go import get_go_dependency_installation_commands, get_go_runtime_setup_steps
+
+    project_root = Path.cwd()
+
+    if project_root == git_root:
+        working_dir = ""
+    else:
+        rel_path = str(project_root.relative_to(git_root))
+        working_dir = f"""defaults:
+      run:
+        working-directory: ./{rel_path}"""
+
+    optimize_yml_content = optimize_yml_content.replace("Optimize new Python code", "Optimize new Go code")
+    optimize_yml_content = optimize_yml_content.replace("{{ working_directory }}", working_dir)
+
+    python_setup = get_dependency_manager_installation_string(DependencyManager.PIP)
+    go_setup = get_go_runtime_setup_steps()
+    setup_runtime = f"""{python_setup}
+      {go_setup}"""
+    optimize_yml_content = optimize_yml_content.replace("{{ setup_runtime_environment }}", setup_runtime)
+
+    install_deps = f"""|
+          python -m pip install --upgrade pip
+          pip install codeflash
+          {get_go_dependency_installation_commands()}"""
+    optimize_yml_content = optimize_yml_content.replace("{{ install_dependencies_command }}", install_deps)
+
+    codeflash_cmd = "codeflash"
+    if benchmark_mode:
+        codeflash_cmd += " --benchmark"
+    return optimize_yml_content.replace("{{ codeflash_command }}", codeflash_cmd)

codeflash/cli_cmds/init_go.py

@@ -40,7 +40,7 @@ def _get_theme() -> Any:
     return CodeflashTheme()
 
 
-def init_go_project() -> None:
+def init_go_project(*, skip_confirm: bool = False, skip_api_key: bool = False) -> None:
     from codeflash.cli_cmds.github_workflow import install_github_actions
     from codeflash.cli_cmds.init_auth import install_github_app, prompt_api_key
 
@@ -54,14 +54,14 @@ def init_go_project() -> None:
     console.print(lang_panel)
     console.print()
 
-    did_add_new_key = prompt_api_key()
+    did_add_new_key = False if skip_api_key else prompt_api_key()
 
-    setup_info = collect_go_setup_info()
+    setup_info = collect_go_setup_info(skip_confirm=skip_confirm)
     git_remote = setup_info.git_remote or "origin"
 
     install_github_app(git_remote)
 
-    install_github_actions(override_formatter_check=True)
+    install_github_actions(override_formatter_check=True, skip_confirm=skip_confirm)
 
     usage_table = Table(show_header=False, show_lines=False, border_style="dim")
     usage_table.add_column("Command", style="cyan")
@@ -95,7 +95,7 @@ def init_go_project() -> None:
     sys.exit(0)
 
 
-def collect_go_setup_info() -> GoSetupInfo:
+def collect_go_setup_info(*, skip_confirm: bool = False) -> GoSetupInfo:
 
     from codeflash.cli_cmds.init_config import ask_for_telemetry
 
@@ -129,14 +129,18 @@ def collect_go_setup_info() -> GoSetupInfo:
     console.print(detection_panel)
     console.print()
 
+    if skip_confirm:
+        git_remote = _get_git_remote_for_setup(skip_confirm=True)
+        return GoSetupInfo(git_remote=git_remote, disable_telemetry=False)
+
     git_remote = _get_git_remote_for_setup()
 
     disable_telemetry = not ask_for_telemetry()
 
     return GoSetupInfo(git_remote=git_remote, disable_telemetry=disable_telemetry)
 
 
-def _get_git_remote_for_setup() -> str:
+def _get_git_remote_for_setup(*, skip_confirm: bool = False) -> str:
     try:
         repo = Repo(Path.cwd(), search_parent_directories=True)
         git_remotes = get_git_remotes(repo)
@@ -145,6 +149,8 @@ def _get_git_remote_for_setup() -> str:
 
         if len(git_remotes) == 1:
             return git_remotes[0]
+        if skip_confirm:
+            return "origin" if "origin" in git_remotes else git_remotes[0]
 
         git_panel = Panel(
             Text(