mlx-code 0.0.9__tar.gz → 0.0.10__tar.gz

{mlx_code-0.0.9 → mlx_code-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlx-code
-Version: 0.0.9
+Version: 0.0.10
 Summary: Coding Agent for Mac
 Home-page: https://github.com/JosefAlbers/mlx-code
 Author: J Joe
@@ -31,7 +31,7 @@ Dynamic: summary
 
 A lightweight coding agent built on Apple's MLX framework.
 
-![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)
+[![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)](https://youtu.be/0lkY7YQCyCo)
 
 ---
 
@@ -48,68 +48,65 @@ A lightweight coding agent built on Apple's MLX framework.
 ## Quick Start
 
 ```bash
-pip install mlx-code[all]
-mlxc
+pip install mlx-code
+mlc
 ```
 
 ---
 
 ## Command Line
 
-### `mlxc`: local server + harness
+### `mlc`: local server + harness
 
 Starts the MLX inference server and launches a harness against it.
 
 ```bash
 # Default: local MLX server + built-in REPL harness
-mlxc
+mlc
 
 # Use a different harness (routes traffic through the local server)
-mlxc --leash claude
-mlxc --leash gemini
-mlxc --leash codex
+mlc --leash claude
+mlc --leash gemini
+mlc --leash codex
 
 # Server only, no harness
-mlxc --leash none
+mlc --leash none
 
 # Specify a model
-mlxc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
+mlc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
 
 # Restrict the tools available to the agent
-mlxc --tools Read Write Bash
+mlc --tools Read Write Bash
 
 # Custom system prompt
-mlxc --system "You are a helpful assistant."
+mlc --system "You are a helpful assistant."
 
 # Load skills from a directory (scans recursively for SKILL.md files)
-mlxc --skill ./my-skills
+mlc --skill ./my-skills
 
 # Resume a previous session from a git commit hash
-mlxc --resume <commit-hash>
+mlc --resume <commit-hash>
 
-# Because `mlxc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
-echo "explain lsp.py" | mlxc -d | cat - PLAN.md | mlxc
+# Because `mlc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
+echo "Here's the solution you proposed: <excerpt>$(mlc -p "write code for a chrome extension to play youtube x5 speed")</excerpt> Now argue against it. What are the edge cases this doesn't handle? What assumptions did you make that might not hold in a production system? What would you change if you knew this code would be read by a senior engineer in a security audit?" | mlc
 ```
 
-### `mlxc-run`: harness only
+### `mlc-run`: harness only
 
 Runs the agent harness against an already-running server or a remote provider.
 
 ```bash
 # Connect to a local server at 127.0.0.1:8000 (default)
-mlxc-run
+mlc-run
 
 # Remote providers
-mlxc-run --api claude
-mlxc-run --api gemini
-mlxc-run --api deepseek --model deepseek-v4-pro
-mlxc-run --api codex
+mlc-run --api claude
+mlc-run --api gemini
+mlc-run --api deepseek --model deepseek-v4-pro
+mlc-run --api codex
 
 # Custom endpoint
-mlxc-run --url http://localhost:9000
-
-# With skills
-mlxc-run --skill ./my-skills
+echo "explain lsp.py" | mlc-run -a deepseek | cat - PLAN.md | mlc-run --url http://localhost:9000
 ```
 
 ---

{mlx_code-0.0.9 → mlx_code-0.0.10}/README.md

@@ -2,7 +2,7 @@
 
 A lightweight coding agent built on Apple's MLX framework.
 
-![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)
+[![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)](https://youtu.be/0lkY7YQCyCo)
 
 ---
 
@@ -19,68 +19,65 @@ A lightweight coding agent built on Apple's MLX framework.
 ## Quick Start
 
 ```bash
-pip install mlx-code[all]
-mlxc
+pip install mlx-code
+mlc
 ```
 
 ---
 
 ## Command Line
 
-### `mlxc`: local server + harness
+### `mlc`: local server + harness
 
 Starts the MLX inference server and launches a harness against it.
 
 ```bash
 # Default: local MLX server + built-in REPL harness
-mlxc
+mlc
 
 # Use a different harness (routes traffic through the local server)
-mlxc --leash claude
-mlxc --leash gemini
-mlxc --leash codex
+mlc --leash claude
+mlc --leash gemini
+mlc --leash codex
 
 # Server only, no harness
-mlxc --leash none
+mlc --leash none
 
 # Specify a model
-mlxc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
+mlc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
 
 # Restrict the tools available to the agent
-mlxc --tools Read Write Bash
+mlc --tools Read Write Bash
 
 # Custom system prompt
-mlxc --system "You are a helpful assistant."
+mlc --system "You are a helpful assistant."
 
 # Load skills from a directory (scans recursively for SKILL.md files)
-mlxc --skill ./my-skills
+mlc --skill ./my-skills
 
 # Resume a previous session from a git commit hash
-mlxc --resume <commit-hash>
+mlc --resume <commit-hash>
 
-# Because `mlxc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
-echo "explain lsp.py" | mlxc -d | cat - PLAN.md | mlxc
+# Because `mlc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
+echo "Here's the solution you proposed: <excerpt>$(mlc -p "write code for a chrome extension to play youtube x5 speed")</excerpt> Now argue against it. What are the edge cases this doesn't handle? What assumptions did you make that might not hold in a production system? What would you change if you knew this code would be read by a senior engineer in a security audit?" | mlc
 ```
 
-### `mlxc-run`: harness only
+### `mlc-run`: harness only
 
 Runs the agent harness against an already-running server or a remote provider.
 
 ```bash
 # Connect to a local server at 127.0.0.1:8000 (default)
-mlxc-run
+mlc-run
 
 # Remote providers
-mlxc-run --api claude
-mlxc-run --api gemini
-mlxc-run --api deepseek --model deepseek-v4-pro
-mlxc-run --api codex
+mlc-run --api claude
+mlc-run --api gemini
+mlc-run --api deepseek --model deepseek-v4-pro
+mlc-run --api codex
 
 # Custom endpoint
-mlxc-run --url http://localhost:9000
-
-# With skills
-mlxc-run --skill ./my-skills
+echo "explain lsp.py" | mlc-run -a deepseek | cat - PLAN.md | mlc-run --url http://localhost:9000
 ```
 
 ---

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/main.py

@@ -879,7 +879,7 @@ def generate_step(
         y, logprobs = (next_y, next_logprobs)
         if n <= _tl:
             _td.append(_yi)
-            if is_stuck(_td) or (n == _tl and _te not in _td):
+            if (is_stuck(_td) or n == _tl) and _te not in _td:
                 y = mx.array([_te])
                 _tl = -1
         n += 1
@@ -1505,8 +1505,19 @@ def serve(
 
 
 def main():
-    parser = argparse.ArgumentParser(
-        description="Serve a local MLX model with an OpenAI-compatible API."
+    parser = argparse.ArgumentParser(description="mlx-code MAIN")
+    parser.add_argument(
+        "-p",
+        "--prompt",
+        default=None,
+        help="Initial prompt sent automatically when the REPL starts",
+    )
+    parser.add_argument(
+        "-r",
+        "--resume",
+        default=None,
+        metavar="COMMIT",
+        help="Resume a previous session from the given git commit hash",
     )
     parser.add_argument(
         "-m",
@@ -1567,17 +1578,6 @@ def main():
         ],
         help="Regex patterns stripped from model output before it is returned to the client",
     )
-    parser.add_argument(
-        "--prompt",
-        default=None,
-        help="Initial prompt sent automatically when the REPL starts",
-    )
-    parser.add_argument(
-        "--resume",
-        default=None,
-        metavar="COMMIT",
-        help="Resume a previous session from the given git commit hash",
-    )
     parser.add_argument("--stream", default=None, help="File to stream log into")
     args, leash_args = parser.parse_known_args()
     logger.debug(f"args={args!r} leash_args={leash_args!r}")

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/mcb.py

@@ -183,7 +183,7 @@ class KB:
 
 class DocThread:
     def __init__(self, kb: KB | None = None):
-        self.kb = KB("mcb.json") if kb is None else kb
+        self.kb = KB(db_path="mcb.json") if kb is None else kb
         self.submissions = []
 
     def submit(

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/mcb_tool.py

@@ -38,9 +38,9 @@ class CommentKBTool(Tool):
         dt = self.ctx.get("dt")
         if dt is None:
             return tout("No knowledge base is available", True)
-        if params.to not in dt.kb:
-            return tout(f"Comment not found: {params.to}", True)
-        return tout(json.dumps({"id": dt.comment(params.content, to=params.to)}))
+        if params.parent not in dt.kb:
+            return tout(f"Comment not found: {params.parent}", True)
+        return tout(json.dumps({"id": dt.comment(params.content, to=params.parent)}))
 
 
 class SubmitKBParams(BaseModel):
@@ -70,4 +70,4 @@ def system_prompt(dt):
 
 
 ALL_TOOLS = [ReadKBTool, CommentKBTool, SubmitKBTool]
-ALL_NAMES = [i.name for i in ALL_TOOLS]
+ALL_NAMES = [i.name for i in ALL_TOOLS] + ["Agent"]

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/repl.py

@@ -234,13 +234,13 @@ class Agent:
         return msg
 
 
-_REPL_HELP = "Commands:\n  /help          — show this message\n  /clear         — clear conversation history\n  /history       — print message history\n  /tools         — list active tools\n  /branch        — spawn a branched sub-agent and run a one-shot prompt\n  /abort         — signal abort after next tool call\n  exit / quit    — end the session\n"
+_REPL_HELP = "Commands:\n  /help          — show this message\n  /clear         — clear conversation history\n  /history       — print message history\n  /tools         — list active tools\n  /branch        — spawn a branched sub-agent and run a one-shot prompt\n  /abort         — signal abort after next tool call\n  /quit / exit    — end the session\n"
 import sys
 import os
 from contextlib import contextmanager
 
 
-def read_input(prompt: str = "\x1b[32m≫\x1b[0m ") -> str:
+def read_input(prompt: str = "\x1b[32m≫ \x1b[0m") -> str:
     if sys.platform == "win32":
         return _read_input_win(prompt)
     else:
@@ -416,7 +416,7 @@ def _read_input_win(prompt: str) -> str:
 
 
 async def repl(agent, init_prompt=None) -> None:
-    is_tty = sys.stdin.isatty()
+    is_tty = sys.stdin.isatty() and sys.stdout.isatty()
     loop = asyncio.get_running_loop()
     _suppress = False
     last_block = ""
@@ -467,9 +467,7 @@ async def repl(agent, init_prompt=None) -> None:
             elif et == "tool_start":
                 name_part = "\x1b[1;43;30;1m" + p["name"] + "\x1b[0m"
                 args_part = (
-                    " \x1b[2;33m" + json.dumps(p["args"]) + "\x1b[0m"
-                    if p["args"]
-                    else ""
+                    " \x1b[33m" + json.dumps(p["args"]) + "\x1b[0m" if p["args"] else ""
                 )
                 emit(name_part + args_part + "\n", "tool")
             elif et == "tool_end":
@@ -504,6 +502,8 @@ async def repl(agent, init_prompt=None) -> None:
         else:
             user_input = sys.stdin.read()
         if not user_input:
+            if not is_tty:
+                break
             continue
         last_block = ""
         logger.debug(user_input)
@@ -556,7 +556,7 @@ async def repl(agent, init_prompt=None) -> None:
                 print(f"Unknown command: {cmd}  (try /help)")
             continue
         if is_tty:
-            if user_input.lower() in {"exit", "quit"}:
+            if user_input.lower() in {"exit", "/quit"}:
                 print("Bye!")
                 break
             print("\x1b[34mπ\x1b[0m ", end="", flush=True)
@@ -682,6 +682,19 @@ def main():
 
     setup_logger(log_file=".log.json")
     parser = argparse.ArgumentParser(description="mlx-code REPL")
+    parser.add_argument(
+        "-p",
+        "--prompt",
+        default=None,
+        help="Initial prompt sent automatically when the REPL starts",
+    )
+    parser.add_argument(
+        "-r",
+        "--resume",
+        default=None,
+        metavar="COMMIT",
+        help="Resume a previous session from the given git commit hash",
+    )
     parser.add_argument(
         "-a",
         "--api",
@@ -720,17 +733,6 @@ def main():
         default=None,
         help="API key; falls back to the relevant *_API_KEY env var",
     )
-    parser.add_argument(
-        "--prompt",
-        default=None,
-        help="Initial prompt sent automatically when the REPL starts",
-    )
-    parser.add_argument(
-        "--resume",
-        default=None,
-        metavar="COMMIT",
-        help="Resume a previous session from the given git commit hash",
-    )
     parser.add_argument("--stream", default=None, help="File to stream log into")
     args = parser.parse_args()
     logger.debug(args)
@@ -743,7 +745,7 @@ def main():
         elif args.api == "gemini":
             api_key = os.environ.get("GEMINI_API_KEY") if api_key is None else api_key
             url = "https://generativelanguage.googleapis.com" if api_key else url
-            model = "gemini-3.1-flash-lite-preview" if model is None else model
+            model = "gemini-3.1-flash-lite" if model is None else model
         tool_names = [] if tool_names is None else tool_names
     run_repl(
         api=args.api,

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/stream_log.py

@@ -45,7 +45,10 @@ class StreamLogger:
                 self.fp.write(self._line_prefix())
                 self._at_line_start = False
             self.fp.write(part)
-        self.fp.flush()
+        try:
+            self.fp.flush()
+        except:
+            pass
 
     @classmethod
     def attach_to_child(cls, child_agent, parent_ctx, tool_name="sub"):

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code/tools.py

@@ -11,7 +11,7 @@ import logging
 import random
 from abc import ABC, abstractmethod
 from typing import Any, Literal
-from pydantic import BaseModel, Field, ValidationError
+from pydantic import BaseModel, Field, ValidationError, field_validator
 
 logger = logging.getLogger(__name__)
 
@@ -438,11 +438,19 @@ class AgentParams(BaseModel):
     system: str | None = Field(
         default=None, description="System prompt override. Defaults to parent."
     )
-    tools: list[str] | str | None = Field(
-        default=None,
-        description="Restrict tools for the sub-agent, e.g. ['Read', 'Bash']. Defaults to all parent tools.",
+    tools: list[str] = Field(
+        description="Tools available to the sub-agent. Must be an explicit subset of the parent's tools."
     )
 
+    @field_validator("tools", mode="before")
+    @classmethod
+    def coerce_tools(cls, v):
+        if isinstance(v, str):
+            v = json.loads(v)
+        if not isinstance(v, list):
+            raise ValueError("tools must be a list of strings")
+        return v
+
 
 class AgentTool(Tool):
     name = "Agent"
@@ -451,19 +459,12 @@ class AgentTool(Tool):
 
     async def execute(self, params: AgentParams, signal=None) -> dict:
         parent = self.ctx["agent"]
-        tool_names: list[str] | None = None
-        if params.tools is not None:
-            raw = params.tools
-            if isinstance(raw, str):
-                raw = json.loads(raw)
-            tool_names = [t for t in raw if isinstance(t, str)]
         overrides = {}
         if params.api is not None:
             overrides["api"] = params.api
         if params.system is not None:
             overrides["system"] = params.system
-        if tool_names is not None:
-            overrides["tool_names"] = tool_names
+        overrides["tool_names"] = params.tools
         child = parent.spawn(**overrides)
         if "_stream_log_fp" in parent.ctx:
             from .stream_log import StreamLogger

{mlx_code-0.0.9 → mlx_code-0.0.10}/mlx_code.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlx-code
-Version: 0.0.9
+Version: 0.0.10
 Summary: Coding Agent for Mac
 Home-page: https://github.com/JosefAlbers/mlx-code
 Author: J Joe
@@ -31,7 +31,7 @@ Dynamic: summary
 
 A lightweight coding agent built on Apple's MLX framework.
 
-![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)
+[![demo](https://raw.githubusercontent.com/JosefAlbers/mlx-code/main/assets/mlx-code.gif)](https://youtu.be/0lkY7YQCyCo)
 
 ---
 
@@ -48,68 +48,65 @@ A lightweight coding agent built on Apple's MLX framework.
 ## Quick Start
 
 ```bash
-pip install mlx-code[all]
-mlxc
+pip install mlx-code
+mlc
 ```
 
 ---
 
 ## Command Line
 
-### `mlxc`: local server + harness
+### `mlc`: local server + harness
 
 Starts the MLX inference server and launches a harness against it.
 
 ```bash
 # Default: local MLX server + built-in REPL harness
-mlxc
+mlc
 
 # Use a different harness (routes traffic through the local server)
-mlxc --leash claude
-mlxc --leash gemini
-mlxc --leash codex
+mlc --leash claude
+mlc --leash gemini
+mlc --leash codex
 
 # Server only, no harness
-mlxc --leash none
+mlc --leash none
 
 # Specify a model
-mlxc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
+mlc --model mlx-community/Qwen3.5-4B-OptiQ-4bit
 
 # Restrict the tools available to the agent
-mlxc --tools Read Write Bash
+mlc --tools Read Write Bash
 
 # Custom system prompt
-mlxc --system "You are a helpful assistant."
+mlc --system "You are a helpful assistant."
 
 # Load skills from a directory (scans recursively for SKILL.md files)
-mlxc --skill ./my-skills
+mlc --skill ./my-skills
 
 # Resume a previous session from a git commit hash
-mlxc --resume <commit-hash>
+mlc --resume <commit-hash>
 
-# Because `mlxc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
-echo "explain lsp.py" | mlxc -d | cat - PLAN.md | mlxc
+# Because `mlc` reads from stdin when it isn't a TTY, it composes naturally with shell pipes:
+echo "Here's the solution you proposed: <excerpt>$(mlc -p "write code for a chrome extension to play youtube x5 speed")</excerpt> Now argue against it. What are the edge cases this doesn't handle? What assumptions did you make that might not hold in a production system? What would you change if you knew this code would be read by a senior engineer in a security audit?" | mlc
 ```
 
-### `mlxc-run`: harness only
+### `mlc-run`: harness only
 
 Runs the agent harness against an already-running server or a remote provider.
 
 ```bash
 # Connect to a local server at 127.0.0.1:8000 (default)
-mlxc-run
+mlc-run
 
 # Remote providers
-mlxc-run --api claude
-mlxc-run --api gemini
-mlxc-run --api deepseek --model deepseek-v4-pro
-mlxc-run --api codex
+mlc-run --api claude
+mlc-run --api gemini
+mlc-run --api deepseek --model deepseek-v4-pro
+mlc-run --api codex
 
 # Custom endpoint
-mlxc-run --url http://localhost:9000
-
-# With skills
-mlxc-run --skill ./my-skills
+echo "explain lsp.py" | mlc-run -a deepseek | cat - PLAN.md | mlc-run --url http://localhost:9000
 ```
 
 ---

{mlx_code-0.0.9 → mlx_code-0.0.10}/setup.py

@@ -2,11 +2,11 @@ from setuptools import setup, find_packages
 
 setup(
     name="mlx-code",
-    url='https://github.com/JosefAlbers/mlx-code',
+    url="https://github.com/JosefAlbers/mlx-code",
     author_email="albersj66@gmail.com",
     author="J Joe",
     license="Apache-2.0",
-    version="0.0.9",
+    version="0.0.10",
     readme="README.md",
     description="Coding Agent for Mac",
     long_description=open("README.md").read(),
@@ -14,14 +14,11 @@ setup(
     python_requires=">=3.12.8",
     install_requires=[
         "mlx-lm>=0.31.3; platform_system=='Darwin'",
-        "httpx", "pydantic", "GitPython",
+        "httpx",
+        "pydantic",
+        "GitPython",
     ],
-    extras_require={
-        "all": [
-            "python-lsp-server[all]",
-            # "tree-sitter>=0.23.0", "tree-sitter-python", # "tree-sitter-javascript", ‥
-        ],
-    },
+    extras_require={"all": ["python-lsp-server[all]"]},
     packages=find_packages(),
     entry_points={
         "console_scripts": [