tass 0.1.12__tar.gz → 0.1.13__tar.gz

{tass-0.1.12 → tass-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.12
+Version: 0.1.13
 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
@@ -44,6 +44,12 @@ You can run it with
 tass
 ```
 
+or if you only want to ask/request a single thing
+
+```
+tass "convert video.mp4 to audio.mp3"
+```
+
 tass has only been tested with llama.cpp with LLMs such as gpt-oss-120b and MiniMax M2.1, but any LLM with tool calling capabilities should work.
 
 By default, tass will try connecting to http://localhost:8080. To use another host, set the `TASS_HOST` environment variable. If your server requires an API key, you can set the `TASS_API_KEY` environment variable. At the moment there's no support for connecting tass to a non-local API, nor are there plans for it. I plan on keeping tass completely local. There's no telemetry, no logs, just a simple REPL loop.
@@ -52,6 +58,8 @@ Once it's running, you can ask questions or give commands like "Create an empty
 
 You can enter multiline input by ending lines with a backslash (\\). The continuation prompt will keep appearing until you enter a line without a trailing backslash.
 
+You can use the --yolo flag to turn off user confirmations for executing commands and editing files, but I would only recommend using this if you're benchmarking tass with an LLM and highly recommend not using it outside of testing/benchmarking scenarios.
+
 ## Upgrade
 
 ### Using uv

{tass-0.1.12 → tass-0.1.13}/README.md

@@ -30,6 +30,12 @@ You can run it with
 tass
 ```
 
+or if you only want to ask/request a single thing
+
+```
+tass "convert video.mp4 to audio.mp3"
+```
+
 tass has only been tested with llama.cpp with LLMs such as gpt-oss-120b and MiniMax M2.1, but any LLM with tool calling capabilities should work.
 
 By default, tass will try connecting to http://localhost:8080. To use another host, set the `TASS_HOST` environment variable. If your server requires an API key, you can set the `TASS_API_KEY` environment variable. At the moment there's no support for connecting tass to a non-local API, nor are there plans for it. I plan on keeping tass completely local. There's no telemetry, no logs, just a simple REPL loop.
@@ -38,6 +44,8 @@ Once it's running, you can ask questions or give commands like "Create an empty
 
 You can enter multiline input by ending lines with a backslash (\\). The continuation prompt will keep appearing until you enter a line without a trailing backslash.
 
+You can use the --yolo flag to turn off user confirmations for executing commands and editing files, but I would only recommend using this if you're benchmarking tass with an LLM and highly recommend not using it outside of testing/benchmarking scenarios.
+
 ## Upgrade
 
 ### Using uv

{tass-0.1.12 → tass-0.1.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tass"
-version = "0.1.12"
+version = "0.1.13"
 description = "A terminal assistant that allows you to ask an LLM to run commands."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -35,8 +35,38 @@ dev = [
     "pytest>=9.0.2",
 ]
 
+[tool.ruff]
+target-version = "py310"
+line-length = 120
+
+[tool.ruff.lint]
+select = [
+    "A",
+    "B",
+    "C4",
+    "E",
+    "F",
+    "I",
+    "ISC",
+    "SIM",
+    "RUF",
+    "UP",
+    "W",
+]
+
+ignore = [
+    "E501",
+]
+
+[tool.ruff.lint.isort]
+known-first-party = ["tass"]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+
 [tool.pyright]
 pythonVersion = "3.10"
-include = ["src/**/*.py"]
+include = ["src/**/*.py", "tests/**/*.py"]
 venvPath = "."
 venv = ".venv"

{tass-0.1.12 → tass-0.1.13}/src/app.py

@@ -28,7 +28,8 @@ from src.utils import (
 
 class TassApp:
 
-    def __init__(self):
+    def __init__(self, yolo_mode: bool = False):
+        self.yolo_mode = yolo_mode
         self.messages: list[dict] = [{"role": "system", "content": SYSTEM_PROMPT}]
         self.llm_client = LLMClient()
         self.key_bindings = create_key_bindings()
@@ -80,7 +81,7 @@ class TassApp:
 
         console.print("\n - Summarizing conversation...")
         response = self.llm_client.get_chat_completions(
-            messages=self.messages + [{"role": "user", "content": prompt}],
+            messages=[*self.messages, {"role": "user", "content": prompt}],
             tools=[
                 EDIT_FILE_TOOL,
                 EXECUTE_TOOL,
@@ -154,7 +155,7 @@ class TassApp:
                     live.update(generate_layout())
 
                 delta = chunk["choices"][0]["delta"]
-                if not any([delta.get(key) for key in ["content", "reasoning_content", "tool_calls"]]):
+                if not any(delta.get(key) for key in ["content", "reasoning_content", "tool_calls"]):
                     continue
 
                 if delta.get("reasoning_content"):
@@ -208,6 +209,7 @@ class TassApp:
             for tool_call in tool_calls_map.values():
                 tool = self.TOOLS_MAP[tool_call["function"]["name"]]
                 tool_args = json.loads(tool_call["function"]["arguments"])
+                tool_args["yolo_mode"] = self.yolo_mode
                 result = tool(**tool_args)
                 self.messages.append(
                     {
@@ -223,7 +225,19 @@ class TassApp:
             console.print(f"   [red]Tool call failed: {str(e).strip()}[/red]")
             return self.call_llm()
 
-    def run(self):
+    def run(self, initial_input: str | None = None):
+        if initial_input:
+            self.messages.append({"role": "user", "content": initial_input})
+            while True:
+                try:
+                    finished = self.call_llm()
+                except Exception as e:
+                    console.print(f"Failed to call LLM: {e}")
+                    break
+
+                if finished:
+                    return
+
         try:
             self.check_llm_host()
         except KeyboardInterrupt:
@@ -258,12 +272,11 @@ class TassApp:
                 break
 
             self.messages.append({"role": "user", "content": user_input})
-
             while True:
                 try:
                     finished = self.call_llm()
                 except Exception as e:
-                    console.print(f"Failed to call LLM: {str(e)}")
+                    console.print(f"Failed to call LLM: {e}")
                     break
 
                 if finished:

tass-0.1.13/src/cli.py

@@ -0,0 +1,27 @@
+import argparse
+
+from src.app import TassApp
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Terminal Assistant - Ask an LLM to run commands"
+    )
+    parser.add_argument(
+        "--yolo",
+        action="store_true",
+        help="YOLO mode: execute all commands and edit files without asking for confirmation",
+    )
+    parser.add_argument(
+        "prompt",
+        nargs="?",
+        help="Prompt to run (enclose in quotes; runs in single-shot mode and exits)",
+    )
+    args = parser.parse_args()
+
+    app = TassApp(yolo_mode=args.yolo)
+
+    if args.prompt:
+        app.run(initial_input=args.prompt)
+    else:
+        app.run()

tass-0.1.13/src/constants.py

@@ -0,0 +1,73 @@
+import os
+import platform
+import subprocess
+from pathlib import Path
+
+from rich.console import Console
+
+console = Console()
+CWD_PATH = Path.cwd().resolve()
+
+
+def get_shell_info() -> str:
+    shell_path = os.environ.get('SHELL', '')
+    shell_name = shell_path.split('/')[-1] if shell_path else 'unknown'
+    return shell_name
+
+
+def get_git_info() -> str:
+    try:
+        git_info = []
+
+        subprocess.run(['git', 'rev-parse', '--git-dir'], capture_output=True, check=True, cwd=CWD_PATH)
+        result = subprocess.run(['git', 'branch', '--show-current'], capture_output=True, text=True, cwd=CWD_PATH)
+        branch = result.stdout.strip()
+        if branch:
+            git_info.append(f"Branch: {branch}")
+
+        result = subprocess.run(['git', 'status', '--porcelain'], capture_output=True, text=True, cwd=CWD_PATH)
+        has_changes = bool(result.stdout.strip())
+        git_info.append(f"Uncommitted changes: {'Yes' if has_changes else 'No'}")
+
+        return '\n'.join(git_info)
+    except Exception:
+        return "Not a git repository"
+
+
+def get_directory_listing() -> str:
+    try:
+        entries = []
+        iterdir = CWD_PATH.iterdir()
+        for count, entry in enumerate(sorted(iterdir)):
+            if count >= 100:
+                entries.append(f"... and {len(list(iterdir)) - 100} more items")
+                break
+            if entry.is_dir():
+                entries.append(f"{entry.name}/")
+            else:
+                entries.append(entry.name)
+        return '\n'.join(entries) if entries else "Empty directory"
+    except PermissionError:
+        return "Permission denied"
+
+
+SYSTEM_PROMPT = f"""You are tass, or Terminal Assistant, a helpful AI that executes shell commands based on natural-language requests.
+
+If the user's request involves making changes to the filesystem such as creating or deleting files or directories, you MUST first check whether the file or directory exists before proceeding.
+
+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.
+
+This app has a feature where the user can refer to files or directories by typing @ which will open an file autocomplete dropdown. When this feature is used, the @ will remain in the filename. When working with said file, ignore the preceding @.
+
+Current working directory: {CWD_PATH}
+
+# Directory Context
+OS: {platform.system()}
+Shell: {get_shell_info()}
+
+# Git Info
+{get_git_info()}
+
+# Directory Listing
+{get_directory_listing()}
+"""

{tass-0.1.12 → tass-0.1.13}/src/llm_client.py

@@ -14,7 +14,6 @@ class LLMClient:
         self,
         method: Literal["get", "post"],
         url: str,
-        *args,
         **kwargs,
     ):
         return requests.request(
@@ -23,15 +22,14 @@ class LLMClient:
             headers={
                 "Authorization": f"Bearer {self.api_key}",
             },
-            *args,
             **kwargs,
         )
 
-    def get(self, url: str, *args, **kwargs):
-        return self.request("get", url, *args, **kwargs)
+    def get(self, url: str, **kwargs):
+        return self.request("get", url, **kwargs)
 
-    def post(self, url: str, *args, **kwargs):
-        return self.request("post", url, *args, **kwargs)
+    def post(self, url: str, **kwargs):
+        return self.request("post", url, **kwargs)
 
     def get_models(self):
         return self.get("/v1/models", timeout=2)

{tass-0.1.12 → tass-0.1.13}/src/tools/edit_file.py

@@ -6,7 +6,6 @@ from rich.markdown import Markdown
 
 from src.constants import console
 
-
 EDIT_FILE_TOOL = {
     "type": "function",
     "function": {
@@ -80,7 +79,7 @@ def fuzzy_match(edit_find: str, lines: list[str]) -> tuple[int, int] | None:
                 if best_ratio == 1.0:
                     return best_start + 1, best_start + best_window_len
 
-    if best_ratio >= 0.95 and best_start >= 0:
+    if best_ratio >= 0.9:
         return best_start + 1, best_start + best_window_len
 
     return None
@@ -100,7 +99,9 @@ def convert_edit_to_line_edit(edit: dict, original_content: str) -> LineEdit:
 
     # Then try matching without spacing
     for i in range(len(lines)):
-        if [line.strip() for line in lines[i : i + len(edit_lines)]] == [line.strip() for line in edit_lines]:
+        stripped_lines = [line.strip() for line in lines[i : i + len(edit_lines)]]
+        stripped_edit_lines = [line.strip() for line in edit_lines]
+        if stripped_lines == stripped_edit_lines:
             return LineEdit(i + 1, i + len(edit_lines), edit["replace"], False)
 
     # Finally try sequence matching while ignoring whitespace
@@ -111,7 +112,7 @@ def convert_edit_to_line_edit(edit: dict, original_content: str) -> LineEdit:
     raise Exception("Edit not found in file")
 
 
-def edit_file(path: str, edits: list[dict]) -> str:
+def edit_file(path: str, edits: list[dict], yolo_mode: bool = False) -> str:
     def find_line_edit(n: int) -> LineEdit | None:
         for line_edit in line_edits:
             if line_edit.line_start <= n <= line_edit.line_end:
@@ -121,7 +122,7 @@ def edit_file(path: str, edits: list[dict]) -> str:
 
     file_exists = Path(path).exists()
     if file_exists:
-        with open(path, "r") as f:
+        with open(path) as f:
             original_content = f.read()
     else:
         original_content = ""
@@ -160,11 +161,12 @@ def edit_file(path: str, edits: list[dict]) -> str:
 
     console.print()
     console.print(Markdown(f"```diff\n{unified_diff_md}\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()
-        return f"User declined: {reason or 'no reason'}"
 
+    if not yolo_mode:
+        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()
+            return f"User declined: {reason or 'no reason'}"
     console.print(" └ Running...")
     try:
         with open(path, "w") as f:

{tass-0.1.12 → tass-0.1.13}/src/tools/execute.py

@@ -64,16 +64,12 @@ def is_read_only_command(command: str) -> bool:
     command = command.replace(";", "|")
 
     pipes = command.split("|")
-    for pipe in pipes:
-        if pipe.strip().split()[0] not in READ_ONLY_COMMANDS:
-            return False
+    return all(pipe.strip().split()[0] in READ_ONLY_COMMANDS for pipe in pipes)
 
-    return True
 
-
-def execute(command: str, explanation: str) -> str:
+def execute(command: str, explanation: str, yolo_mode: bool = False) -> str:
     command = command.strip()
-    requires_confirmation = not is_read_only_command(command)
+    requires_confirmation = not yolo_mode and not is_read_only_command(command)
     if requires_confirmation:
         console.print()
         console.print(Markdown(f"```shell\n{command}\n```"))
@@ -83,11 +79,9 @@ def execute(command: str, explanation: str) -> str:
         if answer not in ("yes", "y", ""):
             reason = console.input("Why not? (optional, press Enter to skip): ").strip()
             return f"User declined: {reason or 'no reason'}"
-
-    if requires_confirmation:
         console.print(" └ Running...")
     else:
-        console.print(f" └ Running [bold]{command}[/] (Explanation: {explanation})...")
+        console.print(f" └ Running [bold]{command}[/] (Explanation: {explanation})")
 
     try:
         result = subprocess.run(

{tass-0.1.12 → tass-0.1.13}/src/tools/read_file.py

@@ -1,6 +1,5 @@
 from src.constants import console
 
-
 READ_FILE_TOOL = {
     "type": "function",
     "function": {
@@ -31,7 +30,7 @@ READ_FILE_TOOL = {
 }
 
 
-def read_file(path: str, start: int = 1, num_lines: int = 1000) -> str:
+def read_file(path: str, start: int = 1, num_lines: int = 1000, yolo_mode: bool = False) -> str:
     if start == 1 and num_lines == 1000:
         console.print(f" └ Reading file [bold]{path}[/]...")
     else:
@@ -59,4 +58,4 @@ def read_file(path: str, start: int = 1, num_lines: int = 1000) -> str:
     except Exception as e:
         console.print("   [red]read_file failed[/red]")
         console.print(f"   [red]{str(e).strip()}[/red]")
-        return f"read_file failed: {str(e)}"
+        return f"read_file failed: {e}"

{tass-0.1.12 → tass-0.1.13}/tests/test_fuzzy_match.py

@@ -1,6 +1,5 @@
 from src.tools.edit_file import fuzzy_match
 
-
 file_contents = """def fib(n):
     '''Return the nth Fibonacci number (iterative).'''
     if n <= 0:
@@ -22,7 +21,25 @@ if __name__ == "__main__":
 def test_fuzzy():
     edit_find = "    return b\n\n\n\nif __name__ == \"__main__\":"
     lines = file_contents.split("\n")
-    # assert fuzzy_match(edit_find, lines) == (10, 13)
+    assert fuzzy_match(edit_find, lines) == (10, 13)
+
+    edit_find = """def fib(n):
+
+'''Return the nth Fibonacci number iterative.'''
+
+    if n<= 0:
+
+         return 0
+
+    if n ==1:
+
+      return 1
+
+    a, b = 0, 1
+
+    for _ in range(2, n + 1):
+
+        a, b = b, a + b
 
-    edit_find = "def fib(n):\n\n    '''Return the nth Fibonacci number (iterative).'''\n\n    if n <= 0:\n\n        return 0\n\n    if n == 1:\n\n        return 1\n\n    a, b = 0, 1\n\n    for _ in range(2, n + 1):\n\n        a, b = b, a + b\n\n    return b"
+    return b"""
     assert fuzzy_match(edit_find, lines) == (1, 10)

{tass-0.1.12 → tass-0.1.13}/tests/test_utils.py

@@ -1,6 +1,6 @@
 import pytest
 
-from src.utils import is_read_only_command
+from src.tools.execute import is_read_only_command
 
 
 @pytest.mark.parametrize(

{tass-0.1.12 → tass-0.1.13}/uv.lock

@@ -273,7 +273,7 @@ wheels = [
 
 [[package]]
 name = "tass"
-version = "0.1.12"
+version = "0.1.13"
 source = { editable = "." }
 dependencies = [
     { name = "prompt-toolkit" },

tass-0.1.12/src/cli.py

@@ -1,6 +0,0 @@
-from src.app import TassApp
-
-
-def main():
-    app = TassApp()
-    app.run()

tass-0.1.12/src/constants.py

@@ -1,17 +0,0 @@
-from pathlib import Path
-
-from rich.console import Console
-
-
-console = Console()
-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.
-
-If the user's request involves making changes to the filesystem such as creating or deleting files or directories, you MUST first check whether the file or directory exists before proceeding.
-
-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.
-
-This app has a feature where the user can refer to files or directories by typing @ which will open an file autocomplete dropdown. When this feature is used, the @ will remain in the filename. When working with said file, ignore the preceding @.
-
-Current working directory: {CWD_PATH}"""

{tass-0.1.12 → tass-0.1.13}/src/utils.py

@@ -1,7 +1,7 @@
 from prompt_toolkit.completion import (
+    CompleteEvent,
     Completer,
     Completion,
-    CompleteEvent,
 )
 from prompt_toolkit.document import Document
 from prompt_toolkit.key_binding import KeyBindings