pdd-cli 0.0.45__py3-none-any.whl → 0.0.90__py3-none-any.whl

pdd/load_prompt_template.py

@@ -23,18 +23,44 @@ def load_prompt_template(prompt_name: str) -> Optional[str]:
         print_formatted("[red]Unexpected error loading prompt template[/red]")
         return None
 
-    # Step 1: Get project path from environment variable
-    project_path = os.getenv('PDD_PATH')
-    if not project_path:
-        print_formatted("[red]PDD_PATH environment variable is not set[/red]")
-        return None
+    # Step 1: Get project path from environment variable (preferred),
+    # else fall back to auto-detect based on this module's location or CWD.
+    project_path_env = os.getenv('PDD_PATH')
+    candidate_paths = []
+    if project_path_env:
+        candidate_paths.append(Path(project_path_env))
+
+    # Fallback 1: repository root inferred from this module (pdd/ => repo root)
+    try:
+        module_root = Path(__file__).resolve().parent  # pdd/
+        repo_root = module_root.parent                 # repo root
+        candidate_paths.append(repo_root)
+    except Exception:
+        pass
 
-    # Construct the full path to the prompt file
-    prompt_path = Path(project_path) / 'prompts' / f"{prompt_name}.prompt"
+    # Fallback 2: current working directory
+    candidate_paths.append(Path.cwd())
+
+    # Build candidate prompt paths to try in order
+    prompt_candidates = []
+    for cp in candidate_paths:
+        # Check both <path>/prompts/ and <path>/pdd/prompts/
+        # The latter handles installed package case where prompts are in pdd/prompts/
+        prompt_candidates.append(cp / 'prompts' / f"{prompt_name}.prompt")
+        prompt_candidates.append(cp / 'pdd' / 'prompts' / f"{prompt_name}.prompt")
 
     # Step 2: Load and return the prompt template
-    if not prompt_path.exists():
-        print_formatted(f"[red]Prompt file not found: {prompt_path}[/red]")
+    prompt_path: Optional[Path] = None
+    for candidate in prompt_candidates:
+        if candidate.exists():
+            prompt_path = candidate
+            break
+
+    if prompt_path is None:
+        tried = "\n".join(str(c) for c in prompt_candidates)
+        print_formatted(
+            f"[red]Prompt file not found in any candidate locations for '{prompt_name}'. Tried:\n{tried}[/red]"
+        )
         return None
 
     try:
@@ -56,4 +82,4 @@ if __name__ == "__main__":
     prompt = load_prompt_template("example_prompt")
     if prompt:
         print_formatted("[blue]Loaded prompt template:[/blue]")
-        print_formatted(prompt)
+        print_formatted(prompt)

pdd/pdd_completion.fish

@@ -10,6 +10,8 @@ complete -c pdd -n "__fish_use_subcommand" -l quiet -d "Decrease output verbosit
 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 context -r -d "Override .pddrc context"
+complete -c pdd -n "__fish_use_subcommand" -l list-contexts -d ".pddrc contexts and exit"
 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"
 
@@ -27,13 +29,18 @@ complete -c pdd -n "__fish_use_subcommand" -a conflicts -d "Analyze conflicts be
 complete -c pdd -n "__fish_use_subcommand" -a crash -d "Fix code causing program crash"
 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 auto-deps -d "Analyze and insert dependencies from directory or glob"
 complete -c pdd -n "__fish_use_subcommand" -a verify -d "Verify functional correctness using LLM judgment"
+complete -c pdd -n "__fish_use_subcommand" -a sync -d "Synchronize prompt, code, examples, tests"
+complete -c pdd -n "__fish_use_subcommand" -a setup -d "Interactive setup and completion install"
+complete -c pdd -n "__fish_use_subcommand" -a install_completion -d "Install shell completion"
+complete -c pdd -n "__fish_use_subcommand" -a pytest-output -d "Run pytest and capture structured output"
 
 # 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" -s e -l env -xa "(env | cut -d= -f1 | sed 's/.*/&=/' | sort -u)" -d "Set template variable (KEY=VALUE) or read KEY from env"
 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"
@@ -125,10 +132,26 @@ complete -c pdd -n "__fish_seen_subcommand_from verify" -l budget -x -d "Max bud
 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
 
+# sync command
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l max-attempts -x -d "Max attempts for loops"
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l budget -x -d "Total budget for sync"
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l skip-verify -d "Skip functional verification"
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l skip-tests -d "Skip unit test generation"
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l target-coverage -x -d "Desired coverage percentage"
+complete -c pdd -n "__fish_seen_subcommand_from sync" -l log -d "Show analysis instead of running"
+
+# setup and install_completion have no options
+complete -c pdd -n "__fish_seen_subcommand_from setup" -d "Run interactive setup"
+complete -c pdd -n "__fish_seen_subcommand_from install_completion" -d "Install shell completion"
+
+# pytest-output command
+complete -c pdd -n "__fish_seen_subcommand_from pytest-output" -l json-only -d "Print only JSON"
+complete -c pdd -n "__fish_seen_subcommand_from pytest-output" -a "(__fish_complete_suffix .py)"
+
 # 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 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 verify" -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 sync setup install_completion pytest-output" -d "Show help for specific command"

pdd/pdd_completion.sh

@@ -15,13 +15,13 @@ _pdd() {
     cword=$COMP_CWORD
 
     # Global options
-    local global_opts="--force --strength --time --temperature --verbose --quiet --output-cost --review-examples --local --help --version"
+    local global_opts="--force --strength --time --temperature --verbose --quiet --output-cost --review-examples --local --context --list-contexts --help --version"
 
     # Commands
-    local commands="generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify"
+    local commands="generate example test preprocess fix split change update detect conflicts crash trace bug auto-deps verify sync setup install_completion pytest-output"
 
     # Command-specific options
-    local generate_opts="--output --original-prompt --incremental"
+    local generate_opts="--output --original-prompt --incremental --env -e"
     local example_opts="--output"
     local test_opts="--output --language --coverage-report --existing-tests --target-coverage --merge"
     local preprocess_opts="--output --xml --recursive --double --exclude"
@@ -36,6 +36,8 @@ _pdd() {
     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"
+    local sync_opts="--max-attempts --budget --skip-verify --skip-tests --target-coverage --log"
+    local pytest_output_opts="--json-only"
 
     # Complete global options before command
     if [[ $cword -eq 1 ]]; then
@@ -46,6 +48,16 @@ _pdd() {
     # Complete command-specific options
     case ${words[1]} in
         generate)
+            # If completing the value for -e/--env, suggest environment variable names (with and without '=')
+            if [[ $prev == "-e" || $prev == "--env" ]]; then
+                local vars
+                vars=$(env | cut -d= -f1 | sort -u)
+                # Offer both KEY and KEY=
+                local vars_with_eq
+                vars_with_eq=$(printf '%s=\n' $vars)
+                COMPREPLY=($(compgen -W "$vars $vars_with_eq" -- "$cur"))
+                return
+            fi
             _complete_files ".prompt"
             COMPREPLY+=($(compgen -W "$generate_opts" -- "$cur"))
             ;;
@@ -131,6 +143,20 @@ _pdd() {
             _complete_files
             COMPREPLY+=($(compgen -W "$verify_opts" -- "$cur"))
             ;;
+        sync)
+            # BASENAME (not a file), offer options
+            COMPREPLY+=($(compgen -W "$sync_opts" -- "$cur"))
+            ;;
+        setup)
+            # no command-specific options
+            ;;
+        install_completion)
+            # no command-specific options
+            ;;
+        pytest-output)
+            _complete_files
+            COMPREPLY+=($(compgen -W "$pytest_output_opts" -- "$cur"))
+            ;;
         *)
             COMPREPLY=($(compgen -W "$global_opts" -- "$cur"))
             ;;
@@ -157,4 +183,4 @@ _complete_files() {
     fi
 }
 
-complete -F _pdd pdd
+complete -F _pdd pdd

pdd/pdd_completion.zsh

@@ -57,6 +57,8 @@ _pdd_global_opts=(
   '--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.]'
+  '--context[Override automatic .pddrc context]:context-name:_guard'
+  '--list-contexts[List available .pddrc contexts and exit]'
   '--help[Show help message and exit.]'
   '--version[Show version and exit.]'
 )
@@ -65,6 +67,14 @@ _pdd_global_opts=(
 # Per-subcommand completion functions
 ##
 
+# Helper: suggest environment variables (KEY and KEY=)
+_pdd_env_vars() {
+  local -a envs envs_eq
+  envs=(${(f)"$(env | cut -d= -f1 | sort -u)"})
+  envs_eq=(${envs/%/=})
+  _describe -t envvars 'environment variables' envs_eq envs
+}
+
 # generate
 # Usage: pdd [GLOBAL OPTIONS] generate [OPTIONS] PROMPT_FILE
 # Options:
@@ -77,6 +87,7 @@ _pdd_generate() {
     '--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.]' \
+    '(-e --env)'{-e,--env}'[Set template variable (KEY=VALUE) or read KEY from env]:template variable:_pdd_env_vars' \
     '1:prompt-file:_files' \
     '*:filename:_files'
 }
@@ -349,7 +360,7 @@ _pdd_bug() {
 #   --force-scan
 # Args:
 #   1: PROMPT_FILE
-#   2: DIRECTORY_PATH
+#   2: DIRECTORY_PATH (directory or glob pattern)
 _pdd_auto_deps() {
   _arguments -s \
     $_pdd_global_opts \
@@ -357,7 +368,7 @@ _pdd_auto_deps() {
     '--csv=[CSV file for dependency info (default: project_dependencies.csv).]:filename:_files' \
     '--force-scan[Force rescanning of all potential dependency files.]' \
     '1:prompt-file:_files' \
-    '2:directory:_files -/' \
+    '2:directory-or-glob:_files -/' \
     '*:filename:_files'
 }
 
@@ -387,6 +398,54 @@ _pdd_verify() {
     '*:filename:_files'
 }
 
+# sync
+# Usage: pdd [GLOBAL OPTIONS] sync [OPTIONS] BASENAME
+# Options:
+#   --max-attempts [INT]
+#   --budget [FLOAT]
+#   --skip-verify
+#   --skip-tests
+#   --target-coverage [FLOAT]
+#   --log
+# Arg:
+#   1: BASENAME
+_pdd_sync() {
+  _arguments -s \
+    $_pdd_global_opts \
+    '--max-attempts=[Maximum attempts for iterative loops (default 3)]:int' \
+    '--budget=[Maximum total cost for sync (default 10.0)]:float' \
+    '--skip-verify[Skip the functional verification step]' \
+    '--skip-tests[Skip unit test generation and fixing]' \
+    '--target-coverage=[Desired code coverage percentage]:float' \
+    '--log[Show analysis instead of executing operations]' \
+    '1:basename: ' \
+    '*: :'
+}
+
+# setup (no options)
+_pdd_setup() {
+  _arguments -s $_pdd_global_opts
+}
+
+# install_completion (no options)
+_pdd_install_completion() {
+  _arguments -s $_pdd_global_opts
+}
+
+# pytest-output
+# Usage: pdd [GLOBAL OPTIONS] pytest-output [OPTIONS] TEST_FILE
+# Options:
+#   --json-only
+# Arg:
+#   1: TEST_FILE
+_pdd_pytest_output() {
+  _arguments -s \
+    $_pdd_global_opts \
+    '--json-only[Print only JSON to stdout]' \
+    '1:test-file:_files' \
+    '*:filename:_files'
+}
+
 ##
 # Main PDD completion dispatcher
 ##
@@ -410,8 +469,12 @@ _pdd() {
     'crash:Fix errors in a code module and its calling program'
     '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'
+    'auto-deps:Analyze a prompt and include deps from a directory or glob'
     'verify:Verify functional correctness using LLM judgment and iteratively fix'
+    'sync:Synchronize prompt, code, examples, tests with analysis'
+    'setup:Interactive setup and completion install'
+    'install_completion:Install shell completion for current shell'
+    'pytest-output:Run pytest and capture structured output'
   )
 
   # If there's no subcommand yet (i.e., user typed only "pdd " or "pdd -<Tab>"), offer global opts or subcommands.
@@ -471,6 +534,18 @@ _pdd() {
     verify)
       _pdd_verify
       ;;
+    sync)
+      _pdd_sync
+      ;;
+    setup)
+      _pdd_setup
+      ;;
+    install_completion)
+      _pdd_install_completion
+      ;;
+    pytest-output)
+      _pdd_pytest_output
+      ;;
     # If the subcommand is unknown or not typed yet, fall back to showing the list of subcommands.
     *)
       _describe -t subcommands 'pdd subcommand' _pdd_subcommands
@@ -487,4 +562,4 @@ else
   echo >&2 "Warning: Could not register pdd completion. Make sure ZSH completion system is working."
 fi
 
-# End of pdd_completion.zsh
+# End of pdd_completion.zsh

pdd/postprocess.py

@@ -3,7 +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
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 
 class ExtractedCode(BaseModel):
     """Pydantic model for the extracted code."""
@@ -36,7 +36,7 @@ def postprocess_0(text: str) -> str:
 def postprocess(
     llm_output: str,
     language: str,
-    strength: float = 0.9,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0,
     time: float = DEFAULT_TIME,
     verbose: bool = False
@@ -99,7 +99,14 @@ def postprocess(
         if not response or 'result' not in response:
             raise ValueError("Failed to get valid response from LLM")
 
-        extracted_code_obj: ExtractedCode = response['result'] # Renamed for clarity
+        result_obj = response['result']
+        if not isinstance(result_obj, ExtractedCode):
+            # If we got a string (likely an error message from llm_invoke), fallback to simple extraction
+            if verbose:
+                print(f"[yellow]Structured extraction failed ({result_obj}). Falling back to simple extraction.[/yellow]")
+            return (postprocess_0(llm_output), response.get('cost', 0.0), response.get('model_name', 'fallback'))
+
+        extracted_code_obj: ExtractedCode = result_obj
         code_text = extracted_code_obj.extracted_code
 
         # Step 3c: Remove triple backticks and language identifier if present