pdd-cli 0.0.48__py3-none-any.whl → 0.0.50__py3-none-any.whl

pdd/__init__.py

@@ -1,20 +1,20 @@
 """PDD - Prompt Driven Development"""
 
-__version__ = "0.0.48"
+__version__ = "0.0.50"
 
 # 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.
-EXTRACTION_STRENGTH = 0.9
+EXTRACTION_STRENGTH = 0.3
 
-DEFAULT_STRENGTH = 0.9
+DEFAULT_STRENGTH = 0.75
 
 DEFAULT_TEMPERATURE = 0.0
 
 DEFAULT_TIME = 0.25
 
 # Define constants used across the package
-DEFAULT_LLM_MODEL = "gpt-4.1-nano"
+DEFAULT_LLM_MODEL = "gpt-5-mini"
 # When going to production, set the following constants:
 # REACT_APP_FIREBASE_API_KEY
 # GITHUB_CLIENT_ID

pdd/bug_to_unit_test.py

@@ -108,6 +108,7 @@ def bug_to_unit_test(  # pylint: disable=too-many-arguments, too-many-locals
             strength=0.89,
             temperature=temperature,
             time=time,
+            language=language,
             verbose=False,
         )
 
@@ -121,6 +122,7 @@ def bug_to_unit_test(  # pylint: disable=too-many-arguments, too-many-locals
                 strength=strength,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=True,
             )
             total_cost += continued_cost

pdd/cli.py

@@ -1001,7 +1001,14 @@ def auto_deps(
     auto_deps_csv_path: Optional[str],
     force_scan: bool,
 ) -> Optional[Tuple[str, float, str]]: # Modified return type
-    """Analyze prompt and insert dependencies from a directory."""
+    """Analyze a prompt and insert dependencies from a directory or glob.
+
+    DIRECTORY_PATH accepts either a directory path or a glob pattern and is
+    expanded recursively when you use patterns like `**/*.py`. Examples:
+      - examples/**/*.py
+      - context/*_example.py
+      - examples/*
+    """
     quiet = ctx.obj.get("quiet", False)
     command_name = "auto-deps"
     try:

pdd/code_generator.py

@@ -83,6 +83,7 @@ def code_generator(
             strength=0.5,
             temperature=0.0,
             time=time,
+            language=language,
             verbose=verbose
         )
         total_cost += check_cost
@@ -97,6 +98,7 @@ def code_generator(
                 strength=strength,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=verbose
             )
             total_cost += continue_cost
@@ -126,4 +128,4 @@ def code_generator(
     except Exception as e:
         if verbose:
             console.print(f"[bold red]Unexpected Error: {str(e)}[/bold red]")
-        raise
+        raise

pdd/context_generator.py

@@ -95,6 +95,7 @@ def context_generator(
                 strength=0.5,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=verbose
             )
         except Exception as e:
@@ -112,6 +113,7 @@ def context_generator(
                 strength=strength,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=verbose
             )
             total_cost = llm_response['cost'] + unfinished_cost + continue_cost
@@ -149,4 +151,4 @@ if __name__ == "__main__":
         print("[bold green]Generated Example Code:[/bold green]")
         print(example_code)
         print(f"[bold blue]Total Cost: ${total_cost:.6f}[/bold blue]")
-        print(f"[bold blue]Model Name: {model_name}[/bold blue]")
+        print(f"[bold blue]Model Name: {model_name}[/bold blue]")

pdd/continue_generation.py

@@ -1,4 +1,5 @@
-from typing import Tuple
+from typing import Tuple, Optional
+import logging
 from rich.console import Console
 from rich.syntax import Syntax
 from pydantic import BaseModel, Field
@@ -9,6 +10,10 @@ from .unfinished_prompt import unfinished_prompt
 from . import EXTRACTION_STRENGTH, DEFAULT_TIME
 
 console = Console()
+logger = logging.getLogger(__name__)
+
+# Maximum number of generation loops to prevent infinite loops
+MAX_GENERATION_LOOPS = 20
 
 class TrimResultsStartOutput(BaseModel):
     explanation: str = Field(description="The explanation of how you determined what to cut out")
@@ -24,6 +29,7 @@ def continue_generation(
     strength: float,
     temperature: float,
     time: float = DEFAULT_TIME,
+    language: Optional[str] = None,
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -84,10 +90,16 @@ def continue_generation(
         code_block = trim_start_response['result'].code_block
 
         # Step 4: Continue generation loop
-        while True:
+        while loop_count < MAX_GENERATION_LOOPS:
             loop_count += 1
             if verbose:
                 console.print(f"[cyan]Generation loop {loop_count}[/cyan]")
+            
+            # Check for maximum loops reached
+            if loop_count >= MAX_GENERATION_LOOPS:
+                logger.warning(f"Reached maximum generation loops ({MAX_GENERATION_LOOPS}), terminating")
+                console.print(f"[yellow]Warning: Reached maximum generation loops ({MAX_GENERATION_LOOPS}), terminating[/yellow]")
+                break
 
             # Generate continuation
             continue_response = llm_invoke(
@@ -106,19 +118,47 @@ def continue_generation(
             model_name = continue_response['model_name']
             continue_result = continue_response['result']
 
-            # Check if generation is complete
-            last_chunk = code_block[-600:] if len(code_block) > 600 else code_block
-            _, is_finished, check_cost, _ = unfinished_prompt(
+            if verbose:
+                try:
+                    preview = (continue_result[:160] + '...') if isinstance(continue_result, str) and len(continue_result) > 160 else continue_result
+                except Exception:
+                    preview = "<non-str>"
+                console.print(f"[blue]Continue model:[/blue] {model_name}")
+                console.print(f"[blue]Continue preview:[/blue] {preview!r}")
+
+            # If the model produced no continuation, avoid an endless loop
+            if not isinstance(continue_result, str) or not continue_result.strip():
+                logger.warning("Empty continuation received; stopping to avoid loop.")
+                break
+
+            # Build prospective new block and check completeness on the updated tail
+            new_code_block = code_block + continue_result
+            last_chunk = new_code_block[-600:] if len(new_code_block) > 600 else new_code_block
+            reasoning, is_finished, check_cost, check_model = unfinished_prompt(
                 prompt_text=last_chunk,
                 strength=0.5,
                 temperature=0,
                 time=time,
+                language=language,
                 verbose=verbose
             )
             total_cost += check_cost
 
+            if verbose:
+                console.print(f"[magenta]Tail length:[/magenta] {len(last_chunk)}")
+                # Show a safe, shortened representation of the tail
+                try:
+                    tail_preview = (last_chunk[-200:] if len(last_chunk) > 200 else last_chunk)
+                except Exception:
+                    tail_preview = "<unprintable tail>"
+                console.print(f"[magenta]Tail preview (last 200 chars):[/magenta]\n{tail_preview}")
+                console.print(f"[magenta]Unfinished check model:[/magenta] {check_model}")
+                console.print(f"[magenta]is_finished:[/magenta] {is_finished}")
+                console.print(f"[magenta]Reasoning:[/magenta] {reasoning}")
+
             if not is_finished:
-                code_block += continue_result
+                code_block = new_code_block
+                # Continue to next iteration
             else:
                 # Trim and append final continuation
                 trim_response = llm_invoke(
@@ -146,4 +186,4 @@ def continue_generation(
 
     except Exception as e:
         console.print(f"[bold red]Error in continue_generation: {str(e)}[/bold red]")
-        raise
+        raise

pdd/data/llm_model.csv

@@ -1,18 +1,17 @@
 provider,model,input,output,coding_arena_elo,base_url,api_key,max_reasoning_tokens,structured_output,reasoning_type
-OpenAI,gpt-4.1-nano,0.1,0.4,1249,,OPENAI_API_KEY,0,True,none
-xai,xai/grok-3-beta,3.0,15.0,1332,https://api.x.ai/v1,XAI_API_KEY,0,False,none
-Anthropic,claude-3-5-haiku-20241022,.8,4,1261,,ANTHROPIC_API_KEY,0,True,none
-OpenAI,deepseek/deepseek-chat,.27,1.1,1353,https://api.deepseek.com/beta,DEEPSEEK_API_KEY,0,False,none
-Google,vertex_ai/gemini-2.5-flash,0.15,0.6,1330,,VERTEX_CREDENTIALS,0,True,effort
+OpenAI,gpt-5-nano,0.05,0.4,1249,,OPENAI_API_KEY,0,True,none
+Google,vertex_ai/gemini-2.5-flash,0.15,0.6,1290,,VERTEX_CREDENTIALS,0,True,effort
 Google,gemini-2.5-pro,1.25,10.0,1360,,GOOGLE_API_KEY,0,True,none
-Anthropic,claude-sonnet-4-20250514,3.0,15.0,1340,,ANTHROPIC_API_KEY,64000,True,budget
-Google,vertex_ai/gemini-2.5-pro,1.25,10.0,1361,,VERTEX_CREDENTIALS,0,True,none
-OpenAI,o4-mini,1.1,4.4,1333,,OPENAI_API_KEY,0,True,effort
-OpenAI,o3,10.0,40.0,1389,,OPENAI_API_KEY,0,True,effort
-OpenAI,gpt-4.1,2.0,8.0,1335,,OPENAI_API_KEY,0,True,none
-OpenAI,deepseek/deepseek-reasoner,0.55,2.19,1337,https://api.deepseek.com/beta,DEEPSEEK_API_KEY,0,False,none
-Fireworks,fireworks_ai/accounts/fireworks/models/deepseek-r1,3.0,8.0,1338,,FIREWORKS_API_KEY,0,False,none
-OpenAI,chatgpt-4o-latest,5,15,1369,,OPENAI_API_KEY,0,False,none
-Anthropic,bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0,3.0,15.0,1339,,,64000,True,budget
-OpenAI,azure/o4-mini,1.1,4.4,1334,,OPENAI_API_KEY,0,True,effort
-OpenAI,openai/mlx-community/Qwen3-30B-A3B-4bit,0,0,1293,http://localhost:8080,,0,False,none
+Google,vertex_ai/claude-sonnet-4,3.0,15.0,1359,,VERTEX_CREDENTIALS,64000,True,budget
+Google,vertex_ai/gemini-2.5-pro,1.25,10.0,1405,,VERTEX_CREDENTIALS,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
+Google,vertex_ai/deepseek-ai/deepseek-r1-0528-maas,0.55,2.19,1391,,VERTEX_CREDENTIALS,0,False,none
+Fireworks,fireworks_ai/accounts/fireworks/models/qwen3-coder-480b-a35b-instruct,3.0,8.0,1363,,FIREWORKS_API_KEY,0,False,none
+Google,vertex_ai/claude-opus-4-1,3.0,15.0,1426,,VERTEX_CREDENTIALS,64000,True,budget
+OpenAI,azure/o4-mini,1.1,4.4,1335,,OPENAI_API_KEY,0,True,effort
+OpenAI,openai/mlx-community/Qwen3-30B-A3B-4bit,0,0,1040,http://localhost:8080,,0,False,none
+OpenAI,lm_studio/openai-gpt-oss-120b-mlx-6,0.0001,0,1082,http://localhost:1234/v1,,0,True,none
+Fireworks,fireworks_ai/accounts/fireworks/models/glm-4p5,3.0,8.0,1364,,FIREWORKS_API_KEY,0,False,none
+OpenAI,groq/moonshotai/kimi-k2-instruct,1.0,3.0,1330,,GROQ_API_KEY,0,True,none

pdd/detect_change_main.py

@@ -6,7 +6,7 @@ from rich import print as rprint
 
 from .construct_paths import construct_paths
 from .detect_change import detect_change
-from . import DEFAULT_TIME
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 
 def detect_change_main(
     ctx: click.Context,
@@ -54,7 +54,7 @@ def detect_change_main(
         prompt_contents = [input_strings[f"prompt_file_{i}"] for i in range(len(prompt_files))]
 
         # Get model parameters from context
-        strength = ctx.obj.get('strength', 0.9)
+        strength = ctx.obj.get('strength', DEFAULT_STRENGTH)
         temperature = ctx.obj.get('temperature', 0)
         time_budget = ctx.obj.get('time', DEFAULT_TIME)
 

pdd/generate_test.py

@@ -98,6 +98,7 @@ def generate_test(
                 strength=strength,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=verbose
             )
         total_cost += check_cost
@@ -112,6 +113,7 @@ def generate_test(
                 strength=strength,
                 temperature=temperature,
                 time=time,
+                language=language,
                 verbose=verbose
             )
             total_cost += continue_cost
@@ -181,4 +183,4 @@ def _validate_inputs(
     if not isinstance(temperature, float):
         raise ValueError("Temperature must be a float")
     if not language or not isinstance(language, str):
-        raise ValueError("Language must be a non-empty string")
+        raise ValueError("Language must be a non-empty string")