pdd-cli 0.0.38__py3-none-any.whl → 0.0.40__py3-none-any.whl

pdd/change.py

@@ -6,7 +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
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH, DEFAULT_TIME
 
 console = Console()
 
@@ -17,8 +17,10 @@ def change(
     input_prompt: str,
     input_code: str,
     change_prompt: str,
-    strength: float,
-    temperature: float,
+    strength: float = DEFAULT_STRENGTH,
+    temperature: float = 0.0,
+    time: float = DEFAULT_TIME,
+    budget: float = 5.0,  # Note: budget is in the spec but not used. Keeping for now.
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -30,26 +32,28 @@ def change(
         change_prompt (str): Instructions for modifying the input prompt
         strength (float): The strength parameter for the LLM model (0-1)
         temperature (float): The temperature parameter for the LLM model
-        verbose (bool): Whether to print detailed information
+        time (float): The time budget for LLM calls.
+        budget (float): The budget for the operation (currently unused directly by llm_invoke).
+        verbose (bool): Whether to print out detailed information.
 
     Returns:
         Tuple[str, float, str]: (modified prompt, total cost, model name)
     """
     try:
         # Step 1: Load prompt templates
-        change_llm_prompt = load_prompt_template("change_LLM")
-        extract_prompt = load_prompt_template("extract_prompt_change_LLM")
+        change_llm_prompt_template = load_prompt_template("change_LLM")
+        extract_prompt_template = load_prompt_template("extract_prompt_change_LLM")
 
-        if not all([change_llm_prompt, extract_prompt]):
+        if not all([change_llm_prompt_template, extract_prompt_template]):
             raise ValueError("Failed to load prompt templates")
 
         # Step 2: Preprocess the change_LLM prompt
-        processed_change_llm = preprocess(change_llm_prompt, recursive=False, double_curly_brackets=False)
-        processed_change_prompt = preprocess(change_prompt, recursive=False, double_curly_brackets=False)
+        processed_change_llm_template = preprocess(change_llm_prompt_template, recursive=False, double_curly_brackets=False)
+        processed_change_prompt_content = preprocess(change_prompt, recursive=False, double_curly_brackets=False)
 
         # Input validation
-        if not all([input_prompt, input_code, change_prompt]):
-            raise ValueError("Missing required input parameters")
+        if not all([input_prompt, input_code, processed_change_prompt_content]):
+            raise ValueError("Missing required input parameters after preprocessing")
         if not (0 <= strength <= 1):
             raise ValueError("Strength must be between 0 and 1")
 
@@ -61,14 +65,15 @@ def change(
             console.print(Panel("Running change prompt through LLM...", style="blue"))
 
         change_response = llm_invoke(
-            prompt=processed_change_llm,
+            prompt=processed_change_llm_template,
             input_json={
                 "input_prompt": input_prompt,
                 "input_code": input_code,
-                "change_prompt": processed_change_prompt
+                "change_prompt": processed_change_prompt_content
             },
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 
@@ -85,10 +90,11 @@ def change(
             console.print(Panel("Extracting modified prompt...", style="blue"))
 
         extract_response = llm_invoke(
-            prompt=extract_prompt,
+            prompt=extract_prompt_template,
             input_json={"llm_output": change_response["result"]},
-            strength=EXTRACTION_STRENGTH,  # Fixed strength for extraction
+            strength=EXTRACTION_STRENGTH,
             temperature=temperature,
+            time=time,
             verbose=verbose,
             output_pydantic=ExtractedPrompt
         )
@@ -110,23 +116,27 @@ def change(
         return modified_prompt, total_cost, final_model_name
 
     except Exception as e:
-        console.print(f"[red]Error in change function: {str(e)}[/red]")
+        # Conditionally print error if verbose
+        if verbose: 
+            console.print(f"[red]Error in change function: {str(e)}[/red]")
         raise
 
 def main():
     """Example usage of the change function"""
     try:
         # Example inputs
-        input_prompt = "Write a function that adds two numbers"
-        input_code = "def add(a, b):\n    return a + b"
-        change_prompt = "Make the function handle negative numbers explicitly"
+        input_prompt_content = "Write a function that adds two numbers"
+        input_code_content = "def add(a, b):\n    return a + b"
+        change_prompt_content = "Make the function handle negative numbers explicitly"
         
         modified_prompt, cost, model = change(
-            input_prompt=input_prompt,
-            input_code=input_code,
-            change_prompt=change_prompt,
+            input_prompt=input_prompt_content,
+            input_code=input_code_content,
+            change_prompt=change_prompt_content,
             strength=0.7,
             temperature=0.7,
+            time=DEFAULT_TIME,
+            budget=10.0,
             verbose=True
         )
 

pdd/change_main.py

@@ -10,7 +10,7 @@ from .construct_paths import construct_paths
 from .change import change as change_func
 from .process_csv_change import process_csv_change
 from .get_extension import get_extension
-from . import DEFAULT_STRENGTH  # Assuming DEFAULT_STRENGTH is defined in __init__.py
+from . import DEFAULT_STRENGTH, DEFAULT_TIME  # Added DEFAULT_TIME
 
 # Import Rich for pretty printing
 from rich import print as rprint
@@ -66,6 +66,7 @@ def change_main(
     quiet: bool = ctx.obj.get("quiet", False)
     strength: float = ctx.obj.get("strength", DEFAULT_STRENGTH)
     temperature: float = ctx.obj.get("temperature", 0.0)
+    time_budget: float = ctx.obj.get("time", DEFAULT_TIME) # Added time_budget
     # --- Get language and extension from context --- 
     # These are crucial for knowing the target code file types, especially in CSV mode
     target_language: str = ctx.obj.get("language", "") # Get from context
@@ -215,6 +216,7 @@ def change_main(
                     csv_file=change_prompt_file,
                     strength=strength,
                     temperature=temperature,
+                    time=time_budget, # Pass time_budget
                     code_directory=input_code, # Pass the directory path
                     language=csv_target_language,
                     extension=extension,
@@ -254,16 +256,18 @@ def change_main(
                  return msg, 0.0, ""
 
             try:
-                modified_prompt, total_cost, model_name = change_func(
-                    input_prompt=input_prompt_content,
-                    input_code=input_code_content,
-                    change_prompt=change_prompt_content,
-                    strength=strength,
-                    temperature=temperature,
-                    verbose=ctx.obj.get("verbose", False),
+                # Call the imported change function
+                result_message, total_cost, model_name = change_func(
+                    change_prompt_content,
+                    input_code_content,
+                    input_prompt_content,
+                    strength,
+                    temperature,
+                    time_budget, # Pass time_budget
+                    budget=budget,
+                    quiet=quiet
                 )
-                result_message = modified_prompt # Store the content for saving
-                success = True # Assume success if no exception
+                success = True  # Assume success if no exception
                 logger.info("Single prompt change successful.")
             except Exception as e:
                 msg = f"Error during prompt modification: {e}"

pdd/cli.py

@@ -11,7 +11,7 @@ from rich.theme import Theme
 from rich.markup import MarkupError, escape
 
 # --- Relative Imports for Internal Modules ---
-from . import DEFAULT_STRENGTH, __version__
+from . import DEFAULT_STRENGTH, __version__, DEFAULT_TIME
 from .auto_deps_main import auto_deps_main
 from .auto_update import auto_update
 from .bug_main import bug_main
@@ -89,6 +89,13 @@ def handle_error(e: Exception, command_name: str, quiet: bool):
     show_default=True,
     help="Set the temperature of the AI model.",
 )
+@click.option(
+    "--time",
+    type=click.FloatRange(0.0, 1.0),
+    default=None,
+    show_default=True,
+    help="Controls reasoning allocation for LLMs (0.0-1.0). Uses DEFAULT_TIME if None.",
+)
 @click.option(
     "--verbose",
     is_flag=True,
@@ -131,6 +138,7 @@ def cli(
     output_cost: Optional[str],
     review_examples: bool,
     local: bool,
+    time: Optional[float], # Type hint is Optional[float]
 ):
     """
     Main entry point for the PDD CLI. Handles global options and initializes context.
@@ -148,6 +156,8 @@ def cli(
     ctx.obj["output_cost"] = output_cost
     ctx.obj["review_examples"] = review_examples
     ctx.obj["local"] = local
+    # Use DEFAULT_TIME if time is not provided
+    ctx.obj["time"] = time if time is not None else DEFAULT_TIME
 
     # Suppress verbose if quiet is enabled
     if quiet:

pdd/cmd_test_main.py

@@ -49,6 +49,7 @@ def cmd_test_main(
     verbose = ctx.obj["verbose"]
     strength = ctx.obj["strength"]
     temperature = ctx.obj["temperature"]
+    time = ctx.obj.get("time")
 
     if verbose:
         print(f"[bold blue]Prompt file:[/bold blue] {prompt_file}")
@@ -97,9 +98,11 @@ def cmd_test_main(
             unit_test, total_cost, model_name = generate_test(
                 input_strings["prompt_file"],
                 input_strings["code_file"],
-                strength,
-                temperature,
-                language,
+                strength=strength,
+                temperature=temperature,
+                time=time,
+                language=language,
+                verbose=verbose
             )
         except Exception as e:
             print(f"[bold red]Error generating tests: {e}[/bold red]")
@@ -121,6 +124,7 @@ def cmd_test_main(
                 language=language,
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose,
             )
         except Exception as e:

pdd/code_generator.py

@@ -1,4 +1,4 @@
-from typing import Tuple
+from typing import Tuple, Optional
 from rich.console import Console
 from . import EXTRACTION_STRENGTH
 from .preprocess import preprocess
@@ -14,6 +14,7 @@ def code_generator(
     language: str,
     strength: float,
     temperature: float = 0.0,
+    time: Optional[float] = None,
     verbose: bool = False,
     preprocess_prompt: bool = True,
 ) -> Tuple[str, float, str]:
@@ -25,6 +26,7 @@ def code_generator(
         language (str): The target programming language
         strength (float): The strength of the LLM model (0 to 1)
         temperature (float, optional): The temperature for the LLM model. Defaults to 0.0
+        time (Optional[float], optional): The time for the LLM model. Defaults to None
         verbose (bool, optional): Whether to print detailed information. Defaults to False
 
     Returns:
@@ -65,6 +67,7 @@ def code_generator(
             input_json={},
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         initial_output = response['result']
@@ -79,6 +82,7 @@ def code_generator(
             prompt_text=last_chunk,
             strength=0.5,
             temperature=0.0,
+            time=time,
             verbose=verbose
         )
         total_cost += check_cost
@@ -92,6 +96,7 @@ def code_generator(
                 llm_output=initial_output,
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose
             )
             total_cost += continue_cost
@@ -107,6 +112,7 @@ def code_generator(
             language=language,
             strength=EXTRACTION_STRENGTH,
             temperature=0.0,
+            time=time,
             verbose=verbose
         )
         total_cost += postprocess_cost

pdd/code_generator_main.py

@@ -410,11 +410,17 @@ def code_generator_main(
             if current_execution_is_local:
                 if verbose: console.print("Executing code generator locally...")
                 generated_code_content, total_cost, model_name = local_code_generator_func(
-                    prompt=prompt_content, language=language, strength=strength,
-                    temperature=temperature, verbose=verbose
+                    prompt=prompt_content,
+                    language=language,
+                    strength=strength,
+                    temperature=temperature,
+                    time=time_budget,
+                    verbose=verbose,
+                    preprocess_prompt=True
                 )
+                was_incremental_operation = False
                 if verbose:
-                    console.print(Panel(f"Local generation successful. Model: {model_name}, Cost: ${total_cost:.6f}", title="[green]Local Success[/green]", expand=False))
+                    console.print(Panel(f"Full generation successful. Model: {model_name}, Cost: ${total_cost:.6f}", title="[green]Local Success[/green]", expand=False))
         
         if generated_code_content is not None:
             if output_path:

pdd/conflicts_in_prompts.py

@@ -4,7 +4,7 @@ from rich import print as rprint
 from rich.markdown import Markdown
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
-from . import EXTRACTION_STRENGTH
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH, DEFAULT_TIME
 
 class ConflictChange(BaseModel):
     prompt_name: str = Field(description="Name of the prompt that needs to be changed")
@@ -16,8 +16,9 @@ class ConflictResponse(BaseModel):
 def conflicts_in_prompts(
     prompt1: str,
     prompt2: str,
-    strength: float = 0.5,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[List[dict], float, str]:
     """
@@ -28,6 +29,7 @@ def conflicts_in_prompts(
         prompt2 (str): Second prompt to compare
         strength (float): Model strength (0-1)
         temperature (float): Model temperature (0-1)
+        time (float): Time budget for LLM calls.
         verbose (bool): Whether to print detailed information
 
     Returns:
@@ -66,6 +68,7 @@ def conflicts_in_prompts(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 
@@ -88,6 +91,7 @@ def conflicts_in_prompts(
             input_json=extract_input,
             strength=EXTRACTION_STRENGTH,
             temperature=temperature,
+            time=time,
             output_pydantic=ConflictResponse,
             verbose=verbose
         )
@@ -125,6 +129,7 @@ def main():
             prompt2=prompt2,
             strength=0.7,
             temperature=0,
+            time=DEFAULT_TIME,
             verbose=True
         )
 

pdd/conflicts_main.py

@@ -6,6 +6,7 @@ from rich import print as rprint
 
 from .construct_paths import construct_paths
 from .conflicts_in_prompts import conflicts_in_prompts
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 
 def conflicts_main(ctx: click.Context, prompt1: str, prompt2: str, output: Optional[str], verbose: bool = False) -> Tuple[List[Dict], float, str]:
     """
@@ -40,13 +41,16 @@ def conflicts_main(ctx: click.Context, prompt1: str, prompt2: str, output: Optio
         prompt2_content = input_strings["prompt2"]
 
         # Analyze conflicts
-        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)
         conflicts, total_cost, model_name = conflicts_in_prompts(
             prompt1_content,
             prompt2_content,
             strength,
-            temperature
+            temperature,
+            time_budget,
+            verbose
         )
 
         # Replace prompt1 and prompt2 with actual file paths

pdd/context_generator.py

@@ -5,9 +5,18 @@ from .llm_invoke import llm_invoke
 from .unfinished_prompt import unfinished_prompt
 from .continue_generation import continue_generation
 from .postprocess import postprocess
-from . import EXTRACTION_STRENGTH
-
-def context_generator(code_module: str, prompt: str, language: str = "python", strength: float = 0.5, temperature: float = 0, verbose: bool = False) -> tuple:
+from . import EXTRACTION_STRENGTH, DEFAULT_TIME
+from typing import Optional
+
+def context_generator(
+    code_module: str,
+    prompt: str,
+    language: str = "python",
+    strength: float = 0.5,
+    temperature: float = 0,
+    time: Optional[float] = DEFAULT_TIME,
+    verbose: bool = False,
+) -> tuple:
     """
     Generates a concise example on how to use a given code module properly.
 
@@ -17,6 +26,7 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
         language (str): The language of the code module. Default is "python".
         strength (float): The strength of the LLM model to use. Default is 0.5. Range is between 0 and 1.
         temperature (float): The temperature of the LLM model to use. Default is 0. Range is between 0 and 1.
+        time (Optional[float], optional): Time allocation for the LLM. Defaults to DEFAULT_TIME.
         verbose (bool): Whether to print out the details of the function. Default is False.
 
     Returns:
@@ -45,6 +55,9 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
         return None, 0.0, None
 
     try:
+        if verbose:
+            print(f"[bold blue]Generating example for language: {language}[/bold blue]")
+
         # Step 1: Load and preprocess the 'example_generator_LLM' prompt template
         prompt_template = load_prompt_template("example_generator_LLM")
         if not prompt_template:
@@ -70,6 +83,7 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
             },
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 
@@ -80,6 +94,7 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
                 prompt_text=last_600_chars,
                 strength=0.5,
                 temperature=temperature,
+                time=time,
                 verbose=verbose
             )
         except Exception as e:
@@ -96,6 +111,7 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
                 llm_output=llm_response['result'],
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose
             )
             total_cost = llm_response['cost'] + unfinished_cost + continue_cost
@@ -113,6 +129,7 @@ def context_generator(code_module: str, prompt: str, language: str = "python", s
             language=language,
             strength=EXTRACTION_STRENGTH,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         total_cost += postprocess_cost

pdd/context_generator_main.py

@@ -40,12 +40,14 @@ def context_generator_main(ctx: click.Context, prompt_file: str, code_file: str,
         # Generate example code
         strength = ctx.obj.get('strength', 0.5)
         temperature = ctx.obj.get('temperature', 0)
+        time = ctx.obj.get('time')
         example_code, total_cost, model_name = context_generator(
             language=language,
             code_module=code_content,
             prompt=prompt_content,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=ctx.obj.get('verbose', False)
         )
 

pdd/continue_generation.py

@@ -6,7 +6,7 @@ from .load_prompt_template import load_prompt_template
 from .preprocess import preprocess
 from .llm_invoke import llm_invoke
 from .unfinished_prompt import unfinished_prompt
-from . import EXTRACTION_STRENGTH
+from . import EXTRACTION_STRENGTH, DEFAULT_TIME
 
 console = Console()
 
@@ -23,6 +23,7 @@ def continue_generation(
     llm_output: str,
     strength: float,
     temperature: float,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -33,6 +34,7 @@ def continue_generation(
         llm_output (str): Current output from the LLM to be checked and continued.
         strength (float): Strength parameter for the LLM model (0-1).
         temperature (float): Temperature parameter for the LLM model (0-1).
+        time (float): Time budget for LLM calls.
         verbose (bool): Whether to print detailed information.
         
     Returns:
@@ -72,8 +74,9 @@ def continue_generation(
         trim_start_response = llm_invoke(
             prompt=processed_prompts['trim_start'],
             input_json={"LLM_OUTPUT": llm_output},
-            strength=0.75,
+            strength=EXTRACTION_STRENGTH,
             temperature=0,
+            time=time,
             output_pydantic=TrimResultsStartOutput,
             verbose=verbose
         )
@@ -95,6 +98,7 @@ def continue_generation(
                 },
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose
             )
             
@@ -108,6 +112,7 @@ def continue_generation(
                 prompt_text=last_chunk,
                 strength=0.5,
                 temperature=0,
+                time=time,
                 verbose=verbose
             )
             total_cost += check_cost
@@ -124,6 +129,7 @@ def continue_generation(
                     },
                     strength=EXTRACTION_STRENGTH,
                     temperature=0,
+                    time=time,
                     output_pydantic=TrimResultsOutput,
                     verbose=verbose
                 )