astra-mcp
Enables AI agents to index local Python repositories and retrieve code intelligence through semantic search, caller lookups, hybrid context expansion, and visualization.
README
<p align="center"> <img src="assets/astra-social-banner.png" alt="Astra: local code intelligence for AI agents" width="100%"> </p>
Astra is dual-licensed under the MIT License or Apache License 2.0, at your option.
Astra is a local code intelligence service for AI agents. It parses repository files into a structural NetworkX graph and a searchable local code-chunk index, with full Python AST analysis and AST-compatible structural extraction across common source, markup, and configuration formats. The same engine is available through the astra CLI and the official MCP Python SDK.
Architecture
The constellation mark represents the relationships Astra discovers across a codebase. Its pipeline has four stages: discover files without executing them, build the structural graph, index code chunks, and retrieve semantic matches with graph expansion.
| Component | Role | Output |
|---|---|---|
| AST and text parser | Uses Python AST for .py files and language-aware structural extraction for common source, markup, and data formats; indexes other readable files as file-level chunks. |
Code chunks and references |
| Structural graph | Builds directed NetworkX relationships between modules, declarations, and callers. | .astra_graph.json |
| Vector index | Stores searchable chunks locally, with optional Sentence Transformers and Chroma acceleration. | .astra_vectors/ |
| Dual interfaces | Makes the same engine available to terminal users and MCP hosts. | astra and astra-mcp |
The MCP surface exposes eleven tools: astra_index_repo, astra_semantic_search, astra_get_callers, astra_path, astra_dipper, astra_tether, astra_get_fragility_hotspots, astra_impact, astra_refactor_plan, astra_hybrid_context, and astra_visualize.
<p align="center"> <img src="assets/astra-pipeline.png" alt="Astra indexing pipeline: parse, graph, index, and retrieve" width="900"> </p>
<p align="center"><em>Indexing pipeline</em></p>
<p align="center"> <img src="assets/astra-components.png" alt="Astra component map connecting the CLI and MCP host to the shared engine" width="900"> </p>
<p align="center"><em>Component map</em></p>
Install
Python 3.10+ is required. Choose one of these installation paths.
Install directly from GitHub
This is the simplest option for using Astra. It installs the astra and astra-mcp commands without placing a checkout in your current folder:
python -m pip install "git+https://github.com/nimelkot/astra.git"
To upgrade an existing installation to the latest GitHub version:
python -m pip install --upgrade --force-reinstall "git+https://github.com/nimelkot/astra.git"
For an isolated command-line installation, use pipx:
pipx install "git+https://github.com/nimelkot/astra.git"
Clone for development
Use this option when you want to edit Astra or run its tests. Run the editable install from the cloned repository root, the directory containing pyproject.toml:
git clone https://github.com/nimelkot/astra.git
cd astra
Get-ChildItem pyproject.toml
python -m venv .venv
.venv\Scripts\Activate.ps1
python -m pip install -e ".[dev]"
If Get-ChildItem pyproject.toml cannot find the file, you are not in the Astra repository root yet. The folder you want to analyze can be anywhere; it does not need to be inside the Astra checkout.
Quick start
Index a project first. Replace the example path with the folder you want Astra to analyze:
astra index C:\Users\YourName\Downloads\my-project
The command recursively scans nested folders for files without importing or executing them. Valid Python files receive AST-level declarations and call relationships. Common source file types (.js, .jsx, .ts, .tsx, .go, .java, .cs, .c, .cc, .cpp, .h, .hpp, .rs, .php, .swift, .kt, .kts, .rb, .dart, .scala, .pl, .r, .m, .lua, .vb, .vbs, .bas, .ps1, .sh, .bash, .zsh, .bat, .cmd, .proto, .graphql, .gql) receive structural declaration and call extraction into the same graph format. SQL files (.sql) receive table/view/function/procedure/trigger extraction and dependency links from statements such as FROM and JOIN. Markdown (.md, .markdown, .mdx) gets section-level extraction, HTML/XML gets element-level extraction, JSON/JSONC and YAML/TOML/INI-style configs get key-level extraction, and BSON files are handled safely with optional key extraction when a BSON decoder is available. Other readable files remain searchable as file-level chunks. Binary files, unsupported encodings, files larger than 2 MB, and generated or dependency directories such as .git, .venv, node_modules, and .astra_vectors are skipped.
For a complete support matrix, see docs/supported-file-types.md.
my-project/
├── .astra_graph.json # structural modules, declarations, and call relationships
└── .astra_vectors/ # local searchable code chunks
Run astra index again after the source code changes. Indexing replaces the previous graph and chunk index for that target directory.
Astra uses a lightweight hash cache (.astra_index_cache.json) so repeated indexing only reparses files whose contents changed. Unchanged files reuse cached structural chunks and references.
CLI
Semantic search
Search by a concept, behavior, symbol name, or docstring text:
astra search "authentication token validation" --path C:\Users\YourName\Downloads\my-project --limit 8
The result table shows the match score, declaration kind, file path, symbol, and source lines. Search results are empty when no indexed chunk shares terms with the query.
Structural callers
Find functions that call a target function:
astra query callers calculate_total --path C:\Users\YourName\Downloads\my-project
The callers query uses the saved AST graph and reports matching caller nodes as JSON. The current CLI structural query type is callers.
Structural shortest path
Find the shortest relationship path between two symbols:
astra path "FastAPI" "ModelField" --path C:\Users\YourName\Downloads\my-project
Example output:
FastAPI
fastapi/applications.py:65
↳ DefaultPlaceholder
fastapi/dependencies/models.py:21
↳ get_request_handler()
fastapi/routing.py:310
↳ ModelField
pydantic/fields.py:740
Shortest path (3 hops):
FastAPI --uses--> DefaultPlaceholder <--references-- get_request_handler() --references--> ModelField
<p align="center"> <img src="docs/astra-path.svg" alt="astra path resolving a call path across the code graph" width="900"> </p>
Dipper sub-graph scoop
Extract a token-optimized, dependency-complete local sub-graph around a concept or symbol:
astra dipper "checkout flow" --path C:\Users\YourName\Downloads\my-project --limit 6 --parent-depth 1 --child-depth 1
The command returns JSON with seeds, nodes, edges, and trimmed source snippets for direct LLM prompting.
Tether structural health sentinel
Run graph-health checks to flag architectural drift risks:
astra tether --path C:\Users\YourName\Downloads\my-project --cycle-limit 25 --fanout-threshold 12
The report includes cycles, orphan declarations, high fan-out symbols, anomaly summaries, and a pass/warn status.
Fragility hotspots
Rank functions and classes by graph centrality, AST complexity, and architectural instability:
astra fragility --path C:\Users\YourName\Downloads\my-project --limit 10 --threshold 75
The matching MCP tool is astra_get_fragility_hotspots. See docs/fragility-hotspots.md for the scoring formula and LLM/CI workflows.
Impact and refactor planning
Inspect the upstream blast radius of a declaration:
astra impact calculate_total --path C:\Users\YourName\Downloads\my-project
Preview a graph-ordered identifier rename without changing files:
astra refactor-plan calculate_total sum_total --path C:\Users\YourName\Downloads\my-project
The MCP equivalents are astra_impact and astra_refactor_plan. astra tether continues to report orphan declarations and circular dependencies, while astra visualize provides the structural anomaly view. See docs/fragility-hotspots.md for the full CLI/MCP workflow.
Tether CI gate policy
Recommended policy for pull-request automation:
- Fail when new structural cycles are introduced.
- Warn when orphan-node count or high-fanout count increases beyond your baseline.
- Pass when cycles do not increase and coupling metrics stay within threshold.
A practical CI sequence:
- Index the repository snapshot for the PR branch.
- Run
astra tetherand store the JSON output. - Compare
summary.cycles,summary.orphans, andsummary.high_fanoutagainst yourmainbaseline. - Block merge only for hard-fail conditions (for example, cycle growth).
- Post warn-level findings as automated PR comments.
Example command:
astra index C:\Users\YourName\Downloads\my-project
astra tether --path C:\Users\YourName\Downloads\my-project --cycle-limit 25 --fanout-threshold 12
Search behavior
The default index is deterministic and works offline using local lexical matching. To enable Sentence Transformers and Chroma vector retrieval, set this before indexing and searching:
$env:ASTRA_ENABLE_EMBEDDINGS = "1"
astra index C:\Users\YourName\Downloads\my-project
astra search "rate limiting and retries" --path C:\Users\YourName\Downloads\my-project
The first embedding-enabled run may download the all-MiniLM-L6-v2 model. If model access or the vector store is unavailable, Astra falls back to the local search implementation.
Visualize an index
After indexing, generate a local HTML report for the target folder:
astra visualize C:\Users\YourName\Downloads\my-project
This writes .astra_visualization.html into the target folder. Open that file in a browser to inspect two views:
- Structural graph: interactive modules, classes, functions, methods, and definition/call edges from
.astra_graph.json. - Vector chunks: searchable file and symbol cards from
.astra_vectors/chunks.json, with expandable source previews.
You can choose a different output path:
astra visualize C:\Users\YourName\Downloads\my-project --output C:\Users\YourName\Desktop\astra-report.html
The report uses the Vis Network browser library from its CDN for the structural canvas. The vector view and generated data remain local; an internet connection is only needed to load the interactive graph library when opening the report.
Visualization examples
The structural graph view renders connected modules, declarations, and relationships across the indexed project:
<p align="center"> <img src="assets/astra_viz_screenshot_2.png" alt="Astra structural graph visualization showing connected code nodes" width="100%"> </p>
The vector chunks view provides searchable cards with file paths, symbols, line ranges, and expandable source previews:
<p align="center"> <img src="assets/astra_viz_screenshot_1.png" alt="Astra vector chunks visualization showing searchable source cards" width="100%"> </p>
MCP
Astra also exposes its engine as an official MCP server over stdio. Configure your MCP host to run astra-mcp; the server must be installed in the environment used by that host.
The native tools are:
| Tool | Purpose |
|---|---|
astra_index_repo(path) |
Index a local repository. |
astra_semantic_search(path, query, limit) |
Search indexed code chunks. |
astra_get_callers(path, target, limit) |
Find callers through the structural graph. |
astra_path(path, source, target, max_hops) |
Find the shortest structural path between two symbols. |
astra_dipper(path, query, limit, parent_depth, child_depth, max_nodes, max_source_chars) |
Scoop a localized dependency-complete sub-graph for token-efficient LLM context. |
astra_tether(path, cycle_limit, fanout_threshold) |
Run structural health checks and return architecture anomaly findings. |
astra_get_fragility_hotspots(path, limit, threshold) |
Rank fragile declarations using graph centrality, AST complexity, and instability. |
astra_impact(path, target, max_nodes) |
Trace incoming calls and dependencies to report a declaration's blast radius. |
astra_refactor_plan(path, target, replacement) |
Preview a graph-ordered, read-only structural identifier rename. |
astra_hybrid_context(path, query, limit, expansion) |
Combine semantic matches with graph expansion. |
astra_visualize(path, output) |
Generate a local HTML graph report and return its file path and URL. |
The repository includes .vscode/mcp.json for VS Code MCP clients. For other hosts, register the server in the host's configuration:
[mcp_servers.astra]
command = "astra-mcp"
args = []
If the host cannot find commands from your shell, use the absolute executable path: .venv\Scripts\astra-mcp.exe on Windows or .venv/bin/astra-mcp on macOS/Linux. The server communicates over stdio, so do not redirect its stdout.
MCP tool orchestration
All analysis and report tools automatically run the incremental index preflight before reading graph or vector artifacts. The hash cache makes unchanged files inexpensive, so an agent can call tools directly after source changes without constructing its own HTML or search index.
For a codebase knowledge-graph visualization, use this sequence:
1. astra_index_repo(path)
2. astra_visualize(path)
For investigation before an edit, use astra_hybrid_context or astra_dipper, then astra_impact and astra_get_fragility_hotspots. For a rename, call astra_refactor_plan first, review its read-only changes and order, apply the approved edit through the host's editing capability, then call astra_index_repo and astra_visualize again. Agents should use Astra's returned paths, URLs, nodes, snippets, and JSON reports rather than recreating equivalent artifacts.
Integrations
Claude Desktop
Use the server over stdio (astra-mcp) and call tools like astra_dipper and astra_tether from Claude. A complete setup guide is available in docs/claude-desktop.md.
Claude CLI
Register Astra with Claude Code from your terminal:
claude mcp add --transport stdio astra -- astra-mcp
Verify the server with claude mcp list, then start Claude Code and ask it to use Astra's MCP tools. If astra-mcp is not on your PATH, replace it with the absolute path to .venv\Scripts\astra-mcp.exe on Windows or .venv/bin/astra-mcp on macOS/Linux.
Gemini CLI
Gemini can consume Astra through any MCP-compatible bridge or local tool host that supports stdio MCP servers. Register astra-mcp, then use astra_dipper to scoop focused context and astra_tether for PR health checks.
Register Astra with Gemini CLI:
gemini mcp add astra astra-mcp
Verify the server with gemini mcp list, then start Gemini CLI and ask it to index or search your project with Astra. If astra-mcp is not on your PATH, use the absolute executable path as the command instead.
Codex CLI
Codex CLI can run Astra directly from terminal commands:
astra index C:\Users\YourName\Downloads\my-project
astra dipper "payment retry flow" --path C:\Users\YourName\Downloads\my-project
astra tether --path C:\Users\YourName\Downloads\my-project
For MCP-based Codex workflows, point your MCP client config to astra-mcp and invoke astra_path, astra_dipper, and astra_tether as needed.
Development
pytest
ruff check .
Astra never imports or executes files it indexes. Parsing failures are skipped and reported only through the resulting counts, which keeps indexing useful for partially broken working trees.
<p align="center"> <img src="assets/astra-social-preview.png" alt="Astra social preview" width="100%"> </p>
推荐服务器
Baidu Map
百度地图核心API现已全面兼容MCP协议,是国内首家兼容MCP协议的地图服务商。
Playwright MCP Server
一个模型上下文协议服务器,它使大型语言模型能够通过结构化的可访问性快照与网页进行交互,而无需视觉模型或屏幕截图。
Magic Component Platform (MCP)
一个由人工智能驱动的工具,可以从自然语言描述生成现代化的用户界面组件,并与流行的集成开发环境(IDE)集成,从而简化用户界面开发流程。
Audiense Insights MCP Server
通过模型上下文协议启用与 Audiense Insights 账户的交互,从而促进营销洞察和受众数据的提取和分析,包括人口统计信息、行为和影响者互动。
VeyraX
一个单一的 MCP 工具,连接你所有喜爱的工具:Gmail、日历以及其他 40 多个工具。
graphlit-mcp-server
模型上下文协议 (MCP) 服务器实现了 MCP 客户端与 Graphlit 服务之间的集成。 除了网络爬取之外,还可以将任何内容(从 Slack 到 Gmail 再到播客订阅源)导入到 Graphlit 项目中,然后从 MCP 客户端检索相关内容。
Kagi MCP Server
一个 MCP 服务器,集成了 Kagi 搜索功能和 Claude AI,使 Claude 能够在回答需要最新信息的问题时执行实时网络搜索。
e2b-mcp-server
使用 MCP 通过 e2b 运行代码。
Neon MCP Server
用于与 Neon 管理 API 和数据库交互的 MCP 服务器
Exa MCP Server
模型上下文协议(MCP)服务器允许像 Claude 这样的 AI 助手使用 Exa AI 搜索 API 进行网络搜索。这种设置允许 AI 模型以安全和受控的方式获取实时的网络信息。