nexcoder 0.1.0__py3-none-any.whl

nex/tools/search.py

@@ -0,0 +1,156 @@
+"""File search tool — ripgrep wrapper with Python fallback."""
+
+from __future__ import annotations
+
+import asyncio
+import re
+from pathlib import Path
+
+from nex.tools import ToolResult
+
+
+async def search_files(
+    pattern: str,
+    path: str,
+    project_dir: Path,
+    max_results: int = 50,
+) -> ToolResult:
+    """Search files using ripgrep or fallback to Python grep.
+
+    Args:
+        pattern: Regex pattern to search for.
+        path: Directory to search (relative to project root).
+        project_dir: Project root directory.
+        max_results: Maximum number of matches to return.
+
+    Returns:
+        ToolResult with matches in file:line:content format.
+    """
+    search_dir = (project_dir / path).resolve()
+    project_root = project_dir.resolve()
+    if not str(search_dir).startswith(str(project_root)):
+        return ToolResult(success=False, output="", error="Path traversal blocked")
+
+    if not search_dir.is_dir():
+        return ToolResult(success=False, output="", error=f"Directory not found: {path}")
+
+    # Try ripgrep first
+    result = await _search_ripgrep(pattern, search_dir, max_results)
+    if result is not None:
+        return result
+
+    # Fallback to Python search
+    return _search_python(pattern, search_dir, project_root, max_results)
+
+
+async def _search_ripgrep(pattern: str, search_dir: Path, max_results: int) -> ToolResult | None:
+    """Attempt search using ripgrep (rg).
+
+    Returns None if rg is not available.
+    """
+    try:
+        process = await asyncio.create_subprocess_exec(
+            "rg",
+            "--no-heading",
+            "--line-number",
+            "--context",
+            "2",
+            "--max-count",
+            str(max_results),
+            "--color",
+            "never",
+            pattern,
+            str(search_dir),
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+
+        stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=15)
+
+        if process.returncode == 2:
+            # rg error (bad pattern, etc.)
+            return ToolResult(
+                success=False,
+                output="",
+                error=f"Search error: {stderr.decode('utf-8', errors='replace')}",
+            )
+
+        output = stdout.decode("utf-8", errors="replace")
+        if not output.strip():
+            return ToolResult(success=True, output="No matches found.")
+
+        return ToolResult(success=True, output=output)
+
+    except FileNotFoundError:
+        return None  # rg not installed, fall back
+    except TimeoutError:
+        return ToolResult(success=False, output="", error="Search timed out")
+
+
+def _search_python(
+    pattern: str, search_dir: Path, project_root: Path, max_results: int
+) -> ToolResult:
+    """Fallback search using Python regex."""
+    try:
+        compiled = re.compile(pattern)
+    except re.error as exc:
+        return ToolResult(success=False, output="", error=f"Invalid regex: {exc}")
+
+    matches: list[str] = []
+    source_extensions = {
+        ".py",
+        ".js",
+        ".ts",
+        ".tsx",
+        ".jsx",
+        ".go",
+        ".rs",
+        ".java",
+        ".rb",
+        ".php",
+        ".c",
+        ".cpp",
+        ".h",
+        ".swift",
+        ".kt",
+        ".md",
+        ".txt",
+        ".toml",
+        ".yaml",
+        ".yml",
+        ".json",
+    }
+
+    for file_path in search_dir.rglob("*"):
+        if len(matches) >= max_results:
+            break
+        if not file_path.is_file():
+            continue
+        if file_path.suffix not in source_extensions:
+            continue
+
+        try:
+            content = file_path.read_text(encoding="utf-8", errors="replace")
+            lines = content.splitlines()
+            rel_path = file_path.relative_to(project_root)
+
+            for i, line in enumerate(lines, start=1):
+                if compiled.search(line):
+                    # Include 2 lines of context
+                    start = max(0, i - 3)
+                    end = min(len(lines), i + 2)
+                    context_lines = []
+                    for j in range(start, end):
+                        prefix = ">" if j == i - 1 else " "
+                        context_lines.append(f"{prefix} {j + 1:4d} | {lines[j]}")
+                    matches.append(f"{rel_path}:{i}\n" + "\n".join(context_lines))
+
+                    if len(matches) >= max_results:
+                        break
+        except (OSError, UnicodeDecodeError):
+            continue
+
+    if not matches:
+        return ToolResult(success=True, output="No matches found.")
+
+    return ToolResult(success=True, output="\n\n".join(matches))

nex/tools/shell.py

@@ -0,0 +1,72 @@
+"""Shell command execution tool."""
+
+from __future__ import annotations
+
+import asyncio
+from pathlib import Path
+
+from nex.tools import ToolResult
+
+_MAX_OUTPUT_LEN = 10_000
+
+
+async def run_command(
+    command: str,
+    project_dir: Path,
+    timeout: int = 30,
+) -> ToolResult:
+    """Execute a shell command with timeout.
+
+    Safety checks are NOT performed here — they happen in the safety layer
+    before this function is called.
+
+    Args:
+        command: The shell command to execute.
+        project_dir: Working directory for the command.
+        timeout: Maximum seconds to wait (default 30).
+
+    Returns:
+        ToolResult with stdout+stderr or error message.
+    """
+    try:
+        process = await asyncio.create_subprocess_shell(
+            command,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+            cwd=str(project_dir),
+        )
+
+        try:
+            stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=timeout)
+        except TimeoutError:
+            process.kill()
+            await process.communicate()
+            return ToolResult(
+                success=False,
+                output="",
+                error=f"Command timed out after {timeout}s: {command}",
+            )
+
+        stdout_str = stdout.decode("utf-8", errors="replace")
+        stderr_str = stderr.decode("utf-8", errors="replace")
+
+        combined = ""
+        if stdout_str:
+            combined += stdout_str
+        if stderr_str:
+            if combined:
+                combined += "\n--- stderr ---\n"
+            combined += stderr_str
+
+        if len(combined) > _MAX_OUTPUT_LEN:
+            total = len(combined)
+            combined = combined[:_MAX_OUTPUT_LEN] + f"\n... (truncated, total {total} chars)"
+
+        return ToolResult(
+            success=process.returncode == 0,
+            output=combined,
+            error=f"Exit code {process.returncode}" if process.returncode != 0 else None,
+        )
+
+    except OSError as exc:
+        return ToolResult(success=False, output="", error=f"Failed to run command: {exc}")

nexcoder-0.1.0.dist-info/METADATA

@@ -0,0 +1,170 @@
+Metadata-Version: 2.4
+Name: nexcoder
+Version: 0.1.0
+Summary: The coding agent that remembers — AI coding assistant with persistent memory and error learning.
+Project-URL: Homepage, https://github.com/nex-ai/nex-ai
+Project-URL: Repository, https://github.com/nex-ai/nex-ai
+Author: Nex AI Contributors
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,anthropic,claude,cli,coding-agent
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Code Generators
+Requires-Python: >=3.12
+Requires-Dist: anthropic>=0.40.0
+Requires-Dist: gitpython>=3.1.0
+Requires-Dist: httpx>=0.28.0
+Requires-Dist: rich>=13.9.0
+Requires-Dist: tree-sitter-javascript>=0.23.0
+Requires-Dist: tree-sitter-python>=0.23.0
+Requires-Dist: tree-sitter-typescript>=0.23.0
+Requires-Dist: tree-sitter>=0.24.0
+Requires-Dist: typer[all]>=0.15.0
+Provides-Extra: dev
+Requires-Dist: build>=1.0.0; extra == 'dev'
+Requires-Dist: mypy>=1.13.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.24.0; extra == 'dev'
+Requires-Dist: pytest>=8.3.0; extra == 'dev'
+Requires-Dist: ruff>=0.8.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Nex AI — The Coding Agent That Remembers
+
+A CLI-based AI coding agent that wraps the Anthropic Claude API with persistent project memory, error pattern learning, and codebase indexing.
+
+## Features
+
+- **Persistent Memory** — Remembers your project context, conventions, and architecture across sessions
+- **Error Learning** — Logs past mistakes and queries them before generating code to avoid repeating errors
+- **Codebase Indexing** — Uses tree-sitter to parse your codebase and select relevant context for each task
+- **Interactive Chat** — Multi-turn conversation mode with full tool access and persistent history
+- **Smart Context Selection** — Two-phase token budgeting with file-level TF-IDF ranking and signature-only fallback
+- **Safety Layer** — Detects destructive operations (rm -rf, DROP TABLE, force push) and requires confirmation
+- **Git Integration** — Creates isolated branches per task, shows diffs, and offers to commit
+
+## Installation
+
+```bash
+pip install nexcoder
+```
+
+## Quick Start
+
+```bash
+# Initialize Nex in your project
+nex init
+
+# Set up your API key
+nex auth
+
+# Build the codebase index
+nex index
+
+# Run a one-shot task
+nex "add a health check endpoint"
+
+# Or start an interactive session
+nex chat
+
+# Check project status
+nex status
+```
+
+## Commands
+
+| Command | Description |
+|---------|-------------|
+| `nex "task"` | Run a coding task |
+| `nex init` | Initialize .nex/ directory |
+| `nex index` | Build codebase index (.nex/index.json) |
+| `nex chat` | Start interactive chat session |
+| `nex status` | Show project stats |
+| `nex auth` | Configure API key |
+| `nex memory show` | View project memory |
+| `nex memory edit` | Edit project memory |
+| `nex rollback` | Undo last agent change |
+| `nex --dry-run "task"` | Preview without executing |
+
+## How It Works
+
+1. **Context Assembly** — Loads project memory, error patterns, and relevant code into the prompt. Files are ranked by TF-IDF relevance, with top files included in full and remaining files as signature summaries.
+2. **Agent Loop** — Iterates: call Claude API → execute tools → feed results back, up to 25 iterations.
+3. **Interactive Chat** — `nex chat` maintains a persistent conversation with the agent, accumulating context across turns. Useful for exploratory work, debugging, or multi-step tasks.
+4. **Codebase Index** — `nex index` parses your source files with tree-sitter, extracting function/class signatures for fast relevance search.
+5. **Git Commit** — Shows diff and offers to commit on an isolated branch.
+
+## Tools
+
+The agent has access to 5 tools:
+
+| Tool | Description |
+|------|-------------|
+| `read_file` | Read file contents with line numbers |
+| `write_file` | Write content, creating directories if needed |
+| `run_command` | Execute shell commands (safety-checked) |
+| `search_files` | Regex search across the codebase |
+| `list_directory` | Recursive directory listing |
+
+## Development
+
+```bash
+# Clone and install in editable mode
+git clone https://github.com/nex-ai/nex-ai.git
+cd nex-ai
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint and format
+ruff check src/ tests/
+ruff format src/ tests/
+
+# Type check
+mypy src/nex/
+```
+
+## Architecture
+
+```
+src/nex/
+├── cli.py          # Typer CLI entry point
+├── agent.py        # Core agent loop + ChatSession
+├── api_client.py   # Anthropic API wrapper
+├── planner.py      # Task decomposition (Haiku)
+├── reviewer.py     # Independent code review
+├── context.py      # Context assembly + token budgeting
+├── safety.py       # Destructive operation detection
+├── config.py       # Configuration management
+├── memory/         # Persistent memory system
+│   ├── project.py  # .nex/memory.md
+│   ├── errors.py   # .nex/errors.db (SQLite)
+│   └── decisions.py # .nex/decisions.md
+├── indexer/        # Codebase indexing
+│   ├── scanner.py  # File discovery
+│   ├── parser.py   # tree-sitter AST parsing
+│   └── index.py    # Index builder + TF-IDF search
+└── tools/          # Agent tools (5 total)
+    ├── file_ops.py # read_file, write_file
+    ├── shell.py    # run_command
+    ├── search.py   # search_files
+    └── git_ops.py  # Git operations
+```
+
+## Environment Variables
+
+| Variable | Description | Default |
+|----------|-------------|---------|
+| `ANTHROPIC_API_KEY` | Anthropic API key (required) | — |
+| `NEX_MODEL` | Override default model | `claude-sonnet-4-20250514` |
+| `NEX_MAX_ITERATIONS` | Max tool calls per task | `25` |
+| `NEX_DRY_RUN` | Default dry-run mode | `false` |
+| `NEX_LOG_LEVEL` | Logging verbosity | `INFO` |
+
+## License
+
+MIT

nexcoder-0.1.0.dist-info/RECORD

@@ -0,0 +1,30 @@
+nex/__init__.py,sha256=L_sbkyem5kYqGkPuCZcd-O2wBI3_SUfhf6YOUZZKKys,130
+nex/agent.py,sha256=1GOUOYYNW5pHgG2C11yxmTRUaDkgFKsIg1YD8BCHK4Y,21757
+nex/api_client.py,sha256=L3uZLq2CcBLzphccl_6Hls8bWbjnqMs30OhQP-LWxC8,6193
+nex/cli.py,sha256=hrOSsE89e15kveFk7CakVqLDhey16t2rwi6rIqGq7UI,15562
+nex/config.py,sha256=U-HH513SnP-Qjwm9r7J-X-cc5mXIvwUcxw8mhpPii6I,5636
+nex/context.py,sha256=3n8tNWlLTkUIR8GuQKmNCNEBnMCdrBRq8eD6rVAoOs4,8273
+nex/exceptions.py,sha256=i8qczPp-mkH94RtNHwfyzjWUXFYuF_EmOwCydfBZXG0,1043
+nex/planner.py,sha256=xFmHvP00BQxfwwPtAoMwE8BiJXzluHFNJuL36736roM,3594
+nex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nex/reviewer.py,sha256=kz9xuuNsMMD_05FNLQSyiM0cODxWLTSIsKJKo9Sl37Q,3348
+nex/safety.py,sha256=keaGodCsHOfVkqHsKKMGYm1XNPsWtfQ8L-h5CZwWVjU,7588
+nex/test_runner.py,sha256=ZLcCGlL0LvboNLY8nClZPkgkw7oK8LiZ4nUaZaJV3WI,6140
+nex/indexer/__init__.py,sha256=AG2KmuT7MWvguV_vkfXM_wuRGCe5B0w-5x4SybQaogs,388
+nex/indexer/index.py,sha256=R92Cq7DGfgXkms_vm-34pJ0j34alwFOOWZHGwef3iok,10570
+nex/indexer/parser.py,sha256=Wo3mEnhztfJKt-tb8-FiI4T8AUqG5JvECP2IwmYMjYs,13877
+nex/indexer/scanner.py,sha256=lIYT_MG3VD0GhxqRCFEYSSG2-hh_MXcXGPsHCSU42Oc,5499
+nex/memory/__init__.py,sha256=p4dQWSY614MC0DhVgtqediIkCO9Lf5adv5F0YF4IzNI,396
+nex/memory/decisions.py,sha256=XvnEb8qwG6lnF-1IFDJ7D6oLc5tN8FvShGx6VO6KE2Y,3905
+nex/memory/errors.py,sha256=DCpE71giIPFD3_90dJILtwNORUj3cMrPFBySKQOiOC8,8174
+nex/memory/project.py,sha256=2OZ2el2T3g-TXcfwqCzWKfasdEWCkLT7aQjx5H5VFhQ,4746
+nex/tools/__init__.py,sha256=WK4gsBGgaWbGz2j8PuBvQtQNTpeWGp5FG_wZLPoqHKU,3443
+nex/tools/file_ops.py,sha256=_QMbWsljexkKkzUqz9llOU1uS2rC4GeMg6nq7C4GD6s,2970
+nex/tools/git_ops.py,sha256=K8yH37Y6vIJUjK-DOZ6afrno_KPOK8xK1VJxfcAlW9w,5406
+nex/tools/search.py,sha256=jSy6uiPzU2VNEv6YJ0e8ZvoYUy8m0YQYqpip9LzVlO0,4648
+nex/tools/shell.py,sha256=3QrzK5C5mq0HX9HS8EssH94d7EEAheJrNEtEc-d6dq0,2113
+nexcoder-0.1.0.dist-info/METADATA,sha256=5D8MaS5V9uwqMNLt92-618yoNgTKdIN6x88mRldPjbg,6021
+nexcoder-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+nexcoder-0.1.0.dist-info/entry_points.txt,sha256=xR6UmNT0yEmlXt7E-Xcv8zZPywI7TmZw-c8WIOl6iVQ,36
+nexcoder-0.1.0.dist-info/licenses/LICENSE,sha256=aOwfjAeGkp6oZxVcTmr_0ae5RoJ2FiJ9HVWAsg-o7tU,1076
+nexcoder-0.1.0.dist-info/RECORD,,

nexcoder-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

nexcoder-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+nex = nex.cli:app

nexcoder-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Nex AI Contributors
+
+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, 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,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.