repo-map is an advanced tool for generating comprehensive, AI-enhanced summaries of software repositories. It provides developers with valuable insights into project structures, file purposes, and potential considerations across various programming languages. Using efficient caching, repo-map only processes files that have changed since the last run, making it ideal for continuous use in evolving projects. This tool not only aids in understanding and documenting codebases but can also assist LLM agents in writing accurate and functional code within your existing project structure.
- 📊 Generates detailed repository structure summaries
- 🧠 AI-powered enhancements:
- 💡 Developer considerations for potential issues or unique aspects
- 🗣️ Concise explanations of file purposes and functionality
- 🔍 Insights into code structure and organization
- 🌐 Analyzes code structure across multiple programming languages
- 🚀 Supports various file types including Python, Java, JavaScript, TypeScript, and more
- 💾 Caching mechanism using SQLite for efficient processing of unchanged files
- 🌳 Tree-like visualization of the repository structure
- 📝 Markdown output for easy sharing and documentation
- 🔒 Respects .gitignore files (including nested ones) within the target directory for file exclusion
- 🚦 Implements rate limiting and exponential backoff for LLM API calls
- ⚡ Asynchronous processing for improved performance
Install repo-map using pip:
pip install repo-map
Additionally, the repo_map.py
file is designed to be a standalone script that can be easily integrated into any existing project repository.
To generate a repository map, run the following command:
repo-map <repository_path> [options]
Replace <repository_path>
with the path to the repository you want to analyze.
-y,
--yes
: Automatically accept the disclaimer and proceed without prompting--model MODEL
: Specify the OpenRouter LLM model to use (default: anthropic/claude-3.5-sonnet)
Examples:
# Basic usage
repo-map /path/to/your/repo
# Use a specific model
repo-map /path/to/your/repo --model anthropic/claude-3-opus
# Auto-accept disclaimer
repo-map /path/to/your/repo -y
Here's an example of a repo-map generated for an advanced Snake game implemented in Python:
/ (SSSnakeGame)
├── main.py (Python)
│ ├── Description: Entry point for the Snake game, initializes game and runs the main loop.
│ ├── Developer Consideration: "Uses pygame for game development, which may require additional setup for cross-platform compatibility."
│ ├── Imports: [pygame, random, time]
│ ├── Functions: [main, game_loop, draw_snake, draw_food]
├── config.py (Python)
│ ├── Description: Centralizes game configuration parameters.
│ ├── Developer Consideration: "Hard-coded values may need adjustment for different screen sizes or game difficulties."
├── assets/
│ ├── images/
│ │ ├── snake_head.png (Image)
│ │ ├── food.png (Image)
│ ├── sounds/
│ │ ├── eat.wav (Audio)
│ │ ├── game_over.mp3 (Audio)
├── requirements.txt (Text)
│ ├── Description: Lists all Python package dependencies for the project.
├── README.md (Markdown)
│ ├── Description: Provides project overview, setup instructions, and gameplay details.
└──────────────
This example demonstrates how repo-map provides a comprehensive overview of a Snake game project, including file descriptions, developer considerations, and key structural information.
- Python 3.7+
- Dependencies:
- requests
- tqdm
- aiohttp
- pathspec
- certifi
Before using repo-map, you need to set up your OpenRouter API key. Set the following environment variable:
export OPENROUTER_API_KEY=your_api_key_here
Replace your_api_key_here
with your actual OpenRouter API key.
- 📂 Walks through the repository directory structure
- 📝 Analyzes file contents and extracts key information (imports, functions, classes)
- 🤖 Utilizes an LLM (via OpenRouter) to generate descriptions and developer considerations
- 🗃️ Caches results in SQLite for efficient processing of unchanged files
- 📊 Generates a comprehensive tree-like structure of the repository
- 💾 Saves the output as a Markdown file for easy viewing and sharing
summarize_repo()
: Generates the initial repository summaryenhance_repo_with_llm()
: Enhances the summary with AI-generated descriptionsget_structure()
: Extracts classes, functions, and constants from filesprint_tree()
: Displays the repository structure in a tree formatsave_tree_map()
: Saves the repository map to a Markdown file
- The tool supports a wide range of file types and programming languages. Check the
SUPPORTED_LANGUAGES
dictionary in the script for a full list. - A pre-enhanced repository summary is saved to
.repo_map_structure.json
as an intermediate step. - The tool uses a manual ignore list for generated files like
.repo_map_structure.json
and.repo-map-cache.db
. - SSL verification is handled using the certifi library for secure API communications.
This project is licensed under the Apache 2.0 License. See the LICENSE file in the root directory of this project for the full license text.
Contributions, issues, and feature requests are welcome! Feel free to check the issues page.
If you encounter any problems or have any questions, please open an issue in the GitHub repository.
Current version: 0.1.0
By using this tool, you acknowledge that files will be sent to the OpenRouter LLM for processing. Ensure you have the necessary permissions and consider any sensitive information in your repository.