gptdiff 0.1.20__tar.gz → 0.1.21__tar.gz

{gptdiff-0.1.20 → gptdiff-0.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.20
+Version: 0.1.21
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -10,7 +10,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: openai>=1.0.0
 Requires-Dist: tiktoken>=0.5.0
-Requires-Dist: ai_agent_toolbox>=0.1.13
+Requires-Dist: ai-agent-toolbox>=0.1.15
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-mock; extra == "test"

{gptdiff-0.1.20 → gptdiff-0.1.21}/gptdiff/gptdiff.py

@@ -985,17 +985,23 @@ def swallow_reasoning(full_response: str) -> (str, str):
          - reasoning: The extracted reasoning block, or an empty string if not found.
     """
     pattern = re.compile(
-        r"(?P<reasoning>>\s*Reasoning.*?Reasoned for \d+\s*seconds)",
+        r"(?P<reasoning>>\s*Reasoning.*?Reasoned.*?seconds)",
         re.DOTALL
     )
     match = pattern.search(full_response)
     if match:
-        reasoning = match.group("reasoning").strip()
-        final_content = full_response.replace(reasoning, "").strip()
+        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
 
 if __name__ == "__main__":
-    main()
+    main()

{gptdiff-0.1.20 → gptdiff-0.1.21}/gptdiff.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gptdiff
-Version: 0.1.20
+Version: 0.1.21
 Summary: A tool to generate and apply git diffs using LLMs
 Author: 255labs
 Classifier: License :: OSI Approved :: MIT License
@@ -10,7 +10,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: openai>=1.0.0
 Requires-Dist: tiktoken>=0.5.0
-Requires-Dist: ai_agent_toolbox>=0.1.13
+Requires-Dist: ai-agent-toolbox>=0.1.15
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-mock; extra == "test"

{gptdiff-0.1.20 → gptdiff-0.1.21}/gptdiff.egg-info/requires.txt

@@ -1,6 +1,6 @@
 openai>=1.0.0
 tiktoken>=0.5.0
-ai_agent_toolbox>=0.1.13
+ai-agent-toolbox>=0.1.15
 
 [docs]
 mkdocs

{gptdiff-0.1.20 → gptdiff-0.1.21}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='gptdiff',
-    version='0.1.20',
+    version='0.1.21',
     description='A tool to generate and apply git diffs using LLMs',
     author='255labs',
     packages=find_packages(),  # Use find_packages() to automatically discover packages
@@ -12,7 +12,7 @@ setup(
     install_requires=[
         'openai>=1.0.0',
         'tiktoken>=0.5.0',
-        'ai_agent_toolbox>=0.1.13'
+        'ai-agent-toolbox>=0.1.15'
     ],
     extras_require={
         'test': ['pytest', 'pytest-mock'],

gptdiff-0.1.21/tests/test_swallow_reasoning.py

@@ -0,0 +1,156 @@
+import pytest
+
+from gptdiff.gptdiff import swallow_reasoning
+
+def test_swallow_reasoning_extraction():
+    llm_response = (
+        "+> Reasoning\n"
+        "+None\n"
+        "+Reasoned about summary drawer button 변경 for 15 seconds\n"
+        "+def new():\n"
+        "```"
+    )
+    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"
+    )
+    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_with_untested_response():
+    llm_response = (
+        "> Reasoning\n"
+        "**Considering the request**\n"
+        "I’m noting that the user wants me to apply a diff to a file and return the result in a block, ensuring the entire file is included.\n"
+        "**Ensuring comprehensive inclusion**\n"
+        "I'm making sure the entire file is included when presenting the result in a block, following the user's request carefully.\n"
+        "**Ensuring clarity**\n"
+        "I’m integrating the diff into the file and ensuring the entire file is returned as requested. This approach maintains precision and clarity in the response.\n"
+        "**Refining the response**\n"
+        "I’m focusing on how to structure the response by carefully integrating the diff and ensuring the entire file is included in a clear block format.\n"
+        "**Connecting the pieces**\n"
+        "I'm mapping out how to apply the diff to the file carefully and ensure the entire file is incorporated into the final block.\n"
+        "Reasoned for a few seconds\n"
+        "\n"
+        "```diff\n"
+        "--- a/file.py\n"
+        "+++ b/file.py\n"
+        "@@ -1,2 +1,2 @@\n"
+        "-def old():\n"
+        "+def new():\n"
+        "```"
+    )
+    final_content, reasoning = swallow_reasoning(llm_response)
+
+    expected_reasoning = (
+        "> Reasoning\n"
+        "**Considering the request**\n"
+        "I’m noting that the user wants me to apply a diff to a file and return the result in a block, ensuring the entire file is included.\n"
+        "**Ensuring comprehensive inclusion**\n"
+        "I'm making sure the entire file is included when presenting the result in a block, following the user's request carefully.\n"
+        "**Ensuring clarity**\n"
+        "I’m integrating the diff into the file and ensuring the entire file is returned as requested. This approach maintains precision and clarity in the response.\n"
+        "**Refining the response**\n"
+        "I’m focusing on how to structure the response by carefully integrating the diff and ensuring the entire file is included in a clear block format.\n"
+        "**Connecting the pieces**\n"
+        "I'm mapping out how to apply the diff to the file carefully and ensure the entire file is incorporated into the final block.\n"
+        "Reasoned for a few 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():
+    llm_response = (
+        "> 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\n"
+        "\n"
+        "```diff\n"
+        "--- a/file.py\n"
+        "+++ b/file.py\n"
+        "@@ -1,2 +1,2 @@\n"
+        "-def old():\n"
+        "+def new():\n"
+        "```"
+    )
+    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"
+    )
+    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_no_reasoning():
+    llm_response = (
+        "```diff\n"
+        "--- a/file.py\n"
+        "+++ b/file.py\n"
+        "@@ -1,2 +1,2 @@\n"
+        "-def old():\n"
+        "+def new():\n"
+        "```"
+    )
+    final_content, reasoning = swallow_reasoning(llm_response)
+    assert reasoning == ""
+    assert final_content == llm_response.strip()
+
+def test_swallow_reasoning_inline_newlines():
+    llm_response = (
+        "Prefix text before reasoning and some inline content "
+        "> Reasoning\n"
+        "Inline line 1\n"
+        "Inline line 2\n"
+        "Reasoned for 2 seconds "
+        "and then suffix text.\n"
+        "```diff\n"
+        "--- a/inline.py\n"
+        "+++ b/inline.py\n"
+        "@@ -1,2 +1,2 @@\n"
+        "-print('Old')\n"
+        "+print('New')\n"
+        "```"
+    )
+    final_content, reasoning = swallow_reasoning(llm_response)
+    expected_reasoning = (
+        "> Reasoning\n"
+        "Inline line 1\n"
+        "Inline line 2\n"
+        "Reasoned for 2 seconds"
+    )
+    # Count the newlines in the extracted reasoning block.
+    newline_count = reasoning.count('\n')
+    # There should be 3 newline characters: after "> Reasoning", after "Inline line 1", and after "Inline line 2"
+    assert newline_count == 3, f"Expected 3 newlines, got {newline_count}"
+    assert reasoning == expected_reasoning
+    # Ensure the reasoning block is removed from the final content.
+    assert expected_reasoning not in final_content
+    # Verify that surrounding content remains.
+    assert "Prefix text before reasoning" in final_content
+    assert "and then suffix text." in final_content
+    # Verify that the diff block is still present.
+    assert "```diff" in final_content

gptdiff-0.1.20/tests/test_swallow_reasoning.py

@@ -1,51 +0,0 @@
-import pytest
-
-from gptdiff.gptdiff import swallow_reasoning
-
-
-def test_swallow_reasoning_extraction():
-    llm_response = (
-        "> 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\n"
-        "\n"
-        "```diff\n"
-        "--- a/file.py\n"
-        "+++ b/file.py\n"
-        "@@ -1,2 +1,2 @@\n"
-        "-def old():\n"
-        "+def new():\n"
-        "```"
-    )
-    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"
-    )
-    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_no_reasoning():
-    llm_response = (
-        "```diff\n"
-        "--- a/file.py\n"
-        "+++ b/file.py\n"
-        "@@ -1,2 +1,2 @@\n"
-        "-def old():\n"
-        "+def new():\n"
-        "```"
-    )
-    final_content, reasoning = swallow_reasoning(llm_response)
-    assert reasoning == ""
-    assert final_content == llm_response.strip()