wrencode 0.1.4__tar.gz → 0.1.4.1__tar.gz

{wrencode-0.1.4 → wrencode-0.1.4.1}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: wrencode
-Version: 0.1.4
+Version: 0.1.4.1
 Summary: A minimal agentic coding assistant in a single Python file
 Project-URL: Homepage, https://github.com/almostly/wrencode
 Project-URL: Repository, https://github.com/almostly/wrencode
 Project-URL: Issues, https://github.com/almostly/wrencode/issues
-Author: Denis Burakov
+Author: Almostly
 License-Expression: MIT
 Keywords: agent,ai,anthropic,cli,coding-assistant,llm,mlx,ollama,openai
 Classifier: Development Status :: 4 - Beta
@@ -55,6 +55,12 @@ saved choice, e.g. for CI.
 The standalone binary can't bundle the heavy ML stack, so the local-weights
 backends (`mlx`, `transformers`) are only offered when running from source.
 
+The default local models are
+[`deburky/gpt-oss-claude-code`](https://huggingface.co/deburky/gpt-oss-claude-code)
+(transformers) and
+[`deburky/gpt-oss-claude-mlx`](https://huggingface.co/deburky/gpt-oss-claude-mlx)
+(MLX) — override either with `MODEL=...`.
+
 ## Tools
 
 The agent has access to seven tools:
@@ -260,4 +266,8 @@ To clear history: use `/c` in the session, or delete `~/.wrencode/history.json`
 
 ## License
 
-MIT - Copyright 2026 Denis Burakov.
+MIT - Copyright 2026 Almostly.
+
+-----
+
+![Almostly](assets/almostly-badge.svg)

{wrencode-0.1.4 → wrencode-0.1.4.1}/README.md

@@ -32,6 +32,12 @@ saved choice, e.g. for CI.
 The standalone binary can't bundle the heavy ML stack, so the local-weights
 backends (`mlx`, `transformers`) are only offered when running from source.
 
+The default local models are
+[`deburky/gpt-oss-claude-code`](https://huggingface.co/deburky/gpt-oss-claude-code)
+(transformers) and
+[`deburky/gpt-oss-claude-mlx`](https://huggingface.co/deburky/gpt-oss-claude-mlx)
+(MLX) — override either with `MODEL=...`.
+
 ## Tools
 
 The agent has access to seven tools:
@@ -237,4 +243,8 @@ To clear history: use `/c` in the session, or delete `~/.wrencode/history.json`
 
 ## License
 
-MIT - Copyright 2026 Denis Burakov.
+MIT - Copyright 2026 Almostly.
+
+-----
+
+![Almostly](assets/almostly-badge.svg)

{wrencode-0.1.4 → wrencode-0.1.4.1}/pyproject.toml

@@ -9,7 +9,7 @@ description = "A minimal agentic coding assistant in a single Python file"
 readme = "README.md"
 requires-python = ">=3.9"
 license = "MIT"
-authors = [{ name = "Denis Burakov" }]
+authors = [{ name = "Almostly" }]
 keywords = [
     "agent",
     "llm",

{wrencode-0.1.4 → wrencode-0.1.4.1}/wrencode.py

@@ -8,14 +8,17 @@ HuggingFace Transformers, Anthropic, OpenAI, OpenRouter, and local proxy.
 Provides a tool-calling agent loop with file read/write/edit, glob, grep,
 and bash — enough to autonomously navigate and modify a codebase.
 
-Copyright 2026 Denis Burakov. All Rights Reserved.
+Copyright (c) 2026 Almostly
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 copies of the Software, and to permit persons to whom the Software is
-furnished to do so.
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
 
 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
@@ -67,7 +70,7 @@ for _dir in (os.path.dirname(os.path.abspath(__file__)), os.getcwd()):
 # -----------------------------------------------------------------------------------------------
 # Backend Configuration
 # -----------------------------------------------------------------------------------------------
-WRENCODE_VERSION = "0.1.4"
+WRENCODE_VERSION = "0.1.4.1"
 
 # Per-backend defaults. "kind" controls how a backend is treated:
 #   api         - hosted HTTP API, needs an API key
@@ -80,42 +83,42 @@ BACKEND_SPECS: dict[str, dict[str, str]] = {
         "model": "claude-haiku-4-5-20251001",
         "key_env": "ANTHROPIC_API_KEY",
         "api_base": "https://api.anthropic.com/v1/messages",
-        "label": "Anthropic Claude  (API key)",
+        "label": "Anthropic Claude (API key)",
     },
     "openai": {
         "kind": "api",
         "model": "gpt-4o-mini",
         "key_env": "OPENAI_API_KEY",
         "api_base": "https://api.openai.com/v1/chat/completions",
-        "label": "OpenAI GPT  (API key)",
+        "label": "OpenAI GPT (API key)",
     },
     "openrouter": {
         "kind": "api",
         "model": "anthropic/claude-3-haiku",
         "key_env": "OPENROUTER_API_KEY",
         "api_base": "https://openrouter.ai/api/v1/chat/completions",
-        "label": "OpenRouter — any model  (API key)",
+        "label": "OpenRouter — any model (API key)",
     },
     "local": {
         "kind": "local-proxy",
         "model": "gpt-oss-20b",
         "key_env": "LOCAL_API_KEY",
-        "label": "Local proxy  (Anthropic-compatible server on localhost)",
+        "label": "Local proxy (Anthropic-compatible server on localhost)",
     },
     "ollama": {
         "kind": "local-proxy",
         "model": "llama3.2",
-        "label": "Ollama  (local models via `ollama serve`)",
+        "label": "Ollama (local models via `ollama serve`)",
     },
     "transformers": {
         "kind": "local-ml",
         "model": "deburky/gpt-oss-claude-code",
-        "label": "HuggingFace Transformers  (CPU/GPU, source install)",
+        "label": "HuggingFace Transformers (CPU/GPU, source install)",
     },
     "mlx": {
         "kind": "local-ml",
-        "model": "mlx-community/Qwen2.5-3B-Instruct-4bit",
-        "label": "Apple Silicon via MLX  (source install)",
+        "model": "deburky/gpt-oss-claude-mlx",
+        "label": "Apple Silicon via MLX (source install)",
     },
 }
 
@@ -575,9 +578,54 @@ def flatten_content(content: Any) -> str:
     return "\n".join(parts)
 
 
+# Lightweight, language-agnostic code coloring — no pygments, keeps the binary lean.
+_CODE_TOKEN = re.compile(
+    r"(?P<comment>#[^\n]*|//[^\n]*)"
+    r"|(?P<string>\"[^\"\n]*\"|'[^'\n]*'|`[^`\n]*`)"
+    r"|(?P<num>\b\d[\d_.]*\b)"
+    r"|(?P<kw>\b(?:def|class|return|import|from|if|elif|else|for|while|try|except|"
+    r"finally|with|as|in|not|and|or|is|lambda|yield|async|await|pass|break|continue|"
+    r"raise|global|nonlocal|assert|True|False|None|const|let|var|function|export|"
+    r"default|new|this|fn|match|struct|enum|pub|use|mut|public|private|static|void)\b)"
+)
+
+
+def _highlight_code(code: str) -> str:
+    """Apply light ANSI syntax coloring to a code block (best-effort, any language)."""
+
+    def color(m: "re.Match[str]") -> str:
+        g = m.lastgroup
+        if g == "comment":
+            return f"{DIM}{m.group()}{RESET}"
+        if g == "string":
+            return f"{GREEN}{m.group()}{RESET}"
+        if g == "num":
+            return f"{YELLOW}{m.group()}{RESET}"
+        if g == "kw":
+            return f"{BLUE}{m.group()}{RESET}"
+        return m.group()
+
+    return _CODE_TOKEN.sub(color, code)
+
+
 def render_markdown(text: str) -> str:
-    """Render basic markdown bold syntax to terminal bold escape codes."""
-    return re.sub(r"\*\*(.+?)\*\*", f"{BOLD}\\1{RESET}", text)
+    """Render fenced code blocks (lightly highlighted), inline code, and bold."""
+    blocks: list[str] = []
+
+    def stash(m: "re.Match[str]") -> str:
+        lang = m.group(1) or ""
+        body = _highlight_code(m.group(2).rstrip("\n"))
+        head = f"{DIM}┌─ {lang}{RESET}\n" if lang else f"{DIM}┌─{RESET}\n"
+        bordered = "\n".join(f"{DIM}│{RESET} {ln}" for ln in body.split("\n"))
+        blocks.append(f"\n{head}{bordered}\n{DIM}└─{RESET}")
+        return f"\x00B{len(blocks) - 1}\x00"
+
+    text = re.sub(r"```(\w*)\n?(.*?)```", stash, text, flags=re.DOTALL)
+    text = re.sub(r"`([^`\n]+)`", f"{CYAN}\\1{RESET}", text)
+    text = re.sub(r"\*\*(.+?)\*\*", f"{BOLD}\\1{RESET}", text)
+    for i, b in enumerate(blocks):
+        text = text.replace(f"\x00B{i}\x00", b)
+    return text
 
 
 # -----------------------------------------------------------------------------------------------
@@ -1486,6 +1534,29 @@ def load_model() -> Optional[tuple[Any, Any]]:
     if BACKEND == "local":
         print(f"{DIM}Local proxy at {API_BASE}{RESET}\n")
         return None
+    if BACKEND == "ollama":
+        base = os.environ.get("OLLAMA_HOST", "http://localhost:11434").rstrip("/")
+        try:
+            with urllib.request.urlopen(f"{base}/api/tags", timeout=3) as r:
+                installed = {
+                    m.get("name", "")
+                    for m in json.loads(r.read().decode()).get("models", [])
+                }
+            want = MODEL if ":" in MODEL else f"{MODEL}:latest"
+            if not (
+                MODEL in installed
+                or want in installed
+                or any(n.split(":")[0] == MODEL for n in installed)
+            ):
+                have = ", ".join(sorted(installed)) or "none"
+                print(f"{YELLOW}⚠ Model '{MODEL}' isn't pulled into Ollama.{RESET}")
+                print(f"{DIM}  Run: ollama pull {MODEL}   (installed: {have}){RESET}")
+        except Exception:
+            print(
+                f"{YELLOW}⚠ Couldn't reach Ollama at {base} — is `ollama serve` running?{RESET}"
+            )
+        print(f"{DIM}{BACKEND} ({MODEL}){RESET}\n")
+        return None
     # Hosted API backends — all require a key.
     if not API_KEY:
         key_env = BACKEND_SPECS[BACKEND]["key_env"]
@@ -1539,6 +1610,7 @@ def main() -> None:
     )
     resolve_configuration(force_chooser=force)
 
+    sys.stdout.write("\033]0;wrencode\007")  # set terminal tab/window title
     print(WREN_BANNER)
     print(f"{BOLD}wrencode{RESET} 🐦 | {DIM}{BACKEND}:{MODEL}{RESET}\n")
     mlx_state = load_model()
@@ -1550,7 +1622,7 @@ def main() -> None:
 
     while True:
         try:
-            user_input = input(f"{BOLD}{BLUE}❯{RESET} ").strip()
+            user_input = input(f"{BRIGHT_CYAN}❯{RESET} ").strip()
             if not user_input:
                 continue
             action = handle_slash_command(user_input, messages, mlx_state)
@@ -1568,8 +1640,17 @@ def main() -> None:
         except EOFError:
             break
         except Exception as err:
-            print(f"{RED}Error: {err}{RESET}")
-            traceback.print_exc()
+            msg = str(err)
+            print(f"{RED}Error: {msg}{RESET}")
+            if BACKEND == "ollama" and ("not found" in msg.lower() or "404" in msg):
+                print(
+                    f"{YELLOW}Model '{MODEL}' isn't pulled. "
+                    f"Run: ollama pull {MODEL}  (or `ollama list`).{RESET}"
+                )
+            if os.environ.get("WRENCODE_DEBUG"):
+                traceback.print_exc()
+            else:
+                print(f"{DIM}(set WRENCODE_DEBUG=1 for the full traceback){RESET}")
 
 
 if __name__ == "__main__":