gptdiff 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

gptdiff/gptdiff.py

@@ -44,6 +44,9 @@ a/file.py b/file.py
 @@ -1,2 +1,2 @@
 -def old():
 +def new():
+
+-
+You must include the '--- file' and/or '+++ file' part of the diff. File modifications should include both.
 """
     )
     return toolbox
@@ -67,6 +70,25 @@ def create_think_toolbox():
     )
     return toolbox
 
+def color_code_diff(diff_text: str) -> str:
+    """
+    Color code lines in a diff. Lines beginning with '-' in red, and
+    lines beginning with '+' in green.
+    """
+    red = "\033[31m"
+    green = "\033[32m"
+    reset = "\033[0m"
+
+    colorized_lines = []
+    for line in diff_text.split('\n'):
+        if line.startswith('-'):
+            colorized_lines.append(f"{red}{line}{reset}")
+        elif line.startswith('+'):
+            colorized_lines.append(f"{green}{line}{reset}")
+        else:
+            colorized_lines.append(line)
+
+    return '\n'.join(colorized_lines)
 
 def load_gitignore_patterns(gitignore_path):
     with open(gitignore_path, 'r') as f:
@@ -308,8 +330,16 @@ def smartapply(diff_text, files, model=None, api_key=None, base_url=None):
             updated = call_llm_for_apply_with_think_tool_available(path, original, patch, model, api_key=api_key, base_url=base_url)
             files[path] = updated.strip()
 
+    threads = []
+
     for path, patch in parsed_diffs:
-        process_file(path, patch)
+        thread = threading.Thread(target=process_file, args=(path, patch))
+        thread.start()
+        threads.append(thread)
+
+    # Wait for all threads to complete
+    for thread in threads:
+        thread.join()
 
     return files
 
@@ -339,6 +369,7 @@ def parse_arguments():
     parser.add_argument('--temperature', type=float, default=0.7, help='Temperature parameter for model creativity (0.0 to 2.0)')
     parser.add_argument('--max_tokens', type=int, default=30000, help='Temperature parameter for model creativity (0.0 to 2.0)')
     parser.add_argument('--model', type=str, default=None, help='Model to use for the API call.')
+    parser.add_argument('--applymodel', type=str, default=None, help='Model to use for applying the diff. Defaults to the value of --model if not specified.')
 
     parser.add_argument('--nowarn', action='store_true', help='Disable large token warning')
 
@@ -404,11 +435,11 @@ def parse_diff_per_file(diff_text):
 
     return diffs
 
-def call_llm_for_apply_with_think_tool_available(file_path, original_content, file_diff, model, api_key=None, base_url=None):
+def call_llm_for_apply_with_think_tool_available(file_path, original_content, file_diff, model, api_key=None, base_url=None, extra_prompt=None, max_tokens=30000):
     parser = FlatXMLParser("think")
     formatter = FlatXMLPromptFormatter(tag="think")
     toolbox = create_think_toolbox()
-    full_response = call_llm_for_apply(file_path, original_content, file_diff, model, api_key=None, base_url=None)
+    full_response = call_llm_for_apply(file_path, original_content, file_diff, model, api_key=api_key, base_url=base_url, extra_prompt=extra_prompt, max_tokens=max_tokens)
     notool_response = ""
     events = parser.parse(full_response)
     is_in_tool = False
@@ -424,7 +455,7 @@ def call_llm_for_apply_with_think_tool_available(file_path, original_content, fi
 
     return notool_response
 
-def call_llm_for_apply(file_path, original_content, file_diff, model, api_key=None, base_url=None):
+def call_llm_for_apply(file_path, original_content, file_diff, model, api_key=None, base_url=None, extra_prompt=None, max_tokens=30000):
     """AI-powered diff application with conflict resolution.
     
     Internal workhorse for smartapply that handles individual file patches.
@@ -474,6 +505,8 @@ Diff to apply:
 {file_diff}
 </diff>"""
 
+    if extra_prompt:
+        user_prompt += f"\n\n{extra_prompt}"
     if model == "gemini-2.0-flash-thinking-exp-01-21":
         user_prompt = system_prompt+"\n"+user_prompt
     messages = [
@@ -490,7 +523,7 @@ Diff to apply:
     response = client.chat.completions.create(model=model,
         messages=messages,
         temperature=0.0,
-        max_tokens=30000)
+        max_tokens=max_tokens)
     full_response = response.choices[0].message.content
 
     elapsed = time.time() - start_time
@@ -590,12 +623,21 @@ def main():
             if confirmation != 'y':
                 print("Request canceled")
                 sys.exit(0)
-        full_text, diff_text, prompt_tokens, completion_tokens, total_tokens, cost = call_llm_for_diff(system_prompt, user_prompt, files_content, args.model, 
+        try:
+            full_text, diff_text, prompt_tokens, completion_tokens, total_tokens, cost = call_llm_for_diff(system_prompt, user_prompt, files_content, args.model, 
                                                                                                     temperature=args.temperature,
                                                                                                     api_key=os.getenv('GPTDIFF_LLM_API_KEY'),
                                                                                                     base_url=os.getenv('GPTDIFF_LLM_BASE_URL', "https://nano-gpt.com/api/v1/"),
                                                                                                     max_tokens=args.max_tokens
                                                                                                     ) 
+        except Exception as e:
+            full_text = f"{e}"
+            diff_text = ""
+            prompt_tokens = 0
+            completion_tokens = 0
+            total_tokens = 0
+            cost = 0
+            print(f"Error in LLM response {e}")
 
     if(diff_text.strip() == ""):
         print(f"\033[1;33mThere was no data in this diff. The LLM may have returned something invalid.\033[0m")
@@ -608,7 +650,7 @@ def main():
     elif args.apply:
         print("\nAttempting apply with the following diff:")
         print("\n<diff>")
-        print(diff_text)
+        print(color_code_diff(diff_text))
         print("\n</diff>")
         print("Saved to patch.diff")
         if apply_diff(project_dir, diff_text):
@@ -651,8 +693,11 @@ def main():
                 print("SMARTAPPLY")
                 print(file_diff)
                 print("-" * 40)
+                if args.applymodel is None:
+                    args.applymodel = args.model
+
                 try:
-                    updated_content = call_llm_for_apply_with_think_tool_available(file_path, original_content, file_diff, args.model)
+                    updated_content = call_llm_for_apply_with_think_tool_available(file_path, original_content, file_diff, args.applymodel, extra_prompt=f"This changeset is from the following instructions:\n{user_prompt}", max_tokens=args.max_tokens)
 
                     if updated_content.strip() == "":
                         print("Cowardly refusing to write empty file to", file_path, "merge failed")

{gptdiff-0.1.8.dist-info → gptdiff-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.8
+Version: 0.1.10
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -27,21 +27,54 @@ Dynamic: summary
 
 # GPTDiff
 
-🚀 **AI-Powered Code Evolution** - Transform your codebase with natural language instructions
+🚀 **Create and apply diffs with AI** - Modify projects with natural language
+
+More docs at [gptdiff.255labs.xyz](gptdiff.255labs.xyz)
+
+### Example Usage of `gptdiff`
+
+#### Apply a Patch Directly
+```
+bash
+gptdiff "Add button animations on press" --apply
+```
+✅ Successfully applied patch
+
+#### Generate a Patch File
+```
+bash
+gptdiff "Add API documentation" --call
+```
+🔧 Patch written to `diff.patch`
+
+#### Generate a Prompt File Without Calling LLM
+```
+bash
+gptdiff "Improve error messages"
+```
+📄 LLM not called, written to `prompt.txt`
+
+---
+
+### Basic Usage
 
 ```bash
 cd myproject
 gptdiff 'add hover effects to the buttons'
 ```
 
-Generates a prompt.txt file that you can copy and paste into a large context gpt to have a conversation with suggested changes. You can also invoke the API and try to directly apply the patch using a smartapply if the git apply fails.
+Generates a prompt.txt file that you can copy and paste into an LLM
 
-## Value Proposition
+### Simple command line agent loops
 
 ```bash
-gptdiff "Update the readme with an api section" --apply
+while
+do
+  gptdiff "Add missing test cases" --apply
+done
 ```
-<span style="color: #00ff00;">Patch applied successfully.</span>
+
+*Requires reasoning model*
 
 ### Why GPTDiff?
 

gptdiff-0.1.10.dist-info/RECORD

@@ -0,0 +1,8 @@
+gptdiff/__init__.py,sha256=yGjgwv7tNvH1ZLPsQyoo1CxpTOl1iCAwwDBp-_17ksQ,89
+gptdiff/gptdiff.py,sha256=78_Y1ifKxCdC-e8TdKm3kKDklyV0K8S0fKyDdXhLNQs,27706
+gptdiff-0.1.10.dist-info/LICENSE.txt,sha256=zCJk7yUYpMjFvlipi1dKtaljF8WdZ2NASndBYYbU8BY,1228
+gptdiff-0.1.10.dist-info/METADATA,sha256=jw4gVLU2Gk7Iuqy-NdkeDlF9loiLfd7825lZxXIbQEY,7602
+gptdiff-0.1.10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+gptdiff-0.1.10.dist-info/entry_points.txt,sha256=0yvXYEVAZFI-p32kQ4-h3qKVWS0a86jsM9FAwF89t9w,49
+gptdiff-0.1.10.dist-info/top_level.txt,sha256=XNkQkQGINaDndEwRxg8qToOrJ9coyfAb-EHrSUXzdCE,8
+gptdiff-0.1.10.dist-info/RECORD,,

gptdiff-0.1.8.dist-info/RECORD

@@ -1,8 +0,0 @@
-gptdiff/__init__.py,sha256=yGjgwv7tNvH1ZLPsQyoo1CxpTOl1iCAwwDBp-_17ksQ,89
-gptdiff/gptdiff.py,sha256=kzLeNc5M3saCdqBOoE-OjkyHztGbQr39XPiVcFJ_pyY,25958
-gptdiff-0.1.8.dist-info/LICENSE.txt,sha256=zCJk7yUYpMjFvlipi1dKtaljF8WdZ2NASndBYYbU8BY,1228
-gptdiff-0.1.8.dist-info/METADATA,sha256=ErNMAtAqN7HDzarpVJi0IVA5i1pryUQwkjVjus9kjkM,7317
-gptdiff-0.1.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-gptdiff-0.1.8.dist-info/entry_points.txt,sha256=0yvXYEVAZFI-p32kQ4-h3qKVWS0a86jsM9FAwF89t9w,49
-gptdiff-0.1.8.dist-info/top_level.txt,sha256=XNkQkQGINaDndEwRxg8qToOrJ9coyfAb-EHrSUXzdCE,8
-gptdiff-0.1.8.dist-info/RECORD,,