ata-coder 2.4.2__py3-none-any.whl

ata_coder/types.py

@@ -0,0 +1,92 @@
+"""
+Shared type aliases for ATA Coder.
+
+Single source of truth for Message, ToolDef, and other
+commonly-referenced types that were previously duplicated
+across llm_client.py, anthropic_client.py, and agent.py.
+
+TypedDict variants are available for static analysis (mypy, pyright).
+The legacy ``Message`` and ``ToolDef`` aliases remain for runtime
+compatibility and minimal-change migrations.
+"""
+
+from abc import ABC, abstractmethod
+from typing import Any, AsyncIterator, Callable, TypedDict
+
+# ── Legacy aliases (widely used, keep for backward compat) ──────────────────
+
+# OpenAI-compatible message dict: {"role": "...", "content": "...", ...}
+Message = dict[str, Any]
+
+# OpenAI-compatible tool definition dict
+ToolDef = dict[str, Any]
+
+
+# ── TypedDict variants (opt-in for stricter type checking) ──────────────────
+
+class MessageDict(TypedDict, total=False):
+    """OpenAI-compatible message. ``content`` may be None for tool_calls."""
+    role: str
+    content: str | None
+    tool_calls: list[dict[str, Any]]
+    tool_call_id: str
+    name: str
+    reasoning_content: str
+
+
+class ToolDefDict(TypedDict, total=False):
+    """OpenAI-compatible tool definition."""
+    type: str
+    function: dict[str, Any]
+
+
+class ToolCallDict(TypedDict, total=False):
+    """A single tool-call within an assistant message."""
+    id: str
+    type: str
+    function: dict[str, Any]
+
+
+class BaseLLMClient(ABC):
+    """Abstract base for async LLM clients (OpenAI, Anthropic, etc.).
+
+    Both LLMClient and AnthropicClient implement this interface so
+    higher-level code (CoderAgent, SubAgent) can be written against
+    the abstraction rather than branching on provider.
+
+    All I/O methods are async — the client runs on the asyncio event loop.
+    """
+
+    @abstractmethod
+    async def chat(self, messages: list[Message], tools: list[ToolDef] | None = None,
+                   system_prompt: str = "") -> Message:
+        """Send messages and return the assistant response."""
+        ...
+
+    @abstractmethod
+    async def chat_stream(self, messages: list[Message], tools: list[ToolDef] | None = None,
+                          system_prompt: str = "") -> AsyncIterator[tuple[str, Any]]:
+        """Stream a chat completion. Yields (delta_type, content) tuples."""
+        ...
+
+    @abstractmethod
+    def register_tools(self, tools: list[ToolDef]) -> None:
+        """Register tool definitions for subsequent requests."""
+        ...
+
+    @abstractmethod
+    def count_tokens_approx(self, messages: list[Message]) -> int:
+        """Estimate token count for a message list."""
+        ...
+
+    @abstractmethod
+    async def close(self) -> None:
+        """Release resources (HTTP sessions, etc.)."""
+        ...
+
+    def on_usage(self, callback: Callable[[int, int], None]) -> None:
+        """Register a callback for token usage: callback(prompt_tokens, completion_tokens).
+
+        Default no-op — override if the client tracks usage.
+        """
+        pass

ata_coder/utils.py

@@ -0,0 +1,113 @@
+"""
+Shared display utilities — used by both the GUI and the API server.
+
+- brief_args: truncate tool-call argument dicts for compact display
+- enhance_api_error: add helpful suggestions to common API errors
+- try_import_yaml: shared conditional yaml import (used by memory + skills)
+"""
+
+from typing import Any
+
+
+# ── YAML availability (shared across memory.py, skills.py, etc.) ──────────
+
+try:
+    import yaml as _yaml
+    HAS_YAML = True
+except ImportError:
+    _yaml = None  # type: ignore
+    HAS_YAML = False
+
+
+def try_import_yaml():
+    """Return (yaml_module, True) if PyYAML is installed, else (None, False).
+    """
+    return _yaml, HAS_YAML
+
+
+def brief_args(args: dict[str, Any] | None, max_str_len: int = 100) -> str:
+    """Format tool-call arguments into a compact single-line summary.
+
+    Returns an empty string for None/empty dicts.  String values longer
+    than *max_str_len* are truncated with an ellipsis.
+
+    Used by:
+      gui.py   — tool-call display in the chat pane
+      server.py — SSE stream event logging
+    """
+    if not args:
+        return ""
+    parts = []
+    for k, v in args.items():
+        if isinstance(v, str):
+            s = f"{k}="
+            if len(v) > max_str_len:
+                s += f'"{v[:max_str_len]}…"'
+            else:
+                s += f'"{v}"'
+            parts.append(s)
+        else:
+            parts.append(f"{k}={v}")
+    return "  ".join(parts)
+
+
+def enhance_api_error(status_code: int, error_message: str, base_url: str = "") -> str:
+    """Add helpful troubleshooting suggestions to common API errors."""
+    msg = error_message
+
+    if "model" in error_message.lower() and ("not found" in error_message.lower() or "not supported" in error_message.lower()):
+        msg += (
+            "\n\n💡 This usually means the model name is incorrect for this provider."
+            "\n   → Check your ATA_CODER_DEFAULT_MODEL in settings.json"
+            "\n   → For DeepSeek: try 'deepseek-chat' or 'deepseek-v4-pro' (no [1m] suffix)"
+            "\n   → Run 'ata --list-models' to see what the API reports as available"
+        )
+    elif "unauthorized" in error_message.lower() or status_code == 401:
+        msg += (
+            "\n\n🔑 Authentication failed. Check your API key:"
+            "\n   → Settings: ~/.ata_coder/settings.json → env.ATA_CODER_API_KEY"
+            "\n   → Or set ATA_CODER_API_KEY environment variable"
+        )
+    elif "rate limit" in error_message.lower() or status_code == 429:
+        msg += (
+            "\n\n⏳ Rate limited — the API is throttling requests."
+            "\n   → Wait a moment and try again"
+            "\n   → Consider switching to a model with higher rate limits"
+        )
+    elif "connection" in error_message.lower() or status_code in (502, 503, 504):
+        msg += (
+            "\n\n🌐 Server connectivity issue:"
+            "\n   → Check: is the API base URL correct? → " + (base_url or "check settings.json")
+            + "\n   → The server may be temporarily overloaded"
+            "\n   → Try again in a few seconds"
+        )
+    elif "context length" in error_message.lower() or "too long" in error_message.lower():
+        msg += (
+            "\n\n📏 Context length exceeded:"
+            "\n   → Try breaking your task into smaller steps"
+            "\n   → Use /compact to shrink conversation history"
+            "\n   → Increase ATA_CODER_MAX_OUTPUT_TOKENS for larger context"
+        )
+
+    return msg
+
+
+# ── Surrogate sanitization ─────────────────────────────────────────────────
+
+
+def sanitize_surrogates(obj: Any) -> Any:
+    """Recursively replace lone surrogates (U+D800–U+DFFF) in all strings.
+
+    Python's ``json.dumps(ensure_ascii=False)`` can emit lone surrogates that
+    are invalid UTF-8.  When httpx or a file write tries ``.encode("utf-8")``
+    on the result, Python raises ``UnicodeEncodeError: surrogates not allowed``.
+
+    This round-trip through UTF-8 with ``errors="replace"`` strips them safely.
+    """
+    if isinstance(obj, str):
+        return obj.encode("utf-8", errors="replace").decode("utf-8")
+    if isinstance(obj, dict):
+        return {sanitize_surrogates(k): sanitize_surrogates(v) for k, v in obj.items()}
+    if isinstance(obj, list):
+        return [sanitize_surrogates(v) for v in obj]
+    return obj

ata_coder/web/css/style.css

@@ -0,0 +1,180 @@
+/* ═══════════════════════════ One Dark Pro Theme ═══════════════════════════ */
+:root {
+    --bg: #282C34;        --bg-panel: #21252B;    --bg-input: #1E2127;
+    --fg: #ABB2BF;        --comment: #5C6370;
+    --red: #E06C75;       --green: #98C379;       --yellow: #D19A66;
+    --blue: #61AFEF;      --purple: #C678DD;      --cyan: #56B6C2;
+    --orange: #E5C07B;    --teal: #7EE787;
+    --border: #3A3F4B;    --radius: 10px;
+    --transition: 0.25s cubic-bezier(0.4, 0, 0.2, 1);
+    --font-mono: 'Cascadia Code', 'Fira Code', 'JetBrains Mono', Consolas, monospace;
+    --font-ui: 'Segoe UI', system-ui, -apple-system, BlinkMacSystemFont, sans-serif;
+}
+
+* { margin: 0; padding: 0; box-sizing: border-box; }
+
+body {
+    font-family: var(--font-ui); font-size: 14px;
+    background: var(--bg); color: var(--fg);
+    height: 100vh; overflow: hidden; display: flex;
+    -webkit-font-smoothing: antialiased;
+}
+
+/* ═══════════════════════════ Sidebar ═══════════════════════════ */
+.sidebar {
+    width: 260px; min-width: 260px; background: var(--bg-panel);
+    border-right: 1px solid var(--border); display: flex;
+    flex-direction: column; transition: transform var(--transition);
+}
+.sidebar-header {
+    padding: 20px 18px 14px; border-bottom: 1px solid var(--border);
+}
+.sidebar-header .logo { font-size: 18px; font-weight: 700; color: var(--blue); }
+.sidebar-header .logo span { color: var(--fg); font-weight: 400; font-size: 12px; }
+.sidebar-body { flex: 1; overflow-y: auto; padding: 12px 0; }
+.sidebar-section { padding: 8px 18px; }
+.sidebar-section h4 { font-size: 11px; text-transform: uppercase;
+    color: var(--comment); letter-spacing: 0.5px; margin-bottom: 6px; }
+.sidebar-section .val { font-size: 13px; color: var(--fg); word-break: break-all; }
+.sidebar-section select { width: 100%; background: var(--bg-input); color: var(--fg);
+    border: 1px solid var(--border); border-radius: 6px; padding: 6px 8px;
+    font-size: 13px; cursor: pointer; }
+.skill-tag { display: inline-block; font-size: 11px; padding: 2px 8px;
+    background: var(--bg); border: 1px solid var(--border); border-radius: 12px;
+    margin: 2px 3px 2px 0; color: var(--cyan); }
+.sidebar-footer { padding: 12px 18px; border-top: 1px solid var(--border); }
+.shortcut { font-size: 12px; color: var(--comment); padding: 4px 0; cursor: pointer; }
+.shortcut:hover { color: var(--fg); }
+.shortcut kbd { font-family: var(--font-mono); font-size: 11px; background: var(--bg);
+    border: 1px solid var(--border); border-radius: 3px; padding: 1px 5px; margin-right: 4px; }
+
+/* ═══════════════════════════ Main Chat ═══════════════════════════ */
+.main { flex: 1; display: flex; flex-direction: column; min-width: 0; }
+
+/* ── Top Bar ── */
+.top-bar {
+    height: 44px; min-height: 44px; background: var(--bg-panel);
+    border-bottom: 1px solid var(--border); display: flex;
+    align-items: center; padding: 0 16px; gap: 8px;
+}
+.hamburger { display: none; background: none; border: none; color: var(--fg);
+    font-size: 20px; cursor: pointer; padding: 4px 8px; }
+.top-bar .title { font-weight: 600; font-size: 14px; color: var(--blue); }
+.top-bar .subtitle { font-size: 12px; color: var(--comment); margin-left: auto; }
+.top-bar .dot { width: 8px; height: 8px; border-radius: 50%; margin-right: 6px; }
+.dot.ready { background: var(--green); }
+.dot.thinking { background: var(--yellow); animation: pulse 1s infinite; }
+.dot.error { background: var(--red); }
+
+@keyframes pulse { 0%,100%{opacity:1} 50%{opacity:0.4} }
+
+/* ── Chat Area ── */
+.chat { flex: 1; overflow-y: auto; padding: 16px 20px;
+    scroll-behavior: smooth; line-height: 1.6; }
+.chat::-webkit-scrollbar { width: 6px; }
+.chat::-webkit-scrollbar-thumb { background: var(--border); border-radius: 3px; }
+
+/* ── Message Bubbles ── */
+.msg { margin-bottom: 14px; max-width: 85%; animation: fadeIn 0.2s ease; }
+@keyframes fadeIn { from{opacity:0;transform:translateY(4px)} to{opacity:1;transform:translateY(0)} }
+
+.msg.user { margin-left: auto; background: var(--blue); color: #fff;
+    padding: 10px 16px; border-radius: 18px 18px 4px 18px; font-weight: 500; }
+.msg.agent { padding: 2px 0; color: var(--fg); }
+.msg.agent p { margin: 4px 0; }
+.msg.agent h1, .msg.agent h2, .msg.agent h3 { font-weight: 700; margin: 10px 0 6px; }
+.msg.agent h1 { font-size: 1.25em; color: var(--fg); }
+.msg.agent h2 { font-size: 1.12em; color: var(--fg); }
+.msg.agent h3 { font-size: 1.05em; color: var(--fg); }
+.msg.agent code { font-family: var(--font-mono); font-size: 13px;
+    background: var(--bg-panel); padding: 2px 6px; border-radius: 4px; }
+.msg.agent pre { font-family: var(--font-mono); font-size: 13px;
+    background: var(--bg-panel); padding: 12px 16px; border-radius: 8px;
+    overflow-x: auto; margin: 8px 0; border: 1px solid var(--border);
+    line-height: 1.5; white-space: pre-wrap; word-break: break-word; }
+
+.msg.think { margin-left: 28px; padding: 8px 14px; background: var(--bg-panel);
+    border-left: 2px solid var(--teal); border-radius: 0 8px 8px 0;
+    font-size: 13px; color: var(--teal); font-style: italic; }
+.msg.tool { margin-left: 32px; padding: 6px 12px; font-size: 12px;
+    font-family: var(--font-mono); color: var(--purple); }
+.msg.tool-ok { color: var(--green); margin-left: 44px; font-size: 12px; }
+.msg.tool-err { color: var(--red); margin-left: 44px; font-size: 12px; }
+.msg.error { padding: 10px 16px; background: rgba(224,108,117,0.1);
+    border: 1px solid var(--red); border-radius: 8px; color: var(--red);
+    font-weight: 600; }
+.msg.status { text-align: center; font-size: 12px; color: var(--comment);
+    margin: 8px 0; }
+.msg.sep { text-align: center; font-size: 11px; color: var(--comment);
+    border-top: 1px solid var(--border); margin: 16px 0; padding-top: 12px; }
+
+/* ── Input Row ── */
+.input-row {
+    display: flex; align-items: center; gap: 10px;
+    padding: 12px 16px; background: var(--bg-panel);
+    border-top: 1px solid var(--border);
+}
+#cmd-input {
+    flex: 1; background: var(--bg-input); color: var(--fg);
+    border: 1px solid var(--border); border-radius: 24px;
+    padding: 10px 18px; font-size: 14px; font-family: var(--font-ui);
+    outline: none; transition: border-color var(--transition);
+}
+#cmd-input:focus { border-color: var(--blue); }
+#send-btn {
+    background: var(--blue); color: #fff; border: none;
+    border-radius: 24px; padding: 10px 24px; font-size: 14px;
+    font-weight: 600; cursor: pointer; transition: opacity var(--transition);
+    white-space: nowrap;
+}
+#send-btn:hover { opacity: 0.85; }
+#send-btn:disabled { opacity: 0.4; cursor: not-allowed; }
+#stop-btn {
+    background: var(--red); color: #fff; border: none;
+    border-radius: 24px; padding: 10px 24px; font-size: 14px;
+    font-weight: 600; cursor: pointer; transition: opacity var(--transition);
+    white-space: nowrap;
+}
+#stop-btn:hover { opacity: 0.85; }
+/* Streaming state: toggle buttons + disable input */
+body:not(.streaming) #stop-btn  { display: none; }
+body.streaming      #send-btn  { display: none; }
+body.streaming      #cmd-input { opacity: 0.5; pointer-events: none; }
+
+/* ── Command Popup ── */
+.cmd-popup {
+    display: none; position: absolute; bottom: 72px; left: 16px; right: 16px;
+    background: var(--bg-panel); border: 1px solid var(--border);
+    border-radius: 8px; max-height: 240px; overflow-y: auto; z-index: 100;
+    box-shadow: 0 8px 24px rgba(0,0,0,0.4);
+}
+.cmd-popup.show { display: block; }
+.cmd-item { padding: 8px 16px; cursor: pointer; display: flex;
+    justify-content: space-between; font-size: 13px; }
+.cmd-item:hover, .cmd-item.active { background: rgba(97,175,239,0.1); }
+.cmd-item .name { font-family: var(--font-mono); color: var(--cyan); }
+.cmd-item .desc { color: var(--comment); font-size: 12px; }
+
+/* ── Status Bar ── */
+.status-bar {
+    height: 26px; min-height: 26px; background: var(--bg-panel);
+    border-top: 1px solid var(--border); display: flex;
+    align-items: center; padding: 0 16px; font-size: 11px; color: var(--comment);
+    gap: 16px;
+}
+.status-bar span { white-space: nowrap; }
+
+/* ═══════════════════════════ Mobile ═══════════════════════════ */
+@media (max-width: 768px) {
+    .sidebar { position: fixed; left: 0; top: 0; bottom: 0; z-index: 200;
+        transform: translateX(-100%); box-shadow: 4px 0 20px rgba(0,0,0,0.3); }
+    .sidebar.open { transform: translateX(0); }
+    .hamburger { display: block; }
+    .msg { max-width: 95%; }
+    .msg.user { margin-left: 4px; }
+    #cmd-input { font-size: 16px; } /* prevent iOS zoom on focus */
+    .overlay { display: none; position: fixed; inset: 0; background: rgba(0,0,0,0.4);
+        z-index: 199; }
+    .overlay.show { display: block; }
+    .top-bar .subtitle { display: none; }
+}

ata_coder/web/index.html

@@ -0,0 +1,84 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+<meta charset="UTF-8">
+<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no">
+<meta name="apple-mobile-web-app-capable" content="yes">
+<meta name="apple-mobile-web-app-status-bar-style" content="black-translucent">
+<meta name="theme-color" content="#282C34">
+<title>ATA Coder</title>
+<link rel="stylesheet" href="css/style.css">
+</head>
+<body>
+
+<!-- ── Sidebar Overlay (mobile) ── -->
+<div class="overlay" id="overlay"></div>
+
+<!-- ── Sidebar ── -->
+<aside class="sidebar" id="sidebar">
+  <div class="sidebar-header">
+    <div class="logo">⚡ ATA Coder <span id="logo-ver">v2.2.0</span></div>
+  </div>
+  <div class="sidebar-body">
+    <div class="sidebar-section">
+      <h4>Model</h4>
+      <select id="model-select">
+        <option>loading…</option>
+      </select>
+    </div>
+    <div class="sidebar-section">
+      <h4>Skills</h4>
+      <div id="skill-tags"><span class="skill-tag">general-coder</span></div>
+    </div>
+    <div class="sidebar-section">
+      <h4>Workspace</h4>
+      <div class="val" id="ws-path">…</div>
+    </div>
+  </div>
+  <div class="sidebar-footer">
+    <div class="shortcut" data-cmd="/clear"><kbd>/clear</kbd> New chat</div>
+    <div class="shortcut" data-cmd="/history"><kbd>/history</kbd> Sessions</div>
+    <div class="shortcut" data-cmd="/skills"><kbd>/skills</kbd> List skills</div>
+    <div class="shortcut" data-cmd="/models"><kbd>/models</kbd> List models</div>
+  </div>
+</aside>
+
+<!-- ── Main Area ── -->
+<div class="main">
+  <!-- Top Bar -->
+  <div class="top-bar">
+    <button class="hamburger" aria-label="Menu">☰</button>
+    <span class="dot ready" id="status-dot"></span>
+    <span class="title">ATA Coder</span>
+    <span class="subtitle" id="top-subtitle"></span>
+  </div>
+
+  <!-- Chat -->
+  <div class="chat" id="chat">
+    <div class="msg status">⚡ ATA Coder — AI-powered coding assistant</div>
+    <div class="msg status" id="init-info">Connecting…</div>
+  </div>
+
+  <!-- Command Popup -->
+  <div class="cmd-popup" id="cmd-popup"></div>
+
+  <!-- Input Row -->
+  <div class="input-row" id="input-row">
+    <input type="text" id="cmd-input" placeholder="Ask anything… / for commands"
+           autocomplete="off" autocapitalize="off" spellcheck="false">
+    <button id="send-btn">Send</button>
+    <button id="stop-btn">Stop</button>
+  </div>
+
+  <!-- Status Bar -->
+  <div class="status-bar">
+    <span id="st-tokens">tokens: ~0</span>
+    <span id="st-skill">skill: general-coder</span>
+    <span id="st-model">model: …</span>
+    <span id="st-time" style="margin-left:auto">0.0s</span>
+  </div>
+</div>
+
+<script src="js/app.js"></script>
+</body>
+</html>