tau-coding-agent 0.1.0__tar.gz → 0.2.0__tar.gz

{tau_coding_agent-0.1.0/tau_coding_agent.egg-info → tau_coding_agent-0.2.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: tau-coding-agent
-Version: 0.1.0
-Summary: Interactive agent CLI with terminal UI for multi-provider LLM support
+Version: 0.2.0
+Summary: A self-extensible agent CLI with terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools and commands
 Requires-Python: ==3.13.*
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -27,9 +27,15 @@ Requires-Dist: pathspec==1.1.1
 Requires-Dist: gitpython==3.1.50
 Dynamic: license-file
 
-# Tau
+# Tau (τ)
 
-Inspired by [Pi](https://pi.dev), Tau brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+Inspired by [Pi](https://pi.dev), **Tau** brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+
+<p align="center">
+  <img src="assets/tui.png" alt="Tau TUI" width="700">
+</p>
+
+Start a conversation with the agent in the terminal.
 
 ---
 
@@ -55,7 +61,7 @@ pip install -e .
 tau
 ```
 
-Authenticate with your LLM provider using `/login` or set environment variables (e.g., `ANTHROPIC_API_KEY`).
+Authenticate with your LLM provider using `/login` or set environment variables (e.g., `NVIDIA_API_KEY`).
 
 ### Common Commands
 

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/README.md

@@ -1,6 +1,12 @@
-# Tau
+# Tau (τ)
 
-Inspired by [Pi](https://pi.dev), Tau brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+Inspired by [Pi](https://pi.dev), **Tau** brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+
+<p align="center">
+  <img src="assets/tui.png" alt="Tau TUI" width="700">
+</p>
+
+Start a conversation with the agent in the terminal.
 
 ---
 
@@ -26,7 +32,7 @@ pip install -e .
 tau
 ```
 
-Authenticate with your LLM provider using `/login` or set environment variables (e.g., `ANTHROPIC_API_KEY`).
+Authenticate with your LLM provider using `/login` or set environment variables (e.g., `NVIDIA_API_KEY`).
 
 ### Common Commands
 

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/pyproject.toml

@@ -1,7 +1,11 @@
+[build-system]
+requires = ["setuptools>=61"]
+build-backend = "setuptools.build_meta"
+
 [project]
 name = "tau-coding-agent"
-version = "0.1.0"
-description = "Interactive agent CLI with terminal UI for multi-provider LLM support"
+version = "0.2.0"
+description = "A self-extensible agent CLI with terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools and commands"
 readme = "README.md"
 requires-python = "==3.13.*"
 dependencies = [
@@ -49,6 +53,12 @@ indent-style = "space"
 [tool.setuptools]
 include-package-data = true
 
+# Only package the `tau` import package. Without this, setuptools auto-discovery
+# trips over sibling top-level dirs (dist/, docs/, assets/, *.egg-info) and
+# aborts editable installs with "Multiple top-level packages discovered".
+[tool.setuptools.packages.find]
+include = ["tau*"]
+
 [tool.setuptools.package-data]
 tau = [
     "builtins/**/*",

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/console/cli.py

@@ -49,7 +49,7 @@ def resolve_model(model: str | None, provider: str | None) -> tuple[str | None,
 @click.option("--provider", default=None, help="Provider to use (e.g. groq, mistral, openrouter).")
 @click.option("--model", default=None, help="Model ID, or provider/model shorthand (e.g. groq/llama-3.3-70b-versatile).")
 @click.option("--theme", "-t", default=None, metavar="NAME",
-              help="UI theme: default, dracula, nord, gruvbox, catppuccin.")
+              help="UI theme name (default: dark). Builtins: dark, light. See /theme for all installed themes.")
 @click.option("--resume", "-r", is_flag=True, default=False,
               help="Resume the most recent session.")
 @click.option("--system", "-s", default=None, metavar="TEXT",

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/inference/provider/oauth/google_antigravity.py

@@ -8,6 +8,7 @@ from __future__ import annotations
 
 import asyncio
 import json
+import os
 import secrets
 import ssl
 import time
@@ -26,10 +27,10 @@ from tau.inference.provider.oauth.utils import parse_authorization_input, start_
 
 __all__ = ["GoogleAntigravityOAuthProvider"]
 
-_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())
+_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())                                                       
 
-CLIENT_ID = "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com"
-CLIENT_SECRET = "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"
+CLIENT_ID = os.getenv("GOOGLE_ANTIGRAVITY_CLIENT_ID", "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com")
+CLIENT_SECRET = os.getenv("GOOGLE_ANTIGRAVITY_CLIENT_SECRET", "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")
 AUTHORIZE_URL = "https://accounts.google.com/o/oauth2/v2/auth"
 TOKEN_URL = "https://oauth2.googleapis.com/token"
 USERINFO_URL = "https://www.googleapis.com/oauth2/v1/userinfo"

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/inference/provider/oauth/utils.py

@@ -119,12 +119,14 @@ async def await_oauth_code(
     """
     code: Optional[str] = None
     recv_state: Optional[str] = None
+    pending: set = set()
     try:
         if callbacks.on_manual_code_input:
             browser_task = asyncio.ensure_future(code_future)
             manual_task = asyncio.ensure_future(callbacks.on_manual_code_input())
+            pending = {browser_task, manual_task}
             done, pending = await asyncio.wait(
-                [browser_task, manual_task],
+                pending,
                 timeout=300,
                 return_when=asyncio.FIRST_COMPLETED,
             )
@@ -148,7 +150,15 @@ async def await_oauth_code(
                 recv_state = state
             except asyncio.TimeoutError:
                 pass
+    except Exception:
+        for task in pending:
+            if not task.done():
+                task.cancel()
+        raise
     finally:
+        for task in pending:
+            if not task.done():
+                task.cancel()
         server.close()
         await server.wait_closed()
     return code, recv_state

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/message/types.py

@@ -41,6 +41,21 @@ class ImageContent:
     images: list[str | Image.Image | bytes] = field(default_factory=list)
     dimension_note: str | None = None
 
+    def __post_init__(self) -> None:
+        # Normalize raw bytes / PIL images to base64 strings so the content is
+        # always JSON-serializable for session persistence (pydantic cannot
+        # encode non-UTF-8 image bytes). Existing base64 and URL strings pass
+        # through unchanged, and to_base64() re-detects the MIME type from the
+        # base64 magic bytes, so downstream API/render code is unaffected.
+        normalized: list[str | Image.Image | bytes] = []
+        for img in self.images:
+            if isinstance(img, str):
+                normalized.append(img)
+            else:
+                b64, _ = image_to_base64(img)
+                normalized.append(b64)
+        self.images = normalized
+
     def to_base64(self) -> list[tuple[str, str]]:
         """Convert all images to (base64_data, mime_type) pairs.
 

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/runtime/service.py

@@ -178,6 +178,8 @@ class Runtime:
                     expanded = prompt_registry.expand(name, " ".join(args))
                     if expanded is not None:
                         await self.invoke(expanded, options)
+                    else:
+                        self.notify(f"Unknown command: /{name}")
             case t:
                 await self.invoke(t, options)
 

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/themes/registry.py

@@ -20,6 +20,11 @@ if TYPE_CHECKING:
     from tau.tui.theme import LayoutTheme
 
 
+#: Name of the theme used when none is configured. Must always resolve to a
+#: builtin so the TUI can start even with no global/project themes installed.
+DEFAULT_THEME = "dark"
+
+
 class ThemeRegistry:
     def __init__(self) -> None:
         """Initialize an empty theme registry."""
@@ -75,6 +80,24 @@ class ThemeRegistry:
             )
         return loader()
 
+    def get_default(self) -> "LayoutTheme":
+        """Return a theme that is guaranteed to load.
+
+        Tries the configured default, then any builtin, then falls back to a
+        bare ``LayoutTheme()`` so the UI can always start — even when no theme
+        files are present at all.
+        """
+        self._ensure_builtins()
+        for name in (DEFAULT_THEME, "light"):
+            loader = self._registry.get(name)
+            if loader is not None:
+                return loader()
+        if self._registry:
+            return next(iter(self._registry.values()))()
+        from tau.tui.theme import LayoutTheme
+
+        return LayoutTheme()
+
     def list(self) -> list[str]:
         """Return all available theme names."""
         self._ensure_builtins()

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/tui/app.py

@@ -45,6 +45,7 @@ class App:
             on_turn_content=self._input.mark_turn_content,
         )
         self._unsubs: list[Callable[[], None]] = []
+        self._pending_tasks: set[asyncio.Task] = set()
         self._last_ctrl_c: float = 0.0
         self._last_escape: float = 0.0
 
@@ -60,7 +61,7 @@ class App:
         keybindings: KeyMap | None = None,
     ) -> App:
         """Build the TUI around an already-constructed Runtime."""
-        from tau.themes.registry import theme_registry
+        from tau.themes.registry import theme_registry, DEFAULT_THEME
 
         cwd = runtime.session_manager.cwd if runtime.session_manager is not None else None
         theme_registry.load_external(cwd=cwd)
@@ -69,18 +70,21 @@ class App:
         prompt_registry.load_external(cwd=cwd)
 
         resolved_theme: LayoutTheme | None
-        theme_name = "default"
+        theme_name = DEFAULT_THEME
         if isinstance(theme, str):
             theme_name = theme
             resolved_theme = theme_registry.get(theme_name)
         elif theme is None:
             sm = runtime.settings_manager
-            theme_name = (sm.get_theme() if sm is not None else None) or "default"
+            theme_name = (sm.get_theme() if sm is not None else None) or DEFAULT_THEME
             try:
                 resolved_theme = theme_registry.get(theme_name)
             except ValueError:
-                theme_name = "default"
-                resolved_theme = theme_registry.get(theme_name)
+                # Configured theme is gone (e.g. an uninstalled theme package)
+                # or the default builtin is missing — fall back to a theme that
+                # is guaranteed to load instead of crashing on startup.
+                theme_name = DEFAULT_THEME
+                resolved_theme = theme_registry.get_default()
         else:
             resolved_theme = theme
 
@@ -172,7 +176,11 @@ class App:
     def _on_model_palette_commit(self, model_id: str, provider: str) -> None:
         import asyncio
         from tau.tui.commands import model as cmd_model
-        asyncio.ensure_future(cmd_model._apply_model(self._ctx(), model_id, provider))
+        self._track_task(asyncio.ensure_future(cmd_model._apply_model(self._ctx(), model_id, provider)))
+
+    def _track_task(self, task: asyncio.Task) -> None:
+        self._pending_tasks.add(task)
+        task.add_done_callback(self._pending_tasks.discard)
 
     # -------------------------------------------------------------------------
     # UI command registration
@@ -300,14 +308,14 @@ class App:
         # before the loop starts so the layout never renders until the user acts.
         trust_needed = self._setup_trust_screen_if_needed()
 
-        asyncio.ensure_future(self._announce_update())
+        self._track_task(asyncio.ensure_future(self._announce_update()))
 
         await self._runtime.hooks.emit(TuiStartEvent())
         try:
             await self._tui.run()
         finally:
             await self._runtime.hooks.emit(TuiExitEvent())
-            self._cleanup()
+            await self._cleanup()
 
     # -------------------------------------------------------------------------
     # Project trust prompt
@@ -354,7 +362,7 @@ class App:
             import asyncio as _asyncio
             async def _reload() -> None:
                 await self._runtime.reload_extensions()
-            _asyncio.ensure_future(_reload())
+            self._track_task(_asyncio.ensure_future(_reload()))
 
         from tau.tui.components.trust_screen import TrustScreen
         screen = TrustScreen(str(cwd), options, _on_commit)
@@ -449,7 +457,7 @@ class App:
                     ctx = ExtensionContext.from_runtime(runtime)
                     result = h(ctx)
                     if asyncio.iscoroutine(result):
-                        asyncio.ensure_future(result)  # type: ignore[arg-type]
+                        self._track_task(asyncio.ensure_future(result))  # type: ignore[arg-type]
                 return on_input
 
             self._unsubs.append(self._tui.on_input(_make_handler(key, handler)))
@@ -532,9 +540,15 @@ class App:
         ])
         self._layout.set_widget("version_update", banner, placement="above_editor")
 
-    def _cleanup(self) -> None:
+    async def _cleanup(self) -> None:
         self._input.save_history()
         self._hooks.unsubscribe()
         for unsub in self._unsubs:
             unsub()
         self._unsubs.clear()
+        for task in self._pending_tasks:
+            task.cancel()
+        self._pending_tasks.clear()
+        sm = self._runtime.settings_manager
+        if sm is not None:
+            await sm.flush()

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/tui/commands/appearance.py

@@ -13,11 +13,11 @@ def _headers_to_str(headers: dict | None) -> str:
 
 def open_theme_selector(ctx: CommandContext) -> None:
     """Open the theme selector modal with live preview."""
-    from tau.themes.registry import theme_registry
+    from tau.themes.registry import theme_registry, DEFAULT_THEME
 
     names = theme_registry.list()
     sm = ctx.runtime.settings_manager
-    original = (sm.get_theme() if sm is not None else None) or "default"
+    original = (sm.get_theme() if sm is not None else None) or DEFAULT_THEME
 
     def preview(name: str) -> None:
         """Preview the selected theme."""
@@ -54,7 +54,7 @@ def open_settings_panel(ctx: CommandContext) -> None:
     from tau.tui.components.settings_modal import SettingItem, SettingsModal
     from tau.inference.types import Transport, ThinkingLevel
     from tau.engine.types import SteeringMode, FollowupMode
-    from tau.themes.registry import theme_registry
+    from tau.themes.registry import theme_registry, DEFAULT_THEME
 
     sm = ctx.runtime.settings_manager
     if sm is None:
@@ -137,7 +137,7 @@ def open_settings_panel(ctx: CommandContext) -> None:
             id="theme",
             label="Theme",
             description="Color theme for the interface",
-            current_value=sm.get_theme() or "default",
+            current_value=sm.get_theme() or DEFAULT_THEME,
             submenu_items=theme_registry.list(),
             submenu_title="Theme",
         ),

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/tui/components/autocomplete_manager.py

@@ -54,6 +54,12 @@ class AutocompleteManager:
 
     def clear(self) -> None:
         """Dismiss both pickers — called when another picker takes over."""
+        if self._ac_pending_task is not None:
+            self._ac_pending_task.cancel()
+            self._ac_pending_task = None
+        if self._cmd_arg_pending_task is not None:
+            self._cmd_arg_pending_task.cancel()
+            self._cmd_arg_pending_task = None
         self._ac_picker.clear()
         self._cmd_arg_picker.clear()
         self._ac_active_trigger = ""

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0}/tau/tui/input_handler.py

@@ -98,7 +98,15 @@ class InputHandler:
             asyncio.ensure_future(self._invoke(text))
             return
 
-        images = self._extract_clipboard_images(text)
+        images, missing_images = self._extract_clipboard_images(text)
+        if missing_images:
+            plural = "s" if missing_images > 1 else ""
+            self._notify(
+                f"{missing_images} image{plural} could not be found — the media file{plural} may have been deleted or moved.",
+                type="error",
+            )
+            return
+
         audio = self._extract_clipboard_audio(text)
         video = self._extract_clipboard_video(text)
         expanded = self._expand_pasted_texts(text)
@@ -121,6 +129,11 @@ class InputHandler:
             asyncio.ensure_future(self._steer(expanded, images))
             return
 
+        # Strip resolved image markers from the text sent to the model so the
+        # LLM doesn't see raw [image:uuid] placeholders alongside actual image bytes.
+        stripped = re.sub(r"\[image(?::[^\]]+| #\d+)\]", "", expanded).strip()
+        model_text = stripped if stripped else expanded
+
         if images:
             user_msg = UserMessage.with_images(text, images)
         elif audio:
@@ -133,10 +146,10 @@ class InputHandler:
         self._last_user_text = text
         self._turn_has_content = False
         self._tui.request_render()
-        asyncio.ensure_future(self._invoke(self._expand_at_mentions(expanded), images, audio, video))
+        asyncio.ensure_future(self._invoke(self._expand_at_mentions(model_text), images, audio, video))
 
     def _on_followup(self, text: str) -> None:
-        images = self._extract_clipboard_images(text)
+        images, _ = self._extract_clipboard_images(text)
         expanded = self._expand_pasted_texts(text)
         asyncio.ensure_future(self._queue_followup(expanded, images, display_text=text))
 
@@ -209,6 +222,7 @@ class InputHandler:
             self._last_user_text = ""
             if last_text:
                 self._layout.input.set_text(last_text)
+            self._clear_clipboard_caches()
 
         self._turn_has_content = False
         # Stop the spinner immediately. The pre-stream branch cancels the invoke
@@ -220,6 +234,18 @@ class InputHandler:
             asyncio.ensure_future(self._run_queued_next(queued))
         self._tui.request_render()
 
+    def _clear_clipboard_caches(self) -> None:
+        """Clear all clipboard media caches."""
+        self._clipboard_images.clear()
+        self._clipboard_image_notes.clear()
+        self._clipboard_image_counter = 0
+        self._clipboard_audio.clear()
+        self._clipboard_audio_counter = 0
+        self._clipboard_video.clear()
+        self._clipboard_video_counter = 0
+        self._pasted_texts.clear()
+        self._paste_counter = 0
+
     async def _run_queued_next(self, texts: list[str]) -> None:
         """Submit queued input as the next task once the aborted task is idle.
 
@@ -551,7 +577,12 @@ class InputHandler:
         result = re.sub(r"\[paste #\d+(?: \+\d+ lines| \d+ chars)\]", "", result)
         return result.strip()
 
-    def _extract_clipboard_images(self, text: str) -> list[bytes]:
+    def _extract_clipboard_images(self, text: str) -> tuple[list[bytes], int]:
+        """Extract image bytes from markers in text.
+
+        Returns (images, missing_count) where missing_count is the number of
+        persistent [image:uuid] markers whose media files could not be found.
+        """
         images: list[bytes] = []
         seen: set[int] = set()
         for m in re.finditer(r"\[image #(\d+)\]", text):
@@ -569,6 +600,7 @@ class InputHandler:
             except OSError:
                 pass
         # Also resolve persistent [image:{uuid}] markers from history
+        missing = 0
         seen_uuids: set[str] = set()
         for m in re.finditer(r"\[image:([^\]]+)\]", text):
             uid = m.group(1)
@@ -580,11 +612,13 @@ class InputHandler:
                 try:
                     images.append(p.read_bytes())
                 except OSError:
-                    pass
+                    missing += 1
+            else:
+                missing += 1
         self._clipboard_images.clear()
         self._clipboard_image_notes.clear()
         self._clipboard_image_counter = 0
-        return images
+        return images, missing
 
     def _extract_clipboard_image_contents(self, text: str) -> "list[Any]":
         """Like _extract_clipboard_images but returns ImageContent with dimension notes."""

{tau_coding_agent-0.1.0 → tau_coding_agent-0.2.0/tau_coding_agent.egg-info}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: tau-coding-agent
-Version: 0.1.0
-Summary: Interactive agent CLI with terminal UI for multi-provider LLM support
+Version: 0.2.0
+Summary: A self-extensible agent CLI with terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools and commands
 Requires-Python: ==3.13.*
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -27,9 +27,15 @@ Requires-Dist: pathspec==1.1.1
 Requires-Dist: gitpython==3.1.50
 Dynamic: license-file
 
-# Tau
+# Tau (τ)
 
-Inspired by [Pi](https://pi.dev), Tau brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+Inspired by [Pi](https://pi.dev), **Tau** brings the same agent framework capabilities to Python developers. A self-extensible agent CLI with a terminal UI, multi-provider LLM support, session management with branching, and a plugin system for tools, commands, and customization.
+
+<p align="center">
+  <img src="assets/tui.png" alt="Tau TUI" width="700">
+</p>
+
+Start a conversation with the agent in the terminal.
 
 ---
 
@@ -55,7 +61,7 @@ pip install -e .
 tau
 ```
 
-Authenticate with your LLM provider using `/login` or set environment variables (e.g., `ANTHROPIC_API_KEY`).
+Authenticate with your LLM provider using `/login` or set environment variables (e.g., `NVIDIA_API_KEY`).
 
 ### Common Commands