Autonomous AI CLI for interactive architecture visualization, call-graph mapping, and hybrid codebase intelligence.
Project description
๐ง Codetrace AI
The Autonomous System Architect for your Terminal and IDE.
Codetrace AI is a privacy-first AI agent that builds a "Hybrid Brain" of your codebase โ combining semantic vector search with structural call-graph analysis โ so it understands what calls what, who owns what, and what breaks if you change something.
๐ฅ See it in Action
๐ What's New in v1.0.0
โจ Interactive Architecture Visualizer (codetrace visualize)
The biggest addition yet. A completely new, interactive HTML architecture map generated directly from your code graph.
- Folder-level dependency map โ groups all symbols by directory and renders cross-folder function call edges
- Hover on any folder โ instantly highlights all its connections and shows a detailed breakdown panel
- Hover on any edge โ reveals the exact function-to-function calls driving that dependency
- Click a folder โ opens a detail sidebar showing all files, symbols, and dependency counts
- Collapsible tree with smooth animations โ
[+]/[-]indicators keep large graphs navigable - Search bar โ find any symbol and highlight its containing folder instantly
- Drag, zoom, and pan โ full interactive layout control
๐ Multi-Language Parser Expansion (11+ Languages)
Codetrace now parses C#, Swift, Kotlin, Bash, HTML, JSON, CSS, YAML, SQL, TOML, and Dockerfiles in addition to the original Python, JS, TS, Java, C++, Go, Rust, and PHP.
โก 2x Faster Ollama Integration
Local Ollama models connected with Codetrace-AI works 2x faster than before due to the optimization made in the code and Call Graph it initally takes time to load the model but after that it works 2x faster than before as now it use single model for the whole session.
Codetrace AI is open-source. Contributions, bug reports, and feature requests are welcome. If you build something with or inspired by Codetrace, we'd love to hear about it โ please credit the original project.
โจ Full Feature Set
Codetrace acts as a highly knowledgeable senior engineer on your project:
| Feature | Description |
|---|---|
| ๐ Autonomous Code Research | Ask a question, and the agent proactively searches, reads files, and traverses the graph to find the exact answer |
| ๐บ๏ธ Interactive Architecture Map | codetrace visualize generates a beautiful, self-contained interactive HTML graph of your code architecture |
| ๐ Structural Call Graph Mapping | Navigates class and function definitions across 11+ languages to see exactly how your application is wired |
| ๐ฅ Blast Radius Analysis | Analyzes the downstream impact of a code change before you make it, preventing unintended breakages |
| โ๏ธ Human-in-the-Loop Code Edits | Proposes code changes with a rich diff preview โ you approve or decline before anything is written to disk |
| โก Smart SHA-256 Delta Sync | Re-indexes only the files that actually changed. Lightning fast on every subsequent run |
| ๐ IDE Context Injection (MCP) | Connects the Hybrid Brain directly into Cursor, Windsurf, or Claude Code for in-editor AI assistance |
| ๐ Persistent Chat Sessions | All conversations are saved. Resume any past session by ID, or export it to Markdown |
| ๐ 100% Local & Air-Gapped | All parsing, embedding, and graph mapping happens on your machine. Zero data leaves without your consent |
๐ Privacy-First Architecture
Codetrace is built with a Privacy-First design. It can operate 100% offline:
- Local LLM: Configure any local provider via Ollama (e.g.,
llama3.2,deepseek-coder). - Local Embeddings: Uses HuggingFace
bge-small+e5-smallmodels, downloaded once and cached. - True Air-Gap: Transfer the HuggingFace cache (
~/.cache/huggingface/hub) via USB. Runcodetrace init --offlineto block all external calls permanently.
[!WARNING] Ollama Users โ Context Window & RAM Dependency When using Ollama, the effective context window of your local model is directly limited by your available system RAM. If the model's context window is larger than what your RAM can load, Ollama may hang, respond extremely slowly, or crash silently.
Recommendations:
- 8 GB RAM:
qwen2.5-coder:7bยทdeepseek-r1:7bยทphi4-mini(best balance of coding + reasoning at this size)- 16 GB RAM:
qwen2.5-coder:14bยทdeepseek-r1:14bยทgemma3:12bยทgemma4:12b (quantized)(recommended sweet spot for most developers)- 32 GB+ RAM:
qwen2.5-coder:32bยทdeepseek-r1:32bยทdevstral:24b(near frontier-level code reasoning locally)If Codetrace hangs during a chat session while using Ollama, the most likely cause is the model running out of RAM to process the context. Switch to a smaller model with
codetrace config.
๐ Installation
Requires Python 3.10+
pip install codetrace-ai
uv pip install codetrace-ai
Note: Python version 3.14 might have problem with installation of dependencies. Use python version 3.10-3.12 for better experience. For GPU users, make sure you have CUDA installed.
โก Quick Start
cd /path/to/your/project
codetrace init
codetrace chat
codetrace initconfigures your LLM provider, downloads embedding models, indexes your codebase, and registers the MCP server โ all in one command.
๐ ๏ธ CLI Command Reference
| Command | Description |
|---|---|
codetrace init [PATH] |
One-command setup: config โ download models โ index โ register MCP |
codetrace chat |
Launch the interactive AI Architect chat loop |
codetrace chat --resume <ID> |
Resume a specific past chat session |
codetrace index <PATH or URL> |
Re-index a local directory or clone + index a GitHub URL |
codetrace config |
View or update your LLM provider and API key |
codetrace visualize |
Generate and open an interactive HTML architecture graph |
codetrace history |
List all past chat sessions for the current project |
codetrace export <ID> |
Export a chat session to terminal or save as a Markdown file |
Flags:
--offlineโ Strict air-gapped mode (blocks all telemetry and external requests)--fastโ Use smaller embedding models for lower RAM usage--llm <provider>โ Pre-select provider:groq,openai,anthropic,gemini,ollama
In-chat commands:
/clearโ Start a fresh session without exitingexit/quitโ Close the chat
๐ค Agentic Tool Suite
The AI has access to 7 specialized tools it invokes autonomously:
| Tool | What it does |
|---|---|
search_codebase |
Hybrid semantic search (BGE + E5 + RRF + FlashRank reranker) |
get_symbol_relations |
Graph traversal โ see callers and dependencies of any symbol |
analyze_impact |
Blast radius โ find every downstream symbol affected by a change |
read_file |
Read full file content from the indexed DB snapshot |
write_file |
Propose a code change with a diff preview for your approval |
inspect_index |
List all indexed files and DB coverage metadata |
git_diff |
Run a safe, injection-protected git diff |
๐ IDE Integration (MCP)
codetrace init automatically registers the MCP server in Cursor and Claude Code. No manual configuration needed.
Your IDE instantly gains access to all 7 tools above for in-editor assistance.
Using Windsurf? Add it manually to your mcp.json:
"codetrace": {
"command": "python",
"args": [
"/absolute/path/to/your/project/codetrace_mcp/server.py",
"--project",
"/absolute/path/to/your/project"
]
}
๐ File Structure
After codetrace init, your project will have:
your-project/
โโโ .codetrace/
โ โโโ chroma/ โ vector embeddings (ChromaDB)
โ โโโ graph_metadata.db โ code call graph (SQLite + NetworkX)
โ โโโ sync_metadata.db โ SHA-256 delta sync state
โ โโโ chat_history.db โ persistent chat sessions
โ โโโ graph_visualization.html โ generated by `codetrace visualize`
โโโ src/
โโโ your code files
Global config is stored at ~/.codetrace/config.json
๐ Changelog
v1.0.0 โ June 2026
- โ
NEW: Interactive Architecture Visualizer (
codetrace visualize) with collapsible tree, hover panels, search, and cross-folder call edges - โ NEW: Expanded language support โ C#, Swift, Kotlin, Bash, HTML, JSON, CSS, YAML, SQL, TOML, Dockerfile
- โ NEW: Token Budget Manager โ 3-tier context window management with auto-history compression
- โ
NEW: Multi-provider Agent Loop via pure
httpx(zero LangChain dependency) - โ
NEW: Live model listing during
codetrace configโ fetches available models from your provider's API - โ
IMPROVED: Parallel file parsing with
ThreadPoolExecutorfor significantly faster indexing - โ
IMPROVED: Path traversal protection on
read_fileandwrite_filetools
v0.1.2
- Initial public release with Hybrid Brain engine (BGE + E5 + ChromaDB + NetworkX)
- Core agentic tool suite (
search_codebase,analyze_impact,write_file,git_diff) - MCP auto-registration for Cursor and Claude Code
- SHA-256 Smart Delta Sync
- GitHub URL cloning + indexing support
- Persistent chat sessions with
historyandexport
๐ค Contributing
We welcome contributions! See CONTRIBUTING.md for guidelines.
๐ฌ Help shape Codetrace: Join the discussion โ
๐ License
MIT License โ Copyright (c) 2026 Viraaj Sawant. See LICENSE for details.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file codetrace_ai-1.0.1.tar.gz.
File metadata
- Download URL: codetrace_ai-1.0.1.tar.gz
- Upload date:
- Size: 88.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
582b3512970e22c34ca25190db360cb0d493afd4ed5d221656a00b99b4fbbd13
|
|
| MD5 |
1cc9e8c9446a7df87edff3a82c17e3a3
|
|
| BLAKE2b-256 |
6b751efa3e03289495a1b7a3132e48ae1caab719d09a080f3f80b36f52a75e92
|
File details
Details for the file codetrace_ai-1.0.1-py3-none-any.whl.
File metadata
- Download URL: codetrace_ai-1.0.1-py3-none-any.whl
- Upload date:
- Size: 96.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0e4801da47eafc76ba7f0b310ff43d0222a083efbee71003ad847c06b00d2018
|
|
| MD5 |
00a3d15b15f00ded90f9ff231e564df2
|
|
| BLAKE2b-256 |
1c435c612019448133ce9ee5f36617bb50c135b8448c801b76dec154ea0e7301
|