gptdiff 0.1.24__py3-none-any.whl → 0.1.28__py3-none-any.whl

gptdiff/applydiff.py

@@ -189,6 +189,7 @@ def parse_diff_per_file(diff_text):
 
     Note:
         Uses 'b/' prefix detection from git diffs to determine target paths
+        This doesn't work all the time and needs to be revised with stronger models
     """
     header_re = re.compile(r'^(?:diff --git\s+)?(a/[^ ]+)\s+(b/[^ ]+)\s*$', re.MULTILINE)
     lines = diff_text.splitlines()

gptdiff/gptdiff.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 from pathlib import Path
+from urllib.parse import urlparse
 import subprocess
 import hashlib
 import re
@@ -14,6 +15,7 @@ import pkgutil
 import contextvars
 from pkgutil import get_data
 import threading
+from threading import Lock
 
 import openai
 from openai import OpenAI
@@ -203,13 +205,26 @@ def load_prepend_file(file):
     with open(file, 'r') as f:
         return f.read()
 
-# Function to call GPT-4 API and calculate the cost
+def domain_for_url(base_url):
+    parsed = urlparse(base_url)
+    if parsed.netloc:
+        if parsed.username:
+            domain = parsed.hostname
+            if parsed.port:
+                domain += f":{parsed.port}"
+        else:
+            domain = parsed.netloc
+    else:
+        domain = base_url
+    return domain
+
 def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperature=0.7, max_tokens=30000, api_key=None, base_url=None):
     enc = tiktoken.get_encoding("o200k_base")
     
     # Use colors in print statements
     red = "\033[91m"
     green = "\033[92m"
+    blue = "\033[94m"
     reset = "\033[0m"
     start_time = time.time()
 
@@ -222,10 +237,13 @@ def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperat
     if 'gemini' in model:
         user_prompt = system_prompt + "\n" + user_prompt
 
+    input_content = system_prompt + "\n" + user_prompt + "\n" + files_content
+    token_count = len(enc.encode(input_content))
     messages = [
-        {"role": "system", "content": f"{green}{system_prompt}{reset}"},
+        {"role": "system", "content": system_prompt},
         {"role": "user", "content": user_prompt + "\n" + files_content},
     ]
+
     if VERBOSE:
         print(f"{green}Using {model}{reset}")
         print(f"{green}SYSTEM PROMPT{reset}")
@@ -233,7 +251,7 @@ def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperat
         print(f"{green}USER PROMPT{reset}")
         print(user_prompt, "+", len(enc.encode(files_content)), "tokens of file content")
     else:
-        print("Generating diff...")
+        print(f"Generating diff using model '{green}{model}{reset}' from '{blue}{domain_for_url(base_url)}{reset}' with {token_count} input tokens...")
 
     if not api_key:
         api_key = os.getenv('GPTDIFF_LLM_API_KEY')
@@ -248,9 +266,9 @@ def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperat
         temperature=temperature)
 
     if VERBOSE:
-        print("RESPONSE RAW-------------")
+        print("Debug: Raw LLM Response\n---")
         print(response.choices[0].message.content.strip())
-        print("/RESPONSE RAW-------------")
+        print("---")
     else:
         print("Diff generated.")
 
@@ -265,9 +283,6 @@ def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperat
     print("-" * 40)
 
     # Now, these rates are updated to per million tokens
-    cost_per_million_prompt_tokens = 30
-    cost_per_million_completion_tokens = 60
-    cost = (prompt_tokens / 1_000_000 * cost_per_million_prompt_tokens) + (completion_tokens / 1_000_000 * cost_per_million_completion_tokens)
 
     full_response = response.choices[0].message.content.strip()
     full_response, reasoning = swallow_reasoning(full_response)
@@ -279,7 +294,7 @@ def call_llm_for_diff(system_prompt, user_prompt, files_content, model, temperat
         toolbox.use(event)
     diff_response = diff_context.get()
 
-    return full_response, diff_response, prompt_tokens, completion_tokens, total_tokens, cost
+    return full_response, diff_response, prompt_tokens, completion_tokens, total_tokens
 
 # New API functions
 def build_environment(files_dict):
@@ -538,14 +553,23 @@ def smart_apply_patch(project_dir, diff_text, user_prompt, args):
     Attempt to apply a diff via smartapply: process each file concurrently using the LLM.
     """
     from pathlib import Path
+    start_time = time.time()
     parsed_diffs = parse_diff_per_file(diff_text)
     print("Found", len(parsed_diffs), "files in diff, processing smart apply concurrently:")
+    green = "\033[92m"
+    red = "\033[91m"
+    blue = "\033[94m"
+    reset = "\033[0m"
+
     if len(parsed_diffs) == 0:
         print(colorize_warning_warning("There were no entries in this diff. The LLM may have returned something invalid."))
         if args.beep:
             print("\a")
         return
     threads = []
+    success_files = []
+    failed_files = []
+    success_lock = Lock()
 
     def process_file(file_path, file_diff):
         full_path = Path(project_dir) / file_path
@@ -559,11 +583,14 @@ def smart_apply_patch(project_dir, diff_text, user_prompt, args):
                 print(colorize_warning_warning(f"File {file_path} not found - skipping deletion"))
             return
 
-        try:
-            original_content = full_path.read_text()
-        except (UnicodeDecodeError, IOError):
-            print(f"Skipping file {file_path} due to read error")
-            return
+        original_content = ""
+        if full_path.exists():
+            try:
+                original_content = full_path.read_text()
+            except (UnicodeDecodeError, IOError) as e:
+                print(f"Cannot read {file_path} due to {str(e)}, treating as new file")
+        else:
+            print(f"File {file_path} does not exist, treating as new file")
 
         # Use SMARTAPPLY-specific environment variables if set, otherwise fallback.
         smart_apply_model = os.getenv("GPTDIFF_SMARTAPPLY_MODEL")
@@ -586,9 +613,7 @@ def smart_apply_patch(project_dir, diff_text, user_prompt, args):
         else:
             base_url = os.getenv("GPTDIFF_LLM_BASE_URL", "https://nano-gpt.com/api/v1/")
 
-        print("-" * 40)
-        print("Running smartapply with", model, "on", file_path)
-        print("-" * 40)
+        print(f"Running smartapply in parallel for '{file_path}' using model '{green}{model}{reset}' from '{blue}{domain_for_url(base_url)}{reset}'...")
         try:
             updated_content = call_llm_for_apply_with_think_tool_available(
                 file_path, original_content, file_diff, model,
@@ -597,12 +622,18 @@ def smart_apply_patch(project_dir, diff_text, user_prompt, args):
                 max_tokens=args.max_tokens)
             if updated_content.strip() == "":
                 print("Cowardly refusing to write empty file to", file_path, "merge failed")
+                with success_lock:
+                    failed_files.append(file_path)
                 return
             full_path.parent.mkdir(parents=True, exist_ok=True)
             full_path.write_text(updated_content)
             print(f"\033[1;32mSuccessful 'smartapply' update {file_path}.\033[0m")
+            with success_lock:
+                success_files.append(file_path)
         except Exception as e:
             print(f"\033[1;31mFailed to process {file_path}: {str(e)}\033[0m")
+            with success_lock:
+                failed_files.append(file_path)
 
     for file_path, file_diff in parsed_diffs:
         thread = threading.Thread(target=process_file, args=(file_path, file_diff))
@@ -610,7 +641,17 @@ def smart_apply_patch(project_dir, diff_text, user_prompt, args):
         threads.append(thread)
     for thread in threads:
         thread.join()
-
+    elapsed = time.time() - start_time
+    minutes, seconds = divmod(int(elapsed), 60)
+    time_str = f"{minutes}m {seconds}s" if minutes else f"{seconds}s"
+    print(f"Smartapply successfully applied changes in {time_str}. Check the updated files to confirm.")
+    if failed_files:
+        print(f"\033[1;31mSmart apply completed in {time_str} with failures for {len(failed_files)} files:\033[0m")
+        for file in failed_files:
+            print(f"  - {file}")
+        print("Please check the errors above for details.")
+    else:
+        print(f"\033[1;32mSmart apply completed successfully in {time_str} for all {len(success_files)} files.\033[0m")
     if args.beep:
         print("\a")
 
@@ -710,9 +751,8 @@ def main():
         with open('prompt.txt', 'w') as f:
             f.write(full_prompt)
         print(f"Total tokens: {token_count:5d}")
-        print(f"\033[1;32mNot calling GPT-4.\033[0m")  # Green color for success message
+        print(f"\033[1;32mWrote full prompt to prompt.txt.\033[0m")
         print('Instead, wrote full prompt to prompt.txt. Use `xclip -selection clipboard < prompt.txt` then paste into chatgpt')
-        print(f"Total cost: ${0.0:.4f}")
         exit(0)
     else:
         # Validate API key presence before any API operations
@@ -729,7 +769,7 @@ def main():
                 print("Request canceled")
                 sys.exit(0)
         try:
-            full_text, diff_text, prompt_tokens, completion_tokens, total_tokens, cost = call_llm_for_diff(system_prompt, user_prompt, files_content, args.model, 
+            full_text, diff_text, prompt_tokens, completion_tokens, total_tokens = 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/"),
@@ -741,35 +781,40 @@ def main():
             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")
-        print("Unable to parse diff text. Full response:", full_text)
+        print(f"\033[1;33mWarning: No valid diff data was generated. This could be due to an unclear prompt or an invalid LLM response.\033[0m")
+        print("Suggested action: Refine your prompt or check the full response below for clues.")
+        print("Full LLM response:\n---\n" + full_text + "\n---")
         if args.beep:
-            print("\a")  # Terminal bell for completion notification
+            print("\a")
         return
 
     elif args.apply:
         print("\nAttempting apply with the following diff:")
-        print("\n<diff>")
         print(color_code_diff(diff_text))
-        print("\n</diff>")
-        print("Saved to patch.diff")
-        if apply_diff(project_dir, diff_text):
-            print(f"\033[1;32mPatch applied successfully with 'git apply'.\033[0m")
+        print("\033[94m**Attempting to apply patch using basic method...**\033[0m")
+        apply_result = apply_diff(project_dir, diff_text)
+        if apply_result:
+            print(f"\033[1;32mPatch applied successfully with basic apply.\033[0m")
         else:
-            print("Apply failed, attempting smart apply.")
+            print("\033[94m**Attempting smart apply with LLM...**\033[0m")
             smart_apply_patch(project_dir, diff_text, user_prompt, args)
 
     if args.beep:
-        print("\a")  # Terminal bell for completion notification
+        print("\a")
 
-    print(f"Prompt tokens: {prompt_tokens}")
-    print(f"Completion tokens: {completion_tokens}")
-    print(f"Total tokens: {total_tokens}")
-    #print(f"Total cost: ${cost:.4f}")
+    green = "\033[92m"
+    reset = "\033[0m"
+    if VERBOSE:
+        print("API Usage Details:")
+        print(f"- Prompt tokens: {prompt_tokens}")
+        print(f"- Completion tokens: {completion_tokens}")
+        print(f"- Total tokens: {total_tokens}")
+        print(f"- Model used: {green}{args.model}{reset}")
+    else:
+        print(f"API Usage: {total_tokens} tokens, Model used: {green}{args.model}{reset}")
 
 def swallow_reasoning(full_response: str) -> (str, str):
     """

gptdiff/gptpatch.py

@@ -13,7 +13,7 @@ This tool uses the same patch-application logic as gptdiff.
 import sys
 import argparse
 from pathlib import Path
-from gptdiff.gptdiff import apply_diff, smart_apply_patch
+from gptdiff.gptdiff import apply_diff, smart_apply_patch, color_code_diff
 
 def parse_arguments():
     parser = argparse.ArgumentParser(
@@ -49,11 +49,14 @@ def parse_arguments():
         default=30000,
         help="Maximum tokens to use for LLM responses"
     )
+    parser.add_argument('--verbose', action='store_true', help='Enable verbose output with detailed information')
     parser.add_argument('--dumb', action='store_true', default=False, help='Attempt dumb apply before trying smart apply')
     return parser.parse_args()
 
 def main():
     args = parse_arguments()
+    import gptdiff.gptdiff as gd
+    gd.VERBOSE = args.verbose
     if args.diff:
         diff_text = args.diff
     else:
@@ -64,15 +67,21 @@ def main():
         diff_text = diff_path.read_text(encoding="utf8")
 
     project_dir = args.project_dir
+
+    if args.verbose:
+        print("\n\033[1;34mDiff to be applied:\033[0m")
+        print(color_code_diff(diff_text))
+        print("")
+
     if args.dumb:
         success = apply_diff(project_dir, diff_text)
         if success:
-            print("✅ Diff applied successfully.")
+            print("\033[1;32m✅ Diff applied successfully.\033[0m")
         else:
-            print("❌ Failed to apply diff using git apply. Attempting smart apply.")
+            print("\033[1;31m❌ Failed to apply diff using git apply. Attempting smart apply.\033[0m")
             smart_apply_patch(project_dir, diff_text, "", args)
     else:
         smart_apply_patch(project_dir, diff_text, "", args)
         
 if __name__ == "__main__":
-    main()
+    main()

{gptdiff-0.1.24.dist-info → gptdiff-0.1.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.24
+Version: 0.1.28
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -82,13 +82,15 @@ done
 
 *Requires reasoning model*
 
-### Why GPTDiff?
+## Why Choose GPTDiff?
 
-- **Understands Your Code** - Describe changes in plain English
-- **Safe Modifications** - Keeps existing code working
-- **Auto-Fix** - `--apply` fixes mistakes in generated changes
-- **Works Instantly** - No complex setup needed
-- **Whole Project View** - Handles multiple files together
+- **Describe changes in plain English**  
+- **AI gets your whole project**  
+- **Auto-fixes conflicts**  
+- **Keeps code functional**  
+- **Fast setup, no fuss**  
+- **You approve every change**  
+- **Costs are upfront**
 
 ## Core Capabilities
 

gptdiff-0.1.28.dist-info/RECORD

@@ -0,0 +1,10 @@
+gptdiff/__init__.py,sha256=o1hrK4GFvbfKcHPlLVArz4OunE3euIicEBYaLrdDo0k,198
+gptdiff/applydiff.py,sha256=_11ITFMcigwvVptaIpEtyfLUTIy_mYPWExcXUqCBfOs,11200
+gptdiff/gptdiff.py,sha256=Lf3utjFM0T-39YyfMXUHZmVzukJk1DxR9941KNcYyIs,33373
+gptdiff/gptpatch.py,sha256=Vqk2vliYs_BxtuTpwdS88n3A8XToh6RvrCA4N8VqOu0,2759
+gptdiff-0.1.28.dist-info/LICENSE.txt,sha256=zCJk7yUYpMjFvlipi1dKtaljF8WdZ2NASndBYYbU8BY,1228
+gptdiff-0.1.28.dist-info/METADATA,sha256=iqKGOSe6vN62loiZ6vA2UZG-fAE80I_ZxHjMHs8CC_E,8723
+gptdiff-0.1.28.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+gptdiff-0.1.28.dist-info/entry_points.txt,sha256=0VlVNr-gc04a3SZD5_qKIBbtg_L5P2x3xlKE5ftcdkc,82
+gptdiff-0.1.28.dist-info/top_level.txt,sha256=XNkQkQGINaDndEwRxg8qToOrJ9coyfAb-EHrSUXzdCE,8
+gptdiff-0.1.28.dist-info/RECORD,,

gptdiff-0.1.24.dist-info/RECORD

@@ -1,10 +0,0 @@
-gptdiff/__init__.py,sha256=o1hrK4GFvbfKcHPlLVArz4OunE3euIicEBYaLrdDo0k,198
-gptdiff/applydiff.py,sha256=nvTerBtFuXWf1j6nisGY7CQ6qJCIM8J9UHxgoiWReoY,11116
-gptdiff/gptdiff.py,sha256=XN05Zbr1H69_iG8Bx8RQ34vTXXg3WHDANRcGo3ihrhA,31518
-gptdiff/gptpatch.py,sha256=opakY6j_I05ZNx2ACYgxB8SxoZ3POf9iFxDkV5Yn1oU,2393
-gptdiff-0.1.24.dist-info/LICENSE.txt,sha256=zCJk7yUYpMjFvlipi1dKtaljF8WdZ2NASndBYYbU8BY,1228
-gptdiff-0.1.24.dist-info/METADATA,sha256=TE_nbtNX0IMjDek5MHxvDDUaAeVhCXw7p5kCmh0TpZg,8785
-gptdiff-0.1.24.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-gptdiff-0.1.24.dist-info/entry_points.txt,sha256=0VlVNr-gc04a3SZD5_qKIBbtg_L5P2x3xlKE5ftcdkc,82
-gptdiff-0.1.24.dist-info/top_level.txt,sha256=XNkQkQGINaDndEwRxg8qToOrJ9coyfAb-EHrSUXzdCE,8
-gptdiff-0.1.24.dist-info/RECORD,,