tass 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

src/app.py

@@ -124,16 +124,30 @@ class TassApp:
             return self.call_llm()
 
     def read_file(self, path: str, start: int = 1) -> str:
-        console.print(f" └ Reading file [bold]{path}[/]...")
+        if start == 1:
+            console.print(f" └ Reading file [bold]{path}[/]...")
+        else:
+            console.print(f" └ Reading file [bold]{path}[/] (from line {start})...")
 
         try:
-            with open(path) as f:
-                out = f.read()
+            result = subprocess.run(
+                f"cat -n {path}",
+                shell=True,
+                capture_output=True,
+                text=True,
+            )
         except Exception as e:
             console.print("   [red]read_file failed[/red]")
             console.print(f"   [red]{str(e)}[/red]")
             return f"read_file failed: {str(e)}"
 
+        out = result.stdout
+        err = result.stderr
+        if result.returncode != 0:
+            console.print("   [red]read_file failed[/red]")
+            console.print(f"   [red]{err}[/red]")
+            return f"read_file failed: {err}"
+
         lines = []
         line_num = 1
         for line in out.split("\n"):
@@ -151,11 +165,22 @@ class TassApp:
         console.print("   [green]Command succeeded[/green]")
         return "".join(lines)
 
-    def edit_file(self, path: str, find: str, replace: str) -> str:
-        find_with_minuses = "\n".join([f"-{line}" for line in find.split("\n")])
+    def edit_file(self, path: str, line_start: int, line_end: int, replace: str) -> str:
+        with open(path, "r") as f:
+            original_content = f.read()
+
+        original_lines = original_content.split("\n")
+        replaced_lines = original_lines[line_start - 1:line_end]
+        new_content = "\n".join(
+            original_lines[:line_start - 1]
+            + replace.split("\n")
+            + original_lines[line_end:]
+        )
+
+        replaced_with_minuses = "\n".join([f"-{line}" for line in replaced_lines])
         replace_with_pluses = "\n".join([f"+{line}" for line in replace.split("\n")])
         console.print()
-        console.print(Markdown(f"```diff\nEditing {path}\n{find_with_minuses}\n{replace_with_pluses}\n```"))
+        console.print(Markdown(f"```diff\nEditing {path}\n{replaced_with_minuses}\n{replace_with_pluses}\n```"))
         answer = console.input("\n[bold]Run?[/] ([bold]Y[/]/n): ").strip().lower()
         if answer not in ("yes", "y", ""):
             reason = console.input("Why not? (optional, press Enter to skip): ").strip()
@@ -163,16 +188,6 @@ class TassApp:
 
         console.print(" └ Running...")
         try:
-            with open(path, "r") as f:
-                original_content = f.read()
-
-            if find not in original_content:
-                console.print("   [red]edit_file failed[/red]")
-                console.print(f"   [red]edit_file failed:\n'{find}'\nnot found in file[/red]")
-                return f"edit_file failed: '{find}' not found in file"
-
-            new_content = original_content.replace(find, replace)
-
             with open(path, "w") as f:
                 f.write(new_content)
         except Exception as e:
@@ -221,14 +236,20 @@ class TassApp:
             console.print(f"   [red]Command failed[/red] (code {result.returncode})")
             console.print(f"   [red]{err}[/red]")
 
-        if len(out.split("\n")) > 5000:
+        if len(out.split("\n")) > 1000:
             out_first_1000 = "\n".join(out.split("\n")[:1000])
             out = f"{out_first_1000}... (Truncated)"
 
-        if len(err) > 5000:
+        if len(err.split("\n")) > 1000:
             err_first_1000 = "\n".join(err.split("\n")[:1000])
             err = f"{err_first_1000}... (Truncated)"
 
+        if len(out) > 20000:
+            out = f"{out[:20000]}... (Truncated)"
+
+        if len(err) > 20000:
+            err = f"{err[:20000]}... (Truncated)"
+
         return f"Command output (exit {result.returncode}):\n{out}\n{err}"
 
     def run(self):

src/constants.py

@@ -37,7 +37,7 @@ TOOLS = [
         "type": "function",
         "function": {
             "name": "edit_file",
-            "description": "Edits a file. Replaces the instance of 'find' with 'replace'. 'find' and 'replace' are exact strings.The file must be read at least once before calling this tool.",
+            "description": "Edits a file. Removes the contents between the lines 'line_start' and 'line_end' inclusive entirely and replaces it with 'replace'.",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -45,13 +45,17 @@ TOOLS = [
                         "type": "string",
                         "description": "Relative path of the file",
                     },
-                    "find": {
-                        "type": "string",
-                        "description": "The string to find",
+                    "line_start": {
+                        "type": "integer",
+                        "description": "The first line to remove the contents of (inclusive)",
+                    },
+                    "line_end": {
+                        "type": "integer",
+                        "description": "The last line to remove the contents of (inclusive)",
                     },
                     "replace": {
                         "type": "string",
-                        "description": "The string to replace with",
+                        "description": "The string to replace with. Must have the correct spacing and indentation for all lines.",
                     },
                 },
                 "required": ["path", "find", "replace"],
@@ -63,7 +67,7 @@ TOOLS = [
         "type": "function",
         "function": {
             "name": "read_file",
-            "description": "Read a file's contents (up to 1000 lines)",
+            "description": "Read a file's contents (up to 1000 lines). The output will be identical to calling `cat -n <path>` with preceding spaces, line number and a tab.",
             "parameters": {
                 "type": "object",
                 "properties": {

{tass-0.1.3.dist-info → tass-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.3
+Version: 0.1.5
 Summary: A terminal assistant that allows you to ask an LLM to run commands.
 Project-URL: Homepage, https://github.com/cetincan0/tass
 Author: Can Cetin

tass-0.1.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
+src/app.py,sha256=Jej2qjyBRqQREpvm6WWvd6cPy9RLzW74rYegeM8ICPI,10224
+src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
+src/constants.py,sha256=gFIMWh38-uyh2XJdiKUsOwAh7yk4jbdfxmeJZ9yl4fw,3847
+src/utils.py,sha256=rKq34DVmFbsWPy7R6Bfdvv1ztzFLPT4hUd8BFpPHjqs,681
+tass-0.1.5.dist-info/METADATA,sha256=HskJ2m7qsulvtK_N5nZ_aJAcq1PH5zGMDnYMm_cjda8,1071
+tass-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+tass-0.1.5.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
+tass-0.1.5.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+tass-0.1.5.dist-info/RECORD,,

tass-0.1.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
-src/app.py,sha256=jy-6wYh-Wb2Z1DOVWKc7stEX3gtt6dM9BfJOv6ZG9SE,9607
-src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
-src/constants.py,sha256=A0_PwEYo1Dpf8UC6HV7JgQwU7GlZQxKhLgwgmyoc3WY,3478
-src/utils.py,sha256=rKq34DVmFbsWPy7R6Bfdvv1ztzFLPT4hUd8BFpPHjqs,681
-tass-0.1.3.dist-info/METADATA,sha256=uS8VJaj57K9T-LlxZkOy1iQdCSTu_UrLhIocIihJtew,1071
-tass-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-tass-0.1.3.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
-tass-0.1.3.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-tass-0.1.3.dist-info/RECORD,,