pdd-cli 0.0.58__py3-none-any.whl → 0.0.60__py3-none-any.whl

pdd/__init__.py

@@ -1,6 +1,6 @@
 """PDD - Prompt Driven Development"""
 
-__version__ = "0.0.58"
+__version__ = "0.0.60"
 
 # Strength parameter used for LLM extraction across the codebase
 # Used in postprocessing, XML tagging, code generation, and other extraction

pdd/cli.py

@@ -155,11 +155,7 @@ def _should_show_onboarding_reminder(ctx: click.Context) -> bool:
 
 def _run_setup_utility() -> None:
     """Execute the interactive setup utility script."""
-    setup_script = Path(__file__).resolve().parent.parent / "utils" / "pdd-setup.py"
-    if not setup_script.exists():
-        raise FileNotFoundError(f"Setup utility not found at {setup_script}")
-
-    result = subprocess.run([sys.executable, str(setup_script)])
+    result = subprocess.run([sys.executable, "-m", "pdd.setup_tool"])
     if result.returncode not in (0, None):
         raise RuntimeError(f"Setup utility exited with status {result.returncode}")
 

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"
 
@@ -29,12 +31,16 @@ complete -c pdd -n "__fish_use_subcommand" -a trace -d "Trace code line to promp
 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 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" -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"
@@ -126,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,10 +15,10 @@ _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 --env -e"
@@ -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
@@ -141,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"))
             ;;

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.]'
 )
@@ -396,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
 ##
@@ -421,6 +471,10 @@ _pdd() {
     'bug:Generate a unit test based on incorrect vs desired outputs'
     '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.
@@ -480,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

pdd_cli-0.0.58.data/data/utils/pdd-setup.py → pdd/setup_tool.py

@@ -9,6 +9,8 @@ import sys
 import subprocess
 import json
 import requests
+import csv
+import importlib.resources
 from pathlib import Path
 from typing import Dict, Optional, Tuple, List
 
@@ -39,26 +41,27 @@ YELLOW = "\033[93m"
 BOLD = "\033[1m"
 
 # Template content inline
-HELLO_PYTHON_TEMPLATE = """Create a Python program that prints "Hello <username>" in ASCII art.
-
-Requirements:
-- <username> is the username of the current session, using the "whoami" command
-- in the code, generate the full english 26 character alphabet in ascii art as a map of character values to ascii art strings, then use those to render
-- Use only the Python standard library (no external dependencies)
-- Create large, bold ASCII art text, using ASCII drawing characters, symbols, or any other characters that are useful
-- The drawn text should be at least 10 rows in height
-- Make it visually appealing with simple characters like #, *, or =
-- Keep the code clean and readable
-- Add a brief comment explaining what the program does
-
-The program should be self-contained and runnable with just `python3 filename.py`."""
-
-LLM_MODEL_CSV_TEMPLATE = """provider,model,input,output,coding_arena_elo,base_url,api_key,max_reasoning_tokens,structured_output,reasoning_type
-OpenAI,gpt-5-nano,0.05,0.4,1249,,OPENAI_API_KEY,0,True,none
-Google,gemini/gemini-2.5-pro,1.25,10.0,1360,,GOOGLE_API_KEY,0,True,none
-OpenAI,gpt-5-mini,0.25,2.0,1325,,OPENAI_API_KEY,0,True,effort
-OpenAI,gpt-5,1.25,10.0,1482,,OPENAI_API_KEY,0,True,effort
-OpenAI,gpt-4.1,2.0,8.0,1253,,OPENAI_API_KEY,0,True,none"""
+SUCCESS_PYTHON_TEMPLATE = """
+Write a python script to print "You did it, <Username>!!!" to the console.  
+Do not write anything except that message.   
+Capitalize the username."""
+
+def _read_packaged_llm_model_csv() -> Tuple[List[str], List[Dict[str, str]]]:
+    """Load the packaged CSV (pdd/data/llm_model.csv) and return header + rows.
+
+    Returns:
+        (header_fields, rows) where header_fields is the list of column names
+        and rows is a list of dictionaries for each CSV row.
+    """
+    try:
+        csv_text = importlib.resources.files('pdd').joinpath('data/llm_model.csv').read_text()
+    except Exception as e:
+        raise FileNotFoundError(f"Failed to load default LLM model CSV from package: {e}")
+
+    reader = csv.DictReader(csv_text.splitlines())
+    header = reader.fieldnames or []
+    rows = [row for row in reader]
+    return header, rows
 
 def print_colored(text: str, color: str = WHITE, bold: bool = False) -> None:
     """Print colored text to console"""
@@ -95,16 +98,60 @@ def print_pdd_logo():
         ]
     )
     print(f"{CYAN}{logo}{RESET}")
-    print_colored("Supported: OpenAI and Google Gemini (non-Vertex)", WHITE)
+    print()
+    print_colored("Let's get set up quickly with a solid basic configuration!", WHITE, bold=True)
+    print()
+    print_colored("Supported: OpenAI, Google Gemini, and Anthropic Claude", WHITE)
     print_colored("from their respective API endpoints (no third-parties, such as Azure)", WHITE)
     print()
 
+def get_csv_variable_names() -> Dict[str, str]:
+    """Inspect packaged CSV to determine API key variable names per provider.
+
+    Focus on direct providers only: OpenAI GPT models (model startswith 'gpt-'),
+    Google Gemini (model startswith 'gemini/'), and Anthropic (model startswith 'anthropic/').
+    """
+    header, rows = _read_packaged_llm_model_csv()
+    variable_names: Dict[str, str] = {}
+
+    for row in rows:
+        model = (row.get('model') or '').strip()
+        api_key = (row.get('api_key') or '').strip()
+        provider = (row.get('provider') or '').strip().upper()
+
+        if not api_key:
+            continue
+
+        if model.startswith('gpt-') and provider == 'OPENAI':
+            variable_names['OPENAI'] = api_key
+        elif model.startswith('gemini/') and provider == 'GOOGLE':
+            # Prefer direct Gemini key, not Vertex
+            variable_names['GOOGLE'] = api_key
+        elif model.startswith('anthropic/') and provider == 'ANTHROPIC':
+            variable_names['ANTHROPIC'] = api_key
+
+    # Fallbacks if not detected (keep prior behavior)
+    variable_names.setdefault('OPENAI', 'OPENAI_API_KEY')
+    # Prefer GEMINI_API_KEY name for Google if present
+    variable_names.setdefault('GOOGLE', 'GEMINI_API_KEY')
+    variable_names.setdefault('ANTHROPIC', 'ANTHROPIC_API_KEY')
+    return variable_names
+
 def discover_api_keys() -> Dict[str, Optional[str]]:
     """Discover API keys from environment variables"""
+    # Get the variable names actually used in CSV template
+    csv_vars = get_csv_variable_names()
+    
     keys = {
         'OPENAI_API_KEY': os.getenv('OPENAI_API_KEY'),
-        'GOOGLE_API_KEY': os.getenv('GOOGLE_API_KEY') or os.getenv('GEMINI_API_KEY'),
+        'ANTHROPIC_API_KEY': os.getenv('ANTHROPIC_API_KEY'),
     }
+    
+    # For Google, check both possible environment variables but use CSV template's variable name
+    google_var_name = csv_vars.get('GOOGLE', 'GEMINI_API_KEY')  # Default to GEMINI_API_KEY
+    google_api_key = os.getenv('GEMINI_API_KEY') or os.getenv('GOOGLE_API_KEY')
+    keys[google_var_name] = google_api_key
+    
     return keys
 
 def test_openai_key(api_key: str) -> bool:
@@ -140,6 +187,26 @@ def test_google_key(api_key: str) -> bool:
     except Exception:
         return False
 
+def test_anthropic_key(api_key: str) -> bool:
+    """Test Anthropic API key validity"""
+    if not api_key or not api_key.strip():
+        return False
+    
+    try:
+        headers = {
+            'x-api-key': api_key.strip(),
+            'Content-Type': 'application/json'
+        }
+        response = requests.get(
+            'https://api.anthropic.com/v1/messages',
+            headers=headers,
+            timeout=10
+        )
+        # Anthropic returns 400 for invalid request structure but 401/403 for bad keys
+        return response.status_code != 401 and response.status_code != 403
+    except Exception:
+        return False
+
 def test_api_keys(keys: Dict[str, Optional[str]]) -> Dict[str, bool]:
     """Test all discovered API keys"""
     results = {}
@@ -153,8 +220,10 @@ def test_api_keys(keys: Dict[str, Optional[str]]) -> Dict[str, bool]:
             print(f"Testing {key_name}...", end=" ", flush=True)
             if key_name == 'OPENAI_API_KEY':
                 valid = test_openai_key(key_value)
-            elif key_name in ['GOOGLE_API_KEY']:
+            elif key_name in ['GEMINI_API_KEY', 'GOOGLE_API_KEY']:
                 valid = test_google_key(key_value)
+            elif key_name == 'ANTHROPIC_API_KEY':
+                valid = test_anthropic_key(key_value)
             else:
                 valid = False
             
@@ -181,13 +250,16 @@ def get_user_keys(current_keys: Dict[str, Optional[str]]) -> Dict[str, Optional[
     print_colored("Get API keys here:", WHITE)
     print_colored(f"  OpenAI {ARROW_RIGHT} https://platform.openai.com/api-keys", CYAN)
     print_colored(f"  Google Gemini {ARROW_RIGHT} https://aistudio.google.com/app/apikey", CYAN)
+    print_colored(f"  Anthropic {ARROW_RIGHT} https://console.anthropic.com/settings/keys", CYAN)
     print()
     print_colored("A free instant starter key is available from Google Gemini (above)", CYAN)
     print()
     
     new_keys = current_keys.copy()
     
-    for key_name in ['OPENAI_API_KEY', 'GOOGLE_API_KEY']:
+    # Get the actual key names from discovered keys
+    key_names = list(current_keys.keys())
+    for key_name in key_names:
         current_value = current_keys.get(key_name, "")
         status = "found" if current_value else "not found"
         
@@ -230,7 +302,8 @@ def get_shell_init_file(shell: str) -> str:
         'fish': home / '.config/fish/config.fish',
         'csh': home / '.cshrc',
         'tcsh': home / '.tcshrc',
-        'ksh': home / '.kshrc'
+        'ksh': home / '.kshrc',
+        'sh': home / '.profile'
     }
     
     return str(shell_files.get(shell, home / '.bashrc'))
@@ -249,13 +322,13 @@ def create_api_env_script(keys: Dict[str, str], shell: str) -> str:
         for key, value in valid_keys.items():
             lines.append(f'setenv {key} "{value}"')
         return '\n'.join(lines) + '\n'
-    else:  # bash, zsh, ksh and others
+    else:  # bash, zsh, ksh, sh and others
         lines = []
         for key, value in valid_keys.items():
             lines.append(f'export {key}="{value}"')
         return '\n'.join(lines) + '\n'
 
-def save_configuration(valid_keys: Dict[str, str]) -> Tuple[List[str], bool]:
+def save_configuration(valid_keys: Dict[str, str]) -> Tuple[List[str], bool, Optional[str]]:
     """Save configuration to ~/.pdd/ directory"""
     home = Path.home()
     pdd_dir = home / '.pdd'
@@ -271,34 +344,60 @@ def save_configuration(valid_keys: Dict[str, str]) -> Tuple[List[str], bool]:
     shell = detect_shell()
     api_env_content = create_api_env_script(valid_keys, shell)
     
-    # Write api-env file
-    api_env_file = pdd_dir / 'api-env'
+    # Write shell-specific api-env file
+    api_env_file = pdd_dir / f'api-env.{shell}'
     api_env_file.write_text(api_env_content)
     api_env_file.chmod(0o755)
     saved_files.append(str(api_env_file))
     
-    # Create llm_model.csv with only valid providers
-    csv_lines = LLM_MODEL_CSV_TEMPLATE.strip().split('\n')
-    header = csv_lines[0]
-    valid_lines = [header]
-    
-    for line in csv_lines[1:]:
-        if 'OPENAI_API_KEY' in line and 'OPENAI_API_KEY' in valid_keys:
-            valid_lines.append(line)
-        elif 'GOOGLE_API_KEY' in line and 'GOOGLE_API_KEY' in valid_keys:
-            valid_lines.append(line)
-    
+    # Create llm_model.csv with models from packaged CSV filtered by provider and available keys
+    header_fields, rows = _read_packaged_llm_model_csv()
+
+    # Keep only direct Google Gemini (model startswith 'gemini/'), OpenAI GPT (gpt-*) and Anthropic (anthropic/*)
+    def _is_supported_model(row: Dict[str, str]) -> bool:
+        model = (row.get('model') or '').strip()
+        if model.startswith('gpt-'):
+            return True
+        if model.startswith('gemini/'):
+            return True
+        if model.startswith('anthropic/'):
+            return True
+        return False
+
+    # Filter rows by supported models and by api_key presence in valid_keys
+    filtered_rows: List[Dict[str, str]] = []
+    for row in rows:
+        if not _is_supported_model(row):
+            continue
+        api_key_name = (row.get('api_key') or '').strip()
+        # Include only if we have a validated key for this row
+        if api_key_name and api_key_name in valid_keys:
+            filtered_rows.append(row)
+
+    # Write out the filtered CSV to ~/.pdd/llm_model.csv preserving column order
     llm_model_file = pdd_dir / 'llm_model.csv'
-    llm_model_file.write_text('\n'.join(valid_lines) + '\n')
+    with llm_model_file.open('w', newline='') as f:
+        writer = csv.DictWriter(f, fieldnames=header_fields)
+        writer.writeheader()
+        for row in filtered_rows:
+            writer.writerow({k: row.get(k, '') for k in header_fields})
     saved_files.append(str(llm_model_file))
     
     # Update shell init file
     init_file_path = get_shell_init_file(shell)
     init_file = Path(init_file_path)
+    init_file_updated = None
     
     source_line = f'[ -f "{api_env_file}" ] && source "{api_env_file}"'
     if shell == 'fish':
         source_line = f'test -f "{api_env_file}"; and source "{api_env_file}"'
+    elif shell in ['csh', 'tcsh']:
+        source_line = f'if ( -f "{api_env_file}" ) source "{api_env_file}"'
+    elif shell == 'sh':
+        source_line = f'[ -f "{api_env_file}" ] && . "{api_env_file}"'
+    
+    # Ensure parent directory exists (important for fish shell)
+    init_file.parent.mkdir(parents=True, exist_ok=True)
     
     # Check if source line already exists
     if init_file.exists():
@@ -306,15 +405,17 @@ def save_configuration(valid_keys: Dict[str, str]) -> Tuple[List[str], bool]:
         if str(api_env_file) not in content:
             with init_file.open('a') as f:
                 f.write(f'\n# PDD API environment\n{source_line}\n')
+            init_file_updated = str(init_file)
     else:
         init_file.write_text(f'# PDD API environment\n{source_line}\n')
+        init_file_updated = str(init_file)
     
-    return saved_files, created_pdd_dir
+    return saved_files, created_pdd_dir, init_file_updated
 
 def create_sample_prompt():
     """Create the sample prompt file"""
-    prompt_file = Path('hello_you_python.prompt')
-    prompt_file.write_text(HELLO_PYTHON_TEMPLATE)
+    prompt_file = Path('success_python.prompt')
+    prompt_file.write_text(SUCCESS_PYTHON_TEMPLATE)
     return str(prompt_file)
 
 def show_menu(keys: Dict[str, Optional[str]], test_results: Dict[str, bool]) -> str:
@@ -325,7 +426,9 @@ def show_menu(keys: Dict[str, Optional[str]], test_results: Dict[str, bool]) ->
     
     # Show current status
     print_colored("Current API Key Status:", WHITE, bold=True)
-    for key_name in ['OPENAI_API_KEY', 'GOOGLE_API_KEY']:
+    # Get the actual key names from discovered keys
+    key_names = list(keys.keys())
+    for key_name in key_names:
         key_value = keys.get(key_name)
         if key_value:
             status = f"{CHECK_MARK} Valid" if test_results.get(key_name) else f"{CROSS_MARK} Invalid"
@@ -356,7 +459,7 @@ def show_menu(keys: Dict[str, Optional[str]], test_results: Dict[str, bool]) ->
             print_colored("\n\nSetup cancelled.", YELLOW)
             sys.exit(0)
 
-def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_prompt_file: str, shell: str) -> str:
+def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_prompt_file: str, shell: str, valid_keys: Dict[str, str], init_file_updated: Optional[str] = None) -> str:
     """Create comprehensive exit summary"""
     summary_lines = [
         "\n\n\n\n\n",
@@ -364,22 +467,37 @@ def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_pr
         "PDD Setup Complete!",
         create_fat_divider(),
         "",
-        "Files created and configured:",
+        "API Keys Configured:",
         ""
     ]
     
+    # Add configured API keys information
+    if valid_keys:
+        for key_name, key_value in valid_keys.items():
+            # Show just the first and last few characters for security
+            masked_key = f"{key_value[:8]}...{key_value[-4:]}" if len(key_value) > 12 else "***"
+            summary_lines.append(f"  {key_name}: {masked_key}")
+        summary_lines.extend(["", "Files created and configured:", ""])
+    else:
+        summary_lines.extend(["  None", "", "Files created and configured:", ""])
+    
     # File descriptions with alignment
     file_descriptions = []
     if created_pdd_dir:
         file_descriptions.append(("~/.pdd/", "PDD configuration directory"))
     
     for file_path in saved_files:
-        if 'api-env' in file_path:
-            file_descriptions.append((file_path, "API environment variables"))
+        if 'api-env.' in file_path:
+            file_descriptions.append((file_path, f"API environment variables ({shell} shell)"))
         elif 'llm_model.csv' in file_path:
             file_descriptions.append((file_path, "LLM model configuration"))
     
     file_descriptions.append((sample_prompt_file, "Sample prompt for testing"))
+    
+    # Add shell init file if it was updated
+    if init_file_updated:
+        file_descriptions.append((init_file_updated, f"Shell startup file (updated to source API environment)"))
+    
     file_descriptions.append(("PDD-SETUP-SUMMARY.txt", "This summary"))
     
     # Find max file path length for alignment
@@ -397,10 +515,11 @@ def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_pr
         f"1. Reload your shell environment:"
     ])
     
-    # Shell-specific source command
-    api_env_path = f"{Path.home()}/.pdd/api-env"
-    if shell == 'fish':
-        source_cmd = f"source {api_env_path}"
+    # Shell-specific source command for manual reloading
+    api_env_path = f"{Path.home()}/.pdd/api-env.{shell}"
+    # Use dot command for sh shell, source for others
+    if shell == 'sh':
+        source_cmd = f". {api_env_path}"
     else:
         source_cmd = f"source {api_env_path}"
     
@@ -408,7 +527,7 @@ def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_pr
         f"   {source_cmd}",
         "",
         f"2. Generate code from the sample prompt:",
-        f"   pdd generate hello_you_python.prompt",
+        f"   pdd generate success_python.prompt",
         "",
         create_divider(),
         "",
@@ -428,7 +547,8 @@ def create_exit_summary(saved_files: List[str], created_pdd_dir: bool, sample_pr
         "",
         f"{BULLET} As you get comfortable, learn configuration settings, including the .pddrc file, PDD_GENERATE_OUTPUT_PATH, and PDD_TEST_OUTPUT_PATH",
         f"{BULLET} For larger projects, use Makefiles and/or 'pdd sync'",
-        f"{BULLET} For ongoing substantial projects, learn about llm_model.csv to optimize model cost, latency, and output quality",
+        f"{BULLET} For ongoing substantial projects, learn about llm_model.csv and the --strength,",
+        f"  --temperature, and --time options to optimize model cost, latency, and output quality",
         "",
         f"{BULLET} Use 'pdd --help' to explore all available commands",
         "",
@@ -478,12 +598,12 @@ def main():
             print_colored(f"{create_divider()}", CYAN)
             
             try:
-                saved_files, created_pdd_dir = save_configuration(valid_keys)
+                saved_files, created_pdd_dir, init_file_updated = save_configuration(valid_keys)
                 sample_prompt_file = create_sample_prompt()
                 shell = detect_shell()
                 
                 # Create and display summary
-                summary = create_exit_summary(saved_files, created_pdd_dir, sample_prompt_file, shell)
+                summary = create_exit_summary(saved_files, created_pdd_dir, sample_prompt_file, shell, valid_keys, init_file_updated)
                 
                 # Write summary to file
                 summary_file = Path('PDD-SETUP-SUMMARY.txt')
@@ -498,7 +618,11 @@ def main():
                         print_colored(line, YELLOW, bold=True)
                     elif line == create_divider():
                         print_colored(line, CYAN)
-                    elif line.startswith("QUICK START:") or line.startswith("LEARN MORE:") or line.startswith("TIPS:"):
+                    elif line.startswith("API Keys Configured:") or line.startswith("Files created and configured:"):
+                        print_colored(line, CYAN, bold=True)
+                    elif line.startswith("QUICK START:"):
+                        print_colored(line, YELLOW, bold=True)
+                    elif line.startswith("LEARN MORE:") or line.startswith("TIPS:"):
                         print_colored(line, CYAN, bold=True)
                     elif "IMPORTANT:" in line or "Problems?" in line:
                         print_colored(line, YELLOW, bold=True)

{pdd_cli-0.0.58.dist-info → pdd_cli-0.0.60.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pdd-cli
-Version: 0.0.58
+Version: 0.0.60
 Summary: PDD (Prompt-Driven Development) Command Line Interface
 Author: Greg Tanaka
 Author-email: glt@alumni.caltech.edu
@@ -44,6 +44,8 @@ Requires-Dist: openai>=1.99.5
 Provides-Extra: dev
 Requires-Dist: commitizen; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: pytest-testmon; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
 Requires-Dist: pytest-mock; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: z3-solver; extra == "dev"
@@ -51,7 +53,7 @@ Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
 Dynamic: license-file
 
-.. image:: https://img.shields.io/badge/pdd--cli-v0.0.58-blue
+.. image:: https://img.shields.io/badge/pdd--cli-v0.0.60-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
@@ -128,7 +130,7 @@ After installation, verify:
 
    pdd --version
 
-You'll see the current PDD version (e.g., 0.0.58).
+You'll see the current PDD version (e.g., 0.0.60).
 
 Getting Started with Examples
 -----------------------------

{pdd_cli-0.0.58.dist-info → pdd_cli-0.0.60.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pdd/__init__.py,sha256=PcN6Adk2Ea2dffgFJe0eyB538DRmYKrwUXjuu7IrSPQ,633
+pdd/__init__.py,sha256=DkLNT664TDBz1q2bKJ1z7tY7zRdVTF6dTex1j5DokZU,633
 pdd/auto_deps_main.py,sha256=cpP3bbzVL3jomrGinpzTxzIDIC8tmDDYOwUAC1TKRaw,3970
 pdd/auto_include.py,sha256=OJcdcwTwJNqHPHKG9P4m9Ij-PiLex0EbuwJP0uiQi_Y,7484
 pdd/auto_update.py,sha256=w6jzTnMiYRNpwQHQxWNiIAwQ0d6xh1iOB3xgDsabWtc,5236
@@ -6,7 +6,7 @@ pdd/bug_main.py,sha256=EtaGTuucQ7VgqOhyg4o6GFG7_QtTsDPTrRdGJWT648M,4841
 pdd/bug_to_unit_test.py,sha256=BoQqNyKQpBQDW8-JwBH_RX4RHRSiU8Kk3EplFrkECt0,6665
 pdd/change.py,sha256=Hg_x0pa370-e6oDiczaTgFAy3Am9ReCPkqFrvqv4U38,6114
 pdd/change_main.py,sha256=04VHiO_D-jlfeRn6rrVH7ZTA5agXPoJGm1StGI8--XY,27804
-pdd/cli.py,sha256=kLmJZ-2ETIbbpclwM7cQuaVfAiUNTBxJ9RtB0CsF2h0,55355
+pdd/cli.py,sha256=c5Gco_Ra1ZCZf1MtxrNZdySDhZqICHBQMSH5VBqVPEw,55162
 pdd/cmd_test_main.py,sha256=M-i5x26ORXurt_pu8x1sgLAyVIItbuRThiux4wBg3Ls,7768
 pdd/code_generator.py,sha256=AxMRZKGIlLh9xWdn2FA6b3zSoZ-5TIZNIAzqjFboAQs,4718
 pdd/code_generator_main.py,sha256=UtoskalEPpMAvCO-zd6xmr1lbQqSWQ7BvYgNJCybqok,35151
@@ -45,9 +45,9 @@ pdd/llm_invoke.py,sha256=US8J9oZZdVgLIjcmOI-YyHu6z2un43ZN811IcQBmAIA,96728
 pdd/load_prompt_template.py,sha256=stt42Og0PzXy0N_bsaivk5e2l5z_BnHiXIJZM14oVWw,2673
 pdd/logo_animation.py,sha256=n6HJWzuFze2csAAW2-zbxfjvWFYRI4hIdwVBtHBOkj4,20782
 pdd/mcp_config.json,sha256=D3ctWHlShvltbtH37zbYb6smVE0V80_lGjDKDIqsSBE,124
-pdd/pdd_completion.fish,sha256=K4qT52-9LG6apKVH0Bgfqk1Kqfketb47ge_6mcacVFE,12089
-pdd/pdd_completion.sh,sha256=DI2gJc7IxqjgKyEgjmoi8ev3f9Gs3LJrRgSQ-u2T6ms,5844
-pdd/pdd_completion.zsh,sha256=f4BxXrLZUEcFeIUmXWdnFOTnTpJBFZ8KGpnIdjDdKFo,15353
+pdd/pdd_completion.fish,sha256=pfEyBWzFp3matC_SW2BTDNaBJybYKUEi3YpPE-D6AKU,13776
+pdd/pdd_completion.sh,sha256=xgx-g6aeCCrlh6loeLyJN5jCsX15YXrWyT1U499p3C0,6490
+pdd/pdd_completion.zsh,sha256=V9-V8jqw3osjlXNOvjYMJf0E771-_EQe-Cboo1xzPvY,17090
 pdd/postprocess.py,sha256=mNw3iSDxE-eTYo3QwJCj_EmdEnnB5ysUN62YPapC_IM,4433
 pdd/postprocess_0.py,sha256=OW17GyCFLYErCyWh2tL4syuho3q2yFf2wyekQ4BLdPM,2168
 pdd/preprocess.py,sha256=75-J1smdi1Uq7gRQRLtVdkIfwltkeIvIZE-TkxxxCz0,12326
@@ -55,6 +55,7 @@ pdd/preprocess_main.py,sha256=WGhOB9qEu7MmFoyXNml_AmqGii73LJWngx4kTlZ526k,3262
 pdd/process_csv_change.py,sha256=ckNqVPRooWVyIvmqjdEgo2PDLnpoQ6Taa2dUaWGRlzU,27926
 pdd/pytest_output.py,sha256=IrRKYneW_F6zv9WaJwKFGnOBLFBFjk1CnhO_EVAjb9E,6612
 pdd/python_env_detector.py,sha256=y-QESoPNiKaD821uz8okX-9qA-oqvH9cQHY2_MwFHzU,5194
+pdd/setup_tool.py,sha256=h3MW7yr5tSlreq3u26aUfQqFTneQ-wNCoq0vwIYxqMY,24087
 pdd/split.py,sha256=9lWrh-JOjOpxRp4-s1VL7bqJMVWlsmY5LxONT7sYM8A,5288
 pdd/split_main.py,sha256=52rcZoeS_wpYRiqbqMUgr_hUY7GS62otwzDfuAGi6YA,4845
 pdd/summarize_directory.py,sha256=BR3-yGcmUdDT26vWLGokBo6mAZzaT7PzoY_qZriH3cc,10061
@@ -110,10 +111,9 @@ pdd/prompts/unfinished_prompt_LLM.prompt,sha256=vud_G9PlVv9Ig64uBC-hPEVFRk5lwpc8
 pdd/prompts/update_prompt_LLM.prompt,sha256=prIc8uLp2jqnLTHt6JvWDZGanPZipivhhYeXe0lVaYw,1328
 pdd/prompts/xml_convertor_LLM.prompt,sha256=YGRGXJeg6EhM9690f-SKqQrKqSJjLFD51UrPOlO0Frg,2786
 pdd/templates/architecture/architecture_json.prompt,sha256=uSNSsKTL-cuMMhi5a4GSpC94DKkOFAlXh7R0CUlo-hg,8126
-pdd_cli-0.0.58.data/data/utils/pdd-setup.py,sha256=sQ-HwwSN-ER4xz77EZqfxH8W0gG-kDtqj7LaR7SomA4,18955
-pdd_cli-0.0.58.dist-info/licenses/LICENSE,sha256=kvTJnnxPVTYlGKSY4ZN1kzdmJ0lxRdNWxgupaB27zsU,1066
-pdd_cli-0.0.58.dist-info/METADATA,sha256=TIrlo_oc6NbiHyi_WqFxqsRSOXrhEkYcuFoN5eCaD3s,12597
-pdd_cli-0.0.58.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pdd_cli-0.0.58.dist-info/entry_points.txt,sha256=Kr8HtNVb8uHZtQJNH4DnF8j7WNgWQbb7_Pw5hECSR-I,36
-pdd_cli-0.0.58.dist-info/top_level.txt,sha256=xjnhIACeMcMeDfVNREgQZl4EbTni2T11QkL5r7E-sbE,4
-pdd_cli-0.0.58.dist-info/RECORD,,
+pdd_cli-0.0.60.dist-info/licenses/LICENSE,sha256=kvTJnnxPVTYlGKSY4ZN1kzdmJ0lxRdNWxgupaB27zsU,1066
+pdd_cli-0.0.60.dist-info/METADATA,sha256=4vSMytmX5LPnADjAlOVkZPCSf--g5FWbATtu9zo7Qqs,12687
+pdd_cli-0.0.60.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pdd_cli-0.0.60.dist-info/entry_points.txt,sha256=Kr8HtNVb8uHZtQJNH4DnF8j7WNgWQbb7_Pw5hECSR-I,36
+pdd_cli-0.0.60.dist-info/top_level.txt,sha256=xjnhIACeMcMeDfVNREgQZl4EbTni2T11QkL5r7E-sbE,4
+pdd_cli-0.0.60.dist-info/RECORD,,