pdd-cli 0.0.24__py3-none-any.whl → 0.0.25__py3-none-any.whl

pdd/__init__.py

@@ -1 +1,7 @@
-__version__ = "0.0.24"
+__version__ = "0.0.25"
+
+# 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.97
+
+DEFAULT_STRENGTH = 0.8

pdd/bug_main.py

@@ -44,13 +44,17 @@ def bug_main(
             "output": output,
             "language": language
         }
-        input_strings, output_file_paths, _ = construct_paths(
+        input_strings, output_file_paths, detected_language = construct_paths(
             input_file_paths=input_file_paths,
             force=ctx.obj.get('force', False),
             quiet=ctx.obj.get('quiet', False),
             command="bug",
             command_options=command_options
         )
+        
+        # Use the language detected by construct_paths if none was explicitly provided
+        if language is None:
+            language = detected_language
 
         # Load input files
         prompt_content = input_strings["prompt_file"]

pdd/bug_to_unit_test.py

@@ -2,6 +2,7 @@ from typing import Tuple, Optional
 from rich import print
 from rich.markdown import Markdown
 from rich.console import Console
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
 from .unfinished_prompt import unfinished_prompt
@@ -17,7 +18,7 @@ def bug_to_unit_test(
     prompt_used_to_generate_the_code: str,
     code_under_test: str,
     program_used_to_run_code_under_test: str,
-    strength: float = 0.97,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0.0,
     language: str = "python"
 ) -> Tuple[str, float, str]:
@@ -30,7 +31,7 @@ def bug_to_unit_test(
         prompt_used_to_generate_the_code (str): Original prompt used to generate the code
         code_under_test (str): Code to be tested
         program_used_to_run_code_under_test (str): Program used to run the code
-        strength (float, optional): Strength of the LLM model. Must be between 0 and 1. Defaults to 0.97.
+        strength (float, optional): Strength of the LLM model. Must be between 0 and 1. Defaults to DEFAULT_STRENGTH.
         temperature (float, optional): Temperature of the LLM model. Defaults to 0.0.
         language (str, optional): Programming language. Defaults to "python".
 
@@ -43,6 +44,11 @@ def bug_to_unit_test(
     # Validate strength parameter
     if not 0 <= strength <= 1:
         raise ValueError("Strength parameter must be between 0 and 1")
+        
+    # Ensure language parameter is not None or empty
+    if not language or not isinstance(language, str):
+        language = "python"  # Default fallback
+        console.print("[yellow]Warning: Invalid language parameter, defaulting to 'python'[/yellow]")
 
     total_cost = 0.0
     final_model_name = ""
@@ -62,7 +68,7 @@ def bug_to_unit_test(
             "desired_output": desired_output,
             "code_under_test": code_under_test,
             "program_used_to_run_code_under_test": program_used_to_run_code_under_test,
-            "language": language
+            "language": language if language and isinstance(language, str) else "python"
         }
 
         console.print("[bold blue]Generating unit test...[/bold blue]")
@@ -85,7 +91,7 @@ def bug_to_unit_test(
         
         reasoning, is_finished, unfinished_cost, unfinished_model = unfinished_prompt(
             prompt_text=last_600_chars,
-            strength=0.97,
+            strength=strength,
             temperature=temperature,
             verbose=False
         )
@@ -108,10 +114,15 @@ def bug_to_unit_test(
             result = response['result']
 
         # Post-process the result
+        # Double-check language is valid before passing to postprocess
+        if not language or not isinstance(language, str):
+            language = "python"  # Ensure language is valid
+            console.print("[yellow]Warning: Language value became invalid during processing, defaulting to 'python'[/yellow]")
+            
         final_code, postprocess_cost, postprocess_model = postprocess(
             result,
             language,
-            strength=0.97,
+            strength=EXTRACTION_STRENGTH,
             temperature=temperature,
             verbose=True
         )

pdd/change.py

@@ -6,6 +6,7 @@ from pydantic import BaseModel, Field
 from .preprocess import preprocess
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
+from . import EXTRACTION_STRENGTH
 
 console = Console()
 
@@ -86,7 +87,7 @@ def change(
         extract_response = llm_invoke(
             prompt=extract_prompt,
             input_json={"llm_output": change_response["result"]},
-            strength=0.97,  # Fixed strength as specified
+            strength=EXTRACTION_STRENGTH,  # Fixed strength for extraction
             temperature=temperature,
             verbose=verbose,
             output_pydantic=ExtractedPrompt