tass 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

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,
@@ -64,11 +70,14 @@ class TassApp:
 
         prompt = (
             "The conversation is becoming long and might soon go beyond the "
-            "context limit. Please provide a concise summary of the conversation, "
-            "preserving all important details. Keep the summary short enough "
-            "to fit within a few paragraphs at the most."
+            "context limit. Please provide a detailed summary of the conversation, "
+            "preserving all important details. Make sure context is not lost so that "
+            "the conversation can continue without needing to reclarify anything. "
+            "You don't have to preserve entire contents of files that have been read "
+            " or edited, they can be read again if necessary."
         )
 
+        console.print("\n - Summarizing conversation...")
         response = requests.post(
             f"{self.host}/v1/chat/completions",
             json={
@@ -82,6 +91,7 @@ class TassApp:
         data = response.json()
         summary = data["choices"][0]["message"]["content"]
         self.messages = [self.messages[0], {"role": "assistant", "content": f"Summary of the conversation so far:\n{summary}"}]
+        console.print("   [green]Summarization completed[/green]")
 
     def call_llm(self) -> bool:
         response = requests.post(
@@ -137,15 +147,29 @@ class TassApp:
                     continue
 
                 chunk = json.loads(line.removeprefix("data:"))
+                if all(k in chunk.get("timings", {}) for k in ["prompt_n", "prompt_per_second", "predicted_n", "predicted_per_second"]):
+                    timings = chunk["timings"]
+                    timings_str = (
+                        f"Input: {timings['prompt_n']:,} tokens, {timings['prompt_per_second']:,.2f} tok/s | "
+                        f"Output: {timings['predicted_n']:,} tokens, {timings['predicted_per_second']:,.2f} tok/s"
+                    )
+
+                if chunk["choices"][0]["finish_reason"]:
+                    live.update(generate_layout())
+
                 delta = chunk["choices"][0]["delta"]
+                if not any([delta.get(key) for key in ["content", "reasoning_content", "tool_calls"]]):
+                    continue
+
+                if delta.get("reasoning_content"):
+                    reasoning_content += delta["reasoning_content"]
+                    live.update(generate_layout())
+
                 if delta.get("content"):
                     content += delta["content"]
                     live.update(generate_layout())
-                if delta.get("reasoning_content" ):
-                    reasoning_content += delta["reasoning_content"]
-                    live.update(generate_layout())
 
-                for tool_call_delta in delta.get("tool_calls", []):
+                for tool_call_delta in delta.get("tool_calls") or []:
                     index = tool_call_delta["index"]
                     if index not in tool_calls_map:
                         tool_calls_map[index] = (
@@ -172,22 +196,12 @@ class TassApp:
                         if function.get("arguments"):
                             tool_call["function"]["arguments"] += function["arguments"]
 
-                if all(k in chunk.get("timings", {}) for k in ["prompt_n", "prompt_per_second", "predicted_n", "predicted_per_second"]):
-                    timings = chunk["timings"]
-                    timings_str = (
-                        f"Input: {timings['prompt_n']:,} tokens, {timings['prompt_per_second']:,.2f} tok/s | "
-                        f"Output: {timings['predicted_n']:,} tokens, {timings['predicted_per_second']:,.2f} tok/s"
-                    )
-
-                if chunk["choices"][0]["finish_reason"]:
-                    live.update(generate_layout())
-
         self.messages.append(
             {
                 "role": "assistant",
-                "content": content,
-                "reasoning_content": reasoning_content,
-                "tool_calls": list(tool_calls_map.values()),
+                "content": content.strip(),
+                "reasoning_content": reasoning_content.strip(),
+                "tool_calls": list(tool_calls_map.values()) or [],
             }
         )
 
@@ -212,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(
@@ -248,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
 
@@ -286,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"]]
 
@@ -295,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()
@@ -385,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!")

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#",
@@ -113,4 +118,5 @@ READ_ONLY_COMMANDS = [
     "which",
     "sed",
     "find",
+    "test",
 ]

src/utils.py

@@ -1,5 +1,16 @@
-from src.constants import READ_ONLY_COMMANDS
+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.
@@ -21,3 +32,67 @@ def is_read_only_command(command: str) -> bool:
             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.8.dist-info → tass-0.1.10.dist-info}/METADATA

@@ -1,18 +1,23 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.8
+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
 
+<p align="center">
+  <img src="assets/tass.gif" alt="Demo" />
+</p>
+
 A terminal assistant that allows you to ask an LLM to run commands.
 
 ## Warning
@@ -39,7 +44,7 @@ You can run it with
 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.
+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. At the moment there's no support for connecting tass to a non-local API, nor are there plans for it. For the time being, I plan on keeping tass completely local. There's no telemetry, no logs, just a simple REPL loop.
 
 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.10.dist-info/RECORD

@@ -0,0 +1,10 @@
+src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
+src/app.py,sha256=45r-M59vjJb1TXdkfSkTVJq1waLyNYVLxkafav5t6oI,16472
+src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
+src/constants.py,sha256=LYNON4xoqCssh4wA7rjkjyY2JhDXUPFmkQlTz_N0oz8,5089
+src/utils.py,sha256=Uoi60eko9ivvnF-se68yAuwUDFEOHb9Y2SepDaOAbTU,3069
+tass-0.1.10.dist-info/METADATA,sha256=xG6XaXmXrLhDE1uo06tK1nXaSWYz2W8nA-Yx2-ZACHA,1717
+tass-0.1.10.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+tass-0.1.10.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
+tass-0.1.10.dist-info/licenses/LICENSE,sha256=Cdr-_YJHgGaf2vJjcoOsRJySkDaogUhu3yIDvpz7GEQ,1066
+tass-0.1.10.dist-info/RECORD,,

tass-0.1.8.dist-info/RECORD

@@ -1,10 +0,0 @@
-src/__init__.py,sha256=tu2q9W5_pkq30l3tRMTGahColBAAubbLP6LaB3l3IFg,89
-src/app.py,sha256=_ukqGrJdaaaZJiGM_XLNL7tZr2SzI_A-zlLRJrRI52I,15346
-src/cli.py,sha256=op3fYcyfek_KqCCiA-Zdlc9jVZSCi036whMmR2ZjjAs,76
-src/constants.py,sha256=pzriopu167r3yOOcnC80sMjPKEZoDmYV8e8i5aK0rvM,4629
-src/utils.py,sha256=rKq34DVmFbsWPy7R6Bfdvv1ztzFLPT4hUd8BFpPHjqs,681
-tass-0.1.8.dist-info/METADATA,sha256=w9zw9A2ARBv09FQpL-I2lRpbUd7PSGpeT51mhYIEksU,1392
-tass-0.1.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-tass-0.1.8.dist-info/entry_points.txt,sha256=pviKuIOuHvaQ7_YiFxatJEY8XYfh3EzVWy4LJh0v-A0,38
-tass-0.1.8.dist-info/licenses/LICENSE,sha256=Cdr-_YJHgGaf2vJjcoOsRJySkDaogUhu3yIDvpz7GEQ,1066
-tass-0.1.8.dist-info/RECORD,,