tau-coding-agent 0.5.2__tar.gz → 0.5.3__tar.gz

{tau_coding_agent-0.5.2/tau_coding_agent.egg-info → tau_coding_agent-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tau-coding-agent
-Version: 0.5.2
+Version: 0.5.3
 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.12
 Description-Content-Type: text/markdown

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/docs/tui.md

@@ -122,6 +122,9 @@ Input is normalized into typed events:
 - `BgColorEvent`
 
 Use `KeyEvent.matches()` instead of comparing raw escape sequences.
+Mouse reporting is enabled while the TUI runs. The stock interactive
+`TextInput` moves its caret to the clicked text position, including across
+hard newlines and soft-wrapped rows.
 
 ## Overlays
 

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tau-coding-agent"
-version = "0.5.2"
+version = "0.5.3"
 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.12"

tau_coding_agent-0.5.3/tau/__init__.py

@@ -0,0 +1,9 @@
+import os
+
+# Skip pydantic's plugin discovery (a full `importlib.metadata.distributions()`
+# scan of every installed package, ~30-50ms) unless the environment already
+# asked for a specific setting. Tau ships no pydantic plugins and none are
+# expected to be installed alongside it; this must run before the first
+# pydantic BaseModel subclass is defined anywhere in the process, so it lives
+# in this package's __init__ (executed before any tau submodule).
+os.environ.setdefault("PYDANTIC_DISABLE_PLUGINS", "1")

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/agent/prompt/builder.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import os
 import platform
 import shutil
+import sys
 from collections.abc import Callable
 from datetime import date
 from pathlib import Path
@@ -163,17 +164,41 @@ _GIT_LOG_COUNT = 5
 
 
 def _detect_os() -> str:
-    """Return a human-readable OS name and version."""
+    """Return a human-readable OS name and version.
+
+    On Windows, ``platform.system()``/``platform.release()`` call
+    ``platform.uname()``, which shells out to two separate WMI COM queries
+    (~150-200ms combined) to determine the release name. ``sys.getwindowsversion()``
+    reads the same information directly from the OS with no WMI round-trip, so we
+    use it here and derive the release name (10 vs 11) from the build number using
+    the same >=22000 heuristic Windows itself uses.
+    """
+    if sys.platform == "win32":
+        v = sys.getwindowsversion()
+        release = "11" if v.build >= 22000 else str(v.major)
+        return f"Windows {release}"
     system = platform.system()
     if system == "Darwin":
         return f"macOS {platform.mac_ver()[0] or platform.release()}"
     if system == "Linux":
         return f"Linux {platform.release()}"
-    if system == "Windows":
-        return f"Windows {platform.release()}"
     return f"{system} {platform.release()}".strip()
 
 
+def _detect_machine() -> str:
+    """Return the machine architecture (e.g. 'x86_64', 'AMD64').
+
+    ``platform.machine()`` is a thin wrapper around ``platform.uname()``, which
+    is expensive on Windows (see ``_detect_os``). ``PROCESSOR_ARCHITECTURE`` is
+    always set by the OS and gives the same value without the WMI round-trip.
+    """
+    if sys.platform == "win32":
+        arch = os.environ.get("PROCESSOR_ARCHITECTURE")
+        if arch:
+            return arch
+    return platform.machine()
+
+
 def _detect_shell() -> str:
     """Detect the user's shell from $SHELL, falling back to common shells on PATH."""
     shell = os.environ.get("SHELL", "")
@@ -198,7 +223,6 @@ def _git_status(cwd: Path) -> str:
     except ImportError:
         return ""
 
-    repo = None
     try:
         repo = Repo(cwd, search_parent_directories=True)
 
@@ -233,9 +257,12 @@ def _git_status(cwd: Path) -> str:
         return ""
     except GitError:
         return ""
-    finally:
-        if repo is not None:
-            repo.close()
+    # No explicit repo.close() here: GitPython's Repo.__del__ already calls it,
+    # and close() forces two gc.collect() passes on Windows (its own workaround
+    # for tempfile handles). Closing explicitly *and* via __del__ paid for that
+    # gc.collect() pass twice for no benefit — we don't touch the object database
+    # in the common (attached-branch) path, so there's nothing time-sensitive to
+    # release synchronously.
 
     parts = [
         "\n\ngitStatus: snapshot taken at session start; it is not updated during the "
@@ -440,6 +467,8 @@ class PromptBuilder:
         """
         if self._opts.project_trusted is False:
             return ""
+        if self._opts.git_snapshot is not None:
+            return self._opts.git_snapshot
         return _git_status(self._opts.cwd)
 
     def _footer(self) -> str:
@@ -449,7 +478,7 @@ class PromptBuilder:
             "\n\n# Environment\n"
             f"Current working directory: {cwd}\n"
             f"OS: {_detect_os()}\n"
-            f"Architecture: {platform.machine()}\n"
+            f"Architecture: {_detect_machine()}\n"
             f"Shell: {_detect_shell()}\n"
             f"Date: {today}"
         )

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/agent/prompt/types.py

@@ -39,3 +39,9 @@ class PromptOptions(BaseModel):
 
     # Preloaded context files. None keeps direct PromptBuilder auto-discovery.
     context_files: tuple[ContextFile, ...] | None = None
+
+    # Precomputed git snapshot (see builder._git_status). None keeps direct,
+    # synchronous computation inside the builder; callers that can compute it
+    # concurrently with other startup work (e.g. RuntimeContext.create) pass
+    # the already-awaited result here to avoid blocking on it a second time.
+    git_snapshot: str | None = None

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/inference/provider/oauth/anthropic_claude_code.py

@@ -10,7 +10,6 @@ from __future__ import annotations
 import asyncio
 import base64
 import json
-import ssl
 import subprocess
 import sys
 import time
@@ -19,8 +18,6 @@ import urllib.parse
 import urllib.request
 from dataclasses import dataclass
 
-import certifi
-
 from tau.inference.provider.oauth.pkce import generate_pkce
 from tau.inference.provider.oauth.types import (
     AbortSignal,
@@ -31,6 +28,7 @@ from tau.inference.provider.oauth.types import (
 )
 from tau.inference.provider.oauth.utils import (
     await_oauth_code,
+    get_oauth_ssl_context,
     parse_authorization_input,
     start_oauth_callback_server,
 )
@@ -38,8 +36,6 @@ from tau.inference.provider.types import OAuthProvider
 
 __all__ = ["AnthropicClaudeCodeOAuthProvider"]
 
-_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())
-
 _RAW_CLIENT_ID = "OWQxYzI1MGEtZTYxYi00NGQ5LTg4ZWQtNTk0NGQxOTYyZjVl"
 CLIENT_ID = base64.b64decode(_RAW_CLIENT_ID).decode()
 AUTHORIZE_URL = "https://claude.ai/oauth/authorize"
@@ -87,7 +83,7 @@ def _post_json(url: str, body: dict) -> dict:
         method="POST",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=30) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=30) as resp:
             return json.loads(resp.read())
     except urllib.error.HTTPError as e:
         body_text = e.read().decode(errors="replace")
@@ -134,7 +130,7 @@ def _validate_token_sync(access_token: str) -> bool:
         method="GET",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=10) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=10) as resp:
             return resp.status == 200
     except urllib.error.HTTPError as e:
         # 401/403 mean the token is genuinely invalid; other codes may be transient

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/inference/provider/oauth/github_copilot.py

@@ -15,15 +15,12 @@ import asyncio
 import base64
 import json
 import re
-import ssl
 import time
 import urllib.error
 import urllib.parse
 import urllib.request
 from dataclasses import dataclass
 
-import certifi
-
 from tau.inference.provider.oauth.types import (
     AbortSignal,
     OAuthAuthInfo,
@@ -31,12 +28,11 @@ from tau.inference.provider.oauth.types import (
     OAuthLoginCallbacks,
     OAuthPrompt,
 )
+from tau.inference.provider.oauth.utils import get_oauth_ssl_context
 from tau.inference.provider.types import OAuthProvider
 
 __all__ = ["GitHubCopilotOAuthProvider", "get_copilot_base_url"]
 
-_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())
-
 _RAW_CLIENT_ID = "SXYxLmI1MDdhMDhjODdlY2ZlOTg="
 CLIENT_ID = base64.b64decode(_RAW_CLIENT_ID).decode()
 
@@ -101,7 +97,7 @@ def _fetch_json(url: str, *, method: str = "GET", headers: dict, body: bytes | N
     """
     req = urllib.request.Request(url, data=body, headers=headers, method=method)
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=30) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=30) as resp:
             return json.loads(resp.read())
     except urllib.error.HTTPError as e:
         body_text = e.read().decode(errors="replace")
@@ -177,7 +173,7 @@ def _enable_model(copilot_token: str, model_id: str, enterprise_domain: str | No
         method="POST",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=10) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=10) as resp:
             return resp.status == 200
     except Exception:
         return False

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/inference/provider/oauth/google_antigravity.py

@@ -11,7 +11,6 @@ import asyncio
 import json
 import os
 import secrets
-import ssl
 import time
 import urllib.error
 import urllib.parse
@@ -19,8 +18,6 @@ import urllib.request
 from dataclasses import dataclass
 from pathlib import Path
 
-import certifi
-
 from tau.inference.provider.oauth.types import (
     AbortSignal,
     OAuthAuthInfo,
@@ -30,6 +27,7 @@ from tau.inference.provider.oauth.types import (
 )
 from tau.inference.provider.oauth.utils import (
     await_oauth_code,
+    get_oauth_ssl_context,
     parse_authorization_input,
     start_oauth_callback_server,
 )
@@ -37,8 +35,6 @@ from tau.inference.provider.types import OAuthProvider
 
 __all__ = ["GoogleAntigravityOAuthProvider"]
 
-_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())
-
 CLIENT_ID = os.getenv(
     "GOOGLE_ANTIGRAVITY_CLIENT_ID",
     "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com",
@@ -88,7 +84,7 @@ def _post_form(url: str, body: dict) -> dict:
         method="POST",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=30) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=30) as resp:
             return json.loads(resp.read())
     except urllib.error.HTTPError as e:
         body_text = e.read().decode(errors="replace")
@@ -130,7 +126,7 @@ def _validate_token_sync(access_token: str) -> bool:
         method="GET",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=10) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=10) as resp:
             return resp.status == 200
     except urllib.error.HTTPError as e:
         return e.code not in (401, 403)
@@ -275,7 +271,7 @@ class GoogleAntigravityOAuthProvider(OAuthProvider):
                 data=b"",
                 method="POST",
             )
-            with urllib.request.urlopen(req, context=_SSL_CONTEXT, timeout=10):
+            with urllib.request.urlopen(req, context=get_oauth_ssl_context(), timeout=10):
                 pass
         except Exception:
             pass

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/inference/provider/oauth/openai_codex.py

@@ -11,7 +11,6 @@ import asyncio
 import base64
 import json
 import secrets
-import ssl
 import time
 import urllib.error
 import urllib.parse
@@ -19,8 +18,6 @@ import urllib.request
 from dataclasses import dataclass
 from pathlib import Path
 
-import certifi
-
 from tau.inference.provider.oauth.pkce import generate_pkce
 from tau.inference.provider.oauth.types import (
     AbortSignal,
@@ -31,13 +28,12 @@ from tau.inference.provider.oauth.types import (
 )
 from tau.inference.provider.oauth.utils import (
     await_oauth_code,
+    get_oauth_ssl_context,
     parse_authorization_input,
     start_oauth_callback_server,
 )
 from tau.inference.provider.types import OAuthProvider
 
-_SSL_CONTEXT = ssl.create_default_context(cafile=certifi.where())
-
 __all__ = ["OpenAICodexOAuthProvider"]
 
 CLIENT_ID = "app_EMoamEEZ73f0CkXaXp7hrann"
@@ -113,7 +109,7 @@ def _post_token(body: dict[str, str]) -> dict:
         method="POST",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context()) as resp:
             return json.loads(resp.read())
     except urllib.error.HTTPError as e:
         body_text = e.read().decode(errors="replace")
@@ -159,7 +155,7 @@ def _revoke_token_sync(token: str) -> None:
         method="POST",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT):
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context()):
             pass
     except urllib.error.HTTPError:
         # best-effort; treat any error as revocation attempt done
@@ -174,7 +170,7 @@ def _validate_token_sync(access_token: str) -> bool:
         method="GET",
     )
     try:
-        with urllib.request.urlopen(req, context=_SSL_CONTEXT) as resp:
+        with urllib.request.urlopen(req, context=get_oauth_ssl_context()) as resp:
             return resp.status == 200
     except urllib.error.HTTPError:
         return False

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/inference/provider/oauth/utils.py

@@ -4,7 +4,9 @@ from __future__ import annotations
 
 import asyncio
 import logging
+import ssl
 import urllib.parse
+from functools import lru_cache
 
 from tau.inference.provider.oauth.types import OAuthLoginCallbacks
 
@@ -13,11 +15,25 @@ _log = logging.getLogger(__name__)
 __all__ = [
     "OAUTH_SUCCESS_HTML",
     "OAUTH_ERROR_HTML",
+    "get_oauth_ssl_context",
     "parse_authorization_input",
     "start_oauth_callback_server",
     "await_oauth_code",
 ]
 
+
+@lru_cache(maxsize=1)
+def get_oauth_ssl_context() -> ssl.SSLContext:
+    """Build (once, on first use) the shared SSL context for OAuth HTTP calls.
+
+    Loading the certifi CA bundle takes ~150-200ms; deferring it until an
+    OAuth provider is actually used (rather than at module import time)
+    keeps that cost off the startup path for users who aren't using OAuth.
+    """
+    import certifi
+
+    return ssl.create_default_context(cafile=certifi.where())
+
 OAUTH_SUCCESS_HTML = b"""<!DOCTYPE html><html><head><title>Auth complete</title></head><body>
 <h2>Authentication successful!</h2>
 <p>You can close this window and return to the application.</p>

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/modes/interactive/components/layout.py

@@ -272,6 +272,8 @@ class Layout(Component):
         self._settings_panel: list[str] | None = None
         self._prompt = TextPrompt()
         self._oauth_status_lines: list[str] | None = None
+        self._editor_row = 0
+        self._editor_row_count = 0
 
         # Model palette — inline filtered model list for '/model <name> <provider>'
 
@@ -366,6 +368,7 @@ class Layout(Component):
         content.append(_divider("─" * width))
 
         if any_modal:
+            self._editor_row_count = 0
             # Modal replaces the input between the two dividers
             if self._selectors.is_active:
                 content.extend(self._selectors.render(width))
@@ -377,7 +380,10 @@ class Layout(Component):
                 content.extend(self._oauth_status_lines)
         else:
             # Normal: show the input editor
-            content.extend(self.input.render(width))
+            self._editor_row = len(content)
+            editor_lines = self.input.render(width)
+            self._editor_row_count = len(editor_lines)
+            content.extend(editor_lines)
 
         content.append(_divider("─" * width))
 
@@ -407,6 +413,20 @@ class Layout(Component):
         so this method deals only with editor and picker navigation.
         """
         if isinstance(event, MouseEvent):
+            if not event.pressed or event.button != 0:
+                return False
+            position = self._tui.mouse_position_for(self, event)
+            if position is None:
+                return False
+            row, column = position
+            editor_row = row - self._editor_row
+            if (
+                0 <= editor_row < self._editor_row_count
+                and isinstance(self.input, TextInput)
+                and self.input.move_cursor_to_visual(editor_row, column)
+            ):
+                self._tui.request_render()
+                return True
             return False
 
         if self._selectors.handle_input(event):

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/runtime/types.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
+import asyncio
 from dataclasses import dataclass
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel, Field
 
-from tau.agent.prompt.builder import build_prompt
+from tau.agent.prompt.builder import _git_status, build_prompt
 from tau.agent.prompt.types import PromptOptions
 from tau.agent.service import Agent
 from tau.agent.types import AgentConfig
@@ -207,6 +208,16 @@ class RuntimeContext:
                 )
             )
 
+        # Kick off the git-status snapshot now (needed later for the system prompt)
+        # so its subprocess calls run in a background thread concurrently with the
+        # rest of startup (settings, extension loading, resource discovery) instead
+        # of blocking the event loop right before the prompt is built.
+        git_task: asyncio.Task[str] | None = (
+            asyncio.create_task(asyncio.to_thread(_git_status, cwd))
+            if project_trusted and not config.system_prompt
+            else None
+        )
+
         # ── LLM ───────────────────────────────────────────────────────────────
         text_ref = settings_manager.get_model_ref("text")
         model_id = config.model_id or (text_ref.id if text_ref else None) or _DEFAULT_MODEL
@@ -351,6 +362,7 @@ class RuntimeContext:
         skills = skill_registry.list()
 
         # ── System prompt ─────────────────────────────────────────────────────
+        git_snapshot = await git_task if git_task is not None else None
         system_prompt = (
             config.system_prompt
             or resources.system_prompt
@@ -363,6 +375,7 @@ class RuntimeContext:
                     disable_context_files=config.disable_context_files,
                     project_trusted=project_trusted,
                     context_files=resources.context_files,
+                    git_snapshot=git_snapshot,
                 )
             )
         )

{tau_coding_agent-0.5.2 → tau_coding_agent-0.5.3}/tau/session/types.py

@@ -42,14 +42,19 @@ class SessionType(StrEnum):
 
 
 class BaseSessionEntry(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    # defer_build: these models (and the ~10-way discriminated union built from
+    # them) cost ~60ms of pydantic core-schema construction at import time even
+    # though session entries are only read/written once a session file is
+    # actually touched. Deferring moves that cost to first real use instead of
+    # every startup.
+    model_config = ConfigDict(arbitrary_types_allowed=True, defer_build=True)
     id: str = Field(default_factory=_generate_id)
     timestamp: float = Field(default_factory=generate_timestamp)
     parent_id: str | None = None
 
 
 class SessionHeader(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, defer_build=True)
     type: Literal[SessionType.SESSION_HEADER] = SessionType.SESSION_HEADER
     version: int = SESSION_VERSION
     id: str = Field(default_factory=_generate_id)
@@ -150,7 +155,7 @@ SessionFileEntry = Annotated[SessionHeader | SessionEntries, Field(discriminator
 
 
 class SessionTreeNode(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, defer_build=True)
     entry: SessionEntry
     children: list[SessionTreeNode] = Field(default_factory=list)
     label: str | None = None