tass 0.1.5__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
 
@@ -165,22 +166,52 @@ class TassApp:
         console.print("   [green]Command succeeded[/green]")
         return "".join(lines)
 
-    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()
+    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")
-        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:]
-        )
+        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
 
-        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{replaced_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()
@@ -189,7 +220,7 @@ class TassApp:
         console.print(" └ Running...")
         try:
             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. Removes the contents between the lines 'line_start' and 'line_end' inclusive entirely and replaces it with 'replace'.",
+            "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,20 +45,30 @@ TOOLS = [
                         "type": "string",
                         "description": "Relative path of the file",
                     },
-                    "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. Must have the correct spacing and indentation for all lines.",
+                    "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#",
             },
         },

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.5
+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.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-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,,