tass 0.1.7__tar.gz → 0.1.8__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.7
+Version: 0.1.8
 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
@@ -21,10 +21,18 @@ 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
 
 ```
@@ -34,3 +42,19 @@ 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 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.8}/README.md

@@ -8,10 +8,18 @@ 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
 
 ```
@@ -21,3 +29,19 @@ 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 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.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tass"
-version = "0.1.7"
+version = "0.1.8"
 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.8}/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()
 
@@ -102,22 +100,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():
@@ -130,12 +140,10 @@ class TassApp:
                 delta = chunk["choices"][0]["delta"]
                 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()))
+                    live.update(generate_layout())
                 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", []):
                     index = tool_call_delta["index"]
@@ -164,9 +172,15 @@ 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"]:
-                    last_three_lines = "\n".join(reasoning_content.rstrip().split("\n")[-3:])
-                    live.update(generate_layout(last_three_lines, content.rstrip()))
+                    live.update(generate_layout())
 
         self.messages.append(
             {
@@ -362,14 +376,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.8}/uv.lock

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