Fix Prompt for checkpoint being displayed correctly in DEBUG mode

[aseembits93]

User description

Currently, posthog log outputs interfere with the console output of resuming for checkpoint

---

PR Type

Enhancement

---

Description

Move checkpoint prompt logic to main CLI
Add args.previous_checkpoint_functions in main
Optimizer uses args.previous_checkpoint_functions now
Remove redundant checkpoint prompt import

---

Changes walkthrough 📝

Relevant files

Enhancement

main.py Add checkpoint prompt call in main                                              codeflash/main.py Import ask_should_use_checkpoint_get_functions Set args.previous_checkpoint_functions in main +3/-0      optimizer.py Refactor optimizer checkpoint logic                                            codeflash/optimization/optimizer.py Remove inline checkpoint prompt call Replace with args.previous_checkpoint_functions usage +2/-3

---

Need help?

Type /help how to ... in the comments thread for any questions about PR-Agent usage.

Check out the documentation for more information.

[github-actions]

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 2 🔵🔵⚪⚪⚪

🧪 No relevant tests

🔒 No security concerns identified

⚡ Recommended focus areas for review Unused Import The import CodeflashRunCheckpoint is added but not used in this diff, which may trigger lint warnings or indicate missing functionality. Consider removing it or using it as intended. from codeflash.code_utils.checkpoint import CodeflashRunCheckpoint Missing Initialization The attribute args.previous_checkpoint_functions is only set in the else branch of main(). If the optimizer is invoked via the args.command path, this attribute will be missing and may cause errors. Ensure it's initialized for all code paths that call the optimizer. args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args)

[github-actions]

PR Code Suggestions ✨

Explore these optional code suggestions:

Category	Suggestion	Impact
Possible issue	Default checkpoint functions to list Guard against a None return from ask_should_use_checkpoint_get_functions by defaulting to an empty list. This avoids passing None into downstream functions and prevents type errors. codeflash/main.py [39] -args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args) +args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args) or [] Suggestion importance[1-10]: 6 __ Why: Providing a fallback empty list prevents potential None values from propagating and causing type errors when previous_checkpoint_functions is used downstream.	Low
	Add fallback for missing attribute Use getattr to safely retrieve previous_checkpoint_functions with a fallback empty list, preventing attribute errors if the property is missing. codeflash/optimization/optimizer.py [98] -previous_checkpoint_functions=self.args.previous_checkpoint_functions, +previous_checkpoint_functions=getattr(self.args, "previous_checkpoint_functions", []), Suggestion importance[1-10]: 5 __ Why: Using getattr avoids attribute errors if previous_checkpoint_functions isn’t set, but the code already ensures this attribute in main.py, so the impact is moderate.	Low