pdd-cli 0.0.19__py3-none-any.whl → 0.0.21__py3-none-any.whl

pdd/auto_deps_main.py

@@ -94,5 +94,4 @@ def auto_deps_main(
     except Exception as e:
         if not ctx.obj.get('quiet', False):
             rprint(f"[bold red]Error:[/bold red] {str(e)}")
-        sys.exit(1)
-        # Removed the "raise" line so that we only exit, satisfying the test.
+        sys.exit(1)

pdd/cli.py

@@ -46,7 +46,7 @@ console = Console()
 @click.option("--review-examples", is_flag=True,
               help="Review and optionally exclude few-shot examples before command execution.")
 @click.option('--local', is_flag=True, help='Run commands locally instead of in the cloud.')
-@click.version_option(version="0.0.19")
+@click.version_option(version="0.0.21")
 @click.pass_context
 def cli(
     ctx,

pdd/crash_main.py

@@ -51,10 +51,14 @@ def crash_main(
             "output": output,
             "output_program": output_program
         }
+
+        force = ctx.params.get("force", ctx.obj.get("force", False))
+        quiet = ctx.params.get("quiet", ctx.obj.get("quiet", False))
+
         input_strings, output_file_paths, _ = construct_paths(
             input_file_paths=input_file_paths,
-            force=ctx.obj.get('force', False),
-            quiet=ctx.obj.get('quiet', False),
+            force=force,
+            quiet=quiet,
             command="crash",
             command_options=command_options
         )
@@ -66,72 +70,62 @@ def crash_main(
         error_content = input_strings["error_file"]
 
         # Get model parameters from context
-        strength = ctx.obj.get('strength', 0.97)
-        temperature = ctx.obj.get('temperature', 0)
+        strength = ctx.obj.get("strength", 0.97)
+        temperature = ctx.obj.get("temperature", 0)
+
+        verbose = ctx.params.get("verbose", ctx.obj.get("verbose", False))
 
         if loop:
             # Use iterative fixing process
             success, final_code, final_program, attempts, cost, model = fix_code_loop(
-                code_file=code_file,
-                prompt=prompt_content,
-                verification_program=program_file,
-                strength=strength,
-                temperature=temperature,
-                max_attempts=max_attempts or 3,
-                budget=budget or 5.0,
-                error_log_file=error_file,
-                verbose=not ctx.obj.get('verbose', False)
+                code_file, prompt_content, program_file, strength, temperature, max_attempts or 3, budget or 5.0, error_file, verbose
             )
         else:
             # Use single fix attempt
             from .fix_code_module_errors import fix_code_module_errors
             update_program, update_code, final_program, final_code, cost, model = fix_code_module_errors(
-                program=program_content,
-                prompt=prompt_content,
-                code=code_content,
-                errors=error_content,
-                strength=strength,
-                temperature=temperature,
-                verbose=not ctx.obj.get('verbose', False)
+                program_content, prompt_content, code_content, error_content, strength, temperature, verbose
             )
             success = True
             attempts = 1
 
-        # Determine if contents were actually updated
-        if final_code != "":
-            update_code = final_code != code_content
-        else:
-            update_code = False
-        if final_program != "":
-            update_program = final_program != program_content
-        else:
-            update_program = False
-            
-        # Save results if contents changed
-        if update_code and output_file_paths.get("output"):
-            with open(output_file_paths["output"], 'w') as f:
+        # Ensure we have content to write, falling back to original content if needed
+        if final_code == "":
+            final_code = code_content
+        
+        if final_program == "":
+            final_program = program_content
+
+        # Determine whether to write the files based on whether paths are provided
+        should_write_code = output_file_paths.get("output") is not None
+        should_write_program = output_file_paths.get("output_program") is not None
+
+        # Write output files
+        if should_write_code:
+            with open(output_file_paths["output"], "w") as f:
                 f.write(final_code)
-        if update_program and output_file_paths.get("output_program"):
-            with open(output_file_paths["output_program"], 'w') as f:
+
+        if should_write_program:
+            with open(output_file_paths["output_program"], "w") as f:
                 f.write(final_program)
 
         # Provide user feedback
-        if not ctx.obj.get('quiet', False):
+        if not quiet:
             if success:
                 rprint("[bold green]Crash fix completed successfully.[/bold green]")
             else:
-                rprint("[bold yellow]Crash fix completed with some issues.[/bold yellow]")
+                rprint("[bold yellow]Crash fix completed with issues.[/bold yellow]")
             rprint(f"[bold]Model used:[/bold] {model}")
             rprint(f"[bold]Total attempts:[/bold] {attempts}")
-            rprint(f"[bold]Total cost:[/bold] ${cost:.6f}")
-            if update_code and output:
+            rprint(f"[bold]Total cost:[/bold] ${cost:.2f}")
+            if should_write_code:
                 rprint(f"[bold]Fixed code saved to:[/bold] {output_file_paths['output']}")
-            if update_program and output_program:
+            if should_write_program:
                 rprint(f"[bold]Fixed program saved to:[/bold] {output_file_paths['output_program']}")
 
         return success, final_code, final_program, attempts, cost, model
-
+    
     except Exception as e:
-        if not ctx.obj.get('quiet', False):
+        if not quiet:
             rprint(f"[bold red]Error:[/bold red] {str(e)}")
         sys.exit(1)

pdd/data/llm_model.csv

@@ -1,17 +1,17 @@
 provider,model,input,output,coding_arena_elo,base_url,api_key,counter,encoder,max_tokens,max_completion_tokens,structured_output
-OpenAI,"gpt-4o-mini",0.15,0.60,1246,,OPENAI_API_KEY,tiktoken,o200k_base,16384,,True
+OpenAI,"gpt-4o-mini",0.15,0.60,1246,,OPENAI_API_KEY,tiktoken,o200k_base,,16384,True
 OpenAI,"grok-2-1212",2,10,1255,"https://api.x.ai/v1",XAI_API_KEY,tiktoken,o200k_base,4096,,False
 Anthropic,"claude-3-5-haiku-20241022",1,5,1259,,ANTHROPIC_API_KEY,anthropic,claude-3-sonnet-20240229,8192,,False
 OpenAI,"deepseek-coder",0.14,0.28,1279,https://api.deepseek.com/beta,DEEPSEEK_API_KEY,autotokenizer,deepseek-coder-7b-instruct-v1.5,8192,,False
 Google,"gemini-2.0-flash-thinking-exp-01-21",.1,.4,1291,,GOOGLE_API_KEY,,,8192,,False
 GoogleVertexAI,"gemini-2.0-pro-exp-02-05",1.25,5,1299,,VERTEX_AI_API_KEY,,,8192,,False
 Anthropic,claude-3-7-sonnet-20250219,3,15,1312,,ANTHROPIC_API_KEY,anthropic,claude-3-sonnet-20240229,64000,,False
-Google,gemini-exp-1206,1.25,5,1313,,GOOGLE_API_KEY,,,8192,,False
+Google,gemini-2.5-pro-exp-03-25,1.25,5,1313,,GOOGLE_API_KEY,,,8192,,False
 OpenAI,"deepseek-r1-distill-llama-70b-specdec",5,5,1314,https://api.groq.com/openai/v1,GROQ_API_KEY,autotokenizer,deepseek-coder-7b-instruct-v1.5,16384,,False
 Ollama,"deepseek-r1:70b-llama-distill-q8_0",0.0,0.0,1315,,PWD,,,,,False
 Ollama,deepseek-r1:32b-qwen-distill-fp16,0.0,0.0,1316,,PWD,,,,,False
 OpenAI,"o3-mini",1.1,4.4,1319,,OPENAI_API_KEY,tiktoken,o200k_base,,100000,True
 OpenAI,"o1-2024-12-17",15,60,1331,,OPENAI_API_KEY,tiktoken,o200k_base,,32768,True
-OpenAI,"gpt-4o-2024-11-20",2.5,10,1332,,OPENAI_API_KEY,tiktoken,o200k_base,16384,,True
+OpenAI,"gpt-4o-2024-11-20",2.5,10,1332,,OPENAI_API_KEY,tiktoken,o200k_base,,16384,True
 OpenAI,"deepseek-reasoner",0.55,2.19,1336,https://api.deepseek.com/beta,DEEPSEEK_API_KEY,autotokenizer,deepseek-coder-7b-instruct-v1.5,8192,,False
 Fireworks,accounts/fireworks/models/deepseek-r1,3,8,1338,,FIREWORKS_API_KEY,,,8192,,False