tass 0.1.9__tar.gz → 0.1.10__tar.gz

{tass-0.1.9 → tass-0.1.10}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.9
+Version: 0.1.10
 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
 License: MIT
 License-File: LICENSE
 Requires-Python: >=3.10
+Requires-Dist: prompt-toolkit>=3.0.52
 Requires-Dist: requests>=2.32.5
 Requires-Dist: rich>=14.2.0
 Description-Content-Type: text/markdown

{tass-0.1.9 → tass-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tass"
-version = "0.1.9"
+version = "0.1.10"
 description = "A terminal assistant that allows you to ask an LLM to run commands."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -8,6 +8,7 @@ license = {text = "MIT"}
 authors = [{name = "Can Cetin"}]
 
 dependencies = [
+    "prompt-toolkit>=3.0.52",
     "requests>=2.32.5",
     "rich>=14.2.0",
 ]

{tass-0.1.9 → tass-0.1.10}/src/app.py

@@ -4,7 +4,7 @@ import subprocess
 from pathlib import Path
 
 import requests
-
+from prompt_toolkit import prompt
 from rich.console import Console, Group
 from rich.live import Live
 from rich.markdown import Markdown
@@ -15,7 +15,11 @@ from src.constants import (
     SYSTEM_PROMPT,
     TOOLS,
 )
-from src.utils import is_read_only_command
+from src.utils import (
+    FileCompleter,
+    create_key_bindings,
+    is_read_only_command,
+)
 
 console = Console()
 
@@ -25,6 +29,8 @@ class TassApp:
     def __init__(self):
         self.messages: list[dict] = [{"role": "system", "content": SYSTEM_PROMPT}]
         self.host = os.environ.get("TASS_HOST", "http://localhost:8080")
+        self.key_bindings = create_key_bindings()
+        self.file_completer = FileCompleter()
         self.TOOLS_MAP = {
             "execute": self.execute,
             "read_file": self.read_file,
@@ -193,9 +199,9 @@ class TassApp:
         self.messages.append(
             {
                 "role": "assistant",
-                "content": content.strip() or None,
-                "reasoning_content": reasoning_content.strip() or None,
-                "tool_calls": list(tool_calls_map.values()) or None,
+                "content": content.strip(),
+                "reasoning_content": reasoning_content.strip(),
+                "tool_calls": list(tool_calls_map.values()) or [],
             }
         )
 
@@ -220,11 +226,12 @@ class TassApp:
             self.messages.append({"role": "user", "content": str(e)})
             return self.call_llm()
 
-    def read_file(self, path: str, start: int = 1) -> str:
-        if start == 1:
+    def read_file(self, path: str, start: int = 1, num_lines: int = 1000) -> str:
+        if start == 1 and num_lines == 1000:
             console.print(f" └ Reading file [bold]{path}[/]...")
         else:
-            console.print(f" └ Reading file [bold]{path}[/] (from line {start})...")
+            last_line = start + num_lines - 1
+            console.print(f" └ Reading file [bold]{path}[/] (lines {start}-{last_line})...")
 
         try:
             result = subprocess.run(
@@ -256,14 +263,15 @@ class TassApp:
             lines.append(line)
             line_num += 1
 
-            if len(lines) >= 1000:
+            if len(lines) >= num_lines:
                 lines.append("... (truncated)")
                 break
 
         console.print("   [green]Command succeeded[/green]")
-        return "".join(lines)
+        return "\n".join(lines)
 
     def edit_file(self, path: str, edits: list[dict]) -> str:
+        console.print(json.dumps(edits, indent=2))
         for edit in edits:
             edit["applied"] = False
 
@@ -294,8 +302,9 @@ class TassApp:
             if edit["applied"]:
                 continue
 
-            replace_lines = edit["replace"].split("\n")
-            final_lines.extend(replace_lines)
+            replace_lines = edit["content"].split("\n")
+            if edit["content"]:
+                final_lines.extend(replace_lines)
             original_lines = original_content.split("\n")
             replaced_lines = original_lines[edit["line_start"] - 1:edit["line_end"]]
 
@@ -303,8 +312,10 @@ class TassApp:
             line_before = "" if i == 0 else f" {original_lines[i - 1]}\n"
             line_after = "" if edit["line_end"] == len(original_lines) else f"\n {original_lines[edit['line_end']]}"
             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}"
+            replaced_with_pluses = ""
+            if edit["content"]:
+                replaced_with_pluses = "\n" + "\n".join([f"+{line}" for line in edit["content"].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}{replaced_with_pluses}{line_after}"
             edit["applied"] = True
 
         console.print()
@@ -393,12 +404,16 @@ class TassApp:
             try:
                 input_lines = []
                 while True:
-                    input_line = console.input("> ")
+                    input_line = prompt(
+                        "> ",
+                        completer=self.file_completer,
+                        complete_while_typing=True,
+                        key_bindings=self.key_bindings,
+                    )
                     if not input_line or input_line[-1] != "\\":
                         input_lines.append(input_line)
                         break
                     input_lines.append(input_line[:-1])
-
                 user_input = "\n".join(input_lines)
             except KeyboardInterrupt:
                 console.print("\nBye!")

{tass-0.1.9 → tass-0.1.10}/src/constants.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-_cwd_path = Path.cwd().resolve()
+cwd_path = Path.cwd().resolve()
 
 SYSTEM_PROMPT = f"""You are tass, or Terminal Assistant, a helpful AI that executes shell commands based on natural-language requests.
 
@@ -8,7 +8,7 @@ If the user's request involves making changes to the filesystem such as creating
 
 If a user asks for an answer or explanation to something instead of requesting to run a command, answer briefly and concisely. Do not supply extra information, suggestions, tips, or anything of the sort.
 
-Current working directory: {_cwd_path}"""
+Current working directory: {cwd_path}"""
 
 TOOLS = [
     {
@@ -37,7 +37,7 @@ TOOLS = [
         "type": "function",
         "function": {
             "name": "edit_file",
-            "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.",
+            "description": "Edits (or creates) a file. Can make multiple edits in one call. Each edit replaces the contents between 'line_start' and 'line_end' inclusive with 'content'. If creating a file, only return a single edit where 'line_start' and 'line_end' are both 1 and 'content' is the entire contents of the file. You must use the read_file tool before editing a file.",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -47,7 +47,7 @@ TOOLS = [
                     },
                     "edits": {
                         "type": "array",
-                        "description": "List of edits to apply. Each edit must contain 'line_start', 'line_end', and 'replace'.",
+                        "description": "List of edits to apply. Each edit must contain 'line_start', 'line_end', and 'content'.",
                         "items": {
                             "type": "object",
                             "properties": {
@@ -59,12 +59,12 @@ TOOLS = [
                                     "type": "integer",
                                     "description": "The last line to remove (inclusive)",
                                 },
-                                "replace": {
+                                "content": {
                                     "type": "string",
-                                    "description": "The string to replace with. Must have the correct spacing and indentation for all lines.",
+                                    "description": "The content to replace with. Must have the correct spacing and indentation for all lines.",
                                 },
                             },
-                            "required": ["line_start", "line_end", "replace"],
+                            "required": ["line_start", "line_end", "content"],
                         },
                     },
                 },
@@ -77,7 +77,7 @@ TOOLS = [
         "type": "function",
         "function": {
             "name": "read_file",
-            "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.",
+            "description": "Read a file's contents (the first 1000 lines by default). When reading a file for the first time, do not change the defaults and always read the first 1000 lines unless you are absolutely certain of which lines need to be read. The output will be identical to calling `cat -n <path>` with preceding spaces, line number and a tab.",
             "parameters": {
                 "type": "object",
                 "properties": {
@@ -90,6 +90,11 @@ TOOLS = [
                         "description": "Which line to start reading from",
                         "default": 1,
                     },
+                    "num_lines": {
+                        "type": "integer",
+                        "description": "Number of lines to read, defaults to 1000",
+                        "default": 1000,
+                    },
                 },
                 "required": ["path"],
                 "$schema": "http://json-schema.org/draft-07/schema#",

tass-0.1.10/src/utils.py

@@ -0,0 +1,98 @@
+from prompt_toolkit.completion import (
+    Completer,
+    Completion,
+    CompleteEvent,
+)
+from prompt_toolkit.document import Document
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.keys import Keys
+
+from src.constants import (
+    READ_ONLY_COMMANDS,
+    cwd_path,
+)
+
+def is_read_only_command(command: str) -> bool:
+    """A simple check to see if the command is only for reading files.
+
+    Not a comprehensive or foolproof check by any means, and will
+    return false negatives to be safe.
+    """
+    if ">" in command:
+        return False
+
+    # Replace everything that potentially runs another command with a pipe
+    command = command.replace("&&", "|")
+    command = command.replace("||", "|")
+    command = command.replace(";", "|")
+
+    pipes = command.split("|")
+    for pipe in pipes:
+        if pipe.strip().split()[0] not in READ_ONLY_COMMANDS:
+            return False
+
+    return True
+
+
+class FileCompleter(Completer):
+
+    def get_completions(self, document: Document, complete_event: CompleteEvent) -> list[Completion]:
+        """Return file completions for text after @ anywhere in input."""
+        text = document.text
+        cursor_pos = document.cursor_position
+        text_before_cursor = document.text_before_cursor
+        last_at_pos = text_before_cursor.rfind("@")
+        if last_at_pos == -1:
+            return []
+
+        text_after_at = text[last_at_pos + 1 : cursor_pos]
+
+        base_dir = cwd_path
+        search_pattern = text_after_at
+        prefix = ""
+
+        if "/" in text_after_at:
+            dir_part, file_part = text_after_at.rsplit("/", 1)
+            base_dir = cwd_path / dir_part
+            search_pattern = file_part
+            prefix = dir_part + "/"
+
+        if not base_dir.is_dir():
+            return []
+
+        try:
+            dirs = []
+            files = []
+            for entry in sorted(base_dir.iterdir()):
+                name = entry.name
+                if search_pattern in name:
+                    full_path = prefix + name
+                    if entry.is_dir():
+                        dirs.append(Completion(full_path, display=name, start_position=-len(text_after_at), style="Blue"))
+                    else:
+                        files.append(Completion(full_path, display=name, start_position=-len(text_after_at)))
+
+            return dirs + files
+        except (PermissionError, OSError):
+            return []
+
+
+def create_key_bindings() -> KeyBindings:
+    bindings = KeyBindings()
+
+    @bindings.add(Keys.Backspace)
+    def backspace_with_completion(event):
+        """Also trigger completion on backspace."""
+        buffer = event.current_buffer
+        buffer.delete_before_cursor(count=1)
+        text_before_cursor = buffer.document.text_before_cursor
+        last_at_pos = text_before_cursor.rfind("@")
+        if last_at_pos != -1:
+            buffer.start_completion()
+
+    @bindings.add(Keys.ControlC)
+    def ctrl_c_handler(event):
+        """Ctrl+C should raise KeyboardInterrupt."""
+        raise KeyboardInterrupt()
+
+    return bindings

{tass-0.1.9 → tass-0.1.10}/uv.lock

@@ -178,6 +178,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/54/20/4d324d65cc6d9205fabedc306948156824eb9f0ee1633355a8f7ec5c66bf/pluggy-1.6.0-py3-none-any.whl", hash = "sha256:e920276dd6813095e9377c0bc5566d94c932c33b27a3e3945d8389c374dd4746", size = 20538, upload-time = "2025-05-15T12:30:06.134Z" },
 ]
 
+[[package]]
+name = "prompt-toolkit"
+version = "3.0.52"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "wcwidth" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/a1/96/06e01a7b38dce6fe1db213e061a4602dd6032a8a97ef6c1a862537732421/prompt_toolkit-3.0.52.tar.gz", hash = "sha256:28cde192929c8e7321de85de1ddbe736f1375148b02f2e17edd840042b1be855", size = 434198, upload-time = "2025-08-27T15:24:02.057Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/84/03/0d3ce49e2505ae70cf43bc5bb3033955d2fc9f932163e84dc0779cc47f48/prompt_toolkit-3.0.52-py3-none-any.whl", hash = "sha256:9aac639a3bbd33284347de5ad8d68ecc044b91a762dc39b7c21095fcd6a19955", size = 391431, upload-time = "2025-08-27T15:23:59.498Z" },
+]
+
 [[package]]
 name = "pygments"
 version = "2.19.2"
@@ -261,9 +273,10 @@ wheels = [
 
 [[package]]
 name = "tass"
-version = "0.1.9"
+version = "0.1.10"
 source = { editable = "." }
 dependencies = [
+    { name = "prompt-toolkit" },
     { name = "requests" },
     { name = "rich" },
 ]
@@ -276,6 +289,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
+    { name = "prompt-toolkit", specifier = ">=3.0.52" },
     { name = "requests", specifier = ">=2.32.5" },
     { name = "rich", specifier = ">=14.2.0" },
 ]
@@ -352,3 +366,12 @@ sdist = { url = "https://files.pythonhosted.org/packages/1e/24/a2a2ed9addd907787
 wheels = [
     { url = "https://files.pythonhosted.org/packages/6d/b9/4095b668ea3678bf6a0af005527f39de12fb026516fb3df17495a733b7f8/urllib3-2.6.2-py3-none-any.whl", hash = "sha256:ec21cddfe7724fc7cb4ba4bea7aa8e2ef36f607a4bab81aa6ce42a13dc3f03dd", size = 131182, upload-time = "2025-12-11T15:56:38.584Z" },
 ]
+
+[[package]]
+name = "wcwidth"
+version = "0.2.14"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/24/30/6b0809f4510673dc723187aeaf24c7f5459922d01e2f794277a3dfb90345/wcwidth-0.2.14.tar.gz", hash = "sha256:4d478375d31bc5395a3c55c40ccdf3354688364cd61c4f6adacaa9215d0b3605", size = 102293, upload-time = "2025-09-22T16:29:53.023Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/af/b5/123f13c975e9f27ab9c0770f514345bd406d0e8d3b7a0723af9d43f710af/wcwidth-0.2.14-py2.py3-none-any.whl", hash = "sha256:a7bb560c8aee30f9957e5f9895805edd20602f2d7f720186dfd906e82b4982e1", size = 37286, upload-time = "2025-09-22T16:29:51.641Z" },
+]

tass-0.1.9/src/utils.py

@@ -1,23 +0,0 @@
-from src.constants import READ_ONLY_COMMANDS
-
-
-def is_read_only_command(command: str) -> bool:
-    """A simple check to see if the command is only for reading files.
-
-    Not a comprehensive or foolproof check by any means, and will
-    return false negatives to be safe.
-    """
-    if ">" in command:
-        return False
-
-    # Replace everything that potentially runs another command with a pipe
-    command = command.replace("&&", "|")
-    command = command.replace("||", "|")
-    command = command.replace(";", "|")
-
-    pipes = command.split("|")
-    for pipe in pipes:
-        if pipe.strip().split()[0] not in READ_ONLY_COMMANDS:
-            return False
-
-    return True