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

pdd/generate_test.py

@@ -2,7 +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
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH, DEFAULT_TIME
 from .load_prompt_template import load_prompt_template
 from .preprocess import preprocess
 from .llm_invoke import llm_invoke
@@ -15,9 +15,10 @@ console = Console()
 def generate_test(
     prompt: str,
     code: str,
-    strength: float,
-    temperature: float,
-    language: str,
+    strength: float=DEFAULT_STRENGTH,
+    temperature: float=0.0,
+    time: float = DEFAULT_TIME,
+    language: str = "python",
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -29,6 +30,7 @@ def generate_test(
         strength (float): The strength of the LLM model (0-1).
         temperature (float): The temperature of the LLM model.
         language (str): The programming language for the unit test.
+        time (float, optional): Time budget for LLM calls. Defaults to DEFAULT_TIME.
         verbose (bool): Whether to print detailed information.
 
     Returns:
@@ -62,6 +64,7 @@ def generate_test(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 
@@ -79,6 +82,7 @@ def generate_test(
             prompt_text=last_600_chars,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         total_cost += check_cost
@@ -92,6 +96,7 @@ def generate_test(
                 llm_output=result,
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose
             )
             total_cost += continue_cost
@@ -104,6 +109,7 @@ def generate_test(
             language=language,
             strength=EXTRACTION_STRENGTH,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         total_cost += post_cost

pdd/git_update.py

@@ -5,7 +5,7 @@ from rich.console import Console
 from rich.panel import Panel
 from .update_prompt import update_prompt
 import git
-
+from . import DEFAULT_TIME
 console = Console()
 
 def git_update(
@@ -13,7 +13,8 @@ def git_update(
     modified_code_file: str,
     strength: float,
     temperature: float,
-    verbose: bool = False
+    verbose: bool = False,
+    time: float = DEFAULT_TIME
 ) -> Tuple[Optional[str], float, str]:
     """
     Read in modified code, restore the prior checked-in version from GitHub,
@@ -61,7 +62,8 @@ def git_update(
             modified_code=modified_code,
             strength=strength,
             temperature=temperature,
-            verbose=verbose
+            verbose=verbose,
+            time=time
         )
 
         # Write back the modified code

pdd/increase_tests.py

@@ -1,6 +1,6 @@
-from typing import Tuple
+from typing import Tuple, Optional
 from rich.console import Console
-from . import EXTRACTION_STRENGTH
+from . import EXTRACTION_STRENGTH, DEFAULT_TIME
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
 from .postprocess import postprocess
@@ -13,6 +13,7 @@ def increase_tests(
     language: str = "python",
     strength: float = 0.5,
     temperature: float = 0.0,
+    time: Optional[float] = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, float, str]:
     """
@@ -26,6 +27,7 @@ def increase_tests(
         language (str, optional): Programming language. Defaults to "python".
         strength (float, optional): LLM model strength. Defaults to 0.5.
         temperature (float, optional): LLM model temperature. Defaults to 0.0.
+        time (Optional[float]): Time allocation for the LLM. Defaults to DEFAULT_TIME.
         verbose (bool, optional): Verbose output flag. Defaults to False.
 
     Returns:
@@ -73,6 +75,7 @@ def increase_tests(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 

pdd/insert_includes.py

@@ -7,6 +7,7 @@ from .llm_invoke import llm_invoke
 from .load_prompt_template import load_prompt_template
 from .auto_include import auto_include
 from .preprocess import preprocess
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 
 class InsertIncludesOutput(BaseModel):
     output_prompt: str = Field(description="The prompt with dependencies inserted")
@@ -15,8 +16,9 @@ def insert_includes(
     input_prompt: str,
     directory_path: str,
     csv_filename: str,
-    strength: float,
-    temperature: float,
+    strength: float = DEFAULT_STRENGTH,
+    temperature: float = 0.0,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, str, float, str]:
     """
@@ -28,6 +30,7 @@ def insert_includes(
         csv_filename (str): Name of the CSV file containing dependencies
         strength (float): Strength parameter for the LLM model
         temperature (float): Temperature parameter for the LLM model
+        time (float): Time budget for the LLM model
         verbose (bool, optional): Whether to print detailed information. Defaults to False.
 
     Returns:
@@ -74,6 +77,7 @@ def insert_includes(
             csv_file=csv_content,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
 
@@ -90,6 +94,7 @@ def insert_includes(
             },
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose,
             output_pydantic=InsertIncludesOutput
         )
@@ -135,6 +140,7 @@ def main():
             csv_filename=csv_filename,
             strength=strength,
             temperature=temperature,
+            time=0.25,
             verbose=True
         )
 

pdd/preprocess_main.py

@@ -7,7 +7,7 @@ from rich import print as rprint
 from .construct_paths import construct_paths
 from .preprocess import preprocess
 from .xml_tagger import xml_tagger
-
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 def preprocess_main(
     ctx: click.Context, prompt_file: str, output: Optional[str], xml: bool, recursive: bool, double: bool, exclude: list
 ) -> Tuple[str, float, str]:
@@ -40,10 +40,17 @@ def preprocess_main(
 
         if xml:
             # Use xml_tagger to add XML delimiters
-            strength = ctx.obj.get("strength", 0.5)
+            strength = ctx.obj.get("strength", DEFAULT_STRENGTH)
             temperature = ctx.obj.get("temperature", 0.0)
             verbose = ctx.obj.get("verbose", False)
-            xml_tagged, total_cost, model_name = xml_tagger(prompt, strength, temperature, verbose)
+            time = ctx.obj.get("time", DEFAULT_TIME)
+            xml_tagged, total_cost, model_name = xml_tagger(
+                prompt,
+                strength,
+                temperature,
+                verbose,
+                time=time
+            )
             processed_prompt = xml_tagged
         else:
             # Preprocess the prompt

pdd/process_csv_change.py

@@ -10,6 +10,7 @@ from .get_extension import get_extension
 # Assuming EXTRACTION_STRENGTH and DEFAULT_STRENGTH might be needed later,
 # or just acknowledging their existence as per the prompt.
 # from .. import EXTRACTION_STRENGTH, DEFAULT_STRENGTH
+from . import DEFAULT_TIME # Added DEFAULT_TIME
 
 # No changes needed in the code_under_test based on these specific errors.
 
@@ -69,7 +70,8 @@ def process_csv_change(
     code_directory: str,
     language: str, # Default language if not specified in prompt filename
     extension: str, # Default extension (unused if language suffix found)
-    budget: float
+    budget: float,
+    time: float = DEFAULT_TIME # Added time parameter
 ) -> Tuple[bool, List[Dict[str, str]], float, Optional[str]]:
     """
     Reads a CSV file, processes each row to modify associated code files using an LLM,
@@ -86,6 +88,7 @@ def process_csv_change(
         extension: Default file extension (including '.') if language cannot be inferred.
                    Note: This is less likely to be used if `get_extension` covers the default language.
         budget: Maximum allowed cost for all LLM operations. Must be non-negative.
+        time: Time budget for each LLM operation.
 
     Returns:
         A tuple containing:
@@ -296,7 +299,10 @@ def process_csv_change(
                             input_code=input_code,
                             change_prompt=change_instructions,
                             strength=strength,
-                            temperature=temperature
+                            temperature=temperature,
+                            time=time, # Pass time
+                            budget=budget - total_cost, # Pass per-row budget
+                            quiet=True # Suppress individual change prints for CSV mode
                         )
                         console.print(f"    [dim]Change cost:[/dim] ${cost:.6f}")
                         console.print(f"    [dim]Model used:[/dim] {current_model_name}")

pdd/split.py

@@ -1,11 +1,12 @@
-from typing import Tuple
+from typing import Tuple, Optional
 from rich import print as rprint
 from rich.markdown import Markdown
 from pydantic import BaseModel, Field
 from .load_prompt_template import load_prompt_template
 from .preprocess import preprocess
 from .llm_invoke import llm_invoke
-from . import EXTRACTION_STRENGTH
+
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH, DEFAULT_TEMPERATURE, DEFAULT_TIME
 
 class PromptSplit(BaseModel):
     extracted_functionality: str = Field(description="The extracted functionality as a sub-module prompt")
@@ -15,10 +16,11 @@ def split(
     input_prompt: str,
     input_code: str,
     example_code: str,
-    strength: float,
-    temperature: float,
+    strength: float = DEFAULT_STRENGTH,
+    temperature: float = DEFAULT_TEMPERATURE,
+    time: Optional[float] = DEFAULT_TIME,
     verbose: bool = False
-) -> Tuple[str, str, float, str]:
+) -> Tuple[Tuple[str, str], float, str]:
     """
     Split a prompt into extracted functionality and remaining prompt.
 
@@ -28,14 +30,18 @@ def split(
         example_code (str): Example code showing usage.
         strength (float): LLM strength parameter (0-1).
         temperature (float): LLM temperature parameter (0-1).
+        time (Optional[float]): Time allocation for the LLM.
         verbose (bool): Whether to print detailed information.
 
     Returns:
-        Tuple[str, str, float, str]: (extracted_functionality, remaining_prompt, model_name, total_cost)
+        Tuple[Tuple[str, str], float, str]: 
+            ((extracted_functionality, remaining_prompt), total_cost, model_name)
             where model_name is the name of the model used (returned as the second to last tuple element)
             and total_cost is the aggregated cost from all LLM invocations.
     """
     total_cost = 0.0
+    model_name = ""
+
 
     # Input validation
     if not all([input_prompt, input_code, example_code]):
@@ -79,6 +85,7 @@ def split(
             },
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         total_cost += split_response["cost"]
@@ -95,7 +102,8 @@ def split(
             strength=EXTRACTION_STRENGTH,  # Fixed strength for extraction
             temperature=temperature,
             output_pydantic=PromptSplit,
-            verbose=verbose
+            verbose=verbose,
+            time=time  # Pass time to the second llm_invoke call
         )
         total_cost += extract_response["cost"]
 

pdd/split_main.py

@@ -59,6 +59,7 @@ def split_main(
         # Get parameters from context
         strength = ctx.obj.get('strength', 0.5)
         temperature = ctx.obj.get('temperature', 0)
+        time = ctx.obj.get('time')
 
         # Call the split function with the standardized return pattern (result_data, cost, model_name)
         result_tuple, total_cost, model_name = split(
@@ -67,6 +68,7 @@ def split_main(
             example_code=input_strings["example_code"],
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=not ctx.obj.get('quiet', False)
         )
         

pdd/summarize_directory.py

@@ -11,6 +11,7 @@ from rich.progress import track
 
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
+from . import DEFAULT_TIME
 
 class FileSummary(BaseModel):
     file_summary: str = Field(description="The summary of the file")
@@ -80,6 +81,7 @@ def summarize_directory(
     strength: float,
     temperature: float,
     verbose: bool,
+    time: float = DEFAULT_TIME,
     csv_file: Optional[str] = None
 ) -> Tuple[str, float, str]:
     """
@@ -90,6 +92,7 @@ def summarize_directory(
         strength (float): Between 0 and 1 that is the strength of the LLM model to use.
         temperature (float): Controls the randomness of the LLM's output.
         verbose (bool): Whether to print out the details of the function.
+        time (float): Time budget for LLM calls.
         csv_file (Optional[str]): Current CSV file contents if it already exists.
 
     Returns:
@@ -184,6 +187,7 @@ def summarize_directory(
                         input_json=input_params,
                         strength=strength,
                         temperature=temperature,
+                        time=time,
                         verbose=verbose,
                         output_pydantic=FileSummary
                     )

pdd/trace.py

@@ -6,7 +6,7 @@ import difflib
 from .load_prompt_template import load_prompt_template
 from .preprocess import preprocess
 from .llm_invoke import llm_invoke
-
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 console = Console()
 
 class PromptLineOutput(BaseModel):
@@ -16,9 +16,10 @@ def trace(
     code_file: str,
     code_line: int,
     prompt_file: str,
-    strength: float = 0.5,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0,
-    verbose: bool = False
+    verbose: bool = False,
+    time: float = DEFAULT_TIME
 ) -> Tuple[Optional[int], float, str]:
     """
     Trace a line of code back to its corresponding line in the prompt file.
@@ -30,6 +31,7 @@ def trace(
         strength (float, optional): Model strength. Defaults to 0.5
         temperature (float, optional): Model temperature. Defaults to 0
         verbose (bool, optional): Whether to print detailed information. Defaults to False
+        time (float, optional): Time parameter for LLM calls. Defaults to 0.25
 
     Returns:
         Tuple[Optional[int], float, str]: (prompt line number, total cost, model name)
@@ -67,7 +69,8 @@ def trace(
             },
             strength=strength,
             temperature=temperature,
-            verbose=verbose
+            verbose=verbose,
+            time=time
         )
 
         total_cost += trace_response['cost']
@@ -89,7 +92,8 @@ def trace(
             strength=strength,
             temperature=temperature,
             verbose=verbose,
-            output_pydantic=PromptLineOutput
+            output_pydantic=PromptLineOutput,
+            time=time
         )
 
         total_cost += extract_response['cost']

pdd/trace_main.py

@@ -5,7 +5,7 @@ import os
 import logging
 from .construct_paths import construct_paths
 from .trace import trace
-
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 logging.basicConfig(level=logging.WARNING)
 logger = logging.getLogger(__name__)
 
@@ -49,11 +49,12 @@ def trace_main(ctx: click.Context, prompt_file: str, code_file: str, code_line:
         logger.debug("Input files loaded")
 
         # Perform trace analysis
-        strength = ctx.obj.get('strength', 0.5)
+        strength = ctx.obj.get('strength', DEFAULT_STRENGTH)
         temperature = ctx.obj.get('temperature', 0.0)
+        time = ctx.obj.get('time', DEFAULT_TIME)
         try:
             prompt_line, total_cost, model_name = trace(
-                code_content, code_line, prompt_content, strength, temperature
+                code_content, code_line, prompt_content, strength, temperature, time=time
             )
             logger.debug(f"Trace analysis completed: prompt_line={prompt_line}, total_cost={total_cost}, model_name={model_name}")
             
@@ -90,7 +91,7 @@ def trace_main(ctx: click.Context, prompt_file: str, code_file: str, code_line:
                 logger.debug(f"Results saved to {output_path}")
             except IOError as e:
                 if not quiet:
-                    rprint(f"[bold red]An unexpected error occurred: {e}[/bold red]")
+                    rprint(f"[bold red]Error saving trace results: {e}[/bold red]")
                 logger.error(f"IOError while saving results: {e}")
                 ctx.exit(1)
 

pdd/unfinished_prompt.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel, Field
 from rich import print as rprint
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
+from . import DEFAULT_STRENGTH, DEFAULT_TIME
 
 class PromptAnalysis(BaseModel):
     reasoning: str = Field(description="Structured reasoning for the completeness assessment")
@@ -10,8 +11,9 @@ class PromptAnalysis(BaseModel):
 
 def unfinished_prompt(
     prompt_text: str,
-    strength: float = 0.5,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, bool, float, str]:
     """
@@ -21,6 +23,7 @@ def unfinished_prompt(
         prompt_text (str): The prompt text to analyze
         strength (float, optional): Strength of the LLM model. Defaults to 0.5.
         temperature (float, optional): Temperature of the LLM model. Defaults to 0.
+        time (float, optional): Time budget for LLM calls. Defaults to DEFAULT_TIME.
         verbose (bool, optional): Whether to print detailed information. Defaults to False.
 
     Returns:
@@ -70,6 +73,7 @@ def unfinished_prompt(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose,
             output_pydantic=PromptAnalysis
         )
@@ -103,6 +107,7 @@ if __name__ == "__main__":
     try:
         reasoning, is_finished, cost, model = unfinished_prompt(
             prompt_text=sample_prompt,
+            time=DEFAULT_TIME,
             verbose=True
         )
         rprint("\n[blue]Results:[/blue]")

pdd/update_main.py

@@ -6,7 +6,7 @@ from rich import print as rprint
 from .construct_paths import construct_paths
 from .update_prompt import update_prompt
 from .git_update import git_update
-
+from . import DEFAULT_TIME
 def update_main(
     ctx: click.Context,
     input_prompt_file: str,
@@ -49,6 +49,7 @@ def update_main(
         input_prompt = input_strings["input_prompt_file"]
         modified_code = input_strings["modified_code_file"]
         input_code = input_strings.get("input_code_file")
+        time = ctx.obj.get('time', DEFAULT_TIME)
 
         # Update prompt using appropriate method
         if git:
@@ -59,7 +60,8 @@ def update_main(
                 modified_code_file=modified_code_file,
                 strength=ctx.obj.get("strength", 0.5),
                 temperature=ctx.obj.get("temperature", 0),
-                verbose=ctx.obj.get("verbose", False)
+                verbose=ctx.obj.get("verbose", False),
+                time=time
             )
         else:
             if input_code is None:
@@ -70,7 +72,8 @@ def update_main(
                 modified_code=modified_code,
                 strength=ctx.obj.get("strength", 0.5),
                 temperature=ctx.obj.get("temperature", 0),
-                verbose=ctx.obj.get("verbose", False)
+                verbose=ctx.obj.get("verbose", False),
+                time=time
             )
 
         # Save the modified prompt

pdd/update_prompt.py

@@ -5,7 +5,7 @@ from pydantic import BaseModel, Field
 from .load_prompt_template import load_prompt_template
 from .preprocess import preprocess
 from .llm_invoke import llm_invoke
-
+from . import DEFAULT_TIME
 class PromptUpdate(BaseModel):
     modified_prompt: str = Field(description="The updated prompt that will generate the modified code")
 
@@ -15,7 +15,8 @@ def update_prompt(
     modified_code: str,
     strength: float,
     temperature: float,
-    verbose: bool = False
+    verbose: bool = False,
+    time: float = DEFAULT_TIME
 ) -> Tuple[str, float, str]:
     """
     Update a prompt based on the original and modified code.
@@ -27,6 +28,7 @@ def update_prompt(
         strength (float): The strength parameter for the LLM model (0-1)
         temperature (float): The temperature parameter for the LLM model (0-1)
         verbose (bool, optional): Whether to print detailed output. Defaults to False.
+        time (float, optional): The time parameter for the LLM model. Defaults to 0.25.
 
     Returns:
         Tuple[str, float, str]: (modified_prompt, total_cost, model_name)
@@ -68,7 +70,8 @@ def update_prompt(
             },
             strength=strength,
             temperature=temperature,
-            verbose=verbose
+            verbose=verbose,
+            time=time
         )
 
         if not first_response or not isinstance(first_response, dict) or 'result' not in first_response:
@@ -84,7 +87,8 @@ def update_prompt(
             strength=0.5,
             temperature=temperature,
             output_pydantic=PromptUpdate,
-            verbose=verbose
+            verbose=verbose,
+            time=time
         )
 
         if not second_response or not isinstance(second_response, dict) or 'result' not in second_response:

pdd/xml_tagger.py

@@ -5,7 +5,7 @@ from pydantic import BaseModel, Field
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
 from . import EXTRACTION_STRENGTH
-
+from . import DEFAULT_TIME
 class XMLOutput(BaseModel):
     xml_tagged: str = Field(description="The XML-tagged version of the prompt")
 
@@ -13,7 +13,8 @@ def xml_tagger(
     raw_prompt: str,
     strength: float,
     temperature: float,
-    verbose: bool = False
+    verbose: bool = False,
+    time: float = DEFAULT_TIME
 ) -> Tuple[str, float, str]:
     """
     Enhance a given LLM prompt by adding XML tags to improve its structure and readability.
@@ -23,6 +24,7 @@ def xml_tagger(
         strength (float): The strength parameter for the LLM model (0-1)
         temperature (float): The temperature parameter for the LLM model (0-1)
         verbose (bool): Whether to print detailed information
+        time (float): The time allocation for the LLM calls
 
     Returns:
         Tuple[str, float, str]: (xml_tagged, total_cost, model_name)
@@ -55,7 +57,8 @@ def xml_tagger(
             input_json={"raw_prompt": raw_prompt},
             strength=strength,
             temperature=temperature,
-            verbose=verbose
+            verbose=verbose,
+            time=time
         )
 
         xml_generated_analysis = conversion_response.get('result', '')
@@ -76,7 +79,8 @@ def xml_tagger(
             strength=EXTRACTION_STRENGTH,  # Fixed strength for extraction
             temperature=temperature,
             verbose=verbose,
-            output_pydantic=XMLOutput
+            output_pydantic=XMLOutput,
+            time=time
         )
 
         result: XMLOutput = extraction_response.get('result')
@@ -109,7 +113,8 @@ def main():
             raw_prompt=sample_prompt,
             strength=0.7,
             temperature=0.8,
-            verbose=True
+            verbose=True,
+            time=0.5
         )
 
         rprint("[blue]XML Tagging Complete[/blue]")

{pdd_cli-0.0.38.dist-info → pdd_cli-0.0.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pdd-cli
-Version: 0.0.38
+Version: 0.0.39
 Summary: PDD (Prompt-Driven Development) Command Line Interface
 Author: Greg Tanaka
 Author-email: glt@alumni.caltech.edu
@@ -35,18 +35,18 @@ Requires-Dist: litellm
 Requires-Dist: rich==14.0.0
 Requires-Dist: semver==3.0.2
 Requires-Dist: setuptools
+Requires-Dist: pytest
 Requires-Dist: boto3==1.35.99
 Requires-Dist: python-Levenshtein
 Provides-Extra: dev
 Requires-Dist: commitizen; extra == "dev"
-Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-mock; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: z3-solver; extra == "dev"
 Dynamic: license-file
 
-.. image:: https://img.shields.io/badge/pdd--cli-v0.0.38-blue
+.. image:: https://img.shields.io/badge/pdd--cli-v0.0.39-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
@@ -134,7 +134,7 @@ After installation, verify:
 
    pdd --version
 
-You'll see the current PDD version (e.g., 0.0.38).
+You'll see the current PDD version (e.g., 0.0.39).
 
 Advanced Installation Tips
 --------------------------