pdd-cli 0.0.32__py3-none-any.whl → 0.0.34__py3-none-any.whl

pdd/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.0.32"
+__version__ = "0.0.34"
 
 # Strength parameter used for LLM extraction across the codebase
 # Used in postprocessing, XML tagging, code generation, and other extraction operations. The module should have a large context window and be affordable.

pdd/cli.py

@@ -194,6 +194,9 @@ def process_commands(ctx: click.Context, results: List[Optional[Tuple[Any, float
                  # Check if it was install_completion (which normally returns None)
                  if command_name == "install_completion":
                      console.print(f"  [info]Step {i+1} ({command_name}):[/info] Command completed.")
+                 # If command name is unknown, and it might be install_completion which prints its own status
+                 elif command_name.startswith("Unknown Command"):
+                     console.print(f"  [info]Step {i+1} ({command_name}):[/info] Command executed (see output above for status details).")
                  # Check if it was preprocess (which returns a dummy tuple on success)
                  # This case handles actual failure for preprocess
                  elif command_name == "preprocess":
@@ -1070,17 +1073,21 @@ def verify(
 @click.pass_context
 # No @track_cost
 def install_completion_cmd(ctx: click.Context) -> None: # Return type remains None
-    """Install shell completion for PDD."""
-    quiet = ctx.obj.get("quiet", False)
-    command_name = "install_completion"
+    """Install shell completion for the PDD CLI."""
+    command_name = "install_completion" # For error handling
+    quiet_mode = ctx.obj.get("quiet", False) # Get quiet from context
+
     try:
-        install_completion(quiet=quiet)
-        # Return None on success (as intended)
-        return None
+        # The actual install_completion function is imported from .install_completion
+        install_completion(quiet=quiet_mode) # Pass quiet_mode
+        # Success messages are handled within install_completion based on quiet_mode
+        # No need to print additional messages here unless specifically required
+        # if not quiet_mode:
+        #     console.print(f"[success]'{command_name}' command completed successfully.[/success]")
     except Exception as e:
-        handle_error(e, command_name, quiet)
-        # Explicitly return None on failure as well, consistent with other commands
-        return None
+        # Use the centralized error handler
+        handle_error(e, command_name, quiet_mode)
+        # Do not return anything, as the callback expects None or a tuple
 
 
 # --- Entry Point ---

pdd/code_generator_main.py

@@ -27,7 +27,7 @@ PDD_APP_NAME = "PDD Code Generator"
 
 # Cloud function URL
 CLOUD_GENERATE_URL = "https://us-central1-prompt-driven-development.cloudfunctions.net/generateCode"
-CLOUD_REQUEST_TIMEOUT = 200  # seconds
+CLOUD_REQUEST_TIMEOUT = 400  # seconds
 
 console = Console()
 
@@ -59,8 +59,8 @@ def is_git_repository(path: Optional[str] = None) -> bool:
     return False
 
 
-def get_git_committed_content(file_path: str) -> Optional[str]:
-    """Gets the content of the file as it was in the last commit (HEAD)."""
+def get_git_content_at_ref(file_path: str, git_ref: str = "HEAD") -> Optional[str]:
+    """Gets the content of the file as it was at the specified git_ref."""
     abs_file_path = pathlib.Path(file_path).resolve()
     if not is_git_repository(str(abs_file_path.parent)):
         return None
@@ -77,13 +77,13 @@ def get_git_committed_content(file_path: str) -> Optional[str]:
         # console.print(f"[yellow]File {file_path} is not under git root {git_root}.[/yellow]")
         return None
 
-    returncode, stdout, stderr = _run_git_command(["git", "show", f"HEAD:{relative_path.as_posix()}"], cwd=str(git_root))
+    returncode, stdout, stderr = _run_git_command(["git", "show", f"{git_ref}:{relative_path.as_posix()}"], cwd=str(git_root))
     if returncode == 0:
         return stdout
     else:
-        # File might be new, or other git error. Not necessarily an error for this function's purpose.
-        # if "does not exist" not in stderr and "exists on disk, but not in 'HEAD'" not in stderr and console.is_terminal:
-        #     console.print(f"[yellow]Git (show) warning for {file_path}: {stderr}[/yellow]")
+        # File might not exist at that ref, or other git error.
+        # if "does not exist" not in stderr and "exists on disk, but not in" not in stderr and console.is_terminal: # Be less noisy for common cases
+        #     console.print(f"[yellow]Git (show) warning for {file_path} at {git_ref}: {stderr}[/yellow]")
         return None
 
 def get_file_git_status(file_path: str) -> str:
@@ -192,14 +192,98 @@ def code_generator_main(
                 if verbose:
                     console.print(f"Using specified original prompt: [cyan]{original_prompt_file_path}[/cyan]")
             elif is_git_repository(str(pathlib.Path(prompt_file).parent)):
-                original_prompt_git_content = get_git_committed_content(prompt_file)
-                if original_prompt_git_content is not None:
-                    original_prompt_content_for_incremental = original_prompt_git_content
-                    can_attempt_incremental = True
+                # prompt_content is the current on-disk version
+                head_prompt_content = get_git_content_at_ref(prompt_file, git_ref="HEAD")
+
+                if head_prompt_content is not None:
+                    # Compare on-disk content (prompt_content) with HEAD content
+                    if prompt_content.strip() != head_prompt_content.strip():
+                        # Uncommitted changes exist. Original is HEAD, new is on-disk.
+                        original_prompt_content_for_incremental = head_prompt_content
+                        can_attempt_incremental = True
+                        if verbose:
+                            console.print(f"On-disk [cyan]{prompt_file}[/cyan] has uncommitted changes. Using HEAD version as original prompt.")
+                    else:
+                        # On-disk is identical to HEAD. Search for a prior *different* version.
+                        if verbose:
+                            console.print(f"On-disk [cyan]{prompt_file}[/cyan] matches HEAD. Searching for a prior *different* version as original prompt.")
+
+                        new_prompt_candidate = head_prompt_content # This is also prompt_content (on-disk)
+                        found_different_prior = False
+                        
+                        git_root_path_obj: Optional[pathlib.Path] = None
+                        prompt_file_rel_to_root_str: Optional[str] = None
+
+                        try:
+                            abs_prompt_file_path = pathlib.Path(prompt_file).resolve()
+                            temp_git_root_rc, temp_git_root_str, temp_git_root_stderr = _run_git_command(
+                                ["git", "rev-parse", "--show-toplevel"], 
+                                cwd=str(abs_prompt_file_path.parent)
+                            )
+                            if temp_git_root_rc == 0:
+                                git_root_path_obj = pathlib.Path(temp_git_root_str)
+                                prompt_file_rel_to_root_str = abs_prompt_file_path.relative_to(git_root_path_obj).as_posix()
+                            elif verbose:
+                                console.print(f"[yellow]Git (rev-parse) failed for {prompt_file}: {temp_git_root_stderr}. Cannot search history for prior different version.[/yellow]")
+                        
+                        except ValueError: # If file is not under git root
+                             if verbose:
+                                console.print(f"[yellow]File {prompt_file} not under a detected git root. Cannot search history.[/yellow]")
+                        except Exception as e_git_setup:
+                            if verbose:
+                                console.print(f"[yellow]Error setting up git info for {prompt_file}: {e_git_setup}. Cannot search history.[/yellow]")
+
+                        if git_root_path_obj and prompt_file_rel_to_root_str:
+                            MAX_COMMITS_TO_SEARCH = 10  # How far back to look
+                            log_cmd = ["git", "log", f"--pretty=format:%H", f"-n{MAX_COMMITS_TO_SEARCH}", "--", prompt_file_rel_to_root_str]
+                            
+                            log_rc, log_stdout, log_stderr = _run_git_command(log_cmd, cwd=str(git_root_path_obj))
+
+                            if log_rc == 0 and log_stdout.strip():
+                                shas = log_stdout.strip().split('\\n')
+                                if verbose:
+                                     console.print(f"Found {len(shas)} commits for [cyan]{prompt_file_rel_to_root_str}[/cyan] in recent history (up to {MAX_COMMITS_TO_SEARCH}).")
+
+                                if len(shas) > 1: # Need at least one commit before the one matching head_prompt_content
+                                    for prior_sha in shas[1:]: # Iterate starting from the commit *before* HEAD's version of the file
+                                        if verbose:
+                                            console.print(f"  Checking commit {prior_sha[:7]} for content of [cyan]{prompt_file}[/cyan]...")
+                                        
+                                        # get_git_content_at_ref uses the original prompt_file path, 
+                                        # which it resolves internally relative to the git root.
+                                        prior_content = get_git_content_at_ref(prompt_file, prior_sha) 
+                                        
+                                        if prior_content is not None:
+                                            if prior_content.strip() != new_prompt_candidate.strip():
+                                                original_prompt_content_for_incremental = prior_content
+                                                can_attempt_incremental = True
+                                                found_different_prior = True
+                                                if verbose:
+                                                    console.print(f"    [green]Found prior different version at commit {prior_sha[:7]}. Using as original prompt.[/green]")
+                                                break 
+                                            elif verbose:
+                                                 console.print(f"    Content at {prior_sha[:7]} is identical to current HEAD. Skipping.")
+                                        elif verbose:
+                                            console.print(f"    Could not retrieve content for [cyan]{prompt_file}[/cyan] at commit {prior_sha[:7]}.")
+                                else: 
+                                    if verbose:
+                                        console.print(f"  File [cyan]{prompt_file_rel_to_root_str}[/cyan] has less than 2 versions in recent history at this path.")
+                            elif verbose:
+                                console.print(f"[yellow]Git (log) failed for {prompt_file_rel_to_root_str} or no history found: {log_stderr}[/yellow]")
+                        
+                        if not found_different_prior:
+                            original_prompt_content_for_incremental = new_prompt_candidate 
+                            can_attempt_incremental = True 
+                            if verbose:
+                                console.print(
+                                    f"[yellow]Warning: Could not find a prior *different* version of {prompt_file} "
+                                    f"within the last {MAX_COMMITS_TO_SEARCH if git_root_path_obj else 'N/A'} relevant commits. "
+                                    f"Using current HEAD version as original (prompts will be identical).[/yellow]"
+                                )
+                else:
+                    # File not in HEAD, cannot determine git-based original prompt.
                     if verbose:
-                        console.print(f"Using last committed version of [cyan]{prompt_file}[/cyan] as original prompt.")
-                elif verbose:
-                    console.print(f"[yellow]Warning: Could not find committed version of {prompt_file} in git for incremental generation.[/yellow]")
+                        console.print(f"[yellow]Warning: Could not find committed version of {prompt_file} in git (HEAD) for incremental generation.[/yellow]")
             
             if force_incremental_flag and existing_code_content:
                 if not (original_prompt_content_for_incremental or "original_prompt_file" in input_strings): # Check if original prompt is actually available

pdd/context_generator.py

@@ -33,11 +33,6 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
             print("[red]Error: prompt is missing.[/red]")
         return None, 0.0, None
 
-    supported_languages = ["python", "javascript", "java"]
-    if language not in supported_languages:
-        if verbose:
-            print(f"[red]Error: Unsupported language '{language}'.[/red]")
-        return None, 0.0, None
 
     if not (0 <= strength <= 1):
         if verbose:

pdd/context_generator_main.py

@@ -25,7 +25,7 @@ def context_generator_main(ctx: click.Context, prompt_file: str, code_file: str,
         command_options = {
             "output": output
         }
-        input_strings, output_file_paths, _ = construct_paths(
+        input_strings, output_file_paths, language = construct_paths(
             input_file_paths=input_file_paths,
             force=ctx.obj.get('force', False),
             quiet=ctx.obj.get('quiet', False),
@@ -41,6 +41,7 @@ def context_generator_main(ctx: click.Context, prompt_file: str, code_file: str,
         strength = ctx.obj.get('strength', 0.5)
         temperature = ctx.obj.get('temperature', 0)
         example_code, total_cost, model_name = context_generator(
+            language=language,
             code_module=code_content,
             prompt=prompt_content,
             strength=strength,

pdd/data/language_format.csv

@@ -12,6 +12,7 @@ Go,//,.go
 Rust,//,.rs
 Kotlin,//,.kt
 TypeScript,//,.ts
+TypeScriptReact,//,.tsx
 C#,//,.cs
 SQL,--,.sql
 Shell,#,.sh

pdd/install_completion.py

@@ -88,7 +88,7 @@ def get_completion_script_extension(shell: str) -> str:
     return mapping.get(shell, shell)
 
 
-def install_completion():
+def install_completion(quiet: bool = False):
     """
     Install shell completion for the PDD CLI by detecting the user's shell,
     copying the relevant completion script, and appending a source command
@@ -97,7 +97,8 @@ def install_completion():
     shell = get_current_shell()
     rc_file = get_shell_rc_path(shell)
     if not rc_file:
-        rprint(f"[red]Unsupported shell: {shell}[/red]")
+        if not quiet:
+            rprint(f"[red]Unsupported shell: {shell}[/red]")
         raise click.Abort()
 
     ext = get_completion_script_extension(shell)
@@ -107,7 +108,8 @@ def install_completion():
     completion_script_path = os.path.join(local_pdd_path, f"pdd_completion.{ext}")
 
     if not os.path.exists(completion_script_path):
-        rprint(f"[red]Completion script not found: {completion_script_path}[/red]")
+        if not quiet:
+            rprint(f"[red]Completion script not found: {completion_script_path}[/red]")
         raise click.Abort()
 
     source_command = f"source {completion_script_path}"
@@ -115,9 +117,12 @@ def install_completion():
     try:
         # Ensure the RC file exists (create if missing).
         if not os.path.exists(rc_file):
-            os.makedirs(os.path.dirname(rc_file), exist_ok=True)
+            # Create parent directories if they don't exist
+            rc_dir = os.path.dirname(rc_file)
+            if rc_dir: # Ensure rc_dir is not an empty string (e.g. if rc_file is in current dir)
+                 os.makedirs(rc_dir, exist_ok=True)
             with open(rc_file, "w", encoding="utf-8") as cf:
-                cf.write("")
+                cf.write("") # Create an empty file
 
         # Read existing content
         with open(rc_file, "r", encoding="utf-8") as cf:
@@ -126,11 +131,14 @@ def install_completion():
         if source_command not in content:
             with open(rc_file, "a", encoding="utf-8") as rf:
                 rf.write(f"\n# PDD CLI completion\n{source_command}\n")
-
-            rprint(f"[green]Shell completion installed for {shell}.[/green]")
-            rprint(f"Please restart your shell or run 'source {rc_file}' to enable completion.")
+            
+            if not quiet:
+                rprint(f"[green]Shell completion installed for {shell}.[/green]")
+                rprint(f"Please restart your shell or run 'source {rc_file}' to enable completion.")
         else:
-            rprint(f"[yellow]Shell completion already installed for {shell}.[/yellow]")
+            if not quiet:
+                rprint(f"[yellow]Shell completion already installed for {shell}.[/yellow]")
     except OSError as exc:
-        rprint(f"[red]Failed to install shell completion: {exc}[/red]")
+        if not quiet:
+            rprint(f"[red]Failed to install shell completion: {exc}[/red]")
         raise click.Abort()

pdd/pdd_completion.fish

@@ -3,11 +3,13 @@
 # Global options
 complete -c pdd -n "__fish_use_subcommand" -l force -d "Overwrite existing files without confirmation"
 complete -c pdd -n "__fish_use_subcommand" -l strength -x -d "Set AI model strength (0.0 to 1.0)"
+complete -c pdd -n "__fish_use_subcommand" -l time -x -d "Set AI model reasoning time (0.0 to 1.0)"
 complete -c pdd -n "__fish_use_subcommand" -l temperature -x -d "Set AI model temperature"
 complete -c pdd -n "__fish_use_subcommand" -l verbose -d "Increase output verbosity"
 complete -c pdd -n "__fish_use_subcommand" -l quiet -d "Decrease output verbosity"
 complete -c pdd -n "__fish_use_subcommand" -l output-cost -r -d "Enable cost tracking and output CSV file"
 complete -c pdd -n "__fish_use_subcommand" -l review-examples -d "Review few-shot examples before execution"
+complete -c pdd -n "__fish_use_subcommand" -l local -d "Run commands locally"
 complete -c pdd -n "__fish_use_subcommand" -l help -d "Show help message"
 complete -c pdd -n "__fish_use_subcommand" -l version -d "Show version information"
 
@@ -26,9 +28,12 @@ complete -c pdd -n "__fish_use_subcommand" -a crash -d "Fix code causing program
 complete -c pdd -n "__fish_use_subcommand" -a trace -d "Trace code line to prompt"
 complete -c pdd -n "__fish_use_subcommand" -a bug -d "Generate unit test from bug report"
 complete -c pdd -n "__fish_use_subcommand" -a auto-deps -d "Analyze and insert dependencies"
+complete -c pdd -n "__fish_use_subcommand" -a verify -d "Verify functional correctness using LLM judgment"
 
 # Command-specific completions
 complete -c pdd -n "__fish_seen_subcommand_from generate" -l output -r -d "Output location for generated code"
+complete -c pdd -n "__fish_seen_subcommand_from generate" -l original-prompt -r -d "Original prompt file for incremental generation"
+complete -c pdd -n "__fish_seen_subcommand_from generate" -l incremental -d "Force incremental patching"
 complete -c pdd -n "__fish_seen_subcommand_from generate" -a "(__fish_complete_suffix .prompt)"
 
 complete -c pdd -n "__fish_seen_subcommand_from example" -l output -r -d "Output location for example code"
@@ -63,10 +68,67 @@ complete -c pdd -n "__fish_seen_subcommand_from fix" -a "(__fish_complete_suffix
 complete -c pdd -n "__fish_seen_subcommand_from fix" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)"
 complete -c pdd -n "__fish_seen_subcommand_from fix" -a "(__fish_complete_suffix .log)"
 
+complete -c pdd -n "__fish_seen_subcommand_from split" -l output-sub -r -d "Output for sub-prompt file"
+complete -c pdd -n "__fish_seen_subcommand_from split" -l output-modified -r -d "Output for modified prompt file"
+complete -c pdd -n "__fish_seen_subcommand_from split" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from split" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For INPUT_CODE and EXAMPLE_CODE
+
+complete -c pdd -n "__fish_seen_subcommand_from change" -l output -r -d "Output location for modified prompt"
+complete -c pdd -n "__fish_seen_subcommand_from change" -l csv -d "Use CSV for batch changes"
+complete -c pdd -n "__fish_seen_subcommand_from change" -l budget -x -d "Maximum budget for change process"
+complete -c pdd -n "__fish_seen_subcommand_from change" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from change" -a "(__fish_complete_suffix .csv)" # For the change prompt file if it's a CSV
+complete -c pdd -n "__fish_seen_subcommand_from change" -a "(__fish_complete_path)" # For INPUT_CODE directory or file
+
+complete -c pdd -n "__fish_seen_subcommand_from update" -l output -r -d "Output for updated prompt file"
+complete -c pdd -n "__fish_seen_subcommand_from update" -l git -d "Use git history for original code"
+complete -c pdd -n "__fish_seen_subcommand_from update" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from update" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For MODIFIED_CODE_FILE and INPUT_CODE_FILE
+
+complete -c pdd -n "__fish_seen_subcommand_from detect" -l output -r -d "Output CSV for analysis results"
+complete -c pdd -n "__fish_seen_subcommand_from detect" -a "(__fish_complete_suffix .prompt)" # For PROMPT_FILES and CHANGE_FILE
+
+complete -c pdd -n "__fish_seen_subcommand_from conflicts" -l output -r -d "Output CSV for conflict analysis"
+complete -c pdd -n "__fish_seen_subcommand_from conflicts" -a "(__fish_complete_suffix .prompt)" # For PROMPT1 and PROMPT2
+
+complete -c pdd -n "__fish_seen_subcommand_from crash" -l output -r -d "Output for fixed code file"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -l output-program -r -d "Output for fixed program file"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -l loop -d "Enable iterative fixing"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -l max-attempts -x -d "Maximum fix attempts"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -l budget -x -d "Maximum budget for fixing"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from crash" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For CODE_FILE and PROGRAM_FILE
+complete -c pdd -n "__fish_seen_subcommand_from crash" -a "(__fish_complete_suffix .log .txt)" # For ERROR_FILE
+
+complete -c pdd -n "__fish_seen_subcommand_from trace" -l output -r -d "Output for trace analysis results"
+complete -c pdd -n "__fish_seen_subcommand_from trace" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from trace" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For CODE_FILE
+# No specific completion for CODE_LINE, it's a number
+
+complete -c pdd -n "__fish_seen_subcommand_from bug" -l output -r -d "Output for generated unit test"
+complete -c pdd -n "__fish_seen_subcommand_from bug" -l language -x -d "Programming language for unit test"
+complete -c pdd -n "__fish_seen_subcommand_from bug" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from bug" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For CODE_FILE and PROGRAM_FILE
+complete -c pdd -n "__fish_seen_subcommand_from bug" -a "(__fish_complete_suffix .txt .log)" # For CURRENT_OUTPUT_FILE and DESIRED_OUTPUT_FILE
+
+complete -c pdd -n "__fish_seen_subcommand_from auto-deps" -l output -r -d "Output for modified prompt file"
+complete -c pdd -n "__fish_seen_subcommand_from auto-deps" -l csv -r -d "CSV file for dependency info"
+complete -c pdd -n "__fish_seen_subcommand_from auto-deps" -l force-scan -d "Force rescanning dependencies"
+complete -c pdd -n "__fish_seen_subcommand_from auto-deps" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from auto-deps" -a "(__fish_complete_path)" # For DIRECTORY_PATH
+
+complete -c pdd -n "__fish_seen_subcommand_from verify" -l output-results -r -d "Output for verification results log"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -l output-code -r -d "Output for verified code file"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -l output-program -r -d "Output for verified program file"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -l max-attempts -x -d "Max fix attempts in verification loop"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -l budget -x -d "Max budget for verification and fixing"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from verify" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)" # For CODE_FILE and PROGRAM_FILE
+
 # File completion for all commands
-complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps" -a "(__fish_complete_suffix .prompt)"
-complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)"
-complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps" -a "(__fish_complete_suffix .log .txt .csv)"
+complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify" -a "(__fish_complete_suffix .prompt)"
+complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify" -a "(__fish_complete_suffix .py .js .java .cpp .rb .go)"
+complete -c pdd -n "__fish_seen_subcommand_from generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify" -a "(__fish_complete_suffix .log .txt .csv)"
 
 # Help completion
-complete -c pdd -n "__fish_seen_subcommand_from help" -a "generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps" -d "Show help for specific command"
+complete -c pdd -n "__fish_seen_subcommand_from help" -a "generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify" -d "Show help for specific command"

pdd/pdd_completion.sh

@@ -15,19 +15,19 @@ _pdd() {
     cword=$COMP_CWORD
 
     # Global options
-    local global_opts="--force --strength --temperature --verbose --quiet --output-cost --review-examples --help --version"
+    local global_opts="--force --strength --time --temperature --verbose --quiet --output-cost --review-examples --local --help --version"
 
     # Commands
-    local commands="generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps"
+    local commands="generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify"
 
     # Command-specific options
-    local generate_opts="--output"
+    local generate_opts="--output --original-prompt --incremental"
     local example_opts="--output"
     local test_opts="--output --language --coverage-report --existing-tests --target-coverage --merge"
     local preprocess_opts="--output --xml --recursive --double --exclude"
     local fix_opts="--output-test --output-code --output-results --loop --verification-program --max-attempts --budget --auto-submit"
     local split_opts="--output-sub --output-modified"
-    local change_opts="--output --csv"
+    local change_opts="--output --csv --budget"
     local update_opts="--output --git"
     local detect_opts="--output"
     local conflicts_opts="--output"
@@ -35,6 +35,7 @@ _pdd() {
     local trace_opts="--output"
     local bug_opts="--output --language"
     local auto_deps_opts="--output --csv --force-scan"
+    local verify_opts="--output-results --output-code --output-program --max-attempts --budget"
 
     # Complete global options before command
     if [[ $cword -eq 1 ]]; then
@@ -78,23 +79,26 @@ _pdd() {
         change)
             _complete_files ".prompt"
             _complete_files
-            _complete_files
+            if [[ $prev != "--csv" && ${COMP_WORDS[COMP_CWORD-2]} != "--csv" ]]; then
+                 _complete_files ".prompt"
+            fi
             COMPREPLY+=($(compgen -W "$change_opts" -- "$cur"))
             ;;
         update)
             _complete_files ".prompt"
             _complete_files
-            _complete_files
+            if [[ ! " ${words[@]} " =~ " --git " ]]; then
+                _complete_files
+            fi
             COMPREPLY+=($(compgen -W "$update_opts" -- "$cur"))
             ;;
         detect)
             _complete_files ".prompt"
-            _complete_files
             COMPREPLY+=($(compgen -W "$detect_opts" -- "$cur"))
             ;;
         conflicts)
             _complete_files ".prompt"
-            _complete_files
+            _complete_files ".prompt"
             COMPREPLY+=($(compgen -W "$conflicts_opts" -- "$cur"))
             ;;
         crash)
@@ -121,6 +125,12 @@ _pdd() {
             _complete_files ".prompt"
             COMPREPLY+=($(compgen -W "$auto_deps_opts" -- "$cur"))
             ;;
+        verify)
+            _complete_files ".prompt"
+            _complete_files
+            _complete_files
+            COMPREPLY+=($(compgen -W "$verify_opts" -- "$cur"))
+            ;;
         *)
             COMPREPLY=($(compgen -W "$global_opts" -- "$cur"))
             ;;
@@ -130,12 +140,21 @@ _pdd() {
 _complete_files() {
     local ext=$1
     local files
+    if [[ "$cur" == -* ]]; then
+        return
+    fi
+
     if [[ -n $ext ]]; then
-        files=$(compgen -f -X "!*$ext" -- "$cur")
+        files=$(compgen -f -X "!*${ext}" -- "$cur")
+        COMPREPLY+=($(compgen -f -o plusdirs -X "!*${ext}" -- "$cur" | awk -v e="$ext" '$0 ~ e"$"{ COMPREPLY+=($0) }'))
+        files=$(echo "${files}" | awk '!seen[$0]++')
+
     else
-        files=$(compgen -f -- "$cur")
+        files=$(compgen -f -o plusdirs -- "$cur")
+    fi
+    if [[ -n "$files" ]]; then
+       COMPREPLY+=($(compgen -W "$files" -- "$cur"))
     fi
-    COMPREPLY+=($(compgen -W "$files" -- "$cur"))
 }
 
 complete -F _pdd pdd

pdd/pdd_completion.zsh

@@ -50,11 +50,13 @@ local -a _pdd_global_opts
 _pdd_global_opts=(
   '--force[Overwrite existing files without asking for confirmation.]'
   '--strength[Set the strength of the AI model (0.0 to 1.0, default: 0.5)]:strength:(0.0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0)'
+  '--time[Controls the reasoning allocation for LLM models (0.0 to 1.0, default: 0.25)]:time:(0.0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0)'
   '--temperature[Set the temperature of the AI model (default: 0.0)]:temperature:(0.0 0.25 0.5 0.75 1.0)'
   '--verbose[Increase output verbosity for more detailed information.]'
   '--quiet[Decrease output verbosity (minimal information).]'
   '--output-cost[Enable cost tracking and output a CSV file with usage details.]:filename:_files'
   '--review-examples[Review and optionally exclude few-shot examples before command execution.]'
+  '--local[Run commands locally instead of in the cloud.]'
   '--help[Show help message and exit.]'
   '--version[Show version and exit.]'
 )
@@ -73,6 +75,8 @@ _pdd_generate() {
   _arguments -s \
     $_pdd_global_opts \
     '--output=[Specify where to save the generated code.]:filename:_files' \
+    '--original-prompt=[The original prompt file used to generate existing code.]:filename:_files' \
+    '--incremental[Force incremental patching even if changes are significant.]' \
     '1:prompt-file:_files' \
     '*:filename:_files'
 }
@@ -209,6 +213,7 @@ _pdd_change() {
     $_pdd_global_opts \
     '--output=[Where to save the modified prompt file.]:filename:_files' \
     '--csv[Use a CSV file for batch changes (columns: prompt_name, change_instructions).]' \
+    '--budget=[Maximum cost allowed for the change process (default 5.0)]:float' \
     '1:change-prompt-file:_files' \
     '2:input-code:_files' \
     '3:optional-prompt-file:_files' \
@@ -356,6 +361,32 @@ _pdd_auto_deps() {
     '*:filename:_files'
 }
 
+# verify
+# Usage: pdd [GLOBAL OPTIONS] verify [OPTIONS] PROMPT_FILE CODE_FILE PROGRAM_FILE
+# Options:
+#   --output-results [LOCATION]
+#   --output-code [LOCATION]
+#   --output-program [LOCATION]
+#   --max-attempts [INT]
+#   --budget [FLOAT]
+# Args:
+#   1: PROMPT_FILE
+#   2: CODE_FILE
+#   3: PROGRAM_FILE
+_pdd_verify() {
+  _arguments -s \
+    $_pdd_global_opts \
+    '--output-results=[Where to save verification and fixing results log.]:filename:_files' \
+    '--output-code=[Where to save the successfully verified code file.]:filename:_files' \
+    '--output-program=[Where to save the successfully verified program file.]:filename:_files' \
+    '--max-attempts=[Maximum fix attempts in verification loop (default 3)]:int' \
+    '--budget=[Maximum cost for verification and fixing (default 5.0)]:float' \
+    '1:prompt-file:_files' \
+    '2:code-file:_files' \
+    '3:program-file:_files' \
+    '*:filename:_files'
+}
+
 ##
 # Main PDD completion dispatcher
 ##
@@ -380,6 +411,7 @@ _pdd() {
     'trace:Find the prompt file line number associated with a code line'
     'bug:Generate a unit test based on incorrect vs desired outputs'
     'auto-deps:Analyze a prompt file and directory for needed dependencies'
+    'verify:Verify functional correctness using LLM judgment and iteratively fix'
   )
 
   # If there's no subcommand yet (i.e., user typed only "pdd " or "pdd -<Tab>"), offer global opts or subcommands.
@@ -436,6 +468,9 @@ _pdd() {
     auto-deps)
       _pdd_auto_deps
       ;;
+    verify)
+      _pdd_verify
+      ;;
     # If the subcommand is unknown or not typed yet, fall back to showing the list of subcommands.
     *)
       _describe -t subcommands 'pdd subcommand' _pdd_subcommands

pdd/postprocess.py

@@ -3,6 +3,7 @@ from rich import print
 from pydantic import BaseModel, Field
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
+from . import DEFAULT_TIME
 
 class ExtractedCode(BaseModel):
     """Pydantic model for the extracted code."""
@@ -18,12 +19,13 @@ def postprocess_0(text: str) -> str:
     in_code_block = False
     
     for line in lines:
-        if line.startswith('```'):
+        if '```' in line: # MODIFIED: Was line.startswith('```')
             if not in_code_block:
-                # Skip the language identifier line
+                # Skip the language identifier line / content on opening delimiter line
                 in_code_block = True
                 continue
             else:
+                # Content on closing delimiter line is skipped
                 in_code_block = False
                 continue
         if in_code_block:
@@ -36,6 +38,7 @@ def postprocess(
     language: str,
     strength: float = 0.9,
     temperature: float = 0,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -46,6 +49,7 @@ def postprocess(
         language (str): The programming language of the code to extract
         strength (float): The strength of the LLM model to use (0-1)
         temperature (float): The temperature parameter for the LLM (0-1)
+        time (float): The thinking effort for the LLM model (0-1)
         verbose (bool): Whether to print detailed processing information
     
     Returns:
@@ -87,6 +91,7 @@ def postprocess(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose,
             output_pydantic=ExtractedCode
         )
@@ -94,14 +99,14 @@ def postprocess(
         if not response or 'result' not in response:
             raise ValueError("Failed to get valid response from LLM")
 
-        extracted_code: ExtractedCode = response['result']
-        code_text = extracted_code.extracted_code
+        extracted_code_obj: ExtractedCode = response['result'] # Renamed for clarity
+        code_text = extracted_code_obj.extracted_code
 
         # Step 3c: Remove triple backticks and language identifier if present
         lines = code_text.split('\n')
         if lines and lines[0].startswith('```'):
             lines = lines[1:]
-        if lines and lines[-1].startswith('```'):
+        if lines and lines[-1].startswith('```'): # Check if lines is not empty again after potentially removing first line
             lines = lines[:-1]
         
         final_code = '\n'.join(lines)

{pdd_cli-0.0.32.dist-info → pdd_cli-0.0.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pdd-cli
-Version: 0.0.32
+Version: 0.0.34
 Summary: PDD (Prompt-Driven Development) Command Line Interface
 Author: Greg Tanaka
 Author-email: glt@alumni.caltech.edu
@@ -46,9 +46,13 @@ Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: z3-solver; extra == "dev"
 Dynamic: license-file
 
-.. image:: https://img.shields.io/badge/pdd--cli-v0.0.32-blue
+.. image:: https://img.shields.io/badge/pdd--cli-v0.0.34-blue
    :alt: PDD-CLI Version
 
+.. image:: https://img.shields.io/badge/Discord-join%20chat-7289DA.svg?logo=discord&logoColor=white&link=https://discord.gg/Yp4RTh8bG7
+   :alt: Join us on Discord
+   :target: https://discord.gg/Yp4RTh8bG7
+
 PDD (Prompt-Driven Development) Command Line Interface
 ======================================================
 
@@ -130,7 +134,7 @@ After installation, verify:
 
    pdd --version
 
-You'll see the current PDD version (e.g., 0.0.32).
+You'll see the current PDD version (e.g., 0.0.34).
 
 Advanced Installation Tips
 --------------------------

{pdd_cli-0.0.32.dist-info → pdd_cli-0.0.34.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pdd/__init__.py,sha256=oBlK2EdKLxWULcsuHgrlD34jImz8M12FUeCXxTkbxIk,632
+pdd/__init__.py,sha256=bBnNpQn3DLgqp8-jon-Ht7N0S9jfAWVFwaoxPpKzEPI,632
 pdd/auto_deps_main.py,sha256=NVLqL5FHxe2eorViXTuh8z2zH9Sb-b6MNN9aZ1hqevY,3552
 pdd/auto_include.py,sha256=aCa2QXDlOdKbh4vS3uDjWptkHB_Qv3QBNCbZe6mGWoo,6074
 pdd/auto_update.py,sha256=Pfav1hrqQIDjZIPuIvryBeM7k-Rc72feVUTJZPtigaU,2889
@@ -6,16 +6,16 @@ pdd/bug_main.py,sha256=cSGBnHmFIA8WrkGiohJFVRuM2086v-wlPvTJqTv00WQ,4631
 pdd/bug_to_unit_test.py,sha256=oejqoKomLseKknYDFlQKQ04TNT3soqOjMPghxty8Guo,6311
 pdd/change.py,sha256=EKmv7WvXNX24rjLCnrcaoo4xOVkNhCa9HLRbpMAxQSw,5036
 pdd/change_main.py,sha256=Tk0aHY8Z2d-d0m7lLfc_HqgOqs4x_dJD1IrVZSxuxIM,25580
-pdd/cli.py,sha256=VFoyx3iJmFeHCc4HhOc_7tiz3FY9oJjCUtS9PR1wwl0,38341
+pdd/cli.py,sha256=xXj88OFD74T6Cb8ySHXwwVnFNl-oD0YNi02geyCTjf4,39072
 pdd/cmd_test_main.py,sha256=aSCxRnSurg15AvPcJDAPp9xy8p_qqnjU1oV14Hi2R54,5301
 pdd/code_generator.py,sha256=DqQNN6jCNjSJvHi0IFyqkSfak6LeDG-yQHPYnvd4AJQ,4424
-pdd/code_generator_main.py,sha256=wup3O2dVh9JzlGsWtXvypqAHgHyBZHOWAe3NJuV_-g4,18648
+pdd/code_generator_main.py,sha256=oCI35RqQ7gBov8mncVg1mhdDNX-0jTsoIeSzrz7WpAk,25109
 pdd/comment_line.py,sha256=sX2hf4bG1fILi_rvI9MkkwCZ2IitgKkW7nOiw8aQKPY,1845
 pdd/conflicts_in_prompts.py,sha256=XaEm9jIMcbENTyhphD8NXs2EmRxuPu1EI8GHrsiRwq0,4687
 pdd/conflicts_main.py,sha256=O87s9baSa9DJMndxPIdsnYO_spoajcv9jii3XYt_-fM,3473
 pdd/construct_paths.py,sha256=AE3vblj2v21ZdReM40pz47gvmXg-RZjuR_2FALj_9fs,17672
-pdd/context_generator.py,sha256=MO483MqtMR5EUanllD7Ru5U9idcVMIMutntcWMn31cI,5847
-pdd/context_generator_main.py,sha256=TtsY3jHictdEjmB4cHyNwXmZW_LfHJp3KW3UXyzR2cU,2735
+pdd/context_generator.py,sha256=KQAaJRwVpsPOF4y5lhP3DpIrlxhR39C0hNyTAi1Kx2U,5618
+pdd/context_generator_main.py,sha256=WSS7uRkS2wi8HCixDA3_xzFuPqFRF_X2tGe6K3loUZc,2773
 pdd/continue_generation.py,sha256=upZw77fKqHZ5IS8Ful7eioTi9VTqeuY1V9vzIwHeOJQ,5419
 pdd/crash_main.py,sha256=yveWWJWxCOzRLxPk1m1rgm4tNwpAk5jYRvYyItAaU2w,7262
 pdd/detect_change.py,sha256=t70NAgakVc2u7XiI-lZjuR2RZ8Mrd3lJwRPu8I8s_P4,5310
@@ -40,14 +40,14 @@ pdd/git_update.py,sha256=Ya7eI7YFtGIpT7FdziFJfnFkiZlj8I9Lh98lqtXfClc,2855
 pdd/increase_tests.py,sha256=dlYd9PosV3g8kpDlQh6gd8WUauClvs_-Z8ERRD-kZk4,3456
 pdd/incremental_code_generator.py,sha256=cWo3DJ0PybnrepFEAMibGjTVY3T8mLVvPt5W8cNhuxU,9402
 pdd/insert_includes.py,sha256=UASoq_46UNL6l7VGB7DW2jb4kcWlP6Hbj2EWuh7210Q,5310
-pdd/install_completion.py,sha256=H_FIid0eHCn7I6HHIsBqdComAxjLIu7CbD6yfMU48aw,5085
+pdd/install_completion.py,sha256=bLMJuMOBDvsEnDAUpgiPesNRGhY_IvBvz8ZvmbTzP4o,5472
 pdd/llm_invoke.py,sha256=457rD3f7KQHJ3BtI4boxTurArCww2zWAl4X77TxVROs,65788
 pdd/load_prompt_template.py,sha256=4NH8_t5eon_vcyTznqtemJ_yAPkTJm_hSdTRgzj3qEQ,1907
 pdd/mcp_config.json,sha256=D3ctWHlShvltbtH37zbYb6smVE0V80_lGjDKDIqsSBE,124
-pdd/pdd_completion.fish,sha256=rs-43fa3kcDBN1uy4oxiofLAWmaqW0U2j5Mu4wCHh5M,6121
-pdd/pdd_completion.sh,sha256=qurWrEksqptjryBZszxHv6i0MqgnIqJenMBDrzMgI98,4535
-pdd/pdd_completion.zsh,sha256=-yz69Rs4lnJ-bRAF8H2Tzk0GuDUo-BWjytE3fv5wTS8,13458
-pdd/postprocess.py,sha256=7Dt4C7hZZbqCpYK0LG2Ui_vIUYw9UTN3w5Wgd_JZYBs,4021
+pdd/pdd_completion.fish,sha256=XgtbuPV11pB8urYXhbpXhWuwKAVLpiAJxOMIes5CV-8,11884
+pdd/pdd_completion.sh,sha256=e0K4Ai1yJRj6Fd6Qpk-brAEicW3ciPSyIuNMQkSG5mI,5342
+pdd/pdd_completion.zsh,sha256=WPcz7BzvjDRmcEZfL-kFvhYmUtz2BHxJK_r3_cPa478,14966
+pdd/postprocess.py,sha256=jWJSUPP7bDChyTEek35UOdy6fhU6c5EoDSoMypGwIdE,4402
 pdd/postprocess_0.py,sha256=OW17GyCFLYErCyWh2tL4syuho3q2yFf2wyekQ4BLdPM,2168
 pdd/preprocess.py,sha256=UB1rqSNscUC-JHujvGb11LJ5OadZ3GVD1Qeq1dI6HMc,9508
 pdd/preprocess_main.py,sha256=dAgFGmjuJB1taZl31c1sY2jMGtQgjnWLbpeB7EFtojY,2977
@@ -64,7 +64,7 @@ pdd/update_main.py,sha256=5a4nsOOaAXULdk0BS9pj4blZ_QHBFeET37uaAqoJI2g,3912
 pdd/update_model_costs.py,sha256=zfGqWoVIjBppKUt1Naq2ZsIdiucwfH6at2DflcwkJf8,22998
 pdd/update_prompt.py,sha256=OdPRIAMu7OBx7E4SOU95hWgdtBY4oO8XOe1dvPChMlU,4351
 pdd/xml_tagger.py,sha256=5AWzOboNuXMJmmIpi1hhPPAB6nxbBOkRqgoPLpoGYT8,4292
-pdd/data/language_format.csv,sha256=xUTmFHXSBVBRfPV-NKG3oWo5_ped5ukP-ekFcIlVzJk,877
+pdd/data/language_format.csv,sha256=X7jj_clC6dA93ij0TUPNd9oJ2tHyKUKXlWo1XVhP6zo,901
 pdd/data/llm_model.csv,sha256=BZioMyxUlDuJdcxFCDV2BTw1ssiHKZBf4OxELIy4JGY,1487
 pdd/prompts/auto_include_LLM.prompt,sha256=0t-Jmm5o6vVTmqsISTUiewqPT8bB389UZnJoHZvgtu4,13967
 pdd/prompts/bug_to_unit_test_LLM.prompt,sha256=KdMkvRVnjVSf0NTYIaDXIMT93xPttXEwkMpjWx5leLs,1588
@@ -101,9 +101,9 @@ pdd/prompts/trim_results_start_LLM.prompt,sha256=OKz8fAf1cYWKWgslFOHEkUpfaUDARh3
 pdd/prompts/unfinished_prompt_LLM.prompt,sha256=-JgBpiPTQZdWOAwOG1XpfpD9waynFTAT3Jo84eQ4bTw,1543
 pdd/prompts/update_prompt_LLM.prompt,sha256=prIc8uLp2jqnLTHt6JvWDZGanPZipivhhYeXe0lVaYw,1328
 pdd/prompts/xml_convertor_LLM.prompt,sha256=YGRGXJeg6EhM9690f-SKqQrKqSJjLFD51UrPOlO0Frg,2786
-pdd_cli-0.0.32.dist-info/licenses/LICENSE,sha256=-1bjYH-CEjGEQ8VixtnRYuu37kN6F9NxmZSDkBuUQ9o,1062
-pdd_cli-0.0.32.dist-info/METADATA,sha256=gnrubUf69rPVr3t01FvagrPuxrJ3B6JRbHXjYQYB3OQ,7778
-pdd_cli-0.0.32.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-pdd_cli-0.0.32.dist-info/entry_points.txt,sha256=Kr8HtNVb8uHZtQJNH4DnF8j7WNgWQbb7_Pw5hECSR-I,36
-pdd_cli-0.0.32.dist-info/top_level.txt,sha256=xjnhIACeMcMeDfVNREgQZl4EbTni2T11QkL5r7E-sbE,4
-pdd_cli-0.0.32.dist-info/RECORD,,
+pdd_cli-0.0.34.dist-info/licenses/LICENSE,sha256=-1bjYH-CEjGEQ8VixtnRYuu37kN6F9NxmZSDkBuUQ9o,1062
+pdd_cli-0.0.34.dist-info/METADATA,sha256=rU1Yu_F0B5Z0eR7ms1PPzZPlVTY6CQ2L3lLsjuvw4wY,7984
+pdd_cli-0.0.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pdd_cli-0.0.34.dist-info/entry_points.txt,sha256=Kr8HtNVb8uHZtQJNH4DnF8j7WNgWQbb7_Pw5hECSR-I,36
+pdd_cli-0.0.34.dist-info/top_level.txt,sha256=xjnhIACeMcMeDfVNREgQZl4EbTni2T11QkL5r7E-sbE,4
+pdd_cli-0.0.34.dist-info/RECORD,,

{pdd_cli-0.0.32.dist-info → pdd_cli-0.0.34.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any