gptdiff 0.1.30__tar.gz → 0.1.31__tar.gz

{gptdiff-0.1.30 → gptdiff-0.1.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.30
+Version: 0.1.31
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -31,30 +31,48 @@ GPTDiff: Create and apply diffs using AI.
 This tool leverages natural language instructions to modify project codebases.
 -->
 
+## Table of Contents
+
+- [Quick Start](#quick-start)
+- [Example Usage](#example-usage-of-gptdiff)
+- [Basic Usage](#basic-usage)
+- [Simple Command Line Agent Loops](#simple-command-line-agent-loops)
+- [Why Choose GPTDiff?](#why-choose-gptdiff)
+- [Core Capabilities](#core-capabilities)
+  - [CLI Excellence](#-cli-excellence)
+  - [Magic Diff Generation](#-magic-diff-generation)
+  - [Smart Apply System](#-smart-apply-system)
+- [Get Started](#get-started)
+  - [Installation](#installation)
+  - [Configuration](#configuration)
+- [Command Line Usage](#command-line-usage)
+
 🚀 **Create and apply diffs with AI**  
 Modify your project using plain English.
 
-More documentation at [gptdiff.255labs.xyz](gptdiff.255labs.xyz)
+More documentation at [gptdiff.255labs.xyz](https://gptdiff.255labs.xyz)
+
+## Quick Start
 
+1. **Install GPTDiff**
+   
+   
 ### Example Usage of `gptdiff`
 
 #### Apply a Patch Directly
-```
-bash
+```bash
 gptdiff "Add button animations on press" --apply
 ```
 ✅ Successfully applied patch
 
 #### Generate a Patch File
-```
-bash
+```bash
 gptdiff "Add API documentation" --call
 ```
 🔧 Patch written to `diff.patch`
 
 #### Generate a Prompt File Without Calling LLM
-```
-bash
+```bash
 gptdiff "Improve error messages"
 ```
 📄 LLM not called, written to `prompt.txt`

{gptdiff-0.1.30 → gptdiff-0.1.31}/README.md

@@ -4,30 +4,48 @@ GPTDiff: Create and apply diffs using AI.
 This tool leverages natural language instructions to modify project codebases.
 -->
 
+## Table of Contents
+
+- [Quick Start](#quick-start)
+- [Example Usage](#example-usage-of-gptdiff)
+- [Basic Usage](#basic-usage)
+- [Simple Command Line Agent Loops](#simple-command-line-agent-loops)
+- [Why Choose GPTDiff?](#why-choose-gptdiff)
+- [Core Capabilities](#core-capabilities)
+  - [CLI Excellence](#-cli-excellence)
+  - [Magic Diff Generation](#-magic-diff-generation)
+  - [Smart Apply System](#-smart-apply-system)
+- [Get Started](#get-started)
+  - [Installation](#installation)
+  - [Configuration](#configuration)
+- [Command Line Usage](#command-line-usage)
+
 🚀 **Create and apply diffs with AI**  
 Modify your project using plain English.
 
-More documentation at [gptdiff.255labs.xyz](gptdiff.255labs.xyz)
+More documentation at [gptdiff.255labs.xyz](https://gptdiff.255labs.xyz)
+
+## Quick Start
 
+1. **Install GPTDiff**
+   
+   
 ### Example Usage of `gptdiff`
 
 #### Apply a Patch Directly
-```
-bash
+```bash
 gptdiff "Add button animations on press" --apply
 ```
 ✅ Successfully applied patch
 
 #### Generate a Patch File
-```
-bash
+```bash
 gptdiff "Add API documentation" --call
 ```
 🔧 Patch written to `diff.patch`
 
 #### Generate a Prompt File Without Calling LLM
-```
-bash
+```bash
 gptdiff "Improve error messages"
 ```
 📄 LLM not called, written to `prompt.txt`
@@ -301,4 +319,4 @@ pip install -e .[test]
 pytest tests/
 ```
 
-This will execute all unit tests verifying core diff generation and application logic.
+This will execute all unit tests verifying core diff generation and application logic.

{gptdiff-0.1.30 → gptdiff-0.1.31}/gptdiff/gptdiff.py

@@ -417,6 +417,10 @@ prepended to the system prompt.
     """
     if model is None:
         model = os.getenv('GPTDIFF_MODEL', 'deepseek-reasoner')
+        # Use ANTHROPIC_BUDGET_TOKENS env var if set and no cli override provided
+        if anthropic_budget_tokens is None:
+            anthropic_budget_tokens = os.getenv('ANTHROPIC_BUDGET_TOKENS')
+    
     if prepend:
         if prepend.startswith("http://") or prepend.startswith("https://"):
             import urllib.request
@@ -438,7 +442,7 @@ prepended to the system prompt.
         max_tokens=max_tokens,
         api_key=api_key,
         base_url=base_url,
-        budget_tokens=anthropic_budget_tokens
+        budget_tokens=int(anthropic_budget_tokens) if anthropic_budget_tokens is not None else None
     )
     return diff_text
 
@@ -929,20 +933,17 @@ def swallow_reasoning(full_response: str) -> (str, str):
         r"(?P<reasoning>>\s*Reasoning.*?Reasoned.*?seconds)",
         re.DOTALL
     )
-    match = pattern.search(full_response)
-    if match:
+    reasoning_list = []
+    def replacer(match):
         raw_reasoning = match.group("reasoning")
-        # Remove any leading '+' characters and extra whitespace from each line
         reasoning_lines = [line.lstrip('+').strip() for line in raw_reasoning.splitlines()]
         reasoning = "\n".join(reasoning_lines).strip()
-
-        # Remove the reasoning block from the response using its exact span
-        final_content = full_response[:match.start()] + full_response[match.end():]
-        final_content = final_content.strip()
-    else:
-        reasoning = ""
-        final_content = full_response.strip()
-    return final_content, reasoning
+        reasoning_list.append(reasoning)
+        return ""
+    
+    final_content = re.sub(pattern, replacer, full_response)
+    reasoning = "\n".join(reasoning_list)
+    return final_content.strip(), reasoning
 
 def strip_bad_output(updated: str, original: str) -> str:
     """

{gptdiff-0.1.30 → gptdiff-0.1.31}/gptdiff.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.30
+Version: 0.1.31
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -31,30 +31,48 @@ GPTDiff: Create and apply diffs using AI.
 This tool leverages natural language instructions to modify project codebases.
 -->
 
+## Table of Contents
+
+- [Quick Start](#quick-start)
+- [Example Usage](#example-usage-of-gptdiff)
+- [Basic Usage](#basic-usage)
+- [Simple Command Line Agent Loops](#simple-command-line-agent-loops)
+- [Why Choose GPTDiff?](#why-choose-gptdiff)
+- [Core Capabilities](#core-capabilities)
+  - [CLI Excellence](#-cli-excellence)
+  - [Magic Diff Generation](#-magic-diff-generation)
+  - [Smart Apply System](#-smart-apply-system)
+- [Get Started](#get-started)
+  - [Installation](#installation)
+  - [Configuration](#configuration)
+- [Command Line Usage](#command-line-usage)
+
 🚀 **Create and apply diffs with AI**  
 Modify your project using plain English.
 
-More documentation at [gptdiff.255labs.xyz](gptdiff.255labs.xyz)
+More documentation at [gptdiff.255labs.xyz](https://gptdiff.255labs.xyz)
+
+## Quick Start
 
+1. **Install GPTDiff**
+   
+   
 ### Example Usage of `gptdiff`
 
 #### Apply a Patch Directly
-```
-bash
+```bash
 gptdiff "Add button animations on press" --apply
 ```
 ✅ Successfully applied patch
 
 #### Generate a Patch File
-```
-bash
+```bash
 gptdiff "Add API documentation" --call
 ```
 🔧 Patch written to `diff.patch`
 
 #### Generate a Prompt File Without Calling LLM
-```
-bash
+```bash
 gptdiff "Improve error messages"
 ```
 📄 LLM not called, written to `prompt.txt`

{gptdiff-0.1.30 → gptdiff-0.1.31}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='gptdiff',
-    version='0.1.30',
+    version='0.1.31',
     description='A tool to generate and apply git diffs using LLMs',
     author='255labs',
     packages=find_packages(),  # Use find_packages() to automatically discover packages

{gptdiff-0.1.30 → gptdiff-0.1.31}/tests/test_multidiff.py

@@ -34,8 +34,7 @@ DIFF 2
     expected = """
 DIFF 1
 
-DIFF 2
-`"""
+DIFF 2"""
 
  
     # Define a dummy call_llm function that returns our fake response

{gptdiff-0.1.30 → gptdiff-0.1.31}/tests/test_swallow_reasoning.py

@@ -2,7 +2,7 @@ import pytest
 
 from gptdiff.gptdiff import swallow_reasoning
 
-def test_swallow_reasoning_extraction():
+def test_swallow_reasoning_extraction_simple():
     llm_response = (
         "+> Reasoning\n"
         "+None\n"
@@ -13,19 +13,39 @@ def test_swallow_reasoning_extraction():
     final_content, reasoning = swallow_reasoning(llm_response)
     expected_reasoning = (
         "> Reasoning\n"
-        "**Applying the diff**\n"
-        "I'm piecing together how to efficiently apply a diff to a file...\n"
-        "**Returning the result**\n"
-        "I'm finalizing the method to apply the diff updates...\n"
-        "Reasoned for 6 seconds"
+        "None\n"
+        "Reasoned about summary drawer button 변경 for 15 seconds"
     )
     assert reasoning == expected_reasoning
     # The final content should no longer contain the reasoning block.
     assert expected_reasoning not in final_content
-    # And it should contain the diff block.
-    assert "```diff" in final_content
 
 
+def test_swallow_reasoning_extraction_multiline():
+    llm_response = (
+        "line 1> Reasoning\n"
+        "+None\n"
+        "+Reasoned about summary drawer button 변경 for 1 seconds\n"
+        "line 2\n"
+        "  > Reasoning\n"
+        "+None\n"
+        "+Reasoned about summary drawer button 변경 for 2 seconds\n"
+        "line 3:"
+    )
+    final_content, reasoning = swallow_reasoning(llm_response)
+    expected_reasoning = (
+        "> Reasoning\n"
+        "None\n"
+        "Reasoned about summary drawer button 변경 for 1 seconds\n"
+        "> Reasoning\n"
+        "None\n"
+        "Reasoned about summary drawer button 변경 for 2 seconds"
+    )
+    assert reasoning == expected_reasoning
+    assert "line 1\nline 2\n  \nline 3:" == final_content
+    # The final content should no longer contain the reasoning block.
+    assert expected_reasoning not in final_content
+
 def test_swallow_reasoning_with_untested_response():
     llm_response = (
         "> Reasoning\n"