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

pdd/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.0.38"
+__version__ = "0.0.39"
 
 # 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.

pdd/auto_deps_main.py

@@ -4,6 +4,7 @@ from typing import Tuple, Optional
 import click
 from rich import print as rprint
 
+from . import DEFAULT_STRENGTH, DEFAULT_TIME
 from .construct_paths import construct_paths
 from .insert_includes import insert_includes
 
@@ -60,8 +61,9 @@ def auto_deps_main(
             Path(csv_path).unlink()
 
         # Get strength and temperature 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)
 
         # Call insert_includes with the prompt content and directory path
         modified_prompt, csv_output, total_cost, model_name = insert_includes(
@@ -70,6 +72,7 @@ def auto_deps_main(
             csv_filename=csv_path,
             strength=strength,
             temperature=temperature,
+            time=time_budget,
             verbose=not ctx.obj.get('quiet', False)
         )
 

pdd/auto_include.py

@@ -8,6 +8,7 @@ from .llm_invoke import llm_invoke
 from .summarize_directory import summarize_directory
 import pandas as pd
 from io import StringIO
+from . import DEFAULT_TIME, DEFAULT_STRENGTH
 
 console = Console()
 
@@ -18,8 +19,9 @@ def auto_include(
     input_prompt: str,
     directory_path: str,
     csv_file: Optional[str] = None,
-    strength: float = 0.7,
+    strength: float = DEFAULT_STRENGTH,
     temperature: float = 0.0,
+    time: float = DEFAULT_TIME,
     verbose: bool = False
 ) -> Tuple[str, str, float, str]:
     """
@@ -31,6 +33,7 @@ def auto_include(
         csv_file (Optional[str]): Contents of existing CSV file
         strength (float): Strength of LLM model (0-1)
         temperature (float): Temperature of LLM model (0-1)
+        time (float): Time budget for LLM calls
         verbose (bool): Whether to print detailed information
 
     Returns:
@@ -68,6 +71,7 @@ def auto_include(
             directory_path=directory_path,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose,
             csv_file=csv_file
         )
@@ -100,6 +104,7 @@ def auto_include(
             },
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=verbose
         )
         total_cost += auto_include_response["cost"]
@@ -115,6 +120,7 @@ def auto_include(
                 input_json={"llm_output": auto_include_response["result"]},
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=verbose,
                 output_pydantic=AutoIncludeOutput
             )
@@ -160,6 +166,7 @@ context/image_utils.py,"Image processing utilities",2023-01-01T10:00:00"""
             csv_file=csv_file,
             strength=0.7,
             temperature=0.0,
+            time=DEFAULT_TIME,
             verbose=True
         )
 

pdd/bug_main.py

@@ -5,6 +5,7 @@ import click
 from rich import print as rprint
 from pathlib import Path
 
+from . import DEFAULT_STRENGTH, DEFAULT_TIME
 from .construct_paths import construct_paths
 from .bug_to_unit_test import bug_to_unit_test
 
@@ -64,8 +65,9 @@ def bug_main(
         desired_output_content = input_strings["desired_output"]
 
         # Generate unit test
-        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)
         unit_test, total_cost, model_name = bug_to_unit_test(
             current_output_content,
             desired_output_content,
@@ -74,7 +76,8 @@ def bug_main(
             program_content,
             strength,
             temperature,
-            language
+            language,
+            time_budget
         )
 
         # Save results if output path is provided

pdd/bug_to_unit_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, DEFAULT_STRENGTH
+from . import EXTRACTION_STRENGTH, DEFAULT_STRENGTH, DEFAULT_TIME
 from .load_prompt_template import load_prompt_template
 from .llm_invoke import llm_invoke
 from .unfinished_prompt import unfinished_prompt
@@ -20,6 +20,7 @@ def bug_to_unit_test(
     program_used_to_run_code_under_test: str,
     strength: float = DEFAULT_STRENGTH,
     temperature: float = 0.0,
+    time: float = DEFAULT_TIME,
     language: str = "python"
 ) -> Tuple[str, float, str]:
     """
@@ -33,6 +34,7 @@ def bug_to_unit_test(
         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 DEFAULT_STRENGTH.
         temperature (float, optional): Temperature of the LLM model. Defaults to 0.0.
+        time (float, optional): Time budget for LLM calls. Defaults to DEFAULT_TIME.
         language (str, optional): Programming language. Defaults to "python".
 
     Returns:
@@ -77,6 +79,7 @@ def bug_to_unit_test(
             input_json=input_json,
             strength=strength,
             temperature=temperature,
+            time=time,
             verbose=True
         )
 
@@ -93,6 +96,7 @@ def bug_to_unit_test(
             prompt_text=last_600_chars,
             strength=0.75,
             temperature=temperature,
+            time=time,
             verbose=False
         )
         
@@ -105,6 +109,7 @@ def bug_to_unit_test(
                 llm_output=response['result'],
                 strength=strength,
                 temperature=temperature,
+                time=time,
                 verbose=True
             )
             total_cost += continued_cost
@@ -124,6 +129,7 @@ def bug_to_unit_test(
             language,
             strength=EXTRACTION_STRENGTH,
             temperature=temperature,
+            time=time,
             verbose=True
         )
         total_cost += postprocess_cost
@@ -154,7 +160,8 @@ def add_numbers(a, b):
             desired_output=desired_output,
             prompt_used_to_generate_the_code=prompt,
             code_under_test=code,
-            program_used_to_run_code_under_test=program
+            program_used_to_run_code_under_test=program,
+            time=DEFAULT_TIME
         )
 
         if unit_test:

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}")
@@ -100,6 +101,7 @@ def cmd_test_main(
                 strength,
                 temperature,
                 language,
+                time=time,
             )
         except Exception as e:
             print(f"[bold red]Error generating tests: {e}[/bold red]")
@@ -121,6 +123,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)
         )