agent-booster 0.1.0__tar.gz

agent_booster-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Sudhi Seshachala
+
+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.

agent_booster-0.1.0/PKG-INFO

@@ -0,0 +1,269 @@
+Metadata-Version: 2.4
+Name: agent-booster
+Version: 0.1.0
+Summary: AST+vector context router for AI coding agents — cut token costs 5-15x
+Author-email: Sudhi Seshachala <Sudhi@b2bsphere.com>
+License: MIT
+Project-URL: Homepage, https://github.com/sseshachala/conductai
+Project-URL: Repository, https://github.com/sseshachala/conductai
+Project-URL: Issues, https://github.com/sseshachala/conductai/issues
+Keywords: ai,llm,claude,cursor,codex,mcp,tokens,ast
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: tree-sitter>=0.23
+Requires-Dist: tree-sitter-python>=0.23
+Requires-Dist: tree-sitter-typescript>=0.23
+Requires-Dist: click>=8.1
+Requires-Dist: mcp>=1.0
+Requires-Dist: numpy>=1.26
+Requires-Dist: anthropic>=0.40
+Dynamic: license-file
+
+# Agent Booster
+
+Cut AI coding agent token costs 5–15x by routing only the code that matters.
+
+Instead of sending full source files to the model on every read, Agent Booster builds a symbol index of your codebase and returns only the functions and classes relevant to the current task.
+
+---
+
+## How it works
+
+```
+Without Booster                    With Booster
+─────────────────                  ────────────────────────────
+Read executor.py                   Read executor.py + task hint
+→ 1,881 lines (~6k tokens)         → 3 matching functions (~200 tokens)
+```
+
+Five layers work together:
+
+| Layer | What it does |
+|---|---|
+| **Symbol index** | tree-sitter parses every `.py` file, extracts functions/classes into SQLite |
+| **Semantic diff** | only re-indexes files that changed since last run |
+| **Smart read** | given a file + task description, returns only matching symbol line ranges |
+| **MCP server** | exposes tools over stdio so any MCP-compatible agent can call them |
+| **Platform init** | one command writes the right config for Claude Code, Cursor, or Codex |
+
+---
+
+## Installation
+
+Requires Python 3.10+.
+
+```bash
+pip install -e tools/booster
+```
+
+Verify:
+
+```bash
+booster --help
+```
+
+---
+
+## Quickstart
+
+**Step 1 — Index your codebase**
+
+Run once from your project root. Re-run after large refactors.
+
+```bash
+booster index
+# Indexed 166 files, 1107 symbols.
+```
+
+The index is stored at `.booster/symbols.db` (SQLite, gitignored).
+
+**Step 2 — Wire up your AI tool**
+
+```bash
+booster init claude    # Claude Code
+booster init cursor    # Cursor
+booster init codex     # OpenAI Codex CLI
+booster init all       # print all three
+```
+
+Each command prints the JSON config snippet to paste into the right file. For Claude Code, copy the output into `.mcp.json` at your project root (or use the pre-wired `.mcp.json` already in this repo).
+
+**Step 3 — Restart your AI tool**
+
+The `agent-booster` MCP server will be available in every session. No further setup needed.
+
+---
+
+## CLI reference
+
+```bash
+booster index
+```
+Scans all `.py` files from the current directory, extracts functions and classes, stores in `.booster/symbols.db`. Skips `node_modules`, `.venv`, `__pycache__`, `.git`, `.booster`.
+
+---
+
+```bash
+booster search "<query>"
+```
+Keyword search across all indexed symbols. Returns file path, line number, kind, name, and signature.
+
+```bash
+booster search "workflow execute"
+# apps/api/app/runtime/executor.py:1165  function _execute_output  def _execute_output(...)
+```
+
+---
+
+```bash
+booster serve
+```
+Starts the MCP server over stdio. Claude Code, Cursor, and Codex connect to this automatically when configured via `booster init`.
+
+---
+
+```bash
+booster init <platform>
+```
+Prints the MCP config snippet for the target platform. `platform` is one of: `claude`, `cursor`, `codex`, `all`.
+
+---
+
+```bash
+booster gain
+```
+Shows token savings statistics from past sessions. (Full analytics in a future release — currently a stub.)
+
+---
+
+## MCP tools
+
+Once `booster serve` is running, these tools are available to the agent:
+
+### `get_symbols(file)`
+Returns all indexed symbols for a file — name, kind, line range, and signature.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py" }
+Output: function run_workflow (lines 42-89): def run_workflow(...)
+        class WorkflowState (lines 91-140): class WorkflowState:
+        ...
+```
+
+### `search_context(task)`
+Keyword search across all symbols in the index. Returns top 10 matches with file locations.
+
+```
+Input:  { "task": "authenticate user token" }
+Output: apps/api/app/auth/middleware.py:34 function verify_token — def verify_token(...)
+        ...
+```
+
+### `smart_read(file, task)`
+Returns only the lines of a file that are relevant to the task. Falls back to full file content if no symbols match.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py", "task": "execute output block" }
+Output: # function _execute_output (lines 1165-1210)
+        def _execute_output(block, state, credentials, ...):
+            ...
+```
+
+---
+
+## Platform config details
+
+### Claude Code
+
+Add to `.mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### Cursor
+
+Add to `.cursor/mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### OpenAI Codex CLI
+
+Add to `~/.codex/config.json`:
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "agent-booster": {
+        "command": "booster",
+        "args": ["serve"]
+      }
+    }
+  }
+}
+```
+
+---
+
+## Where it fits
+
+Agent Booster is the third layer in a three-layer token reduction stack — each layer is independent and addable separately:
+
+```
+Layer 3 — Agent Booster     AST+semantic routing, smart file reads
+Layer 2 — RTK               Token compression on tool output
+Layer 1 — Prompt caching    Stable context reuse (native to Claude Code + API)
+```
+
+---
+
+## Project layout
+
+```
+tools/booster/
+├── README.md
+├── pyproject.toml
+└── booster/
+    ├── __init__.py
+    ├── cli.py          # click commands: index, search, serve, init, gain
+    ├── indexer.py      # tree-sitter parser + SQLite symbol store
+    ├── retriever.py    # smart_read: task description → relevant line slice
+    └── mcp_server.py   # MCP server exposing get_symbols, search_context, smart_read
+```
+
+---
+
+## Roadmap
+
+- [ ] TypeScript/TSX support (tree-sitter-typescript)
+- [ ] Vector embeddings for semantic (not just keyword) search
+- [ ] Smart model routing (`route_model` tool: Haiku / Sonnet / Opus)
+- [ ] `booster gain` — real token savings analytics
+- [ ] Watch mode: auto re-index on file save
+- [ ] Publish to PyPI as `agent-booster`

agent_booster-0.1.0/README.md

@@ -0,0 +1,239 @@
+# Agent Booster
+
+Cut AI coding agent token costs 5–15x by routing only the code that matters.
+
+Instead of sending full source files to the model on every read, Agent Booster builds a symbol index of your codebase and returns only the functions and classes relevant to the current task.
+
+---
+
+## How it works
+
+```
+Without Booster                    With Booster
+─────────────────                  ────────────────────────────
+Read executor.py                   Read executor.py + task hint
+→ 1,881 lines (~6k tokens)         → 3 matching functions (~200 tokens)
+```
+
+Five layers work together:
+
+| Layer | What it does |
+|---|---|
+| **Symbol index** | tree-sitter parses every `.py` file, extracts functions/classes into SQLite |
+| **Semantic diff** | only re-indexes files that changed since last run |
+| **Smart read** | given a file + task description, returns only matching symbol line ranges |
+| **MCP server** | exposes tools over stdio so any MCP-compatible agent can call them |
+| **Platform init** | one command writes the right config for Claude Code, Cursor, or Codex |
+
+---
+
+## Installation
+
+Requires Python 3.10+.
+
+```bash
+pip install -e tools/booster
+```
+
+Verify:
+
+```bash
+booster --help
+```
+
+---
+
+## Quickstart
+
+**Step 1 — Index your codebase**
+
+Run once from your project root. Re-run after large refactors.
+
+```bash
+booster index
+# Indexed 166 files, 1107 symbols.
+```
+
+The index is stored at `.booster/symbols.db` (SQLite, gitignored).
+
+**Step 2 — Wire up your AI tool**
+
+```bash
+booster init claude    # Claude Code
+booster init cursor    # Cursor
+booster init codex     # OpenAI Codex CLI
+booster init all       # print all three
+```
+
+Each command prints the JSON config snippet to paste into the right file. For Claude Code, copy the output into `.mcp.json` at your project root (or use the pre-wired `.mcp.json` already in this repo).
+
+**Step 3 — Restart your AI tool**
+
+The `agent-booster` MCP server will be available in every session. No further setup needed.
+
+---
+
+## CLI reference
+
+```bash
+booster index
+```
+Scans all `.py` files from the current directory, extracts functions and classes, stores in `.booster/symbols.db`. Skips `node_modules`, `.venv`, `__pycache__`, `.git`, `.booster`.
+
+---
+
+```bash
+booster search "<query>"
+```
+Keyword search across all indexed symbols. Returns file path, line number, kind, name, and signature.
+
+```bash
+booster search "workflow execute"
+# apps/api/app/runtime/executor.py:1165  function _execute_output  def _execute_output(...)
+```
+
+---
+
+```bash
+booster serve
+```
+Starts the MCP server over stdio. Claude Code, Cursor, and Codex connect to this automatically when configured via `booster init`.
+
+---
+
+```bash
+booster init <platform>
+```
+Prints the MCP config snippet for the target platform. `platform` is one of: `claude`, `cursor`, `codex`, `all`.
+
+---
+
+```bash
+booster gain
+```
+Shows token savings statistics from past sessions. (Full analytics in a future release — currently a stub.)
+
+---
+
+## MCP tools
+
+Once `booster serve` is running, these tools are available to the agent:
+
+### `get_symbols(file)`
+Returns all indexed symbols for a file — name, kind, line range, and signature.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py" }
+Output: function run_workflow (lines 42-89): def run_workflow(...)
+        class WorkflowState (lines 91-140): class WorkflowState:
+        ...
+```
+
+### `search_context(task)`
+Keyword search across all symbols in the index. Returns top 10 matches with file locations.
+
+```
+Input:  { "task": "authenticate user token" }
+Output: apps/api/app/auth/middleware.py:34 function verify_token — def verify_token(...)
+        ...
+```
+
+### `smart_read(file, task)`
+Returns only the lines of a file that are relevant to the task. Falls back to full file content if no symbols match.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py", "task": "execute output block" }
+Output: # function _execute_output (lines 1165-1210)
+        def _execute_output(block, state, credentials, ...):
+            ...
+```
+
+---
+
+## Platform config details
+
+### Claude Code
+
+Add to `.mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### Cursor
+
+Add to `.cursor/mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### OpenAI Codex CLI
+
+Add to `~/.codex/config.json`:
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "agent-booster": {
+        "command": "booster",
+        "args": ["serve"]
+      }
+    }
+  }
+}
+```
+
+---
+
+## Where it fits
+
+Agent Booster is the third layer in a three-layer token reduction stack — each layer is independent and addable separately:
+
+```
+Layer 3 — Agent Booster     AST+semantic routing, smart file reads
+Layer 2 — RTK               Token compression on tool output
+Layer 1 — Prompt caching    Stable context reuse (native to Claude Code + API)
+```
+
+---
+
+## Project layout
+
+```
+tools/booster/
+├── README.md
+├── pyproject.toml
+└── booster/
+    ├── __init__.py
+    ├── cli.py          # click commands: index, search, serve, init, gain
+    ├── indexer.py      # tree-sitter parser + SQLite symbol store
+    ├── retriever.py    # smart_read: task description → relevant line slice
+    └── mcp_server.py   # MCP server exposing get_symbols, search_context, smart_read
+```
+
+---
+
+## Roadmap
+
+- [ ] TypeScript/TSX support (tree-sitter-typescript)
+- [ ] Vector embeddings for semantic (not just keyword) search
+- [ ] Smart model routing (`route_model` tool: Haiku / Sonnet / Opus)
+- [ ] `booster gain` — real token savings analytics
+- [ ] Watch mode: auto re-index on file save
+- [ ] Publish to PyPI as `agent-booster`

agent_booster-0.1.0/agent_booster.egg-info/PKG-INFO

@@ -0,0 +1,269 @@
+Metadata-Version: 2.4
+Name: agent-booster
+Version: 0.1.0
+Summary: AST+vector context router for AI coding agents — cut token costs 5-15x
+Author-email: Sudhi Seshachala <Sudhi@b2bsphere.com>
+License: MIT
+Project-URL: Homepage, https://github.com/sseshachala/conductai
+Project-URL: Repository, https://github.com/sseshachala/conductai
+Project-URL: Issues, https://github.com/sseshachala/conductai/issues
+Keywords: ai,llm,claude,cursor,codex,mcp,tokens,ast
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: tree-sitter>=0.23
+Requires-Dist: tree-sitter-python>=0.23
+Requires-Dist: tree-sitter-typescript>=0.23
+Requires-Dist: click>=8.1
+Requires-Dist: mcp>=1.0
+Requires-Dist: numpy>=1.26
+Requires-Dist: anthropic>=0.40
+Dynamic: license-file
+
+# Agent Booster
+
+Cut AI coding agent token costs 5–15x by routing only the code that matters.
+
+Instead of sending full source files to the model on every read, Agent Booster builds a symbol index of your codebase and returns only the functions and classes relevant to the current task.
+
+---
+
+## How it works
+
+```
+Without Booster                    With Booster
+─────────────────                  ────────────────────────────
+Read executor.py                   Read executor.py + task hint
+→ 1,881 lines (~6k tokens)         → 3 matching functions (~200 tokens)
+```
+
+Five layers work together:
+
+| Layer | What it does |
+|---|---|
+| **Symbol index** | tree-sitter parses every `.py` file, extracts functions/classes into SQLite |
+| **Semantic diff** | only re-indexes files that changed since last run |
+| **Smart read** | given a file + task description, returns only matching symbol line ranges |
+| **MCP server** | exposes tools over stdio so any MCP-compatible agent can call them |
+| **Platform init** | one command writes the right config for Claude Code, Cursor, or Codex |
+
+---
+
+## Installation
+
+Requires Python 3.10+.
+
+```bash
+pip install -e tools/booster
+```
+
+Verify:
+
+```bash
+booster --help
+```
+
+---
+
+## Quickstart
+
+**Step 1 — Index your codebase**
+
+Run once from your project root. Re-run after large refactors.
+
+```bash
+booster index
+# Indexed 166 files, 1107 symbols.
+```
+
+The index is stored at `.booster/symbols.db` (SQLite, gitignored).
+
+**Step 2 — Wire up your AI tool**
+
+```bash
+booster init claude    # Claude Code
+booster init cursor    # Cursor
+booster init codex     # OpenAI Codex CLI
+booster init all       # print all three
+```
+
+Each command prints the JSON config snippet to paste into the right file. For Claude Code, copy the output into `.mcp.json` at your project root (or use the pre-wired `.mcp.json` already in this repo).
+
+**Step 3 — Restart your AI tool**
+
+The `agent-booster` MCP server will be available in every session. No further setup needed.
+
+---
+
+## CLI reference
+
+```bash
+booster index
+```
+Scans all `.py` files from the current directory, extracts functions and classes, stores in `.booster/symbols.db`. Skips `node_modules`, `.venv`, `__pycache__`, `.git`, `.booster`.
+
+---
+
+```bash
+booster search "<query>"
+```
+Keyword search across all indexed symbols. Returns file path, line number, kind, name, and signature.
+
+```bash
+booster search "workflow execute"
+# apps/api/app/runtime/executor.py:1165  function _execute_output  def _execute_output(...)
+```
+
+---
+
+```bash
+booster serve
+```
+Starts the MCP server over stdio. Claude Code, Cursor, and Codex connect to this automatically when configured via `booster init`.
+
+---
+
+```bash
+booster init <platform>
+```
+Prints the MCP config snippet for the target platform. `platform` is one of: `claude`, `cursor`, `codex`, `all`.
+
+---
+
+```bash
+booster gain
+```
+Shows token savings statistics from past sessions. (Full analytics in a future release — currently a stub.)
+
+---
+
+## MCP tools
+
+Once `booster serve` is running, these tools are available to the agent:
+
+### `get_symbols(file)`
+Returns all indexed symbols for a file — name, kind, line range, and signature.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py" }
+Output: function run_workflow (lines 42-89): def run_workflow(...)
+        class WorkflowState (lines 91-140): class WorkflowState:
+        ...
+```
+
+### `search_context(task)`
+Keyword search across all symbols in the index. Returns top 10 matches with file locations.
+
+```
+Input:  { "task": "authenticate user token" }
+Output: apps/api/app/auth/middleware.py:34 function verify_token — def verify_token(...)
+        ...
+```
+
+### `smart_read(file, task)`
+Returns only the lines of a file that are relevant to the task. Falls back to full file content if no symbols match.
+
+```
+Input:  { "file": "apps/api/app/runtime/executor.py", "task": "execute output block" }
+Output: # function _execute_output (lines 1165-1210)
+        def _execute_output(block, state, credentials, ...):
+            ...
+```
+
+---
+
+## Platform config details
+
+### Claude Code
+
+Add to `.mcp.json` in your project root:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### Cursor
+
+Add to `.cursor/mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "agent-booster": {
+      "command": "booster",
+      "args": ["serve"]
+    }
+  }
+}
+```
+
+### OpenAI Codex CLI
+
+Add to `~/.codex/config.json`:
+
+```json
+{
+  "mcp": {
+    "servers": {
+      "agent-booster": {
+        "command": "booster",
+        "args": ["serve"]
+      }
+    }
+  }
+}
+```
+
+---
+
+## Where it fits
+
+Agent Booster is the third layer in a three-layer token reduction stack — each layer is independent and addable separately:
+
+```
+Layer 3 — Agent Booster     AST+semantic routing, smart file reads
+Layer 2 — RTK               Token compression on tool output
+Layer 1 — Prompt caching    Stable context reuse (native to Claude Code + API)
+```
+
+---
+
+## Project layout
+
+```
+tools/booster/
+├── README.md
+├── pyproject.toml
+└── booster/
+    ├── __init__.py
+    ├── cli.py          # click commands: index, search, serve, init, gain
+    ├── indexer.py      # tree-sitter parser + SQLite symbol store
+    ├── retriever.py    # smart_read: task description → relevant line slice
+    └── mcp_server.py   # MCP server exposing get_symbols, search_context, smart_read
+```
+
+---
+
+## Roadmap
+
+- [ ] TypeScript/TSX support (tree-sitter-typescript)
+- [ ] Vector embeddings for semantic (not just keyword) search
+- [ ] Smart model routing (`route_model` tool: Haiku / Sonnet / Opus)
+- [ ] `booster gain` — real token savings analytics
+- [ ] Watch mode: auto re-index on file save
+- [ ] Publish to PyPI as `agent-booster`

agent_booster-0.1.0/agent_booster.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+LICENSE
+README.md
+pyproject.toml
+agent_booster.egg-info/PKG-INFO
+agent_booster.egg-info/SOURCES.txt
+agent_booster.egg-info/dependency_links.txt
+agent_booster.egg-info/entry_points.txt
+agent_booster.egg-info/requires.txt
+agent_booster.egg-info/top_level.txt
+booster/__init__.py
+booster/cli.py
+booster/indexer.py
+booster/mcp_server.py
+booster/retriever.py

agent_booster-0.1.0/agent_booster.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

agent_booster-0.1.0/agent_booster.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+booster = booster.cli:main

agent_booster-0.1.0/agent_booster.egg-info/requires.txt

@@ -0,0 +1,7 @@
+tree-sitter>=0.23
+tree-sitter-python>=0.23
+tree-sitter-typescript>=0.23
+click>=8.1
+mcp>=1.0
+numpy>=1.26
+anthropic>=0.40

agent_booster-0.1.0/agent_booster.egg-info/top_level.txt

@@ -0,0 +1 @@
+booster

agent_booster-0.1.0/booster/cli.py

@@ -0,0 +1,91 @@
+from __future__ import annotations
+
+import asyncio
+import json
+from pathlib import Path
+
+import click
+
+from booster.indexer import SymbolIndexer
+
+_CLAUDE_SNIPPET = {
+    "mcpServers": {
+        "agent-booster": {
+            "command": "booster",
+            "args": ["serve"],
+        }
+    }
+}
+
+_CURSOR_SNIPPET = {
+    "mcpServers": {
+        "agent-booster": {
+            "command": "booster",
+            "args": ["serve"],
+        }
+    }
+}
+
+_CODEX_SNIPPET = {
+    "mcp": {
+        "servers": {
+            "agent-booster": {
+                "command": "booster",
+                "args": ["serve"],
+            }
+        }
+    }
+}
+
+
+@click.group()
+def main() -> None:
+    pass
+
+
+@main.command("index")
+def cmd_index() -> None:
+    root = Path.cwd()
+    indexer = SymbolIndexer(root)
+    files, symbols = indexer.index_all()
+    click.echo(f"Indexed {files} files, {symbols} symbols.")
+
+
+@main.command("search")
+@click.argument("query")
+def cmd_search(query: str) -> None:
+    root = Path.cwd()
+    indexer = SymbolIndexer(root)
+    results = indexer.search(query)
+    if not results:
+        click.echo("No matches.")
+        return
+    for r in results:
+        click.echo(f"{r['file']}:{r['start_line']}  {r['kind']} {r['name']}  {r['signature']}")
+
+
+@main.command("serve")
+def cmd_serve() -> None:
+    from booster.mcp_server import serve
+    asyncio.run(serve())
+
+
+@main.command("init")
+@click.argument("platform", type=click.Choice(["claude", "cursor", "codex", "all"]))
+def cmd_init(platform: str) -> None:
+    def _print(label: str, snippet: dict) -> None:
+        click.echo(f"# {label}")
+        click.echo(json.dumps(snippet, indent=2))
+        click.echo()
+
+    if platform in ("claude", "all"):
+        _print("Add to .mcp.json (project root)", _CLAUDE_SNIPPET)
+    if platform in ("cursor", "all"):
+        _print("Add to .cursor/mcp.json", _CURSOR_SNIPPET)
+    if platform in ("codex", "all"):
+        _print("Add to ~/.codex/config.json", _CODEX_SNIPPET)
+
+
+@main.command("gain")
+def cmd_gain() -> None:
+    click.echo("No stats yet. Run some sessions first.")

agent_booster-0.1.0/booster/indexer.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+import sqlite3
+from pathlib import Path
+
+import tree_sitter_python as tspython
+from tree_sitter import Language, Node, Parser
+
+PY_LANGUAGE = Language(tspython.language())
+
+_SKIP_DIRS = {"node_modules", ".venv", "__pycache__", ".git", ".booster"}
+
+_CREATE_TABLE = """
+CREATE TABLE IF NOT EXISTS symbols (
+    id        INTEGER PRIMARY KEY AUTOINCREMENT,
+    file      TEXT NOT NULL,
+    name      TEXT NOT NULL,
+    kind      TEXT NOT NULL,
+    start_line INTEGER NOT NULL,
+    end_line   INTEGER NOT NULL,
+    signature  TEXT NOT NULL DEFAULT ''
+)
+"""
+
+_KIND_MAP = {
+    "function_definition": "function",
+    "class_definition": "class",
+}
+
+
+def _extract_name(node: Node, source: bytes) -> str:
+    for child in node.children:
+        if child.type == "identifier":
+            return source[child.start_byte:child.end_byte].decode("utf-8", errors="replace")
+    return ""
+
+
+def _extract_signature(node: Node, source: bytes) -> str:
+    first_line_end = source.find(b"\n", node.start_byte)
+    if first_line_end == -1:
+        first_line_end = node.end_byte
+    return source[node.start_byte:first_line_end].decode("utf-8", errors="replace").strip()
+
+
+class SymbolIndexer:
+    def __init__(self, root: Path) -> None:
+        self.root = root
+        db_dir = root / ".booster"
+        db_dir.mkdir(exist_ok=True)
+        self._conn = sqlite3.connect(str(db_dir / "symbols.db"))
+        self._conn.row_factory = sqlite3.Row
+        self._conn.execute(_CREATE_TABLE)
+        self._conn.commit()
+        self._parser = Parser(PY_LANGUAGE)
+
+    def index_file(self, path: Path) -> int:
+        source = path.read_bytes()
+        tree = self._parser.parse(source)
+        rel = str(path.relative_to(self.root))
+
+        self._conn.execute("DELETE FROM symbols WHERE file = ?", (rel,))
+
+        inserted = 0
+        cursor = [tree.root_node]
+        while cursor:
+            node = cursor.pop()
+            if node.type in _KIND_MAP:
+                name = _extract_name(node, source)
+                if name:
+                    sig = _extract_signature(node, source)
+                    self._conn.execute(
+                        "INSERT INTO symbols (file, name, kind, start_line, end_line, signature) VALUES (?, ?, ?, ?, ?, ?)",
+                        (rel, name, _KIND_MAP[node.type], node.start_point[0] + 1, node.end_point[0] + 1, sig),
+                    )
+                    inserted += 1
+            cursor.extend(node.children)
+
+        self._conn.commit()
+        return inserted
+
+    def index_all(self) -> tuple[int, int]:
+        files = 0
+        symbols = 0
+        for path in self.root.rglob("*.py"):
+            if any(part in _SKIP_DIRS for part in path.parts):
+                continue
+            try:
+                symbols += self.index_file(path)
+                files += 1
+            except Exception:
+                pass
+        return files, symbols
+
+    def get_symbols(self, file: str) -> list[dict]:
+        rows = self._conn.execute(
+            "SELECT * FROM symbols WHERE file = ? ORDER BY start_line", (file,)
+        ).fetchall()
+        return [dict(r) for r in rows]
+
+    def search(self, query: str, limit: int = 10) -> list[dict]:
+        terms = query.lower().split()
+        if not terms:
+            return []
+        conditions = " AND ".join(
+            "(LOWER(name) LIKE ? OR LOWER(signature) LIKE ?)" for _ in terms
+        )
+        params: list[str] = []
+        for t in terms:
+            params.extend([f"%{t}%", f"%{t}%"])
+        params.append(str(limit))
+        rows = self._conn.execute(
+            f"SELECT * FROM symbols WHERE {conditions} ORDER BY name LIMIT ?",
+            params,
+        ).fetchall()
+        return [dict(r) for r in rows]

agent_booster-0.1.0/booster/mcp_server.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from mcp.server import Server
+from mcp.server.stdio import stdio_server
+from mcp.types import TextContent, Tool
+
+from booster.indexer import SymbolIndexer
+from booster.retriever import smart_read as _smart_read
+
+_ROOT = Path.cwd()
+_indexer: SymbolIndexer | None = None
+
+
+def _get_indexer() -> SymbolIndexer:
+    global _indexer
+    if _indexer is None:
+        _indexer = SymbolIndexer(_ROOT)
+    return _indexer
+
+
+app = Server("agent-booster")
+app.version = "0.1.0"
+
+
+@app.list_tools()
+async def list_tools() -> list[Tool]:
+    return [
+        Tool(
+            name="get_symbols",
+            description="Return all indexed symbols for a given file path.",
+            inputSchema={
+                "type": "object",
+                "properties": {"file": {"type": "string", "description": "Relative file path"}},
+                "required": ["file"],
+            },
+        ),
+        Tool(
+            name="search_context",
+            description="Keyword search across all indexed symbols. Returns top 10 matches.",
+            inputSchema={
+                "type": "object",
+                "properties": {"task": {"type": "string", "description": "Task description or keywords"}},
+                "required": ["task"],
+            },
+        ),
+        Tool(
+            name="smart_read",
+            description="Return only the relevant slice of a file based on a task description.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "file": {"type": "string", "description": "Relative file path"},
+                    "task": {"type": "string", "description": "Task description"},
+                },
+                "required": ["file", "task"],
+            },
+        ),
+    ]
+
+
+@app.call_tool()
+async def call_tool(name: str, arguments: dict) -> list[TextContent]:
+    indexer = _get_indexer()
+
+    if name == "get_symbols":
+        symbols = indexer.get_symbols(arguments["file"])
+        lines = [
+            f"{s['kind']} {s['name']} (lines {s['start_line']}-{s['end_line']}): {s['signature']}"
+            for s in symbols
+        ]
+        text = "\n".join(lines) if lines else "No symbols found."
+        return [TextContent(type="text", text=text)]
+
+    if name == "search_context":
+        results = indexer.search(arguments["task"], limit=10)
+        lines = [
+            f"{s['file']}:{s['start_line']} {s['kind']} {s['name']} — {s['signature']}"
+            for s in results
+        ]
+        text = "\n".join(lines) if lines else "No matches found."
+        return [TextContent(type="text", text=text)]
+
+    if name == "smart_read":
+        file_path = _ROOT / arguments["file"]
+        text = _smart_read(file_path, arguments["task"], indexer)
+        return [TextContent(type="text", text=text)]
+
+    return [TextContent(type="text", text=f"Unknown tool: {name}")]
+
+
+async def serve() -> None:
+    async with stdio_server() as (read_stream, write_stream):
+        await app.run(read_stream, write_stream, app.create_initialization_options())

agent_booster-0.1.0/booster/retriever.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from booster.indexer import SymbolIndexer
+
+
+def smart_read(file_path: Path, task: str, indexer: SymbolIndexer) -> str:
+    rel = str(file_path.relative_to(indexer.root))
+    keywords = [w.lower() for w in task.split() if len(w) > 2]
+
+    all_symbols = indexer.get_symbols(rel)
+    matched = [
+        s for s in all_symbols
+        if any(kw in s["name"].lower() or kw in s["signature"].lower() for kw in keywords)
+    ]
+
+    source_lines = file_path.read_text(encoding="utf-8", errors="replace").splitlines()
+
+    if not matched:
+        return "\n".join(source_lines)
+
+    chunks: list[str] = []
+    for sym in matched:
+        start = sym["start_line"] - 1
+        end = sym["end_line"]
+        header = f"# {sym['kind']} {sym['name']} (lines {sym['start_line']}-{sym['end_line']})"
+        body = "\n".join(source_lines[start:end])
+        chunks.append(f"{header}\n{body}")
+
+    return "\n\n".join(chunks)

agent_booster-0.1.0/pyproject.toml

@@ -0,0 +1,44 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "agent-booster"
+version = "0.1.0"
+description = "AST+vector context router for AI coding agents — cut token costs 5-15x"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.10"
+authors = [{ name = "Sudhi Seshachala", email = "Sudhi@b2bsphere.com" }]
+keywords = ["ai", "llm", "claude", "cursor", "codex", "mcp", "tokens", "ast"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "tree-sitter>=0.23",
+    "tree-sitter-python>=0.23",
+    "tree-sitter-typescript>=0.23",
+    "click>=8.1",
+    "mcp>=1.0",
+    "numpy>=1.26",
+    "anthropic>=0.40",
+]
+
+[project.urls]
+Homepage = "https://github.com/sseshachala/conductai"
+Repository = "https://github.com/sseshachala/conductai"
+Issues = "https://github.com/sseshachala/conductai/issues"
+
+[project.scripts]
+booster = "booster.cli:main"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["booster*"]

agent_booster-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+