tass 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

src/app.py

@@ -1,6 +1,7 @@
 import json
 import os
 import subprocess
+from pathlib import Path
 
 import requests
 from rich.console import Console
@@ -56,7 +57,7 @@ class TassApp:
             console.print(f"[red]Unable to verify new host {self.host}. Continuing with it anyway.[/red]")
 
     def summarize(self):
-        max_messages = 10
+        max_messages = 20
         if len(self.messages) <= max_messages:
             return
 
@@ -124,16 +125,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 +166,52 @@ 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")])
-        replace_with_pluses = "\n".join([f"+{line}" for line in replace.split("\n")])
+    def edit_file(self, path: str, edits: list[dict]) -> str:
+        for edit in edits:
+            edit["applied"] = False
+
+        def find_edit(n: int) -> dict | None:
+            for edit in edits:
+                if edit["line_start"] <= n <= edit["line_end"]:
+                    return edit
+
+            return None
+
+        file_exists = Path(path).exists()
+        if file_exists:
+            with open(path, "r") as f:
+                original_content = f.read()
+        else:
+            original_content = ""
+
+        final_lines = []
+        original_lines = original_content.split("\n")
+        diff_text = f"{'Editing' if file_exists else 'Creating'} {path}"
+        for i, line in enumerate(original_lines):
+            line_num = i + 1
+            edit = find_edit(line_num)
+            if not edit:
+                final_lines.append(line)
+                continue
+
+            if edit["applied"]:
+                continue
+
+            replace_lines = edit["replace"].split("\n")
+            final_lines.extend(replace_lines)
+            original_lines = original_content.split("\n")
+            replaced_lines = original_lines[edit["line_start"] - 1:edit["line_end"]]
+
+            prev_line_num = line_num if line_num == 1 else line_num - 1
+            line_before = "" if i == 0 else f" {original_lines[i - 1]}\n"
+            line_after = "" if i == len(original_lines) - 1 else f"\n {original_lines[i + 1]}"
+            replaced_with_minuses = "\n".join([f"-{line}" for line in replaced_lines]) if file_exists else ""
+            replace_with_pluses = "\n".join([f"+{line}" for line in edit["replace"].split("\n")])
+            diff_text = f"{diff_text}\n\n@@ -{prev_line_num},{len(replaced_lines)} +{prev_line_num},{len(replace_lines)} @@\n{line_before}{replaced_with_minuses}\n{replace_with_pluses}{line_after}"
+            edit["applied"] = True
+
         console.print()
-        console.print(Markdown(f"```diff\nEditing {path}\n{find_with_minuses}\n{replace_with_pluses}\n```"))
+        console.print(Markdown(f"```diff\n{diff_text}\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,18 +219,8 @@ 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)
+                f.write("\n".join(final_lines))
         except Exception as e:
             console.print("   [red]edit_file failed[/red]")
             console.print(f"   [red]{str(e)}[/red]")

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 (or creates) a file. Makes multiple replacements in one call. Each edit removes the contents between 'line_start' and 'line_end' inclusive and replaces it with 'replace'. If creating a file, only return a single edit where the line_start and line_end are both 1 and replace is the entire contents of the file.",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -45,16 +45,30 @@ TOOLS = [
                         "type": "string",
                         "description": "Relative path of the file",
                     },
-                    "find": {
-                        "type": "string",
-                        "description": "The string to find",
-                    },
-                    "replace": {
-                        "type": "string",
-                        "description": "The string to replace with",
+                    "edits": {
+                        "type": "array",
+                        "description": "List of edits to apply. Each edit must contain 'line_start', 'line_end', and 'replace'.",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "line_start": {
+                                    "type": "integer",
+                                    "description": "The first line to remove (inclusive)",
+                                },
+                                "line_end": {
+                                    "type": "integer",
+                                    "description": "The last line to remove (inclusive)",
+                                },
+                                "replace": {
+                                    "type": "string",
+                                    "description": "The string to replace with. Must have the correct spacing and indentation for all lines.",
+                                },
+                            },
+                            "required": ["line_start", "line_end", "replace"],
+                        },
                     },
                 },
-                "required": ["path", "find", "replace"],
+                "required": ["path", "edits"],
                 "$schema": "http://json-schema.org/draft-07/schema#",
             },
         },
@@ -63,7 +77,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.4.dist-info → tass-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.4
+Version: 0.1.6
 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
@@ -33,4 +33,4 @@ tass
 
 tass has only been tested with gpt-oss-120b using llama.cpp so far, but in theory any LLM with tool calling capabilities should work. By default, it will try connecting to http://localhost:8080. If you want to use another host, set the `TASS_HOST` environment variable.
 
-Once it's running, you can ask questions like "Can you create an empty file called test.txt?" and it will propose a command to run after user confirmation.
+Once it's running, you can ask questions or give commands like "Create an empty file called test.txt" and it will propose a command to run after user confirmation.

tass-0.1.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
+src/app.py,sha256=SZGIStkRskTraOARKR-sh8hjfQT7EXwJBG-oymIABhU,11466
+src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
+src/constants.py,sha256=2MWn3-tvZjJ2xW68BE7S1V8CgqDuBt3cBG5Bx8ILrKY,4620
+src/utils.py,sha256=rKq34DVmFbsWPy7R6Bfdvv1ztzFLPT4hUd8BFpPHjqs,681
+tass-0.1.6.dist-info/METADATA,sha256=xu-OHc1sIrlDrxXVCUSdPmzYnjVam5tYB96EYJiTWCc,1079
+tass-0.1.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+tass-0.1.6.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
+tass-0.1.6.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+tass-0.1.6.dist-info/RECORD,,

tass-0.1.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
-src/app.py,sha256=sTPTwDc8KjF9RpL7W5Ix0D0n8ERcWsP1WEVzLWGOKco,9779
-src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
-src/constants.py,sha256=A0_PwEYo1Dpf8UC6HV7JgQwU7GlZQxKhLgwgmyoc3WY,3478
-src/utils.py,sha256=rKq34DVmFbsWPy7R6Bfdvv1ztzFLPT4hUd8BFpPHjqs,681
-tass-0.1.4.dist-info/METADATA,sha256=K1hq1AlWam-s671Nhmb0-X4jj6qAOi8Lfw27wuDxAXA,1071
-tass-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-tass-0.1.4.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
-tass-0.1.4.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-tass-0.1.4.dist-info/RECORD,,