wcgw 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

wcgw/client/openai_client.py

@@ -185,6 +185,7 @@ def loop(
 - Interact with running program using this tool
 - Special keys like arrows, interrupts, enter, etc.
 - Send text input to the running program.
+- Send send_specials=["Enter"] to recheck status of a running program.
 - Only one of send_text, send_specials, send_ascii should be provided.""",
         ),
         openai.pydantic_function_tool(

wcgw/client/tools.py

@@ -453,6 +453,14 @@ def find_least_edit_distance_substring(
     content_lines = [
         line.strip() for line in orig_content_lines
     ]  # Remove trailing and leading space for calculating edit distance
+    new_to_original_indices = {}
+    new_content_lines = []
+    for i in range(len(content_lines)):
+        if not content_lines[i]:
+            continue
+        new_content_lines.append(content_lines[i])
+        new_to_original_indices[len(new_content_lines) - 1] = i
+    content_lines = new_content_lines
     find_lines = find_str.split("\n")
     find_lines = [
         line.strip() for line in find_lines
@@ -470,8 +478,9 @@ def find_least_edit_distance_substring(
                 edit_distance_sum += len(find_lines[j])
         if edit_distance_sum < min_edit_distance:
             min_edit_distance = edit_distance_sum
-            min_edit_distance_lines = orig_content_lines[i: i + len(
-                find_lines)]
+            orig_start_index = new_to_original_indices[i]
+            orig_end_index = new_to_original_indices.get(i + len(find_lines) - 1, len(orig_content_lines) - 1) + 1
+            min_edit_distance_lines = orig_content_lines[orig_start_index:orig_end_index]
     return "\n".join(min_edit_distance_lines), min_edit_distance
 
 
@@ -481,12 +490,12 @@ def edit_content(content: str, find_lines: str, replace_with_lines: str) -> str:
         closest_match, min_edit_distance = find_least_edit_distance_substring(
             content, find_lines
         )
-        print(
-            f"Exact match not found, found with whitespace removed edit distance: {min_edit_distance}"
-        )
-        if min_edit_distance / len(find_lines) < 1 / 100:
-            print("Editing file with closest match")
+        if min_edit_distance == 0:
             return edit_content(content, closest_match, replace_with_lines)
+        else:
+            print(
+                f"Exact match not found, found with whitespace removed edit distance: {min_edit_distance}"
+            )
         raise Exception(
             f"Error: no match found for the provided `find_lines` in the file. Closest match:\n---\n{closest_match}\n---\nFile not edited"
         )

{wcgw-1.2.0.dist-info → wcgw-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 1.2.0
+Version: 1.2.1
 Summary: What could go wrong giving full shell access to chatgpt?
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>

{wcgw-1.2.0.dist-info → wcgw-1.2.1.dist-info}/RECORD

@@ -6,12 +6,12 @@ wcgw/client/anthropic_client.py,sha256=g5dj7tP_wkLMjQawil4PxhrEemlrIKYttWUXBI5SZ
 wcgw/client/cli.py,sha256=Oja42CHkVO8puqOXflko9NeephYCMa85aBmQTEjBZtI,932
 wcgw/client/common.py,sha256=grH-yV_4tnTQZ29xExn4YicGLxEq98z-HkEZwH0ReSg,1410
 wcgw/client/diff-instructions.txt,sha256=I_qroE6_aMpWtOeywPSDCTxmz-P6VPopdp1kR2r0D8w,1375
-wcgw/client/openai_client.py,sha256=Vsj8TzYj22fawU1LagAU7bn21Va50KXwpYZpGZ0BUZ0,17310
+wcgw/client/openai_client.py,sha256=Pg72I4UfUQ-xrOkKlr4w1F1rc4KRvG0gePDO6E25-FM,17381
 wcgw/client/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
-wcgw/client/tools.py,sha256=NeGIZ8AJ9-XAiW6gUXf7GWs1UJjw8f7ecxUBBEAtyEI,25994
+wcgw/client/tools.py,sha256=zAWK0qAhVyVtWsW_v4fq2aU-l1I4c9i819VhFBAQ6-8,26416
 wcgw/relay/serve.py,sha256=sMbERQTM1GhpnPWLzdtpcb23TIsgbP-ZuqVj-vjp5Rw,8186
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-1.2.0.dist-info/METADATA,sha256=1AJa0M-m_tvHNpANqak4-q7bKTgg07_qgjddGwQaOhs,5255
-wcgw-1.2.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-wcgw-1.2.0.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
-wcgw-1.2.0.dist-info/RECORD,,
+wcgw-1.2.1.dist-info/METADATA,sha256=CxTKyE6VJqJAlvnWY6RmxgrYmd__UVc3ciRfnd4-oSw,5255
+wcgw-1.2.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+wcgw-1.2.1.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
+wcgw-1.2.1.dist-info/RECORD,,