tass 0.1.7__tar.gz → 0.1.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.7
+Version: 0.1.9
 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
@@ -13,6 +13,10 @@ 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
@@ -21,16 +25,40 @@ This tool can run commands including ones that can modify, move, or delete files
 
 ## Installation
 
+### Using uv
+
 ```
 uv tool install tass
 ```
 
+### Using pip
+
+```
+pip install tass
+```
+
 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.
+
+You can enter multiline input by ending lines with a backslash (\\). The continuation prompt will appear until you enter a line without a trailing backslash.
+
+## Upgrade
+
+### Using uv
+
+```
+uv tool upgrade tass
+```
+
+### Using pip
+
+```
+pip install --upgrade tass
+```

{tass-0.1.7 → tass-0.1.9}/README.md

@@ -1,5 +1,9 @@
 # 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
@@ -8,16 +12,40 @@ This tool can run commands including ones that can modify, move, or delete files
 
 ## Installation
 
+### Using uv
+
 ```
 uv tool install tass
 ```
 
+### Using pip
+
+```
+pip install tass
+```
+
 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.
+
+You can enter multiline input by ending lines with a backslash (\\). The continuation prompt will appear until you enter a line without a trailing backslash.
+
+## Upgrade
+
+### Using uv
+
+```
+uv tool upgrade tass
+```
+
+### Using pip
+
+```
+pip install --upgrade tass
+```

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

@@ -1,6 +1,6 @@
 [project]
 name = "tass"
-version = "0.1.7"
+version = "0.1.9"
 description = "A terminal assistant that allows you to ask an LLM to run commands."
 readme = "README.md"
 requires-python = ">=3.10"

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

@@ -15,9 +15,7 @@ from src.constants import (
     SYSTEM_PROMPT,
     TOOLS,
 )
-from src.utils import (
-    is_read_only_command,
-)
+from src.utils import is_read_only_command
 
 console = Console()
 
@@ -66,11 +64,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={
@@ -84,6 +85,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(
@@ -102,22 +104,34 @@ class TassApp:
         content = ""
         reasoning_content = ""
         tool_calls_map = {}
+        timings_str = ""
 
-        def generate_layout(reasoning_content: str, content: str):
+        def generate_layout():
             groups = []
 
             if reasoning_content:
+                last_three_lines = "\n".join(reasoning_content.rstrip().split("\n")[-3:])
                 groups.append(Text(""))
-                groups.append(Panel(Text(reasoning_content, style="grey50"), title="Thought process", title_align="left", style="grey50"))
+                groups.append(
+                    Panel(
+                        Text(
+                            last_three_lines,
+                            style="grey50",
+                        ),
+                        title="Thought process",
+                        title_align="left",
+                        subtitle=timings_str,
+                        style="grey50",
+                    )
+                )
 
             if content:
                 groups.append(Text(""))
-                groups.append(Markdown(content))
-                groups.append(Text(""))
+                groups.append(Markdown(content.rstrip()))
 
             return Group(*groups)
 
-        with Live(generate_layout(reasoning_content, content), refresh_per_second=10) as live:
+        with Live(generate_layout(), refresh_per_second=10) as live:
             for line in response.iter_lines():
                 line = line.decode("utf-8")
                 if not line.strip():
@@ -127,17 +141,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"]
-                    last_three_lines = "\n".join(reasoning_content.rstrip().split("\n")[-3:])
-                    live.update(generate_layout(last_three_lines, content.rstrip()))
-                if delta.get("reasoning_content" ):
-                    reasoning_content += delta["reasoning_content"]
-                    last_three_lines = "\n".join(reasoning_content.rstrip().split("\n")[-3:])
-                    live.update(generate_layout(last_three_lines, content.rstrip()))
+                    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] = (
@@ -164,16 +190,12 @@ class TassApp:
                         if function.get("arguments"):
                             tool_call["function"]["arguments"] += function["arguments"]
 
-                if chunk["choices"][0]["finish_reason"]:
-                    last_three_lines = "\n".join(reasoning_content.rstrip().split("\n")[-3:])
-                    live.update(generate_layout(last_three_lines, content.rstrip()))
-
         self.messages.append(
             {
                 "role": "assistant",
-                "content": content,
-                "reasoning_content": reasoning_content,
-                "tool_calls": list(tool_calls_map.values()),
+                "content": content.strip() or None,
+                "reasoning_content": reasoning_content.strip() or None,
+                "tool_calls": list(tool_calls_map.values()) or None,
             }
         )
 
@@ -362,14 +384,22 @@ class TassApp:
     def run(self):
         try:
             self._check_llm_host()
-            console.print()
         except KeyboardInterrupt:
             console.print("\nBye!")
             return
 
         while True:
+            console.print()
             try:
-                user_input = console.input("> ").strip()
+                input_lines = []
+                while True:
+                    input_line = console.input("> ")
+                    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!")
                 break

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

@@ -113,4 +113,5 @@ READ_ONLY_COMMANDS = [
     "which",
     "sed",
     "find",
+    "test",
 ]

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

@@ -261,7 +261,7 @@ wheels = [
 
 [[package]]
 name = "tass"
-version = "0.1.7"
+version = "0.1.9"
 source = { editable = "." }
 dependencies = [
     { name = "requests" },