Skip to main content

Autonomous AI CLI for interactive architecture visualization, call-graph mapping, and hybrid codebase intelligence.

Project description

Codetrace-ai Banner

PyPI Version Downloads MIT License GitHub Stars PRs Welcome Open Issues Open Pull Requests Build Status Join Slack

๐Ÿง  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

Watch Demo


๐Ÿ†• 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:

  1. Local LLM: Configure any local provider via Ollama (e.g., llama3.2, deepseek-coder).
  2. Local Embeddings: Uses HuggingFace bge-small + e5-small models, downloaded once and cached.
  3. True Air-Gap: Transfer the HuggingFace cache (~/.cache/huggingface/hub) via USB. Run codetrace init --offline to 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 init configures 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 exiting
  • exit / 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 ThreadPoolExecutor for significantly faster indexing
  • โœ… IMPROVED: Path traversal protection on read_file and write_file tools

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 history and export

๐Ÿค 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

codetrace_ai-1.0.1.tar.gz (88.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

codetrace_ai-1.0.1-py3-none-any.whl (96.5 kB view details)

Uploaded Python 3

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

Hashes for codetrace_ai-1.0.1.tar.gz
Algorithm Hash digest
SHA256 582b3512970e22c34ca25190db360cb0d493afd4ed5d221656a00b99b4fbbd13
MD5 1cc9e8c9446a7df87edff3a82c17e3a3
BLAKE2b-256 6b751efa3e03289495a1b7a3132e48ae1caab719d09a080f3f80b36f52a75e92

See more details on using hashes here.

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

Hashes for codetrace_ai-1.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 0e4801da47eafc76ba7f0b310ff43d0222a083efbee71003ad847c06b00d2018
MD5 00a3d15b15f00ded90f9ff231e564df2
BLAKE2b-256 1c435c612019448133ce9ee5f36617bb50c135b8448c801b76dec154ea0e7301

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page