mlx-code 0.0.30__tar.gz → 0.0.31__tar.gz

{mlx_code-0.0.30 → mlx_code-0.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlx-code
-Version: 0.0.30
+Version: 0.0.31
 Summary: Coding Agent for Mac
 Home-page: https://josefalbers.github.io/mlx-code/
 Author: J Joe
@@ -15,13 +15,11 @@ License-File: LICENSE
 Requires-Dist: mlx-lm>=0.31.3; platform_system == "Darwin"
 Requires-Dist: httpx
 Requires-Dist: pydantic
-Requires-Dist: textual>=8.2.7
-Requires-Dist: rich>=15.0.0
-Requires-Dist: starlette
-Requires-Dist: uvicorn
 Provides-Extra: all
-Requires-Dist: python-lsp-server[all]; extra == "all"
-Requires-Dist: GitPython; extra == "all"
+Requires-Dist: starlette; extra == "all"
+Requires-Dist: uvicorn; extra == "all"
+Requires-Dist: textual>=8.2.7; extra == "all"
+Requires-Dist: rich>=15.0.0; extra == "all"
 Requires-Dist: pygments; extra == "all"
 Dynamic: author
 Dynamic: author-email
@@ -76,7 +74,7 @@ Agents:           ├───────────────────
              │  │ Gemini         │  │ Edit    Bash   │  │
              │  │ Claude         │  │ Grep    Find   │  │
              │  │ Codex          │  │ Ls      Skill  │  │
-             │  │ DeepSeek       │  │ Agent ─────────┼──┼───► Recursively spawns sub-Agents
+             │  │ DeepSeek       │  │ Agent ────────────────► Recursively spawns sub-Agents
              │  └────────────────┘  └────────────────┘  │     
              │  Git worktree                            │
              │  (isolation + session state)             │
@@ -108,10 +106,10 @@ result = await agent.run('refactor utils.py to use dataclasses')
 
 ```bash
 # ephemeral run (no installation)
-uvx --from mlx-code mlc
+uvx --from mlx-code[all] mlc
 
 # or install into the current environment
-pip install mlx-code
+pip install mlx-code[all]
 
 # launch
 mlc                              # with a local MLX model

{mlx_code-0.0.30 → mlx_code-0.0.31}/README.md

@@ -38,7 +38,7 @@ Agents:           ├───────────────────
              │  │ Gemini         │  │ Edit    Bash   │  │
              │  │ Claude         │  │ Grep    Find   │  │
              │  │ Codex          │  │ Ls      Skill  │  │
-             │  │ DeepSeek       │  │ Agent ─────────┼──┼───► Recursively spawns sub-Agents
+             │  │ DeepSeek       │  │ Agent ────────────────► Recursively spawns sub-Agents
              │  └────────────────┘  └────────────────┘  │     
              │  Git worktree                            │
              │  (isolation + session state)             │
@@ -70,10 +70,10 @@ result = await agent.run('refactor utils.py to use dataclasses')
 
 ```bash
 # ephemeral run (no installation)
-uvx --from mlx-code mlc
+uvx --from mlx-code[all] mlc
 
 # or install into the current environment
-pip install mlx-code
+pip install mlx-code[all]
 
 # launch
 mlc                              # with a local MLX model

{mlx_code-0.0.30 → mlx_code-0.0.31}/mlx_code/bats.py

@@ -294,6 +294,26 @@ def make_batch_app(model_name: str, cache_dir: str='.cache'):
             n_cached = sum((1 for _ in pc.cache_dir.glob('*.safetensors')))
         return JSONResponse({'status': 'ok', 'model': model_name, 'active_sequences': len(state['active']), 'prefix_cache_files': n_cached})
     return Starlette(routes=[Route('/v1/models', list_models, methods=['GET']), Route('/v1/messages/count_tokens', count_tokens, methods=['POST']), Route('/v1/chat/completions', generate_endpoint, methods=['POST']), Route('/v1/messages', generate_endpoint, methods=['POST']), Route('/v1/responses', generate_endpoint, methods=['POST']), Route('/v1beta/models/{rest:path}', generate_endpoint, methods=['POST']), Route('/generate', simple_generate, methods=['POST']), Route('/health', health, methods=['GET'])], lifespan=lifespan)
+
+class BatchServer:
+    import uvicorn
+
+    def __init__(self, app, host: str, port: int):
+        config = uvicorn.Config(app, host=host, port=port, loop='asyncio', log_level='warning')
+        self._server = uvicorn.Server(config)
+        self.host = host
+        self.port = port
+
+    def serve_forever(self):
+        self._server.run()
+
+    @property
+    def started(self) -> bool:
+        return self._server.started
+
+def make_batch_server(host: str, port: int, model, cache_dir: str='.cache') -> BatchServer:
+    app = make_batch_app(model, cache_dir=cache_dir)
+    return BatchServer(app, host, port)
 if __name__ == '__main__':
     import uvicorn
     uvicorn.run(make_batch_app('mlx-community/Qwen3.5-4B-OptiQ-4bit'), host='0.0.0.0', port=8000)

{mlx_code-0.0.30 → mlx_code-0.0.31}/mlx_code/gits.py

@@ -9,7 +9,6 @@ import uuid
 from dataclasses import dataclass
 from pathlib import Path
 logger = logging.getLogger(__name__)
-_ADD_EXCLUDES = ['_*', '*.bin', '*.gguf', '*.safetensors', '*.pt', '*.pth', '.cache/', '.log.json', '*.egg-info/', '.eggs/', 'build/', 'dist/', '__pycache__/', '*.pyc', '*.pyo', '*.pyd', '.pytest_cache/', '.tox/', '.nox/', '.coverage', 'htmlcov/', '.venv/', 'venv/', 'env/', '.DS_Store', 'Thumbs.db']
 
 class GitError(RuntimeError):
     pass
@@ -57,11 +56,22 @@ def _count_user_turns(commit_body: str) -> int:
     if messages:
         return sum((1 for m in messages if m.get('role') == 'user'))
     return 0
+_ADD_EXCLUDES = ['_*', '_*/', '*.bin', '*.gguf', '*.safetensors', '*.pt', '*.pth', '.cache/', '.log.json', '*.egg-info/', '.eggs/', 'build/', 'dist/', '__pycache__/', '*.pyc', '*.pyo', '*.pyd', '.pytest_cache/', '.tox/', '.nox/', '.coverage', 'htmlcov/', '.venv/', 'venv/', 'env/', '.DS_Store', 'Thumbs.db']
+
+def _exclude_pathspecs(patterns: list[str]) -> list[str]:
+    specs = []
+    for p in patterns:
+        if p.endswith('/'):
+            name = p[:-1]
+            specs.append(f':(exclude,glob)**/{name}/**')
+        else:
+            specs.append(f':(exclude,glob)**/{p}')
+    return specs
 
 def git_add_filtered(cwd: str) -> None:
-    excludes = [f':(exclude){p}' for p in _ADD_EXCLUDES]
+    excludes = _exclude_pathspecs(_ADD_EXCLUDES)
     try:
-        _git(cwd, 'add', '-A', '--', '.', *excludes)
+        _git(cwd, '-c', 'advice.addIgnoredFile=false', 'add', '-A', '--', '.', *excludes)
     except GitError as e:
         logger.warning('git add warning (ignored): %s', e)
 
@@ -83,7 +93,7 @@ def create_worktree(repo_dir: str, *, worktree_dir: str | None=None, ref: str='H
             root = repo_dir
             gi = os.path.join(root, '.gitignore')
             if not os.path.exists(gi):
-                Path(gi).write_text('\n'.join(['_log.json']))
+                Path(gi).write_text('\n'.join(['.log.json']))
         if not _git(root, 'config', 'user.email', check=False):
             _git(root, 'config', 'user.email', 'agent@local')
         if not _git(root, 'config', 'user.name', check=False):

{mlx_code-0.0.30 → mlx_code-0.0.31}/mlx_code/main.py

@@ -890,11 +890,12 @@ def _serve_cache(host, port, model, cache, system, tools, skips, *, fixed_port=F
                 raise
 
 def _serve_batch(host, port, model, cache_dir='.cache', *, fixed_port=False):
-    import uvicorn
-    from .bats import make_batch_app
+    try:
+        from .bats import make_batch_server
+    except ImportError:
+        print('[warning] uvicorn not installed — batch engine requires: pip install mlx-code[all]')
+        return _serve_cache(host, port, model, cache_dir, None, None, None, fixed_port=fixed_port)
     import socket
-    import time
-    app = make_batch_app(model, cache_dir=cache_dir)
     while True:
         try:
             with socket.socket() as s:
@@ -909,24 +910,10 @@ def _serve_batch(host, port, model, cache_dir='.cache', *, fixed_port=False):
                 raise
         else:
             break
-    config = uvicorn.Config(app, host=host, port=port, loop='asyncio', log_level='warning')
-    uv_server = uvicorn.Server(config)
-    t = threading.Thread(target=uv_server.run, daemon=True)
-    t.start()
-    start_time = time.time()
-    notified = False
-    while True:
-        try:
-            with socket.create_connection((host, port), timeout=0.1):
-                break
-        except OSError:
-            if not notified and time.time() - start_time > 3.0:
-                logger.info('Waiting for batch server to start (model may be downloading)...')
-                notified = True
-            time.sleep(0.2)
+    server = make_batch_server(host, port, model, cache_dir=cache_dir)
     url = f'http://{host}:{port}'
     logger.debug(f'Batch server bound to {url}')
-    return (uv_server, url)
+    return (server, url)
 
 def main():
     parser = argparse.ArgumentParser(description='mlx-code MAIN')
@@ -967,28 +954,20 @@ def main():
         else:
             server, url = _serve_cache(host=args.host, port=port, model=args.model, cache=cache, system=None if args.leash in ('none', 'noapi') else args.system, tools=args.tools, skips=args.skips, fixed_port=fixed_port, gwt=gwt)
         if args.leash == 'none':
-            if args.engine == 'batch':
-                try:
-                    threading.Event().wait()
-                except KeyboardInterrupt:
-                    print('\nShutting down server...')
-            else:
-                try:
-                    server.serve_forever()
-                except KeyboardInterrupt:
-                    print('\nShutting down server...')
-                    server.server_close()
+            try:
+                server.serve_forever()
+            except KeyboardInterrupt:
+                print('\nShutting down server...')
+                server.server_close()
         else:
-            if args.engine == 'cache':
-                threading.Thread(target=server.serve_forever, daemon=True).start()
+            threading.Thread(target=server.serve_forever, daemon=True).start()
+            while not getattr(server, 'started', True):
+                time.sleep(0.05)
             if args.leash == 'noapi':
-                if args.web:
-                    from .web import run_web
-                    web_port = args.web_port if args.web_port is not None else port + 80
-                    run_web(base_url=url, api=args.leash, repo=cwd, env=env, system=args.system, tool_names=args.tools, sdir=args.skill, init_prompt=args.prompt, resume=args.resume, stream=args.stream, host=args.host, port=web_port)
-                else:
-                    from .repl import run_repl
-                    run_repl(base_url=url, api=args.leash, repo=cwd, env=env, system=args.system, tool_names=args.tools, sdir=args.skill, init_prompt=args.prompt, resume=args.resume, stream=args.stream, bare=args.bare)
+                ui_mode = 'web' if args.web else 'bare' if args.bare else 'tui'
+                web_port = args.web_port if args.web_port is not None else port + 80
+                from .repl import run_repl
+                run_repl(base_url=url, api=args.leash, repo=cwd, env=env, system=args.system, tool_names=args.tools, sdir=args.skill, init_prompt=args.prompt, resume=args.resume, stream=args.stream, ui_mode=ui_mode, web_host=args.host, web_port=web_port)
             else:
                 env['GOOGLE_GEMINI_BASE_URL'] = url
                 env['GEMINI_API_KEY'] = 'mc'

{mlx_code-0.0.30 → mlx_code-0.0.31}/mlx_code/repl.py

@@ -742,9 +742,9 @@ _AGENT_ENV_ALLOWLIST: re.Pattern = re.compile('\n    ^(\n    PATH\n    | MANPATH
 def _make_agent_env(base: dict[str, str]) -> dict[str, str]:
     return {k: v for k, v in base.items() if _AGENT_ENV_ALLOWLIST.match(k)}
 
-async def repl(engine: CommandEngine, init_prompt=None, bare=False):
+async def repl(engine: CommandEngine, init_prompt=None, ui_mode='tui'):
     is_tty = sys.stdin.isatty() and sys.stdout.isatty()
-    if bare and is_tty:
+    if ui_mode == 'bare' and is_tty:
         from .bare import BareRepl
         r = BareRepl(engine, init_prompt=init_prompt)
         await r.run()
@@ -754,12 +754,21 @@ async def repl(engine: CommandEngine, init_prompt=None, bare=False):
         if user_input:
             await _stream_to_stdout(engine.active_tab.agent, user_input)
         return None
-    from .tui import ReplApp
-    app = ReplApp(engine, init_prompt=init_prompt)
-    await app.run_async()
-    return app
+    if ui_mode == 'tui':
+        try:
+            from .tui import ReplApp
+        except ImportError:
+            print('[warning] textual/rich not installed — falling back to bare.\n         Install TUI deps with:  pip install mlx-code[all]')
+            from .bare import BareRepl
+            r = BareRepl(engine, init_prompt=init_prompt)
+            await r.run()
+            return None
+        app = ReplApp(engine, init_prompt=init_prompt)
+        await app.run_async()
+        return app
+    return None
 
-def run_repl(*, base_url=None, model=None, api: Literal['claude', 'codex', 'gemini', 'deepseek', 'noapi']='noapi', system='', sdir=None, skills=None, env=None, tool_names=None, extra_tool_classes=None, api_key=None, gwt=None, ctx=None, init_prompt=None, resume_messages=None, repo=None, resume=None, stream=None, bare=False):
+def run_repl(*, base_url=None, model=None, api='noapi', system='', sdir=None, skills=None, env=None, tool_names=None, extra_tool_classes=None, api_key=None, gwt=None, ctx=None, init_prompt=None, resume_messages=None, repo=None, resume=None, stream=None, ui_mode='tui', web_host='127.0.0.1', web_port=8080):
     repo = os.path.abspath(repo or os.getcwd())
     with tempfile.TemporaryDirectory(dir=tempfile.gettempdir()) as _home:
         if gwt is None:
@@ -785,9 +794,6 @@ def run_repl(*, base_url=None, model=None, api: Literal['claude', 'codex', 'gemi
         system = '\n\n'.join(filter(None, [system, skill_prompt]))
         merged_ctx = {'cwd': cwd, 'user_cwd': user_cwd, 'skills': skills, 'gwt': gwt, 'env': agent_env, **(ctx or {})}
         agent = Agent(system=system, api=api, model=model, tool_names=tool_names, extra_tool_classes=extra_tool_classes, api_key=api_key, base_url=base_url, ctx=merged_ctx)
-        engine = CommandEngine()
-        main_tab = TabModel(agent, title='main', is_main=True)
-        engine.tabs = [main_tab]
         log_fp = None
         if stream is not None:
             from .stream_log import StreamLogger
@@ -800,26 +806,35 @@ def run_repl(*, base_url=None, model=None, api: Literal['claude', 'codex', 'gemi
             agent.messages = list(resume_messages)
             print(f'[resumed {len(resume_messages)} messages from checkpoint]')
         try:
-            asyncio.run(repl(engine, init_prompt=init_prompt, bare=bare))
+            if ui_mode == 'web':
+                try:
+                    from .web import run_web
+                except ImportError:
+                    print('[warning] starlette/uvicorn not installed — falling back to bare.\n         Install web deps with:  pip install mlx-code[all]')
+                    engine = CommandEngine()
+                    main_tab = TabModel(agent, title='main', is_main=True)
+                    engine.tabs = [main_tab]
+                    asyncio.run(repl(engine, init_prompt=init_prompt, ui_mode='bare'))
+                else:
+                    run_web(agent=agent, init_prompt=init_prompt, web_host=web_host, web_port=web_port)
+            else:
+                engine = CommandEngine()
+                main_tab = TabModel(agent, title='main', is_main=True)
+                engine.tabs = [main_tab]
+                asyncio.run(repl(engine, init_prompt=init_prompt, ui_mode=ui_mode))
         finally:
             if log_fp:
                 log_fp.close()
-            cleaned: set[str] = set()
-            for tab in engine.tabs:
-                gwt_ref = tab.agent.ctx.get('gwt')
-                if gwt_ref and getattr(gwt_ref, 'worktree', None) and (gwt_ref.worktree not in cleaned):
-                    cleaned.add(gwt_ref.worktree)
-                    try:
-                        cleanup_worktree(gwt_ref)
-                    except Exception:
-                        pass
-            if engine.exit_summary:
-                print('\n--- Session Exit Summary ---')
-                for item in engine.exit_summary:
-                    title = item['title']
-                    branch = item['branch'] or '(no branch)'
-                    marker = ' * <-- exit origin' if item['is_exit_tab'] else ''
-                    print(f'  {title} ({branch}){marker}')
+            if 'engine' in locals() and hasattr(engine, 'tabs'):
+                cleaned: set[str] = set()
+                for tab in engine.tabs:
+                    gwt_ref = tab.agent.ctx.get('gwt')
+                    if gwt_ref and getattr(gwt_ref, 'worktree', None) and (gwt_ref.worktree not in cleaned):
+                        cleaned.add(gwt_ref.worktree)
+                        try:
+                            cleanup_worktree(gwt_ref)
+                        except Exception:
+                            pass
 
 def main():
     import argparse