tass 0.1.8__tar.gz → 0.1.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tass
-Version: 0.1.8
+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
@@ -39,7 +43,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.8 → 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
@@ -26,7 +30,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.8 → tass-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tass"
-version = "0.1.8"
+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.8 → tass-0.1.9}/src/app.py

@@ -64,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={
@@ -82,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(
@@ -137,15 +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"]
                     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 +190,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() or None,
+                "reasoning_content": reasoning_content.strip() or None,
+                "tool_calls": list(tool_calls_map.values()) or None,
             }
         )
 

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

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

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

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